From f916007788479cf1c63a590a56719d420e395303 Mon Sep 17 00:00:00 2001 From: anigasan Date: Sun, 6 Jul 2025 12:52:47 -0700 Subject: [PATCH] commits --- .../remote/tool_runtime/tavily_search/config.py | 6 +++--- .../remote/tool_runtime/tavily_search/tavily_search.py | 9 ++++----- 2 files changed, 7 insertions(+), 8 deletions(-) diff --git a/llama_stack/providers/remote/tool_runtime/tavily_search/config.py b/llama_stack/providers/remote/tool_runtime/tavily_search/config.py index 30d29d34b..ae7b41d50 100644 --- a/llama_stack/providers/remote/tool_runtime/tavily_search/config.py +++ b/llama_stack/providers/remote/tool_runtime/tavily_search/config.py @@ -20,11 +20,11 @@ class TavilySearchToolConfig(BaseModel): ) timeout: float = Field( default=30.0, - description="HTTP request timeout in seconds", + description="HTTP request timeout for the API", ) connect_timeout: float = Field( default=10.0, - description="HTTP connection timeout in seconds", + description="HTTP connection timeout in seconds for the API", ) @classmethod @@ -32,6 +32,6 @@ class TavilySearchToolConfig(BaseModel): return { "api_key": "${env.TAVILY_SEARCH_API_KEY:=}", "max_results": 3, - "timeout": 30.0, + "timeout:" 30.0, "connect_timeout": 10.0, } diff --git a/llama_stack/providers/remote/tool_runtime/tavily_search/tavily_search.py b/llama_stack/providers/remote/tool_runtime/tavily_search/tavily_search.py index e8ca36cf4..cdaf78d4a 100644 --- a/llama_stack/providers/remote/tool_runtime/tavily_search/tavily_search.py +++ b/llama_stack/providers/remote/tool_runtime/tavily_search/tavily_search.py @@ -69,11 +69,10 @@ class TavilySearchToolRuntimeImpl(ToolGroupsProtocolPrivate, ToolRuntime, NeedsR async def invoke_tool(self, tool_name: str, kwargs: dict[str, Any]) -> ToolInvocationResult: api_key = self._get_api_key() - - # Configure timeout for external API calls using config values - timeout = httpx.Timeout(timeout=self.config.timeout, connect=self.config.connect_timeout) - - async with httpx.AsyncClient(timeout=timeout) as client: + + timeout = https.Timeout(timeout=self.config.timeout, connect=self.config.connect_timeout) + + async with httpx.AsyncClient() as client: response = await client.post( "https://api.tavily.com/search", json={"api_key": api_key, "query": kwargs["query"]},