diff --git a/surfsense_backend/app/agents/new_chat/llm_config.py b/surfsense_backend/app/agents/new_chat/llm_config.py index f9f92959c..12b389a2d 100644 --- a/surfsense_backend/app/agents/new_chat/llm_config.py +++ b/surfsense_backend/app/agents/new_chat/llm_config.py @@ -32,7 +32,7 @@ PROVIDER_MAP = { "GROQ": "groq", "COHERE": "cohere", "GOOGLE": "gemini", - "OLLAMA": "ollama", + "OLLAMA": "ollama_chat", "MISTRAL": "mistral", "AZURE_OPENAI": "azure", "OPENROUTER": "openrouter", diff --git a/surfsense_backend/app/services/llm_service.py b/surfsense_backend/app/services/llm_service.py index f0198d91f..e21259990 100644 --- a/surfsense_backend/app/services/llm_service.py +++ b/surfsense_backend/app/services/llm_service.py @@ -94,7 +94,7 @@ async def validate_llm_config( "GROQ": "groq", "COHERE": "cohere", "GOOGLE": "gemini", - "OLLAMA": "ollama", + "OLLAMA": "ollama_chat", "MISTRAL": "mistral", "AZURE_OPENAI": "azure", "OPENROUTER": "openrouter", @@ -241,7 +241,7 @@ async def get_search_space_llm_instance( "GROQ": "groq", "COHERE": "cohere", "GOOGLE": "gemini", - "OLLAMA": "ollama", + "OLLAMA": "ollama_chat", "MISTRAL": "mistral", "AZURE_OPENAI": "azure", "OPENROUTER": "openrouter", @@ -311,7 +311,7 @@ async def get_search_space_llm_instance( "GROQ": "groq", "COHERE": "cohere", "GOOGLE": "gemini", - "OLLAMA": "ollama", + "OLLAMA": "ollama_chat", "MISTRAL": "mistral", "AZURE_OPENAI": "azure", "OPENROUTER": "openrouter",