databoysu commited on
Commit
f341e79
·
1 Parent(s): 985e10f

protect sandbox

Browse files
__pycache__/client.cpython-312.pyc CHANGED
Binary files a/__pycache__/client.cpython-312.pyc and b/__pycache__/client.cpython-312.pyc differ
 
__pycache__/context.cpython-312.pyc CHANGED
Binary files a/__pycache__/context.cpython-312.pyc and b/__pycache__/context.cpython-312.pyc differ
 
__pycache__/environment.cpython-312.pyc CHANGED
Binary files a/__pycache__/environment.cpython-312.pyc and b/__pycache__/environment.cpython-312.pyc differ
 
__pycache__/models.cpython-312.pyc CHANGED
Binary files a/__pycache__/models.cpython-312.pyc and b/__pycache__/models.cpython-312.pyc differ
 
client.py CHANGED
@@ -22,8 +22,12 @@ class TraceFixRLEnv(
22
 
23
  def _parse_result(self, payload: Dict) -> StepResult[CodeObservation]:
24
  obs_data = payload.get("observation", {})
 
 
 
 
25
  observation = CodeObservation(
26
- code_lines=obs_data.get("code_lines", []),
27
  localized_context=obs_data.get("localized_context", ""),
28
  last_execution_output=obs_data.get("last_execution_output", ""),
29
  syntax_error=obs_data.get("syntax_error", False),
 
22
 
23
  def _parse_result(self, payload: Dict) -> StepResult[CodeObservation]:
24
  obs_data = payload.get("observation", {})
25
+ raw_code_dict = obs_data.get("code_dict", {})
26
+ code_dict = {
27
+ int(k): v for k, v in raw_code_dict.items()
28
+ } if isinstance(raw_code_dict, dict) else {}
29
  observation = CodeObservation(
30
+ code_dict=code_dict,
31
  localized_context=obs_data.get("localized_context", ""),
32
  last_execution_output=obs_data.get("last_execution_output", ""),
33
  syntax_error=obs_data.get("syntax_error", False),