From ed6003cf459d56c438311c45e78184989d4fa3d0 Mon Sep 17 00:00:00 2001 From: anigasan Date: Mon, 7 Jul 2025 12:39:33 -0700 Subject: [PATCH] made changes to tavily_search, simplified config logic --- .../remote/tool_runtime/tavily_search/tavily_search.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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 d89c9b101..5158f4d40 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 @@ -70,9 +70,9 @@ class TavilySearchToolRuntimeImpl(ToolGroupsProtocolPrivate, ToolRuntime, NeedsR async def invoke_tool(self, tool_name: str, kwargs: dict[str, Any]) -> ToolInvocationResult: api_key = self._get_api_key() - httpx.Timeout(timeout=self.config.timeout, connect=self.config.connect_timeout) + timeout = httpx.Timeout(timeout=self.config.timeout) - async with httpx.AsyncClient() as client: + async with httpx.AsyncClient(timeout=timeout) as client: response = await client.post( "https://api.tavily.com/search", json={"api_key": api_key, "query": kwargs["query"]},