Make main orchestrator-compliant

Author: AnatoliB

samples-v2/openai_agents/customer_service/customer_service.py

@@ -138,20 +138,18 @@ async def on_seat_booking_handoff(context: RunContextWrapper[AirlineAgentContext
 ### RUN
 
 
-async def main():
+def main(context):
     current_agent: Agent[AirlineAgentContext] = triage_agent
     input_items: list[TResponseInputItem] = []
     context = AirlineAgentContext()
 
-    # Normally, each input from the user would be an API request to your app, and you can wrap the request in a trace()
-    # Here, we'll just use a random UUID for the conversation ID
-    conversation_id = uuid.uuid4().hex[:16]
+    conversation_id = context.instance_id
 
     while True:
-        user_input = input("Enter your message: ")
+        user_input = context.wait_for_external_event("UserInput")
         with trace("Customer service", group_id=conversation_id):
             input_items.append({"content": user_input, "role": "user"})
-            result = await Runner.run(current_agent, input_items, context=context)
+            result = Runner.run_sync(current_agent, input_items, context=context)
 
             for new_item in result.new_items:
                 agent_name = new_item.agent.name

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()
+    return customer_service.customer_service.main(context)