diff --git a/modules/llm.py b/modules/llm.py
index 80d0ea9..5810039 100644
--- a/modules/llm.py
+++ b/modules/llm.py
@@ -178,16 +178,21 @@ def get_google_context(input, num_results):
def send_ollama_query(llmQuery):
# Send the query to the Ollama API and return the response
- result = requests.post(ollamaAPI, data=json.dumps(llmQuery))
- if result.status_code == 200:
- result_json = result.json()
- result = result_json.get("response", "")
- # deepseek has added tags to the response
- if "" in result:
- result = result.split("")[1]
- else:
- raise Exception(f"HTTP Error: {result.status_code}")
- return result
+ try:
+ result = requests.post(ollamaAPI, data=json.dumps(llmQuery), timeout=5)
+ if result.status_code == 200:
+ result_json = result.json()
+ result = result_json.get("response", "")
+ # deepseek has added tags to the response
+ if "" in result:
+ result = result.split("")[1]
+ else:
+ logger.warning(f"System: LLM Query: Ollama API returned status code {result.status_code}")
+ return f"⛔️ Request Error"
+ return result
+ except requests.exceptions.RequestException as e:
+ logger.warning(f"System: LLM Query: Ollama API request failed: {e}")
+ return f"⛔️ Request Error"
def send_ollama_tooling_query(prompt, functions, model=None, max_tokens=450):
"""