Use yield for user input handling

Author: AnatoliB

samples-v2/openai_agents/customer_service/customer_service.py

@@ -146,7 +146,7 @@ def main(context: DurableAIAgentContext):
     conversation_id = context.instance_id
 
     while True:
-        user_input = context.wait_for_external_event("UserInput")
+        user_input = yield context.wait_for_external_event("UserInput")
         with trace("Customer service", group_id=conversation_id):
             input_items.append({"content": user_input, "role": "user"})
             result = Runner.run_sync(current_agent, input_items, context=airline_agent_context)

samples-v2/openai_agents/function_app.py

@@ -120,4 +120,4 @@ def message_filter(context):
 @app.durable_openai_agent_orchestrator
 def customer_service(context):
     import customer_service.customer_service
-    return customer_service.customer_service.main(context)
+    return (yield from customer_service.customer_service.main(context))