Update src/llm_client.py

src/llm_client.py

@@ -14,7 +14,7 @@ def ask_granite(query, context):
     payload = {
         "text": f"USER QUESTION: {query}\n\nDOCUMENT CONTEXT:\n{context[:6000]}",
         "persona": "You are a question answering Subject Matter Expert. Answer strictly based on the provided context.",
-        "model": "granite4:latest", # Or gemma3:latest
+        "model": "gemma3:latest", # granite4:latest or gemma3:latest
         "max_tokens": 5000
     }
     
@@ -24,7 +24,7 @@ def ask_granite(query, context):
     }
     
     try:
-        response = requests.post(api_url, json=payload, headers=headers, timeout=120)
+        response = requests.post(api_url, json=payload, headers=headers, timeout=600)
         if response.status_code == 200:
             return response.json().get("response", "Error: Empty response.")
         else: