mirror of
https://github.com/trycua/computer.git
synced 2026-05-02 05:01:22 -05:00
Restore gemma
This commit is contained in:
@@ -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
|
||||
"""
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user