diff --git a/surfsense_backend/app/agents/new_chat/llm_config.py b/surfsense_backend/app/agents/new_chat/llm_config.py index 12b389a2d..b7305cb4b 100644 --- a/surfsense_backend/app/agents/new_chat/llm_config.py +++ b/surfsense_backend/app/agents/new_chat/llm_config.py @@ -45,6 +45,7 @@ PROVIDER_MAP = { "ALIBABA_QWEN": "openai", "MOONSHOT": "openai", "ZHIPU": "openai", + "GITHUB_MODELS": "openai", "REPLICATE": "replicate", "PERPLEXITY": "perplexity", "ANYSCALE": "anyscale", diff --git a/surfsense_backend/app/services/llm_router_service.py b/surfsense_backend/app/services/llm_router_service.py index 3a19f0b8e..0e7d0b1a1 100644 --- a/surfsense_backend/app/services/llm_router_service.py +++ b/surfsense_backend/app/services/llm_router_service.py @@ -56,6 +56,7 @@ PROVIDER_MAP = { "ALIBABA_QWEN": "openai", "MOONSHOT": "openai", "ZHIPU": "openai", + "GITHUB_MODELS": "openai", "HUGGINGFACE": "huggingface", "CUSTOM": "custom", } diff --git a/surfsense_backend/app/services/llm_service.py b/surfsense_backend/app/services/llm_service.py index e21259990..8bd4da57b 100644 --- a/surfsense_backend/app/services/llm_service.py +++ b/surfsense_backend/app/services/llm_service.py @@ -119,6 +119,7 @@ async def validate_llm_config( "ALIBABA_QWEN": "openai", "MOONSHOT": "openai", "ZHIPU": "openai", # GLM needs special handling + "GITHUB_MODELS": "openai", } provider_prefix = provider_map.get(provider, provider.lower()) model_string = f"{provider_prefix}/{model_name}" @@ -335,6 +336,7 @@ async def get_search_space_llm_instance( "ALIBABA_QWEN": "openai", "MOONSHOT": "openai", "ZHIPU": "openai", + "GITHUB_MODELS": "openai", } provider_prefix = provider_map.get( llm_config.provider.value, llm_config.provider.value.lower()