phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 07:14:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

made changes to tavily_search, simplified config logic

Browse source
This commit is contained in:
anigasan 2025-07-07 12:39:33 -07:00
parent b662924057
commit ed6003cf45
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/remote/tool_runtime/tavily_search/tavily_search.py
View file

@ -70,9 +70,9 @@ class TavilySearchToolRuntimeImpl(ToolGroupsProtocolPrivate, ToolRuntime, NeedsR
async def invoke_tool(self, tool_name: str, kwargs: dict[str, Any]) -> ToolInvocationResult: async def invoke_tool(self, tool_name: str, kwargs: dict[str, Any]) -> ToolInvocationResult:
api_key = self._get_api_key() 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( response = await client.post(
"https://api.tavily.com/search", "https://api.tavily.com/search",
json={"api_key": api_key, "query": kwargs["query"]}, json={"api_key": api_key, "query": kwargs["query"]},