Codexzzz/sql-env

1# from openenv.core.env_server.types import Action, Observation2# from pydantic import Field3 4# class SqlAction(Action):5#     """The agent's action: a SQL query string."""6#     sql_query: str = Field(..., description="SQL query to execute against the database")7 8# class SqlObservation(Observation):9#     """What the agent sees after each reset() or step()."""10#     task_description: str   = Field(default="",  description="Natural language task")11#     schema_info:      str   = Field(default="",  description="DDL schema of the database")12#     query_result:     list  = Field(default_factory=list, description="Rows returned by query")13#     error_message:    str   = Field(default="",  description="SQL error string if query failed")14#     feedback:         str   = Field(default="",  description="Human-readable grader feedback")15#     score_breakdown:  dict  = Field(default_factory=dict, description="Partial score components")16#     attempts_remaining: int = Field(default=5,   description="Steps remaining in episode")17 18 19 20from openenv.core.env_server.types import Action, Observation21from pydantic import Field22 23 24class SqlAction(Action):25    """The agent's action: a SQL query string."""26    sql_query: str = Field(..., description="SQL query to execute against the database")27 28 29class SqlObservation(Observation):30    """What the agent sees after each reset() or step()."""31    task_description:   str   = Field(default="",           description="Natural language task")32    schema_info:        str   = Field(default="",           description="DDL schema of the database")33    query_result:       list  = Field(default_factory=list, description="Rows returned by query")34    error_message:      str   = Field(default="",           description="SQL error string if query failed")35    feedback:           str   = Field(default="",           description="Human-readable grader feedback")36    score_breakdown:    dict  = Field(default_factory=dict, description="Partial score components")37    attempts_remaining: int   = Field(default=5,            description="Steps remaining in episode")38    done:               bool  = Field(default=False,        description="Whether the episode has ended")39    reward:             float = Field(default=0.0,          description="Reward for this step")