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

Fix GitHub Models provider: use github prefix, fix migration, add Zod enum

Browse source
This commit is contained in:
CREDO23 2026-02-09 19:14:16 +02:00
parent 8bc4b255b4
commit a96d18ec31
5 changed files with 7 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

16
surfsense_backend/alembic/versions/96_add_github_models_to_litellmprovider_enum.py
View file

@ -15,20 +15,8 @@ depends_on: str | Sequence[str] | None = None
def upgrade() -> None: def upgrade() -> None:
op.execute( op.execute("COMMIT")
""" op.execute("ALTER TYPE litellmprovider ADD VALUE IF NOT EXISTS 'GITHUB_MODELS'")
DO $$
BEGIN
IF NOT EXISTS (
SELECT 1 FROM pg_enum
WHERE enumtypid = 'litellmprovider'::regtype
AND enumlabel = 'GITHUB_MODELS'
) THEN
ALTER TYPE litellmprovider ADD VALUE 'GITHUB_MODELS';
END IF;
END$$;
"""
)
def downgrade() -> None: def downgrade() -> None:

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

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

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

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

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

@ -119,7 +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", "GITHUB_MODELS": "github",
} }
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}"
@ -336,7 +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", "GITHUB_MODELS": "github",
} }
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()

1
surfsense_web/contracts/types/new-llm-config.types.ts
View file

@ -33,6 +33,7 @@ export const liteLLMProviderEnum = z.enum([
"DATABRICKS", "DATABRICKS",
"COMETAPI", "COMETAPI",
"HUGGINGFACE", "HUGGINGFACE",
"GITHUB_MODELS",
"CUSTOM", "CUSTOM",
]); ]);