| """Typed models for the Model Release environment.""" |
|
|
| from __future__ import annotations |
|
|
| from typing import Any, Dict, List, Literal, Optional |
|
|
| from pydantic import Field |
|
|
| try: |
| from openenv.core.env_server.types import Action, Observation, State |
| except ImportError: |
| from openenv_core.env_server.types import Action, Observation, State |
|
|
| ActionType = Literal["inspect", "set_field", "set_decision", "submit"] |
|
|
|
|
| class ModelReleaseAction(Action): |
| """Action schema for release-readiness workflows.""" |
|
|
| action_type: ActionType = Field( |
| ..., description="One of inspect, set_field, set_decision, or submit" |
| ) |
| target: str = Field( |
| default="", |
| description="Document name for inspect or field name for set_field", |
| ) |
| value: str = Field( |
| default="", |
| description="Field value for set_field or release decision for set_decision", |
| ) |
|
|
|
|
| class ModelReleaseObservation(Observation): |
| """Observation returned after reset and step operations.""" |
|
|
| task_name: str = Field(default="", description="Current task identifier") |
| difficulty: str = Field(default="easy", description="Task difficulty label") |
| goal: str = Field(default="", description="What the agent must accomplish") |
| document_index: Dict[str, str] = Field( |
| default_factory=dict, |
| description="Available documents mapped to short summaries", |
| ) |
| visible_documents: Dict[str, str] = Field( |
| default_factory=dict, |
| description="Full contents of documents that have been inspected", |
| ) |
| package_snapshot: Dict[str, Any] = Field( |
| default_factory=dict, |
| description="Current structured release package fields", |
| ) |
| checklist_status: Dict[str, bool] = Field( |
| default_factory=dict, |
| description="Progress on deterministic grader checks", |
| ) |
| critical_gaps: List[str] = Field( |
| default_factory=list, |
| description="Unsatisfied checks in priority order", |
| ) |
| available_fields: List[str] = Field( |
| default_factory=list, |
| description="Fields the agent may edit with set_field", |
| ) |
| available_decisions: List[str] = Field( |
| default_factory=list, |
| description="Allowed release decisions for the task", |
| ) |
| inspected_documents: List[str] = Field( |
| default_factory=list, |
| description="Documents already revealed to the agent", |
| ) |
| remaining_steps: int = Field(default=0, description="Steps left in the episode") |
| last_action_error: Optional[str] = Field( |
| default=None, |
| description="Validation or execution error from the last action", |
| ) |
| score: float = Field( |
| default=0.0, |
| description="Current normalized task score in the range [0, 1]", |
| ) |
|
|
|
|
| class ModelReleaseState(State): |
| """Episode state for release-readiness tasks.""" |
|
|
| task_name: str = Field(default="", description="Current task identifier") |
| difficulty: str = Field(default="easy", description="Task difficulty label") |
| completed_checks: List[str] = Field( |
| default_factory=list, |
| description="Checklist items currently satisfied", |
| ) |
| inspected_documents: List[str] = Field( |
| default_factory=list, |
| description="Document names revealed during the episode", |
| ) |
| release_decision: str = Field( |
| default="undecided", |
| description="Current release channel or gate decision", |
| ) |
| score: float = Field(default=0.0, description="Current normalized task score") |
