Enabling replay (WIP)

Author: AnatoliB

samples-v2/openai_agents/durable_model_stub.py

@@ -132,12 +132,16 @@ def make_tool_info(tool: Tool) -> ToolInput:
         # Serialize activity_input to JSON
         activity_input_json = activity_input.to_json()
 
-        activity_output = self.durable_orchestration_context.call_activity(
+        task = self.durable_orchestration_context.call_activity(
             "invoke_model_activity",
             activity_input_json
         )
 
-        raise OrchestratorYielded(activity_output)
+        if not self.durable_orchestration_context.is_replaying:
+            raise OrchestratorYielded(task)
+        
+        result = task.result
+        return result
 
     def stream_response(
         self,