diff --git a/src/ai_generators/ollama_md_generator.py b/src/ai_generators/ollama_md_generator.py index 86b4156..a5ca090 100644 --- a/src/ai_generators/ollama_md_generator.py +++ b/src/ai_generators/ollama_md_generator.py @@ -140,8 +140,8 @@ class OllamaGenerator: ) return response["message"]["content"] - # Retry mechanism with 30-minute timeout (same as the original) - timeout_seconds = 30 * 60 + # Retry mechanism with 60-minute timeout (bumped for large context models) + timeout_seconds = 60 * 60 max_retries = 3 for attempt in range(max_retries): diff --git a/src/ai_generators/tools/ollama_web_search_tool.py b/src/ai_generators/tools/ollama_web_search_tool.py index 55c27f9..dc1fb2b 100644 --- a/src/ai_generators/tools/ollama_web_search_tool.py +++ b/src/ai_generators/tools/ollama_web_search_tool.py @@ -102,7 +102,7 @@ class OllamaWebSearchTool(BaseTool): "Authorization": f"Bearer {api_key}", "Content-Type": "application/json", }, - timeout=30.0, + timeout=60.0, ) # Raise for HTTP errors so we can catch them with specific messages