VarshaChukka/SQL_Debugging_REPL_Environment

1from pydantic import BaseModel
2from typing import Optional
3
4class Observation(BaseModel):
5    problem: str
6    db_schema: str
7    last_error: Optional[str] = None
8    query_result: Optional[str] = None
9
10class Action(BaseModel):
11    sql_query: str
12
13class Reward(BaseModel):
14    score: float
15    feedback: str