Changed default model to gemma2:9b

antoninoLorenzo · Aug 8, 2024 · fda14a4 · fda14a4
1 parent f9f3585
commit fda14a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/api.py b/src/api.py
@@ -44,7 +44,7 @@
 # --- Get AI-OPS Settings
 class AgentSettings(BaseSettings):
     """Setup for AI Agent"""
-    MODEL: str = os.environ.get('MODEL', 'gemma:7b')
+    MODEL: str = os.environ.get('MODEL', 'gemma2:9b')
     ENDPOINT: str = os.environ.get('ENDPOINT', 'http://localhost:11434')
     PROVIDER: str = os.environ.get('PROVIDER', 'ollama')
     PROVIDER_KEY: str = os.environ.get('PROVIDER_KEY', '')