Imaginephoenix commited on
Commit
7e241be
·
verified ·
1 Parent(s): d259149

Delete models.py

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