17 lines
669 B
Python
17 lines
669 B
Python
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")
|