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)