tavily-python 0.7.16__tar.gz → 0.7.18__tar.gz

{tavily_python-0.7.16 → tavily_python-0.7.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tavily-python
-Version: 0.7.16
+Version: 0.7.18
 Summary: Python wrapper for the Tavily API
 Home-page: https://github.com/tavily-ai/tavily-python
 Author: Tavily AI

{tavily_python-0.7.16 → tavily_python-0.7.18}/setup.py

@@ -5,7 +5,7 @@ with open('README.md', 'r', encoding='utf-8') as f:
 
 setup(
     name='tavily-python',
-    version='0.7.16',
+    version='0.7.18',
     url='https://github.com/tavily-ai/tavily-python',
     author='Tavily AI',
     author_email='support@tavily.com',

{tavily_python-0.7.16 → tavily_python-0.7.18}/tavily/async_tavily.py

@@ -18,7 +18,8 @@ class AsyncTavilyClient:
                  company_info_tags: Sequence[str] = ("news", "general", "finance"),
                  proxies: Optional[dict[str, str]] = None,
                  api_base_url: Optional[str] = None,
-                 client_source: Optional[str] = None):
+                 client_source: Optional[str] = None,
+                 project_id: Optional[str] = None):
         if api_key is None:
             api_key = os.getenv("TAVILY_API_KEY")
 
@@ -40,12 +41,16 @@ class AsyncTavilyClient:
             else None
         )
 
+        tavily_project = project_id or os.getenv("TAVILY_PROJECT")
+
         self._api_base_url = api_base_url or "https://api.tavily.com"
+        
         self._client_creator = lambda: httpx.AsyncClient(
             headers={
                 "Content-Type": "application/json",
                 "Authorization": f"Bearer {api_key}",
-                "X-Client-Source": client_source or "tavily-python"
+                "X-Client-Source": client_source or "tavily-python",
+                **({"X-Project-ID": tavily_project} if tavily_project else {})
             },
             base_url=self._api_base_url,
             mounts=proxy_mounts
@@ -55,7 +60,7 @@ class AsyncTavilyClient:
     async def _search(
             self,
             query: str,
-            search_depth: Literal["basic", "advanced"] = None,
+            search_depth: Literal["basic", "advanced", "fast", "ultra-fast"] = None,
             topic: Literal["general", "news", "finance"] = None,
             time_range: Literal["day", "week", "month", "year"] = None,
             start_date: str = None,
@@ -132,7 +137,7 @@ class AsyncTavilyClient:
 
     async def search(self,
                      query: str,
-                     search_depth: Literal["basic", "advanced"] = None,
+                     search_depth: Literal["basic", "advanced", "fast", "ultra-fast"] = None,
                      topic: Literal["general", "news", "finance"] = None,
                      time_range: Literal["day", "week", "month", "year"] = None,
                      start_date: str = None,

{tavily_python-0.7.16 → tavily_python-0.7.18}/tavily/tavily.py

@@ -12,7 +12,7 @@ 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, client_source: Optional[str] = None):
+    def __init__(self, api_key: Optional[str] = None, proxies: Optional[dict[str, str]] = None, api_base_url: Optional[str] = None, client_source: Optional[str] = None, project_id: Optional[str] = None):
         if api_key is None:
             api_key = os.getenv("TAVILY_API_KEY")
 
@@ -25,19 +25,22 @@ class TavilyClient:
         }
 
         resolved_proxies = {k: v for k, v in resolved_proxies.items() if v} or None
-
+        tavily_project = project_id or os.getenv("TAVILY_PROJECT")
+        
         self.base_url = api_base_url or "https://api.tavily.com"
         self.api_key = api_key
         self.proxies = resolved_proxies
+        
         self.headers = {
             "Content-Type": "application/json",
             "Authorization": f"Bearer {self.api_key}",
-            "X-Client-Source": client_source or "tavily-python"
+            "X-Client-Source": client_source or "tavily-python",
+            **({"X-Project-ID": tavily_project} if tavily_project else {})
         }
 
     def _search(self,
                 query: str,
-                search_depth: Literal["basic", "advanced"] = None,
+                search_depth: Literal["basic", "advanced", "fast", "ultra-fast"] = None,
                 topic: Literal["general", "news", "finance"] = None,
                 time_range: Literal["day", "week", "month", "year"] = None,
                 start_date: str = None,
@@ -116,7 +119,7 @@ class TavilyClient:
 
     def search(self,
                query: str,
-               search_depth: Literal["basic", "advanced"] = None,
+               search_depth: Literal["basic", "advanced", "fast", "ultra-fast"] = None,
                topic: Literal["general", "news", "finance" ] = None,
                time_range: Literal["day", "week", "month", "year"] = None,
                start_date: str = None,

{tavily_python-0.7.16 → tavily_python-0.7.18}/tavily_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tavily-python
-Version: 0.7.16
+Version: 0.7.18
 Summary: Python wrapper for the Tavily API
 Home-page: https://github.com/tavily-ai/tavily-python
 Author: Tavily AI