unique-web-search 1.8.2__py3-none-any.whl → 1.8.3__py3-none-any.whl

unique_web_search/services/search_engine/custom_api.py

@@ -1,5 +1,5 @@
 import json
-from typing import Literal, TypeVar
+from typing import Any, Literal, TypeVar
 
 from httpx import AsyncClient
 from pydantic import Field
@@ -86,7 +86,10 @@ class CustomAPI(SearchEngine[CustomAPIConfig]):
     async def search(self, query: str, **kwargs) -> list[WebSearchResult]:
         params, body = self._prepare_request_params_and_body(query)
 
-        async with AsyncClient(timeout=self.config.timeout) as client:
+        async_client_params = self._client_config | {
+            "timeout": self.config.timeout,
+        }
+        async with AsyncClient(**async_client_params) as client:
             response = await client.request(
                 method=self._request_method,
                 headers=self._headers,
@@ -118,6 +121,12 @@ class CustomAPI(SearchEngine[CustomAPIConfig]):
     def _additional_body_params(self) -> dict[str, str]:
         return json.loads(self.config.api_additional_body_params)
 
+    @property
+    def _client_config(self) -> dict[str, Any]:
+        if env_settings.custom_web_search_api_client_config is None:
+            return {}
+        return json.loads(env_settings.custom_web_search_api_client_config)
+
     def _prepare_request_params_and_body(self, query: str) -> tuple[dict, dict]:
         params = self._additional_query_params
         body = self._additional_body_params

unique_web_search/settings.py

@@ -66,6 +66,7 @@ class Base(BaseSettings):
     custom_web_search_api_headers: str | None = None
     custom_web_search_api_additional_query_params: str | None = None
     custom_web_search_api_additional_body_params: str | None = None
+    custom_web_search_api_client_config: str | None = None
 
     # Proxy settings
     ## Shared settings
@@ -139,6 +140,7 @@ class Base(BaseSettings):
         "custom_web_search_api_headers",
         "custom_web_search_api_additional_query_params",
         "custom_web_search_api_additional_body_params",
+        "custom_web_search_api_client_config",
         mode="after",
     )
     def validate_json(cls, v: str | None, info: ValidationInfo) -> str | None:

{unique_web_search-1.8.2.dist-info → unique_web_search-1.8.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-web-search
-Version: 1.8.2
+Version: 1.8.3
 Summary: 
 Author: Andreas Hauri, Gustav Hartz, Rami Azouz
 Author-email: Andreas Hauri <andreas@unique.ch>, Gustav Hartz <gustav.hartz.ext@unique.ch>, Rami Azouz <rami.ext@unique.ch>
@@ -17,7 +17,7 @@ Requires-Dist: tavily-python>=0.7.11,<1
 Requires-Dist: unidecode>=1.4.0,<2
 Requires-Dist: azure-ai-projects>=1.0.0,<2
 Requires-Dist: azure-identity>=1.25.0,<2
-Requires-Dist: unique-toolkit>=1.42.9,<2
+Requires-Dist: unique-toolkit>=1.45.0,<2
 Requires-Dist: azure-core>=1.36.0,<2
 Requires-Dist: google-cloud-aiplatform>=1.128.0,<2
 Requires-Dist: google-auth>=2.43.0,<3

{unique_web_search-1.8.2.dist-info → unique_web_search-1.8.3.dist-info}/RECORD

@@ -32,7 +32,7 @@ unique_web_search/services/search_engine/__init__.py,sha256=Wzq26mFLy43cZtr-kbau
 unique_web_search/services/search_engine/base.py,sha256=MBQWiozcIAnXvH7vuMpLo6EaWYlAgI30c-ROb91wZRU,2069
 unique_web_search/services/search_engine/bing.py,sha256=qBhQeeOOt9bxZK_5C3NeLcfHLgbFnMdhhwNAFjqrJEM,3743
 unique_web_search/services/search_engine/brave.py,sha256=qJg7efQCuIU1l1kyb_H_0AvA-zboOg2Ip0QnnMhCpN0,3764
-unique_web_search/services/search_engine/custom_api.py,sha256=d6rM2gKrqKvXqLtA_pfwaON_x6tsZkYvCi4PQ4yGE8Q,4631
+unique_web_search/services/search_engine/custom_api.py,sha256=nW4a1ruIBiMHaF4kjJuVNImBM8rklTjSWFhkIN1DX0E,4968
 unique_web_search/services/search_engine/firecrawl.py,sha256=CEhnpyxYvr8LO10Xxl54hWpIElCLBRQBH-x8o_vMjPA,4611
 unique_web_search/services/search_engine/google.py,sha256=SKdj2uR6nwOT18ktCkAcOJLoQULq7GeHILDFFewG8dU,5423
 unique_web_search/services/search_engine/jina.py,sha256=1I4C18W-UnjoNe4NXK3K5jfeHe9-2u9RBMkWp9Rgo60,10009
@@ -50,8 +50,8 @@ unique_web_search/services/search_engine/utils/vertexai/gemini.py,sha256=lpu9COy
 unique_web_search/services/search_engine/utils/vertexai/prompts.py,sha256=-gq9R6btaubDgPjENVhmlJUBqPYX6AYCBg2MF_o0KZY,1748
 unique_web_search/services/search_engine/utils/vertexai/response_handler.py,sha256=ddlYzNQ8SFR1I31tittED4U6zp4qoBBcYSYFpMVgyd4,2537
 unique_web_search/services/search_engine/vertexai.py,sha256=4AlNP7NHkSMD1lZROBFDcm61stjXDRTfzAkLBRfFy4A,4664
-unique_web_search/settings.py,sha256=Yj8LqYzylqv3qA8yZ5XZSQ-CrTOOEx2O81Z3yk2cs-k,5795
+unique_web_search/settings.py,sha256=EnMmQZGfn6S8_TFY6fCuz_DrL1dTabPuIOzRbQwAO-I,5901
 unique_web_search/utils.py,sha256=5yLyzKyqy1v-Ut_TEYb4fggpiUj-nPf84YsO7db75so,4069
-unique_web_search-1.8.2.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-unique_web_search-1.8.2.dist-info/METADATA,sha256=nKvXFqfohz6PYfvXeYCftT34M4nvVewXWiiH6D6tWOA,5565
-unique_web_search-1.8.2.dist-info/RECORD,,
+unique_web_search-1.8.3.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+unique_web_search-1.8.3.dist-info/METADATA,sha256=2VlbrJC86vSQ-I16zzWJGqg2R8CyulMMhxRNCNG8ViI,5565
+unique_web_search-1.8.3.dist-info/RECORD,,