Implement wait_for_external_event

Author: AnatoliB

samples-v2/openai_agents/durable_ai_agent_context.py

@@ -31,6 +31,12 @@ def call_activity(self, activity_name, input: str):
         self.activities_called += 1
         return task
 
+    def set_custom_status(self, status: str):
+        self.context.set_custom_status(status)
+
+    def wait_for_external_event(self, event_name: str):
+        return self.context.wait_for_external_event(event_name)
+
     def yield_and_clear_tasks(self):
         """Yield all accumulated tasks and clear the tasks list."""
         for task in self.tasks_to_yield:

samples-v2/openai_agents/function_app.py

@@ -82,3 +82,13 @@ def hello_cities(context):
 @app.activity_trigger(input_name="name")
 async def hello(name: str):
     return f"Hello {name}!"
+
+
+
+@app.orchestration_trigger(context_name="context")
+@durable_openai_agent_orchestrator
+def human_in_the_loop(context):
+    context.set_custom_status("Waiting for approval...")
+    event_data = yield context.wait_for_external_event("Approve")
+    context.set_custom_status("")
+    return event_data