apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-06-28 21:49:40 +02:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #802 from CREDO23/sur-161-feat-add-github-models-as-llm-provider

Browse source 
[Improvements] Add GitHub models as llm provider
This commit is contained in:
Rohan Verma 2026-02-09 15:07:40 -08:00 • committed by GitHub
commit dcd1497630
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 108 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

23
surfsense_backend/alembic/versions/96_add_github_models_to_litellmprovider_enum.py Normal file
View file

@ -0,0 +1,23 @@
"""Add GITHUB_MODELS to LiteLLMProvider enum
Revision ID: 96
Revises: 95
"""
from collections.abc import Sequence
from alembic import op
revision: str = "96"
down_revision: str | None = "95"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None
def upgrade() -> None:
op.execute("COMMIT")
op.execute("ALTER TYPE litellmprovider ADD VALUE IF NOT EXISTS 'GITHUB_MODELS'")
def downgrade() -> None:
pass

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

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

1
surfsense_backend/app/db.py
View file

@ -211,6 +211,7 @@ class LiteLLMProvider(str, Enum):
DATABRICKS = "DATABRICKS"
COMETAPI = "COMETAPI"
HUGGINGFACE = "HUGGINGFACE"
GITHUB_MODELS = "GITHUB_MODELS"
CUSTOM = "CUSTOM"

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

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

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

@ -119,6 +119,7 @@ async def validate_llm_config(
"ALIBABA_QWEN": "openai",
"MOONSHOT": "openai",
"ZHIPU": "openai", # GLM needs special handling
"GITHUB_MODELS": "github",
}
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": "github",
}
provider_prefix = provider_map.get(
llm_config.provider.value, llm_config.provider.value.lower()