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

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

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

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

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

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

@@ -157,7 +157,7 @@ class AsyncTavilyClient:
                      **kwargs,  # Accept custom arguments
                      ) -> dict:
         """
-        Combined search method. Set search_depth to either "basic" or "advanced".
+        Combined search method. Set search_depth to either "basic", "advanced", "fast", or "ultra-fast".
         """
         timeout = min(timeout, 120)
         response_dict = await self._search(query,
@@ -508,7 +508,7 @@ class AsyncTavilyClient:
 
     async def get_search_context(self,
                                  query: str,
-                                 search_depth: Literal["basic", "advanced"] = "basic",
+                                 search_depth: Literal["basic", "advanced", "fast", "ultra-fast"] = "basic",
                                  topic: Literal["general", "news", "finance"] = "general",
                                  days: int = 7,
                                  max_results: int = 5,
@@ -550,7 +550,7 @@ class AsyncTavilyClient:
 
     async def qna_search(self,
                          query: str,
-                         search_depth: Literal["basic", "advanced"] = "advanced",
+                         search_depth: Literal["basic", "advanced", "fast", "ultra-fast"] = "advanced",
                          topic: Literal["general", "news", "finance"] = "general",
                          days: int = 7,
                          max_results: int = 5,
@@ -584,7 +584,7 @@ class AsyncTavilyClient:
 
     async def get_company_info(self,
                                query: str,
-                               search_depth: Literal["basic", "advanced"] = "advanced",
+                               search_depth: Literal["basic", "advanced", "fast", "ultra-fast"] = "advanced",
                                max_results: int = 5,
                                timeout: float = 60,
                                country: str = None,

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

@@ -3,7 +3,6 @@ import json
 import os
 import warnings
 from typing import Literal, Sequence, Optional, List, Union, Generator
-from concurrent.futures import ThreadPoolExecutor, as_completed
 from .utils import get_max_items_from_list
 from .errors import UsageLimitExceededError, InvalidAPIKeyError, MissingAPIKeyError, BadRequestError, ForbiddenError, TimeoutError
 
@@ -38,6 +37,21 @@ class TavilyClient:
             **({"X-Project-ID": tavily_project} if tavily_project else {})
         }
 
+        self.session = requests.Session()
+        self.session.headers.update(self.headers)
+        if self.proxies:
+            self.session.proxies.update(self.proxies)
+
+    def close(self):
+        """Close the session and release resources."""
+        self.session.close()
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.close()
+
     def _search(self,
                 query: str,
                 search_depth: Literal["basic", "advanced", "fast", "ultra-fast"] = None,
@@ -89,9 +103,11 @@ class TavilyClient:
             data.update(kwargs)
 
         timeout = min(timeout, 120)
+        url = self.base_url + "/search"
+        payload = json.dumps(data)
 
         try:
-            response = requests.post(self.base_url + "/search", data=json.dumps(data), headers=self.headers, timeout=timeout, proxies=self.proxies)
+            response = self.session.post(url, data=payload, timeout=timeout)
         except requests.exceptions.Timeout:
             raise TimeoutError(timeout)
 
@@ -106,13 +122,12 @@ class TavilyClient:
 
             if response.status_code == 429:
                 raise UsageLimitExceededError(detail)
-            elif response.status_code in [403,432,433]:
+            elif response.status_code in [403, 432, 433]:
                 raise ForbiddenError(detail)
             elif response.status_code == 401:
                 raise InvalidAPIKeyError(detail)
             elif response.status_code == 400:
                 raise BadRequestError(detail)
-            
             else:
                 raise response.raise_for_status()
 
@@ -160,13 +175,8 @@ class TavilyClient:
                                      auto_parameters=auto_parameters,
                                      include_favicon=include_favicon,
                                      include_usage=include_usage,
-                                     **kwargs,
-                                     )
-
-        tavily_results = response_dict.get("results", [])
-
-        response_dict["results"] = tavily_results
-
+                                     **kwargs)
+        response_dict.setdefault("results", [])
         return response_dict
 
     def _extract(self,
@@ -202,7 +212,7 @@ class TavilyClient:
             data.update(kwargs)
 
         try:
-            response = requests.post(self.base_url + "/extract", data=json.dumps(data), headers=self.headers, timeout=timeout, proxies=self.proxies)
+            response = self.session.post(self.base_url + "/extract", data=json.dumps(data), timeout=timeout)
         except requests.exceptions.Timeout:
             raise TimeoutError(timeout)
 
@@ -217,7 +227,7 @@ class TavilyClient:
 
             if response.status_code == 429:
                 raise UsageLimitExceededError(detail)
-            elif response.status_code in [403,432,433]:
+            elif response.status_code in [403, 432, 433]:
                 raise ForbiddenError(detail)
             elif response.status_code == 401:
                 raise InvalidAPIKeyError(detail)
@@ -251,13 +261,8 @@ class TavilyClient:
                                       query=query,
                                       chunks_per_source=chunks_per_source,
                                       **kwargs)
-
-        tavily_results = response_dict.get("results", [])
-        failed_results = response_dict.get("failed_results", [])
-
-        response_dict["results"] = tavily_results
-        response_dict["failed_results"] = failed_results
-
+        response_dict.setdefault("results", [])
+        response_dict.setdefault("failed_results", [])
         return response_dict
 
     def _crawl(self,
@@ -310,8 +315,7 @@ class TavilyClient:
         data = {k: v for k, v in data.items() if v is not None}
 
         try:
-            response = requests.post(
-                self.base_url + "/crawl", data=json.dumps(data), headers=self.headers, timeout=timeout, proxies=self.proxies)
+            response = self.session.post(self.base_url + "/crawl", data=json.dumps(data), timeout=timeout)
         except requests.exceptions.Timeout:
             raise TimeoutError(timeout)
 
@@ -326,7 +330,7 @@ class TavilyClient:
 
             if response.status_code == 429:
                 raise UsageLimitExceededError(detail)
-            elif response.status_code in [403,432,433]:
+            elif response.status_code in [403, 432, 433]:
                 raise ForbiddenError(detail)
             elif response.status_code == 401:
                 raise InvalidAPIKeyError(detail)
@@ -359,26 +363,24 @@ class TavilyClient:
         Combined crawl method.
         include_favicon: If True, include the favicon in the crawl results.
         """
-        response_dict = self._crawl(url,
-                                    max_depth=max_depth,
-                                    max_breadth=max_breadth,
-                                    limit=limit,
-                                    instructions=instructions,
-                                    select_paths=select_paths,
-                                    select_domains=select_domains,
-                                    exclude_paths=exclude_paths,
-                                    exclude_domains=exclude_domains,
-                                    allow_external=allow_external,
-                                    include_images=include_images,
-                                    extract_depth=extract_depth,
-                                    format=format,
-                                    timeout=timeout,
-                                    include_favicon=include_favicon,
-                                    include_usage=include_usage,
-                                    chunks_per_source=chunks_per_source,
-                                    **kwargs)
-
-        return response_dict
+        return self._crawl(url,
+                           max_depth=max_depth,
+                           max_breadth=max_breadth,
+                           limit=limit,
+                           instructions=instructions,
+                           select_paths=select_paths,
+                           select_domains=select_domains,
+                           exclude_paths=exclude_paths,
+                           exclude_domains=exclude_domains,
+                           allow_external=allow_external,
+                           include_images=include_images,
+                           extract_depth=extract_depth,
+                           format=format,
+                           timeout=timeout,
+                           include_favicon=include_favicon,
+                           include_usage=include_usage,
+                           chunks_per_source=chunks_per_source,
+                           **kwargs)
     
     def _map(self,
             url: str,
@@ -421,8 +423,7 @@ class TavilyClient:
         data = {k: v for k, v in data.items() if v is not None}
 
         try:
-            response = requests.post(
-                self.base_url + "/map", data=json.dumps(data), headers=self.headers, timeout=timeout, proxies=self.proxies)
+            response = self.session.post(self.base_url + "/map", data=json.dumps(data), timeout=timeout)
         except requests.exceptions.Timeout:
             raise TimeoutError(timeout)
 
@@ -437,7 +438,7 @@ class TavilyClient:
 
             if response.status_code == 429:
                 raise UsageLimitExceededError(detail)
-            elif response.status_code in [403,432,433]:
+            elif response.status_code in [403, 432, 433]:
                 raise ForbiddenError(detail)
             elif response.status_code == 401:
                 raise InvalidAPIKeyError(detail)
@@ -466,26 +467,24 @@ class TavilyClient:
         Combined map method.
         
         """
-        response_dict = self._map(url,
-                                    max_depth=max_depth,
-                                    max_breadth=max_breadth,
-                                    limit=limit,
-                                    instructions=instructions,
-                                    select_paths=select_paths,
-                                    select_domains=select_domains,
-                                    exclude_paths=exclude_paths,
-                                    exclude_domains=exclude_domains,
-                                    allow_external=allow_external,
-                                    include_images=include_images,
-                                    timeout=timeout,
-                                    include_usage=include_usage,
-                                    **kwargs)
-
-        return response_dict
+        return self._map(url,
+                         max_depth=max_depth,
+                         max_breadth=max_breadth,
+                         limit=limit,
+                         instructions=instructions,
+                         select_paths=select_paths,
+                         select_domains=select_domains,
+                         exclude_paths=exclude_paths,
+                         exclude_domains=exclude_domains,
+                         allow_external=allow_external,
+                         include_images=include_images,
+                         timeout=timeout,
+                         include_usage=include_usage,
+                         **kwargs)
 
     def get_search_context(self,
                            query: str,
-                           search_depth: Literal["basic", "advanced"] = "basic",
+                           search_depth: Literal["basic", "advanced", "fast", "ultra-fast"] = "basic",
                            topic: Literal["general", "news", "finance"] = "general",
                            days: int = 7,
                            max_results: int = 5,
@@ -530,7 +529,7 @@ class TavilyClient:
 
     def qna_search(self,
                    query: str,
-                   search_depth: Literal["basic", "advanced"] = "advanced",
+                   search_depth: Literal["basic", "advanced", "fast", "ultra-fast"] = "advanced",
                    topic: Literal["general", "news", "finance"] = "general",
                    days: int = 7,
                    max_results: int = 5,
@@ -563,45 +562,6 @@ class TavilyClient:
                                      )
         return response_dict.get("answer", "")
 
-    def get_company_info(self,
-                         query: str,
-                         search_depth: Literal["basic",
-                                               "advanced"] = "advanced",
-                         max_results: int = 5,
-                         timeout: float = 60,
-                         country: str = None,
-                         ) -> Sequence[dict]:
-        """ Company information search method. Search depth is advanced by default to get the best answer. """
-        warnings.warn("get_company_info is deprecated and will be removed in future versions.",
-                      DeprecationWarning, stacklevel=2)
-        def _perform_search(topic):
-            return self._search(query,
-                                search_depth=search_depth,
-                                topic=topic,
-                                max_results=max_results,
-                                include_answer=False,
-                                timeout=timeout,
-                                country=country)
-
-        with ThreadPoolExecutor() as executor:
-            # Initiate the search for each topic in parallel
-            future_to_topic = {executor.submit(_perform_search, topic): topic for topic in
-                               ["news", "general", "finance"]}
-
-            all_results = []
-
-            # Process the results as they become available
-            for future in as_completed(future_to_topic):
-                data = future.result()
-                if 'results' in data:
-                    all_results.extend(data['results'])
-
-        # Sort all the results by score in descending order and take the top 'max_results' items
-        sorted_results = sorted(all_results, key=lambda x: x['score'], reverse=True)[
-            :max_results]
-
-        return sorted_results
-
     def _research(self,
                   input: str,
                   model: Literal["mini", "pro", "auto"] = None,
@@ -629,12 +589,10 @@ class TavilyClient:
 
         if stream:
             try:
-                response = requests.post(
+                response = self.session.post(
                     self.base_url + "/research",
                     data=json.dumps(data),
-                    headers=self.headers,
                     timeout=timeout,
-                    proxies=self.proxies,
                     stream=True
                 )
             except requests.exceptions.Timeout:
@@ -649,7 +607,7 @@ class TavilyClient:
 
                 if response.status_code == 429:
                     raise UsageLimitExceededError(detail)
-                elif response.status_code in [403,432,433]:
+                elif response.status_code in [403, 432, 433]:
                     raise ForbiddenError(detail)
                 elif response.status_code == 401:
                     raise InvalidAPIKeyError(detail)
@@ -669,12 +627,10 @@ class TavilyClient:
             return stream_generator()
         else:
             try:
-                response = requests.post(
+                response = self.session.post(
                     self.base_url + "/research",
                     data=json.dumps(data),
-                    headers=self.headers,
-                    timeout=timeout,
-                    proxies=self.proxies
+                    timeout=timeout
                 )
             except requests.exceptions.Timeout:
                 raise TimeoutError(timeout)
@@ -690,7 +646,7 @@ class TavilyClient:
 
                 if response.status_code == 429:
                     raise UsageLimitExceededError(detail)
-                elif response.status_code in [403,432,433]:
+                elif response.status_code in [403, 432, 433]:
                     raise ForbiddenError(detail)
                 elif response.status_code == 401:
                     raise InvalidAPIKeyError(detail)
@@ -724,8 +680,7 @@ class TavilyClient:
             dict: Response containing request_id, created_at, status, input, and model.
         """
 
-        
-        response_dict = self._research(
+        return self._research(
             input=input,
             model=model,
             output_schema=output_schema,
@@ -735,8 +690,6 @@ class TavilyClient:
             **kwargs
         )
 
-        return response_dict
-
     def get_research(self,
                      request_id: str
                      ) -> dict:
@@ -750,17 +703,12 @@ class TavilyClient:
             dict: Research response containing request_id, created_at, completed_at, status, content, and sources.
         """
         try:
-            response = requests.get(
-                self.base_url + f"/research/{request_id}",
-                headers=self.headers,
-                proxies=self.proxies,
-            )
+            response = self.session.get(self.base_url + f"/research/{request_id}")
         except Exception as e:
             raise Exception(f"Error getting research: {e}")
 
         if response.status_code in (200, 202):
-            data = response.json()
-            return data
+            return response.json()
         else:
             detail = ""
             try:
@@ -770,7 +718,7 @@ class TavilyClient:
 
             if response.status_code == 429:
                 raise UsageLimitExceededError(detail)
-            elif response.status_code in [403,432,433]:
+            elif response.status_code in [403, 432, 433]:
                 raise ForbiddenError(detail)
             elif response.status_code == 401:
                 raise InvalidAPIKeyError(detail)

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

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