Spaces:
Runtime error
Runtime error
File size: 1,443 Bytes
02e973e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 | """Data models for the OpenEnv email triage environment."""
from typing import Literal
from pydantic import BaseModel
class EmailObservation(BaseModel):
"""Represents the email context visible to the agent at each step."""
email_id: str
subject: str
body: str
sender: str
timestamp: str
thread_history: list[str]
task_id: str
step_number: int
total_emails: int
class TriageAction(BaseModel):
"""Represents the action chosen by the agent for an email."""
label: Literal["urgent", "normal", "spam", "archive"]
summary: str
route_to: str
class RewardResult(BaseModel):
"""Represents deterministic grading output before reward shaping."""
score: float
breakdown: dict[str, float]
feedback: str
class EnvironmentState(BaseModel):
"""Represents full internal environment state for debugging and evaluation."""
task_id: str
current_step: int
total_steps: int
done: bool
action_history: list[TriageAction]
reward_history: list[float]
class StepResult(BaseModel):
"""Represents the standardized output of environment step calls."""
observation: EmailObservation
reward: float
done: bool
info: dict[str, str | int | float | bool]
class ResetResult(BaseModel):
"""Represents the standardized output of environment reset calls."""
observation: EmailObservation
info: dict[str, str | int | float | bool]
|