fix: ensure RunState serialization compatibility with openai-agents-js

Author: mjschock

examples/agent_patterns/human_in_the_loop.py

@@ -104,7 +104,7 @@ async def main():
         with open("result.json") as f:
             stored_state_json = json.load(f)
 
-        state = RunState.from_json(agent, stored_state_json)
+        state = await RunState.from_json(agent, stored_state_json)
 
         # Process each interruption
         for interruption in result.interruptions:

src/agents/_run_impl.py

@@ -240,6 +240,9 @@ class SingleStepResult:
     tool_output_guardrail_results: list[ToolOutputGuardrailResult]
     """Tool output guardrail results from this step."""
 
+    processed_response: ProcessedResponse | None = None
+    """The processed model response. This is needed for resuming from interruptions."""
+
     @property
     def generated_items(self) -> list[RunItem]:
         """Items generated during the agent run (i.e. everything generated after
@@ -331,6 +334,7 @@ async def execute_tools_and_side_effects(
                 next_step=NextStepInterruption(interruptions=interruptions),
                 tool_input_guardrail_results=tool_input_guardrail_results,
                 tool_output_guardrail_results=tool_output_guardrail_results,
+                processed_response=processed_response,
             )
 
         new_step_items.extend([result.run_item for result in approved_function_results])

src/agents/result.py

@@ -29,7 +29,7 @@
 )
 
 if TYPE_CHECKING:
-    from ._run_impl import QueueCompleteSentinel
+    from ._run_impl import ProcessedResponse, QueueCompleteSentinel
     from .agent import Agent
     from .tool_guardrails import ToolInputGuardrailResult, ToolOutputGuardrailResult
 
@@ -116,6 +116,8 @@ def last_response_id(self) -> str | None:
 @dataclass
 class RunResult(RunResultBase):
     _last_agent: Agent[Any]
+    _last_processed_response: ProcessedResponse | None = field(default=None, repr=False)
+    """The last processed model response. This is needed for resuming from interruptions."""
 
     @property
     def last_agent(self) -> Agent[Any]:
@@ -162,6 +164,7 @@ def to_state(self) -> Any:
         state._model_responses = self.raw_responses
         state._input_guardrail_results = self.input_guardrail_results
         state._output_guardrail_results = self.output_guardrail_results
+        state._last_processed_response = self._last_processed_response
 
         # If there are interruptions, set the current step
         if self.interruptions:
@@ -202,6 +205,9 @@ class RunResultStreaming(RunResultBase):
     is_complete: bool = False
     """Whether the agent has finished running."""
 
+    _last_processed_response: ProcessedResponse | None = field(default=None, repr=False)
+    """The last processed model response. This is needed for resuming from interruptions."""
+
     # Queues that the background run_loop writes to
     _event_queue: asyncio.Queue[StreamEvent | QueueCompleteSentinel] = field(
         default_factory=asyncio.Queue, repr=False
@@ -443,6 +449,7 @@ def to_state(self) -> Any:
         state._input_guardrail_results = self.input_guardrail_results
         state._output_guardrail_results = self.output_guardrail_results
         state._current_turn = self.current_turn
+        state._last_processed_response = self._last_processed_response
 
         # If there are interruptions, set the current step
         if self.interruptions: