tavily-python 0.7.12__tar.gz → 0.7.13__tar.gz

{tavily_python-0.7.12 → tavily_python-0.7.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tavily-python
-Version: 0.7.12
+Version: 0.7.13
 Summary: Python wrapper for the Tavily API
 Home-page: https://github.com/tavily-ai/tavily-python
 Author: Tavily AI
@@ -8,7 +8,7 @@ Author-email: support@tavily.com
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
-Requires-Python: >=3.6
+Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: requests

{tavily_python-0.7.12 → tavily_python-0.7.13}/setup.py

@@ -5,7 +5,7 @@ with open('README.md', 'r', encoding='utf-8') as f:
 
 setup(
     name='tavily-python',
-    version='0.7.12',
+    version='0.7.13',
     url='https://github.com/tavily-ai/tavily-python',
     author='Tavily AI',
     author_email='support@tavily.com',
@@ -19,5 +19,5 @@ setup(
         'License :: OSI Approved :: MIT License',
         'Operating System :: OS Independent',
     ],
-    python_requires='>=3.6',
+    python_requires='>=3.8',
 )

{tavily_python-0.7.12 → tavily_python-0.7.13}/tavily/async_tavily.py

@@ -66,7 +66,7 @@ class AsyncTavilyClient:
             include_answer: Union[bool, Literal["basic", "advanced"]] = None,
             include_raw_content: Union[bool, Literal["markdown", "text"]] = None,
             include_images: bool = None,
-            timeout: int = 60,
+            timeout: float = 60,
             country: str = None,
             auto_parameters: bool = None,
             include_favicon: bool = None,
@@ -141,7 +141,7 @@ class AsyncTavilyClient:
                      include_answer: Union[bool, Literal["basic", "advanced"]] = None,
                      include_raw_content: Union[bool, Literal["markdown", "text"]] = None,
                      include_images: bool = None,
-                     timeout: int = 60,
+                     timeout: float = 60,
                      country: str = None,
                      auto_parameters: bool = None,
                      include_favicon: bool = None,
@@ -183,7 +183,7 @@ class AsyncTavilyClient:
             include_images: bool = None,
             extract_depth: Literal["basic", "advanced"] = None,
             format: Literal["markdown", "text"] = None,
-            timeout: int = 30,
+            timeout: float = 30,
             include_favicon: bool = None,
             **kwargs
     ) -> dict:
@@ -205,8 +205,6 @@ class AsyncTavilyClient:
         if kwargs:
             data.update(kwargs)
 
-        timeout = min(timeout, 120)
-
         async with self._client_creator() as client:
             try:
                 response = await client.post("/extract", content=json.dumps(data), timeout=timeout)
@@ -239,7 +237,7 @@ class AsyncTavilyClient:
                       include_images: bool = None,
                       extract_depth: Literal["basic", "advanced"] = None,
                       format: Literal["markdown", "text"] = None,
-                      timeout: int = 30,
+                      timeout: float = 30,
                       include_favicon: bool = None,
                       **kwargs,  # Accept custom arguments
                       ) -> dict:
@@ -247,7 +245,6 @@ class AsyncTavilyClient:
         Combined extract method.
         include_favicon: If True, include the favicon in the extraction results.
         """
-        timeout = min(timeout, 120)
         response_dict = await self._extract(urls,
                                             include_images,
                                             extract_depth,
@@ -279,7 +276,7 @@ class AsyncTavilyClient:
                include_images: bool = None,
                extract_depth: Literal["basic", "advanced"] = None,
                format: Literal["markdown", "text"] = None,
-               timeout: int = 60,
+               timeout: float = 150,
                include_favicon: bool = None,
                **kwargs
                ) -> dict:
@@ -300,6 +297,7 @@ class AsyncTavilyClient:
             "include_images": include_images,
             "extract_depth": extract_depth,
             "format": format,
+            "timeout": timeout,
             "include_favicon": include_favicon,
         }
 
@@ -308,8 +306,6 @@ class AsyncTavilyClient:
 
         data = {k: v for k, v in data.items() if v is not None}
 
-        timeout = min(timeout, 120)
-
         async with self._client_creator() as client:
             try:
                 response = await client.post("/crawl", content=json.dumps(data), timeout=timeout)
@@ -350,7 +346,7 @@ class AsyncTavilyClient:
                     extract_depth: Literal["basic", "advanced"] = None,
                     include_images: bool = None,
                     format: Literal["markdown", "text"] = None,
-                    timeout: int = 60,
+                    timeout: float = 150,
                     include_favicon: bool = None,
                     **kwargs
                     ) -> dict:
@@ -358,7 +354,6 @@ class AsyncTavilyClient:
         Combined crawl method.
         
         """
