mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-04-28 10:26:33 +02:00
633ea3ac0f
- RBAC soon?? - Updated various services and routes to handle search space-specific LLM preferences. - Modified frontend components to pass search space ID for LLM configuration management. - Removed onboarding page and settings page as part of the refactor.
62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
from datetime import datetime
|
|
from typing import Any
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
from app.db import LiteLLMProvider
|
|
|
|
from .base import IDModel, TimestampModel
|
|
|
|
|
|
class LLMConfigBase(BaseModel):
|
|
name: str = Field(
|
|
..., max_length=100, description="User-friendly name for the LLM configuration"
|
|
)
|
|
provider: LiteLLMProvider = Field(..., description="LiteLLM provider type")
|
|
custom_provider: str | None = Field(
|
|
None, max_length=100, description="Custom provider name when provider is CUSTOM"
|
|
)
|
|
model_name: str = Field(
|
|
..., max_length=100, description="Model name without provider prefix"
|
|
)
|
|
api_key: str = Field(..., description="API key for the provider")
|
|
api_base: str | None = Field(
|
|
None, max_length=500, description="Optional API base URL"
|
|
)
|
|
litellm_params: dict[str, Any] | None = Field(
|
|
default=None, description="Additional LiteLLM parameters"
|
|
)
|
|
|
|
|
|
class LLMConfigCreate(LLMConfigBase):
|
|
search_space_id: int = Field(
|
|
..., description="Search space ID to associate the LLM config with"
|
|
)
|
|
|
|
|
|
class LLMConfigUpdate(BaseModel):
|
|
name: str | None = Field(
|
|
None, max_length=100, description="User-friendly name for the LLM configuration"
|
|
)
|
|
provider: LiteLLMProvider | None = Field(None, description="LiteLLM provider type")
|
|
custom_provider: str | None = Field(
|
|
None, max_length=100, description="Custom provider name when provider is CUSTOM"
|
|
)
|
|
model_name: str | None = Field(
|
|
None, max_length=100, description="Model name without provider prefix"
|
|
)
|
|
api_key: str | None = Field(None, description="API key for the provider")
|
|
api_base: str | None = Field(
|
|
None, max_length=500, description="Optional API base URL"
|
|
)
|
|
litellm_params: dict[str, Any] | None = Field(
|
|
None, description="Additional LiteLLM parameters"
|
|
)
|
|
|
|
|
|
class LLMConfigRead(LLMConfigBase, IDModel, TimestampModel):
|
|
id: int
|
|
created_at: datetime
|
|
search_space_id: int
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|