Spaces:
Paused
Paused
| """Model management and system prompt schemas.""" | |
| from __future__ import annotations | |
| from datetime import datetime | |
| from typing import Literal | |
| from pydantic import BaseModel, Field, field_validator | |
| Provider = Literal[ | |
| "groq", "openai", "anthropic", "gemini", "deepseek", "qwen", "ollama", "custom" | |
| ] | |
| class ModelCreate(BaseModel): | |
| name: str = Field(min_length=1, max_length=128) | |
| provider: Provider | |
| endpoint: str | None = None | |
| apiKey: str | None = None | |
| temperature: float = Field(default=0.7, ge=0.0, le=2.0) | |
| maxTokens: int = Field(default=4096, ge=1, le=200_000) | |
| topP: float = Field(default=1.0, ge=0.0, le=1.0) | |
| systemPromptId: str | None = None | |
| enabled: bool = True | |
| description: str | None = None | |
| displayName: str | None = None | |
| avatar: str | None = None | |
| tags: list[str] = Field(default_factory=list) | |
| class ModelUpdate(BaseModel): | |
| name: str | None = None | |
| endpoint: str | None = None | |
| apiKey: str | None = None | |
| temperature: float | None = Field(default=None, ge=0.0, le=2.0) | |
| maxTokens: int | None = Field(default=None, ge=1, le=200_000) | |
| topP: float | None = Field(default=None, ge=0.0, le=1.0) | |
| systemPromptId: str | None = None | |
| enabled: bool | None = None | |
| description: str | None = None | |
| displayName: str | None = None | |
| avatar: str | None = None | |
| tags: list[str] | None = None | |
| class ModelOut(BaseModel): | |
| """Public-facing model representation. NEVER includes apiKey.""" | |
| id: str | |
| name: str | |
| provider: Provider | |
| endpoint: str | None = None | |
| temperature: float | |
| maxTokens: int | |
| topP: float | |
| systemPromptId: str | None = None | |
| systemPromptName: str | None = None | |
| enabled: bool | |
| description: str | None = None | |
| displayName: str | None = None | |
| avatar: str | None = None | |
| tags: list[str] | |
| createdAt: datetime | |
| updatedAt: datetime | |
| hasApiKey: bool = False | |
| class SystemPromptCreate(BaseModel): | |
| name: str = Field(min_length=1, max_length=128) | |
| content: str = Field(min_length=1, max_length=200_000) | |
| description: str | None = None | |
| tags: list[str] = Field(default_factory=list) | |
| active: bool = True | |
| class SystemPromptUpdate(BaseModel): | |
| name: str | None = None | |
| content: str | None = None | |
| description: str | None = None | |
| tags: list[str] | None = None | |
| active: bool | None = None | |
| changelog: str | None = None | |
| class SystemPromptVersion(BaseModel): | |
| version: int | |
| content: str | |
| changelog: str | None = None | |
| createdAt: datetime | |
| class SystemPromptOut(BaseModel): | |
| id: str | |
| name: str | |
| description: str | None = None | |
| tags: list[str] | |
| active: bool | |
| currentVersion: int | |
| versions: list[SystemPromptVersion] = Field(default_factory=list) | |
| createdAt: datetime | |
| updatedAt: datetime | |
| class SystemPromptSummary(BaseModel): | |
| """Minimal info for non-admin users (content is NEVER sent).""" | |
| id: str | |
| name: str | |
| description: str | None = None | |
| class ProviderKeyCreate(BaseModel): | |
| provider: Provider | |
| apiKey: str = Field(min_length=1) | |
| endpoint: str | None = None | |
| class ProviderKeyOut(BaseModel): | |
| id: str | |
| provider: Provider | |
| endpoint: str | None = None | |
| status: str = "pending" | |
| hasApiKey: bool = True | |
| lastSyncAt: datetime | None = None | |
| modelsImported: int = 0 | |
| createdAt: datetime | |
| updatedAt: datetime | |