Restore gemma

This commit is contained in:
f-trycua
2025-04-04 15:48:29 -07:00
parent 0458aaa1e7
commit 3ecdf2bbef
4 changed files with 4 additions and 4 deletions
+1 -1
View File
@@ -6,7 +6,7 @@ from ..providers.omni.types import LLMProvider
DEFAULT_MODELS = {
LLMProvider.OPENAI: "gpt-4o",
LLMProvider.ANTHROPIC: "claude-3-7-sonnet-20250219",
LLMProvider.OLLAMA: "qwen2.5:7b",
LLMProvider.OLLAMA: "gemma3:4b-it-q4_K_M",
}
# Map providers to their environment variable names
@@ -20,7 +20,7 @@ class OllamaClient(BaseOmniClient):
Args:
api_key: Not used
model: Ollama model name (e.g. "qwen2.5:7b")
model: Ollama model name (e.g. "gemma3:4b-it-q4_K_M")
max_retries: Maximum number of retries for API calls
retry_delay: Base delay between retries in seconds
"""
+1 -1
View File
@@ -36,7 +36,7 @@ Model = LLM
PROVIDER_TO_DEFAULT_MODEL: Dict[LLMProvider, str] = {
LLMProvider.ANTHROPIC: "claude-3-7-sonnet-20250219",
LLMProvider.OPENAI: "gpt-4o",
LLMProvider.OLLAMA: "qwen2.5:7b",
LLMProvider.OLLAMA: "gemma3:4b-it-q4_K_M",
}
# Environment variable names for each provider