From ee217ee9f6866c0fecf238826c0d77f2e276ae42 Mon Sep 17 00:00:00 2001 From: budhewarvijay0407 Date: Fri, 25 Jul 2025 15:56:55 +0530 Subject: [PATCH] adding verify argument for requests --- tavily/tavily.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tavily/tavily.py b/tavily/tavily.py index 4c7851b..ac893d7 100644 --- a/tavily/tavily.py +++ b/tavily/tavily.py @@ -9,11 +9,11 @@ from .config import AllowedCategory class TavilyClient: - """ + """ Tavily API client class. """ - def __init__(self, api_key: Optional[str] = None, proxies: Optional[dict[str, str]] = None, api_base_url: Optional[str] = None): + def __init__(self, api_key: Optional[str] = None, proxies: Optional[dict[str, str]] = None, api_base_url: Optional[str] = None, verify: bool = True): if api_key is None: api_key = os.getenv("TAVILY_API_KEY") @@ -35,6 +35,7 @@ def __init__(self, api_key: Optional[str] = None, proxies: Optional[dict[str, st "Authorization": f"Bearer {self.api_key}", "X-Client-Source": "tavily-python" } + self.verify = verify def _search(self, query: str, @@ -87,7 +88,7 @@ def _search(self, timeout = min(timeout, 120) try: - response = requests.post(self.base_url + "/search", data=json.dumps(data), headers=self.headers, timeout=timeout, proxies=self.proxies) + response = requests.post(self.base_url + "/search", data=json.dumps(data), headers=self.headers, timeout=timeout, proxies=self.proxies , verify=self.verify) except requests.exceptions.Timeout: raise TimeoutError(timeout)