mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-05-29 19:35:20 +02:00
409fec94c3
- Added model eligibility checks to ensure automations can only use billable models (premium or BYOK). - Introduced new API endpoint to report model eligibility status for search spaces. - Updated frontend components to display eligibility alerts and disable creation options when models are not billable. - Enhanced automation creation forms to reflect model eligibility, preventing users from submitting invalid configurations. - Implemented server-side logic to capture and preserve model preferences across automation edits, ensuring consistent behavior during execution.
45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
"""``AutomationDefinition`` — top-level envelope persisted in ``automations.definition``."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
from .execution import Execution
|
|
from .inputs import Inputs
|
|
from .metadata import Metadata
|
|
from .plan_step import PlanStep
|
|
from .trigger_spec import TriggerSpec
|
|
|
|
|
|
class AutomationModels(BaseModel):
|
|
"""Captured model profile for an automation.
|
|
|
|
Snapshotted from the search space's preferences at create time so runs are
|
|
insulated from later chat/search-space model changes. Config-id conventions
|
|
match the shared scheme (``0`` Auto, ``< 0`` global, ``> 0`` BYOK).
|
|
"""
|
|
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
agent_llm_id: int = 0
|
|
image_generation_config_id: int = 0
|
|
vision_llm_config_id: int = 0
|
|
|
|
|
|
class AutomationDefinition(BaseModel):
|
|
"""Top-level shape of an automation."""
|
|
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
schema_version: str = "1.0"
|
|
name: str = Field(..., min_length=1, max_length=200)
|
|
goal: str | None = None
|
|
inputs: Inputs | None = None
|
|
triggers: list[TriggerSpec] = Field(default_factory=list)
|
|
plan: list[PlanStep] = Field(..., min_length=1)
|
|
execution: Execution = Field(default_factory=Execution)
|
|
metadata: Metadata = Field(default_factory=Metadata)
|
|
# Captured server-side at create() and preserved across update(); resolved
|
|
# at runtime instead of the live search space. Optional so drafts/builder
|
|
# payloads validate without it.
|
|
models: AutomationModels | None = None
|