Distinguish between DurableAIAgentContext and AirlineAgentContext

Author: AnatoliB

samples-v2/openai_agents/customer_service/customer_service.py

@@ -1,8 +1,8 @@
 from __future__ import annotations as _annotations
 
 import random
-import uuid
 
+from azure.durable_functions.openai_agents import DurableAIAgentContext
 from pydantic import BaseModel
 
 from agents import (
@@ -138,18 +138,18 @@ async def on_seat_booking_handoff(context: RunContextWrapper[AirlineAgentContext
 ### RUN
 
 
-def main(context):
+def main(context: DurableAIAgentContext):
     current_agent: Agent[AirlineAgentContext] = triage_agent
     input_items: list[TResponseInputItem] = []
-    context = AirlineAgentContext()
+    airline_agent_context = AirlineAgentContext()
 
     conversation_id = context.instance_id
 
     while True:
         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 = Runner.run_sync(current_agent, input_items, context=context)
+            result = Runner.run_sync(current_agent, input_items, context=airline_agent_context)
 
             for new_item in result.new_items:
                 agent_name = new_item.agent.name