Spaces:
Runtime error
Runtime error
| """ | |
| CodeReviewEnv — OpenEnv MCP-compliant typed models. | |
| Follows the same MCP tool-calling pattern as the reference environments | |
| (calendar_env, repl_env). Agents interact via tool calls: | |
| ListToolsAction — discover available tools | |
| ToolCallAction — call a tool by name with arguments | |
| Available tools: | |
| get_code — get the buggy code snippet for review | |
| analyze_code — get structural analysis of the code | |
| check_line — check if a specific line contains a bug (+/- reward) | |
| get_hint — get a progressive hint (costs efficiency) | |
| submit_review — submit final structured review (ends episode) | |
| """ | |
| from typing import Any, Dict, List, Literal, Optional | |
| from pydantic import ConfigDict, Field | |
| from openenv.core.env_server.types import Action, Observation, State | |
| # ─── MCP Action (matches calendar_env pattern) ────────────────────────────── | |
| class MCPAction(Action): | |
| """MCP tool-calling action. Agents either list tools or call a tool.""" | |
| action_type: Literal["ListToolsAction", "ToolCallAction"] = Field( | |
| ..., description="Type of action to perform" | |
| ) | |
| tool_name: Optional[str] = Field( | |
| None, description="Name of tool to call (required for ToolCallAction)" | |
| ) | |
| arguments: Optional[Dict[str, Any]] = Field( | |
| default_factory=dict, description="Arguments for the tool" | |
| ) | |
| class CodeReviewAction(MCPAction): | |
| """Code review action — inherits MCP tool-calling interface.""" | |
| pass | |
| # ─── MCP Observation (matches calendar_env pattern) ───────────────────────── | |
| class MCPObservation(Observation): | |
| """MCP observation returned after tool calls.""" | |
| success: bool = Field(True, description="Whether the action succeeded") | |
| error_message: Optional[str] = Field(None, description="Error details if failed") | |
| tools_list: Optional[List[Dict[str, Any]]] = Field( | |
| None, description="Available tools (for ListToolsAction)" | |
| ) | |
| tool_result: Optional[Dict[str, Any]] = Field( | |
| None, description="Result from tool call" | |
| ) | |
| metadata: Dict[str, Any] = Field(default_factory=dict) | |
| done: bool = Field(False, description="Whether the episode is over") | |
| reward: Optional[float] = Field(None, description="Reward signal") | |
| class CodeReviewObservation(MCPObservation): | |
| """Code review observation — inherits MCP observation.""" | |
| pass | |
| # ─── State ─────────────────────────────────────────────────────────────────── | |
| class CodeReviewState(State): | |
| """Full environment state — includes gold answers (hidden from agent).""" | |
| original_code: str = Field(default="", description="Clean code before injection") | |
| buggy_code: str = Field(default="", description="Code with injected bugs") | |
| gold_bugs: List[Dict[str, Any]] = Field( | |
| default_factory=list, | |
| description="Injected bugs: [{description, lines, fix, bug_type}, ...]", | |
| ) | |
| language: str = Field(default="python", description="Source language") | |
| difficulty: str = Field(default="easy", description="Difficulty tier") | |
| hint_count: int = Field(default=0, description="Hints requested") | |
| snippet_name: str = Field(default="", description="Which snippet was used") | |
| flagged_lines: List[int] = Field(default_factory=list, description="Lines flagged so far") | |