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

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,
}
response = requests.get(url=url, params=params, headers=headers)
response = requests.get(url=url, params=params, headers=headers, timeout=60)
response.raise_for_status()
clean = self._clean_response(response.json())
return json.dumps(clean)
@ -156,7 +156,7 @@ class BraveSearch:
"Accept": "application/json",
}
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()))
def _clean_brave_response(self, search_response, top_k=3):
@ -275,7 +275,7 @@ class WolframAlphaTool(SingleMessageBuiltinTool):
response = requests.get(
self.url,
params=params,
)
timeout=60)
return json.dumps(self._clean_wolfram_alpha_response(response.json()))