converting to pydantic-settings

2026-06-14 21:29:45 -04:00
parent a5d7c3be4f
commit 68b3a38b81
10 changed files with 111 additions and 110 deletions
@@ -29,7 +29,7 @@ def request_embeddings(texts: Sequence[str], config: EbookSearchConfig) -> list[
            f"{config.embedding_base_url.rstrip('/')}/embeddings",
            headers=auth_headers(config.embedding_api_key),
            json={"model": config.embedding_model, "input": list(texts)},
-            timeout=60,
+            timeout=config.embedding_timeout_seconds,
        )
        response.raise_for_status()
        return embedding_vectors_from_response(response.json())
@@ -106,7 +106,7 @@ def request_chat_completion(
                "messages": list(messages),
                "temperature": 0,
            },
-            timeout=60,
+            timeout=config.chat_timeout_seconds,
        )
        response.raise_for_status()
        return chat_content_from_response(response.json())