-        timeout = min(timeout, 120)
         response_dict = await self._crawl(url,
                                     max_depth=max_depth,
                                     max_breadth=max_breadth,
@@ -390,7 +385,7 @@ class AsyncTavilyClient:
                exclude_domains: Sequence[str] = None,
                allow_external: bool = None,
                include_images: bool = None,
-               timeout: int = 60,
+               timeout: float = 150,
                **kwargs
                ) -> dict:
         """
@@ -408,6 +403,7 @@ class AsyncTavilyClient:
             "exclude_domains": exclude_domains,
             "allow_external": allow_external,
             "include_images": include_images,
+            "timeout": timeout,
         }
 
         if kwargs:
@@ -415,8 +411,6 @@ class AsyncTavilyClient:
 
         data = {k: v for k, v in data.items() if v is not None}
 
-        timeout = min(timeout, 120)
-
         async with self._client_creator() as client:
             try:
                 response = await client.post("/map", content=json.dumps(data), timeout=timeout)
@@ -455,14 +449,13 @@ class AsyncTavilyClient:
                     exclude_domains: Sequence[str] = None,
                     allow_external: bool = None,
                     include_images: bool = None,
-                    timeout: int = 60,
+                    timeout: float = 150,
                     **kwargs
                     ) -> dict:
         """
         Combined map method.
 
         """
-        timeout = min(timeout, 120)
         response_dict = await self._map(url,
                                     max_depth=max_depth,
                                     max_breadth=max_breadth,
@@ -488,7 +481,7 @@ class AsyncTavilyClient:
                                  include_domains: Sequence[str] = None,
                                  exclude_domains: Sequence[str] = None,
                                  max_tokens: int = 4000,
-                                 timeout: int = 60,
+                                 timeout: float = 60,
                                  country: str = None,
                                  include_favicon: bool = None,
                                  **kwargs,  # Accept custom arguments
@@ -529,7 +522,7 @@ class AsyncTavilyClient:
                          max_results: int = 5,
                          include_domains: Sequence[str] = None,
                          exclude_domains: Sequence[str] = None,
-                         timeout: int = 60,
+                         timeout: float = 60,
                          country: str = None,
                          include_favicon: bool = None,
                          **kwargs,  # Accept custom arguments
@@ -559,7 +552,7 @@ class AsyncTavilyClient:
                                query: str,
                                search_depth: Literal["basic", "advanced"] = "advanced",
                                max_results: int = 5,
-                               timeout: int = 60,
+                               timeout: float = 60,
                                country: str = None,
                                ) -> Sequence[dict]:
         """ Company information search method. Search depth is advanced by default to get the best answer. """

{tavily_python-0.7.12 → tavily_python-0.7.13}/tavily/errors.py

@@ -17,7 +17,7 @@ class InvalidAPIKeyError(Exception):
         super().__init__(message)
 
 class TimeoutError(Exception):
-    def __init__(self, timeout: int):
+    def __init__(self, timeout: float):
         super().__init__(f"Request timed out after {timeout} seconds.")
 
 class MissingAPIKeyError(Exception):

{tavily_python-0.7.12 → tavily_python-0.7.13}/tavily/tavily.py

@@ -49,7 +49,7 @@ class TavilyClient:
                 include_answer: Union[bool, Literal["basic", "advanced"]] = None,
                 include_raw_content: Union[bool, Literal["markdown", "text"]] = None,
                 include_images: bool = None,
-                timeout: int = 60,
+                timeout: float = 60,
                 country: str = None,
                 auto_parameters: bool = None,
                 include_favicon: bool = None,
@@ -126,7 +126,7 @@ class TavilyClient:
                include_answer: Union[bool, Literal["basic", "advanced"]] = None,
                include_raw_content: Union[bool, Literal["markdown", "text"]] = None,
                include_images: bool = None,
-               timeout: int = 60,
+               timeout: float = 60,
                country: str = None,
                auto_parameters: bool = None,
                include_favicon: bool = None,
@@ -135,7 +135,7 @@ class TavilyClient:
         """
         Combined search method.
         """
-        timeout = min(timeout, 120)
+
         response_dict = self._search(query,
                                      search_depth=search_depth,
                                      topic=topic,
@@ -167,7 +167,7 @@ class TavilyClient:
                  include_images: bool = None,
                  extract_depth: Literal["basic", "advanced"] = None,
                  format: Literal["markdown", "text"] = None,
-                 timeout: int = 30,
+                 timeout: float = 30,
                  include_favicon: bool = None,
                  **kwargs
                  ) -> dict:
@@ -188,8 +188,6 @@ class TavilyClient:
         if kwargs:
             data.update(kwargs)
 
-        timeout = min(timeout, 120)
-
         try:
             response = requests.post(self.base_url + "/extract", data=json.dumps(data), headers=self.headers, timeout=timeout, proxies=self.proxies)
         except requests.exceptions.Timeout:
@@ -220,14 +218,13 @@ class TavilyClient:
                 include_images: bool = None,
                 extract_depth: Literal["basic", "advanced"] = None,
                 format: Literal["markdown", "text"] = None,
-                timeout: int = 30,
+                timeout: float = 30,
                 include_favicon: bool = None,
                 **kwargs,  # Accept custom arguments
                 ) -> dict:
         """
         Combined extract method.
         """
-        timeout = min(timeout, 120)
         response_dict = self._extract(urls,
                                       include_images,
                                       extract_depth,
@@ -258,7 +255,7 @@ class TavilyClient:
             include_images: bool = None,
             extract_depth: Literal["basic", "advanced"] = None,
             format: Literal["markdown", "text"] = None,
-            timeout: int = 60,
+            timeout: float = 150,
             include_favicon: bool = None,
             **kwargs
             ) -> dict:
@@ -280,6 +277,7 @@ class TavilyClient:
             "include_images": include_images,
             "extract_depth": extract_depth,
             "format": format,
+            "timeout": timeout,
             "include_favicon": include_favicon,
         }
 
@@ -287,8 +285,6 @@ class TavilyClient:
             data.update(kwargs)
         
         data = {k: v for k, v in data.items() if v is not None}
-    
-        timeout = min(timeout, 120)
 
         try:
             response = requests.post(
@@ -330,7 +326,7 @@ class TavilyClient:
               include_images: bool = None,
               extract_depth: Literal["basic", "advanced"] = None,
               format: Literal["markdown", "text"] = None,
-              timeout: int = 60,
+              timeout: float = 150,
               include_favicon: bool = None,
               **kwargs
               ) -> dict:
@@ -338,7 +334,6 @@ class TavilyClient:
         Combined crawl method.
         include_favicon: If True, include the favicon in the crawl results.
         """
-        timeout = min(timeout, 120)
         response_dict = self._crawl(url,
                                     max_depth=max_depth,
                                     max_breadth=max_breadth,
@@ -370,7 +365,7 @@ class TavilyClient:
             exclude_domains: Sequence[str] = None,
             allow_external: bool = None,
             include_images: bool = None,
-            timeout: int = 60,
+            timeout: float = 150,
             **kwargs
             ) -> dict:
         """
@@ -388,6 +383,7 @@ class TavilyClient:
             "exclude_domains": exclude_domains,
             "allow_external": allow_external,
             "include_images": include_images,
+            "timeout": timeout,
         }
 
         if kwargs:
@@ -395,8 +391,6 @@ class TavilyClient:
         
         data = {k: v for k, v in data.items() if v is not None}
 
-        timeout = min(timeout, 120)
-
         try:
             response = requests.post(
                 self.base_url + "/map", data=json.dumps(data), headers=self.headers, timeout=timeout, proxies=self.proxies)
@@ -435,14 +429,13 @@ class TavilyClient:
               exclude_domains: Sequence[str] = None,
               allow_external: bool = None,
               include_images: bool = None,
-              timeout: int = 60,
+              timeout: float = 150,
               **kwargs
               ) -> dict:
         """
         Combined map method.
         
         """
-        timeout = min(timeout, 120)
         response_dict = self._map(url,
                                     max_depth=max_depth,
                                     max_breadth=max_breadth,
@@ -468,7 +461,7 @@ class TavilyClient:
                            include_domains: Sequence[str] = None,
                            exclude_domains: Sequence[str] = None,
                            max_tokens: int = 4000,
-                           timeout: int = 60,
+                           timeout: float = 60,
                            country: str = None,
                            include_favicon: bool = None,
                            **kwargs,  # Accept custom arguments
@@ -481,7 +474,7 @@ class TavilyClient:
 
         Returns a string of JSON containing the search context up to context limit.
         """
-        timeout = min(timeout, 120)
+
         response_dict = self._search(query,
                                      search_depth=search_depth,
                                      topic=topic,
@@ -510,7 +503,7 @@ class TavilyClient:
                    max_results: int = 5,
                    include_domains: Sequence[str] = None,
                    exclude_domains: Sequence[str] = None,
-                   timeout: int = 60,
+                   timeout: float = 60,
                    country: str = None,
                    include_favicon: bool = None,
                    **kwargs,  # Accept custom arguments
@@ -518,7 +511,6 @@ class TavilyClient:
         """
         Q&A search method. Search depth is advanced by default to get the best answer.
         """
-        timeout = min(timeout, 120)
         response_dict = self._search(query,
                                      search_depth=search_depth,
                                      topic=topic,
@@ -541,11 +533,10 @@ class TavilyClient:
                          search_depth: Literal["basic",
                                                "advanced"] = "advanced",
                          max_results: int = 5,
-                         timeout: int = 60,
+                         timeout: float = 60,
                          country: str = None,
                          ) -> Sequence[dict]:
         """ Company information search method. Search depth is advanced by default to get the best answer. """
-        timeout = min(timeout, 120)
         def _perform_search(topic):
             return self._search(query,
                                 search_depth=search_depth,

{tavily_python-0.7.12 → tavily_python-0.7.13}/tavily_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tavily-python
-Version: 0.7.12
+Version: 0.7.13
 Summary: Python wrapper for the Tavily API
 Home-page: https://github.com/tavily-ai/tavily-python
 Author: Tavily AI
@@ -8,7 +8,7 @@ Author-email: support@tavily.com
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
-Requires-Python: >=3.6
+Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: requests