File size: 1,774 Bytes
b215601 | 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 | from pydantic import BaseModel, ConfigDict, Field
from typing import List, Literal, Optional
import openenv.core.env_server.types as openenv_types
class Observation(openenv_types.Observation):
ticket_id: str = Field(default="", description="Unique ID of the ticket")
customer_message: str = Field(default="", description="The current message from the customer")
history: List[str] = Field(default_factory=list, description="Conversation history")
missing_info: List[str] = Field(default_factory=list, description="Fields required before routing")
status: str = Field(default="OPEN", description="Ticket status")
refund_processed: bool = Field(default=False, description="True if a refund was already executed")
class Action(openenv_types.Action):
action_type: Literal["ROUTE", "ASK_INFO", "REFUND", "CLOSE"] = Field(..., description="Type of action to take")
argument: str = Field(..., description="The category to route to, the question to ask, or the order ID to refund")
class State(openenv_types.State):
ticket_id: str = Field(default="", description="Unique ID of the ticket")
customer_message: str = Field(default="", description="The current message from the customer")
history: List[str] = Field(default_factory=list, description="Conversation history")
missing_info: List[str] = Field(default_factory=list, description="Fields required before routing")
status: str = Field(default="OPEN", description="Ticket status")
refund_processed: bool = Field(default=False, description="True if a refund was already executed")
class Reward(BaseModel):
value: float = Field(..., description="Numerical reward between -1.0 and 1.0")
feedback: str = Field(..., description="Incremental feedback for the agent") |