apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-21 18:55:16 +02:00
Code Issues Projects Releases Packages Wiki Activity

Add GITHUB_MODELS to provider maps

Browse source
This commit is contained in:
CREDO23 2026-02-09 17:34:25 +02:00
parent fc946ee1e5
commit a9d6c41964
3 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
surfsense_backend/app/agents/new_chat/llm_config.py
View file

@ -45,6 +45,7 @@ PROVIDER_MAP = {
"ALIBABA_QWEN": "openai", "ALIBABA_QWEN": "openai",
"MOONSHOT": "openai", "MOONSHOT": "openai",
"ZHIPU": "openai", "ZHIPU": "openai",
"GITHUB_MODELS": "openai",
"REPLICATE": "replicate", "REPLICATE": "replicate",
"PERPLEXITY": "perplexity", "PERPLEXITY": "perplexity",
"ANYSCALE": "anyscale", "ANYSCALE": "anyscale",

1
surfsense_backend/app/services/llm_router_service.py
View file

@ -56,6 +56,7 @@ PROVIDER_MAP = {
"ALIBABA_QWEN": "openai", "ALIBABA_QWEN": "openai",
"MOONSHOT": "openai", "MOONSHOT": "openai",
"ZHIPU": "openai", "ZHIPU": "openai",
"GITHUB_MODELS": "openai",
"HUGGINGFACE": "huggingface", "HUGGINGFACE": "huggingface",
"CUSTOM": "custom", "CUSTOM": "custom",
} }

2
surfsense_backend/app/services/llm_service.py
View file

@ -119,6 +119,7 @@ async def validate_llm_config(
"ALIBABA_QWEN": "openai", "ALIBABA_QWEN": "openai",
"MOONSHOT": "openai", "MOONSHOT": "openai",
"ZHIPU": "openai", # GLM needs special handling "ZHIPU": "openai", # GLM needs special handling
"GITHUB_MODELS": "openai",
} }
provider_prefix = provider_map.get(provider, provider.lower()) provider_prefix = provider_map.get(provider, provider.lower())
model_string = f"{provider_prefix}/{model_name}" model_string = f"{provider_prefix}/{model_name}"
@ -335,6 +336,7 @@ async def get_search_space_llm_instance(
"ALIBABA_QWEN": "openai", "ALIBABA_QWEN": "openai",
"MOONSHOT": "openai", "MOONSHOT": "openai",
"ZHIPU": "openai", "ZHIPU": "openai",
"GITHUB_MODELS": "openai",
} }
provider_prefix = provider_map.get( provider_prefix = provider_map.get(
llm_config.provider.value, llm_config.provider.value.lower() llm_config.provider.value, llm_config.provider.value.lower()