| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657 |
- """Schemas for plan generation and rule engine."""
- from typing import Optional, Any
- from pydantic import BaseModel, Field
- class ParameterInfo(BaseModel):
- """Information about a scannable parameter."""
- name: str
- display_name: str
- unit: str
- category: str
- default_start: float
- default_stop: float
- default_step: float
- min_allowed: float
- max_allowed: float
- description: str = ""
- topology_supported: list[str] = Field(default_factory=list)
- class ParameterRegistryResponse(BaseModel):
- """Response containing all scannable parameters."""
- total: int
- parameters: list[ParameterInfo]
- class RangeRecommendation(BaseModel):
- """Recommended scan range for a parameter."""
- start: float
- stop: float
- step: float
- notes: str = ""
- class RangeRecommendRequest(BaseModel):
- """Request for range recommendation."""
- parameter_name: str
- boundary_conditions: dict[str, Any] = Field(default_factory=dict)
- class PlanGenerateRequest(BaseModel):
- """Request to generate a plan from boundary conditions."""
- boundary_conditions: dict[str, Any] = Field(default_factory=dict)
- parameter_names: Optional[list[str]] = None
- model_path: str = ""
- class PlanGenerateResponse(BaseModel):
- """Response containing a generated plan draft."""
- name: str
- model_path: str
- topology: str
- variables: list[dict[str, Any]]
- cases: list[dict[str, Any]] = Field(default_factory=list)
- estimated_points: int
- estimated_time_min: float
- generation_summary: dict[str, Any] = Field(default_factory=dict)
|
