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

Add timeout to requests calls (#1)

Browse source 
Co-authored-by: pixeebot[bot] <104101892+pixeebot[bot]@users.noreply.github.com>
This commit is contained in:
pixeeai 2024-09-29 07:24:20 -04:00
parent 43744455d7
commit fdadfb6afb
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/providers/impls/meta_reference/agents/tools/builtin.py
View file

@ -117,7 +117,7 @@ class BingSearch:
"q": query, "q": query,
} }
response = requests.get(url=url, params=params, headers=headers) response = requests.get(url=url, params=params, headers=headers, timeout=60)
response.raise_for_status() response.raise_for_status()
clean = self._clean_response(response.json()) clean = self._clean_response(response.json())
return json.dumps(clean) return json.dumps(clean)
@ -156,7 +156,7 @@ class BraveSearch:
"Accept": "application/json", "Accept": "application/json",
} }
payload = {"q": query} payload = {"q": query}
response = requests.get(url=url, params=payload, headers=headers) response = requests.get(url=url, params=payload, headers=headers, timeout=60)
return json.dumps(self._clean_brave_response(response.json())) return json.dumps(self._clean_brave_response(response.json()))
def _clean_brave_response(self, search_response, top_k=3): def _clean_brave_response(self, search_response, top_k=3):
@ -275,7 +275,7 @@ class WolframAlphaTool(SingleMessageBuiltinTool):
response = requests.get( response = requests.get(
self.url, self.url,
params=params, params=params,
) timeout=60)
return json.dumps(self._clean_wolfram_alpha_response(response.json())) return json.dumps(self._clean_wolfram_alpha_response(response.json()))