From fdadfb6afb211be6443fafa7f4c22489107c1a71 Mon Sep 17 00:00:00 2001 From: pixeeai Date: Sun, 29 Sep 2024 07:24:20 -0400 Subject: [PATCH] Add timeout to `requests` calls (#1) Co-authored-by: pixeebot[bot] <104101892+pixeebot[bot]@users.noreply.github.com> --- .../providers/impls/meta_reference/agents/tools/builtin.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/llama_stack/providers/impls/meta_reference/agents/tools/builtin.py b/llama_stack/providers/impls/meta_reference/agents/tools/builtin.py index 4c9cdfcd2..1fa9693d2 100644 --- a/llama_stack/providers/impls/meta_reference/agents/tools/builtin.py +++ b/llama_stack/providers/impls/meta_reference/agents/tools/builtin.py @@ -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()))