diff options
Diffstat (limited to 'backend/app/schemas.py')
| -rw-r--r-- | backend/app/schemas.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/backend/app/schemas.py b/backend/app/schemas.py index ac90bc1..bd8ebe7 100644 --- a/backend/app/schemas.py +++ b/backend/app/schemas.py @@ -24,6 +24,11 @@ class ModelProvider(str, Enum): OPENAI = "openai" GOOGLE = "google" +class ReasoningEffort(str, Enum): + LOW = "low" + MEDIUM = "medium" + HIGH = "high" + class LLMConfig(BaseModel): provider: ModelProvider model_name: str @@ -31,6 +36,8 @@ class LLMConfig(BaseModel): max_tokens: int = 1000 system_prompt: Optional[str] = None api_key: Optional[str] = None # Optional override, usually from env + enable_google_search: bool = False + reasoning_effort: ReasoningEffort = ReasoningEffort.MEDIUM # For OpenAI reasoning models class MergeStrategy(str, Enum): RAW = "raw" |