fix linting errors for module text llm

ls1intum · Aug 10, 2024 · 24b951c · 24b951c
1 parent 14dcc7e
commit 24b951c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/modules/text/module_text_llm/module_text_llm/helpers/models/openai.py b/modules/text/module_text_llm/module_text_llm/helpers/models/openai.py
@@ -34,9 +34,10 @@ def _get_azure_openai_deployments() -> List[str]:
             "api-key": os.environ["AZURE_OPENAI_API_KEY"]
         }
 
-        models_response = requests.get(f"{base_url}/models?api-version=2023-03-15-preview", headers=headers)
+        models_response = requests.get(f"{base_url}/models?api-version=2023-03-15-preview", headers=headers, timeout=30)
         models_data = models_response.json()["data"]
-        deployments_response = requests.get(f"{base_url}/deployments?api-version=2023-03-15-preview", headers=headers)
+        deployments_response = requests.get(f"{base_url}/deployments?api-version=2023-03-15-preview", headers=headers,
+                                            timeout=30)
         deployments_data = deployments_response.json()["data"]
 
         # Check if deployment["model"] is a substring of model["id"], i.e. "gpt-4o" is substring "gpt-4o-2024-05-13"