Merge branch 'fix/llm-debug' into test/release-2.2.0

Security-Tools-Alliance · Feb 11, 2025 · 1646edc · 1646edc
2 parents 23a9fb5 + d04410f
commit 1646edc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/web/reNgine/llm/config.py b/web/reNgine/llm/config.py
@@ -112,20 +112,18 @@
             'temperature': 0.7,
         },
         'ollama': {
+            'url': OLLAMA_INSTANCE,
             'default_model': 'llama2',
             'models': [
                 'llama2',
                 'mistral',
                 'codellama',
                 'gemma'
             ],
-            'timeout': 30,
+            'timeout': 120,
             'max_retries': 3,
         }
     },
-    'ollama_url': OLLAMA_INSTANCE,
-    'timeout': 30,
-    'max_retries': 3,
     'prompts': {
         'vulnerability': {
             'context': VULNERABILITY_CONTEXT,

diff --git a/web/reNgine/llm/llm.py b/web/reNgine/llm/llm.py
@@ -38,7 +38,7 @@ def _setup_ollama(self) -> None:
         """Setup Ollama client with configuration"""
         ollama_config = self.config['providers']['ollama']
         self.ollama = Ollama(
-            base_url=self.config['ollama_url'],
+            base_url=ollama_config['url'],
             model=self.model_name,
             timeout=ollama_config['timeout']
         )