models.py

from typing import Optional
from openenv.core.env_server.types import Action, Observation, State


class SFDAction(Action):
    message: str


class SFDObservation(Observation):
    session_id: str = ""
    step: int = 0
    total_steps: int = 0
    item_id: str = ""
    domain: str = ""
    difficulty: str = ""
    question: str = ""
    ai_response: str = ""
    message: str = ""
    echoed_message: str = ""


class SFDState(State):
    task: str = ""
    flags: list = []
    confidences: list = []