diff options
Diffstat (limited to 'backend/app/schemas.py')
| -rw-r--r-- | backend/app/schemas.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/backend/app/schemas.py b/backend/app/schemas.py index ac90bc1..54c0560 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" @@ -42,6 +49,10 @@ class NodeRunRequest(BaseModel): user_prompt: str config: LLMConfig merge_strategy: MergeStrategy = MergeStrategy.SMART + attached_file_ids: List[str] = Field(default_factory=list) + # Scopes for file_search filtering: ["project_path/node_id", ...] + # Contains all project/node combinations in the current trace + scopes: List[str] = Field(default_factory=list) class NodeRunResponse(BaseModel): node_id: str |