Update app.py

app.py

@@ -110,7 +110,8 @@ def call_ollama_api(model, prompt):
                     "content": prompt
                 }
             ],
-            timeout=180
+            timeout=config.API_TIMEOUT,
+            max_tokens=config.MAX_TOKENS
         )
         logger.info(f"Received response for model: {model}")
         
@@ -150,10 +151,10 @@ def call_ollama_api(model, prompt):
         ]
         
     except requests.exceptions.Timeout:
-        logger.error(f"Timeout error after 180 seconds for model: {model}")
+        logger.error(f"Timeout error after {config.API_TIMEOUT} seconds for model: {model}")
         return [
             {"role": "user", "content": prompt},
-            {"role": "assistant", "content": "Error: Model response timed out after 180 seconds"}
+            {"role": "assistant", "content": f"Error: Model response timed out after {config.API_TIMEOUT} seconds"}
         ]
     except openai.BadRequestError as e:
         error_msg = str(e)