diff --git a/lib/crewai-tools/src/crewai_tools/tools/tavily_extractor_tool/tavily_extractor_tool.py b/lib/crewai-tools/src/crewai_tools/tools/tavily_extractor_tool/tavily_extractor_tool.py index a89689e0aa..57f3f89aaf 100644 --- a/lib/crewai-tools/src/crewai_tools/tools/tavily_extractor_tool/tavily_extractor_tool.py +++ b/lib/crewai-tools/src/crewai_tools/tools/tavily_extractor_tool/tavily_extractor_tool.py @@ -87,9 +87,11 @@ def __init__(self, **kwargs: Any): """ super().__init__(**kwargs) if TAVILY_AVAILABLE: - self.client = TavilyClient(api_key=self.api_key, proxies=self.proxies) + self.client = TavilyClient( + api_key=self.api_key, proxies=self.proxies, client_name="crewai" + ) self.async_client = AsyncTavilyClient( - api_key=self.api_key, proxies=self.proxies + api_key=self.api_key, proxies=self.proxies, client_name="crewai" ) else: try: diff --git a/lib/crewai-tools/src/crewai_tools/tools/tavily_get_research_tool/tavily_get_research_tool.py b/lib/crewai-tools/src/crewai_tools/tools/tavily_get_research_tool/tavily_get_research_tool.py index c3d6787b39..1b91391377 100644 --- a/lib/crewai-tools/src/crewai_tools/tools/tavily_get_research_tool/tavily_get_research_tool.py +++ b/lib/crewai-tools/src/crewai_tools/tools/tavily_get_research_tool/tavily_get_research_tool.py @@ -54,8 +54,10 @@ def __init__(self, **kwargs: Any): super().__init__(**kwargs) if TAVILY_AVAILABLE: api_key = os.getenv("TAVILY_API_KEY") - self._client = TavilyClient(api_key=api_key) - self._async_client = AsyncTavilyClient(api_key=api_key) + self._client = TavilyClient(api_key=api_key, client_name="crewai") + self._async_client = AsyncTavilyClient( + api_key=api_key, client_name="crewai" + ) else: try: import subprocess diff --git a/lib/crewai-tools/src/crewai_tools/tools/tavily_research_tool/tavily_research_tool.py b/lib/crewai-tools/src/crewai_tools/tools/tavily_research_tool/tavily_research_tool.py index 084fefdf1c..b4bb8e0afc 100644 --- a/lib/crewai-tools/src/crewai_tools/tools/tavily_research_tool/tavily_research_tool.py +++ b/lib/crewai-tools/src/crewai_tools/tools/tavily_research_tool/tavily_research_tool.py @@ -90,8 +90,10 @@ def __init__(self, **kwargs: Any): super().__init__(**kwargs) if TAVILY_AVAILABLE: api_key = os.getenv("TAVILY_API_KEY") - self._client = TavilyClient(api_key=api_key) - self._async_client = AsyncTavilyClient(api_key=api_key) + self._client = TavilyClient(api_key=api_key, client_name="crewai") + self._async_client = AsyncTavilyClient( + api_key=api_key, client_name="crewai" + ) else: try: import subprocess diff --git a/lib/crewai-tools/src/crewai_tools/tools/tavily_search_tool/tavily_search_tool.py b/lib/crewai-tools/src/crewai_tools/tools/tavily_search_tool/tavily_search_tool.py index ef8d7412d4..1b5ebfe79b 100644 --- a/lib/crewai-tools/src/crewai_tools/tools/tavily_search_tool/tavily_search_tool.py +++ b/lib/crewai-tools/src/crewai_tools/tools/tavily_search_tool/tavily_search_tool.py @@ -115,9 +115,11 @@ class TavilySearchTool(BaseTool): def __init__(self, **kwargs: Any): super().__init__(**kwargs) if TAVILY_AVAILABLE: - self.client = TavilyClient(api_key=self.api_key, proxies=self.proxies) + self.client = TavilyClient( + api_key=self.api_key, proxies=self.proxies, client_name="crewai" + ) self.async_client = AsyncTavilyClient( - api_key=self.api_key, proxies=self.proxies + api_key=self.api_key, proxies=self.proxies, client_name="crewai" ) else: try: