Throw to yield

Author: AnatoliB

samples-v2/openai_agents/durable_model_stub.py

@@ -124,7 +124,7 @@ def make_tool_info(tool: Tool) -> ToolInput:
             activity_input
         )
 
-        return activity_output
+        raise OrchestratorYielded(activity_output)
 
     def stream_response(
         self,

samples-v2/openai_agents/function_app.py

@@ -20,9 +20,12 @@ async def hello_orchestration_starter(req: func.HttpRequest, client):
 def basic_hello_world_orchestrator(context):
     set_default_agent_runner(DurableOpenAIRunner(durable_orchestration_context=context))
 
-    from basic.hello_world import main
-    result = main()
-    return result
+    try:
+        from basic.hello_world import main
+        result = main()
+        return result
+    except OrchestratorYielded as e:
+        yield e.activity_output
 
 @app.activity_trigger(input_name="input")
 def invoke_model_activity(input: Any):

samples-v2/openai_agents/orchestrator_exceptions.py

@@ -0,0 +1,12 @@
+"""Exceptions related to durable orchestrator operations."""
+
+
+class OrchestratorYielded(Exception):
+    """Exception raised when the orchestrator yields to an activity.
+    
+    This exception stores the activity output for later processing.
+    """
+    
+    def __init__(self, activity_output):
+        self.activity_output = activity_output
+        super().__init__("Orchestrator yielded to activity")