from pydantic import BaseModel, Field


class CodeExecution(BaseModel):
    """Result of executing a code block in the RLM sandbox."""

    code: str = Field(description="The Python code that was executed")
    stdout: str = Field(description="Standard output captured during execution")
    stderr: str = Field(description="Standard error captured during execution")
    success: bool = Field(description="Whether execution completed without error")


class RLMResult(BaseModel):
    """Result from RLM agent execution."""

    answer: str = Field(description="The answer to the user's question")
    program: str = Field(description="The final consolidated program")