Use custom status for user prompts

Author: AnatoliB

samples-v2/openai_agents/customer_service/customer_service.py

@@ -145,8 +145,10 @@ def main(context: DurableAIAgentContext):
 
     conversation_id = context.instance_id
 
+    context.set_custom_status("How can I help you today?")
     while True:
         user_input = yield context.wait_for_external_event("UserInput")
+        context.set_custom_status("Thinking...")
         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)
@@ -167,3 +169,5 @@ def main(context: DurableAIAgentContext):
                     print(f"{agent_name}: Skipping item: {new_item.__class__.__name__}")
             input_items = result.to_input_list()
             current_agent = result.last_agent
+
+            context.set_custom_status(result.final_output)