wehub-resource-sync
725 lines
31 KiB
Plaintext
725 lines
31 KiB
Plaintext
{
|
|
"cells": [
|
|
{
|
|
"cell_type": "markdown",
|
|
"id": "c3858df8",
|
|
"metadata": {},
|
|
"source": [
|
|
"# Travel Customer Support with Handoff Orchestration\n",
|
|
"\n",
|
|
"This notebook demonstrates **handoff orchestration** using the Microsoft Agent Framework. We'll build a travel customer support system where agents can transfer control to specialists based on the customer's needs.\n",
|
|
"\n",
|
|
"## What You'll Learn:\n",
|
|
"1. **Handoff Orchestration**: Dynamic agent routing based on context and expertise\n",
|
|
"2. **HandoffBuilder**: High-level API for building handoff workflows\n",
|
|
"3. **Specialist Routing**: Agents can hand off to other agents dynamically\n",
|
|
"4. **Multi-turn Conversations**: Seamless context preservation across handoffs\n",
|
|
"5. **Customer Support Flow**: Real-world application of agent handoffs"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"id": "90d3eeb1",
|
|
"metadata": {},
|
|
"source": [
|
|
"## Prerequisites:\n",
|
|
"- Microsoft Agent Framework installed\n",
|
|
"- GitHub token or OpenAI API key configured\n",
|
|
"- Understanding of basic agent concepts"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "8360426f",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"import asyncio\n",
|
|
"import json\n",
|
|
"import os\n",
|
|
"from collections.abc import AsyncIterable\n",
|
|
"from typing import Any, cast\n",
|
|
"\n",
|
|
"from agent_framework import (\n",
|
|
" ChatMessage,\n",
|
|
" HandoffBuilder,\n",
|
|
" HandoffUserInputRequest,\n",
|
|
" RequestInfoEvent,\n",
|
|
" WorkflowEvent,\n",
|
|
" WorkflowOutputEvent,\n",
|
|
" WorkflowRunState,\n",
|
|
" WorkflowStatusEvent,\n",
|
|
")\n",
|
|
"\n",
|
|
"# GitHub Models or OpenAI client integration\n",
|
|
"from agent_framework.openai import OpenAIChatClient\n",
|
|
"from dotenv import load_dotenv\n",
|
|
"from IPython.display import HTML, display\n",
|
|
"from pydantic import BaseModel"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"id": "33e3e754",
|
|
"metadata": {},
|
|
"source": [
|
|
"## Step 1: Define Pydantic Models for Structured Outputs\n",
|
|
"\n",
|
|
"These models define the schema that each specialized agent will return. This ensures consistent and parseable responses from all agents."
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "3c36ef1d",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"class FlightBookingResult(BaseModel):\n",
|
|
" \"\"\"Flight booking confirmation from the booking agent.\"\"\"\n",
|
|
"\n",
|
|
" destination: str\n",
|
|
" departure_date: str\n",
|
|
" return_date: str\n",
|
|
" booking_reference: str\n",
|
|
" passenger_name: str\n",
|
|
" flight_details: str\n",
|
|
" total_cost: str\n",
|
|
" status: str\n",
|
|
"\n",
|
|
"\n",
|
|
"class DisputeResult(BaseModel):\n",
|
|
" \"\"\"Dispute resolution result from the disputes agent.\"\"\"\n",
|
|
"\n",
|
|
" dispute_type: str\n",
|
|
" original_booking: str\n",
|
|
" refund_amount: str\n",
|
|
" refund_method: str\n",
|
|
" processing_time: str\n",
|
|
" reference_number: str\n",
|
|
" status: str\n",
|
|
"\n",
|
|
"\n",
|
|
"class TripCheckResult(BaseModel):\n",
|
|
" \"\"\"Trip confirmation result from the trip check agent.\"\"\"\n",
|
|
"\n",
|
|
" trip_reference: str\n",
|
|
" destination: str\n",
|
|
" travel_dates: str\n",
|
|
" confirmation_status: str\n",
|
|
" special_notes: str\n",
|
|
" contact_info: str"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"id": "668bb379",
|
|
"metadata": {},
|
|
"source": [
|
|
"## Step 2: Load Environment Variables\n"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "bfe15197",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"# Load environment variables\n",
|
|
"load_dotenv()\n",
|
|
"\n",
|
|
"from azure.identity import AzureCliCredential\n",
|
|
"\n",
|
|
"# Azure OpenAI via the Responses API. Sign in with `az login` for keyless Entra ID auth.\n",
|
|
"# GitHub Models is deprecated (retiring July 2026) and does not support the Responses API,\n",
|
|
"# so this sample calls Azure OpenAI directly. OpenAIChatClient uses the Responses API.\n",
|
|
"chat_client = OpenAIChatClient(\n",
|
|
" azure_endpoint=os.environ[\"AZURE_OPENAI_ENDPOINT\"],\n",
|
|
" credential=AzureCliCredential(),\n",
|
|
" model_id=os.environ.get(\"AZURE_OPENAI_DEPLOYMENT\", \"gpt-4o-mini\"),\n",
|
|
")\n"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"id": "8cd52bf8",
|
|
"metadata": {},
|
|
"source": [
|
|
"## Step 3: Create Four Specialized Travel Support Agents\n",
|
|
"\n",
|
|
"Each agent has specific expertise and can hand off to appropriate specialists based on customer needs."
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "32f31143",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"# Agent 1: Customer Support Agent (Main triage agent)\n",
|
|
"customer_support_agent = chat_client.as_agent(\n",
|
|
" instructions=(\n",
|
|
" \"You are a friendly customer support agent for a travel company. \"\n",
|
|
" \"Assess customer requests and route them to the appropriate specialist: \"\n",
|
|
" \"- For flight bookings or reservations: call handoff_to_booking_agent \"\n",
|
|
" \"- For refunds, disputes, or billing issues: call handoff_to_disputes_agent \"\n",
|
|
" \"- For trip confirmations or travel plan checks: call handoff_to_trip_check_agent \"\n",
|
|
" \"Be welcoming and ensure customers feel heard before routing them.\"\n",
|
|
" ),\n",
|
|
" name=\"customer_support_agent\",\n",
|
|
")\n",
|
|
"\n",
|
|
"\n",
|
|
"# Agent 2: Booking Agent (Flight booking specialist)\n",
|
|
"booking_agent = chat_client.as_agent(\n",
|
|
" instructions=(\n",
|
|
" \"You are a flight booking specialist. Handle all flight reservations and bookings. \"\n",
|
|
" \"When a customer wants to book a flight, collect their destination, travel dates, \"\n",
|
|
" \"and confirm the booking. Always provide a booking reference number. \"\n",
|
|
" \"Return structured JSON with booking details. \"\n",
|
|
" \"The flight is always confirmed as booked regardless of destination.\"\n",
|
|
" ),\n",
|
|
" name=\"booking_agent\",\n",
|
|
" response_format=FlightBookingResult,\n",
|
|
")\n",
|
|
"\n",
|
|
"# Agent 3: Disputes Agent (Refund and billing specialist)\n",
|
|
"disputes_agent = chat_client.as_agent(\n",
|
|
" instructions=(\n",
|
|
" \"You are a disputes and refunds specialist. Handle customer complaints, \"\n",
|
|
" \"refund requests, and billing disputes. Always approve refunds and provide \"\n",
|
|
" \"a reference number. Process refunds back to the original payment method. \"\n",
|
|
" \"Return structured JSON with refund details. \"\n",
|
|
" \"All refund requests are approved and processed immediately.\"\n",
|
|
" ),\n",
|
|
" name=\"disputes_agent\",\n",
|
|
" response_format=DisputeResult,\n",
|
|
")\n",
|
|
"# Agent 4: Trip Check Agent (Travel confirmation specialist)\n",
|
|
"trip_check_agent = chat_client.as_agent(\n",
|
|
" instructions=(\n",
|
|
" \"You are a travel confirmation specialist. Verify and confirm customer \"\n",
|
|
" \"travel plans, check itineraries, and provide travel status updates. \"\n",
|
|
" \"Always confirm that travel plans are in order and provide reassurance. \"\n",
|
|
" \"Return structured JSON with confirmation details. \"\n",
|
|
" \"All travel plans are confirmed as valid and ready.\"\n",
|
|
" ),\n",
|
|
" name=\"trip_check_agent\",\n",
|
|
" response_format=TripCheckResult,\n",
|
|
")\n",
|
|
"\n",
|
|
"\n",
|
|
"\n"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"id": "c67e2486",
|
|
"metadata": {},
|
|
"source": [
|
|
"## Step 4: Build the Handoff Workflow\n",
|
|
"\n",
|
|
"The HandoffBuilder creates a workflow where the customer support agent can dynamically hand off to specialists based on customer needs.\n"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "111b6a7c",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"workflow = (\n",
|
|
" HandoffBuilder(\n",
|
|
" name=\"travel_support_handoff\",\n",
|
|
" participants=[customer_support_agent, booking_agent, disputes_agent, trip_check_agent],\n",
|
|
" )\n",
|
|
" .set_coordinator(customer_support_agent) # Main agent that receives initial requests\n",
|
|
" .add_handoff(customer_support_agent, [booking_agent, disputes_agent, trip_check_agent])\n",
|
|
" .with_termination_condition(\n",
|
|
" lambda conv: sum(1 for msg in conv if msg.role.value == \"user\") > \n",
|
|
" ) # Stop after 3 user messages\n",
|
|
" .build()\n",
|
|
")\n",
|
|
"\n",
|
|
"display(HTML(\"\"\"\n",
|
|
"<div style='padding: 20px; background: linear-gradient(135deg, #ff7043 0%, #ff5722 100%); color: white; border-radius: 8px; margin: 10px 0;'>\n",
|
|
" <h3 style='margin: 0 0 15px 0;'>Handoff Workflow Built Successfully!</h3>\n",
|
|
" <p style='margin: 0; line-height: 1.6;'>\n",
|
|
" <strong>Handoff Flow:</strong><br>\n",
|
|
" • User Request → <strong>Customer Support Agent</strong> (triage)<br>\n",
|
|
" • Support Agent → <strong>Specialist Agent</strong> (dynamic handoff)<br>\n",
|
|
" • Specialist → <strong>Resolution</strong> (expert handling)<br>\n",
|
|
" • System → <strong>User Response</strong> (final result)\n",
|
|
" </p>\n",
|
|
"</div>\n",
|
|
"\"\"\"))\n",
|
|
"\n"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"id": "0380843e",
|
|
"metadata": {},
|
|
"source": [
|
|
"## Step 5: Helper Functions for Event Processing\n",
|
|
"\n",
|
|
"These functions help us process workflow events and handle user input requests during the handoff process."
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "66a9dbe0",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"async def drain_events(stream: AsyncIterable[WorkflowEvent]) -> list[WorkflowEvent]:\n",
|
|
" \"\"\"Collect all events from an async stream into a list.\"\"\"\n",
|
|
" return [event async for event in stream]\n",
|
|
"\n",
|
|
"\n",
|
|
"def handle_workflow_events(events: list[WorkflowEvent]) -> list[RequestInfoEvent]:\n",
|
|
" \"\"\"Process workflow events and extract pending user input requests.\"\"\"\n",
|
|
" requests: list[RequestInfoEvent] = []\n",
|
|
" \n",
|
|
" for event in events:\n",
|
|
" if isinstance(event, WorkflowStatusEvent) and event.state in {\n",
|
|
" WorkflowRunState.IDLE,\n",
|
|
" WorkflowRunState.IDLE_WITH_PENDING_REQUESTS,\n",
|
|
" }:\n",
|
|
" print(f\"[Workflow Status] {event.state.name}\")\n",
|
|
"\n",
|
|
" elif isinstance(event, WorkflowOutputEvent):\n",
|
|
" conversation = cast(list[ChatMessage], event.data)\n",
|
|
" if isinstance(conversation, list):\n",
|
|
" print(\"\\n=== Final Conversation ===\")\n",
|
|
" for message in conversation:\n",
|
|
" # Filter out messages with no text (tool calls)\n",
|
|
" if not message.text.strip():\n",
|
|
" continue\n",
|
|
" speaker = message.author_name or message.role.value\n",
|
|
" print(f\"- {speaker}: {message.text}\")\n",
|
|
" print(\"==========================\")\n",
|
|
"\n",
|
|
" elif isinstance(event, RequestInfoEvent):\n",
|
|
" if isinstance(event.data, HandoffUserInputRequest):\n",
|
|
" print_handoff_request(event.data)\n",
|
|
" requests.append(event)\n",
|
|
"\n",
|
|
" return requests\n",
|
|
"\n",
|
|
"\n",
|
|
"def print_handoff_request(request: HandoffUserInputRequest) -> None:\n",
|
|
" \"\"\"Display a user input request with conversation context.\"\"\"\n",
|
|
" print(\"\\n=== User Input Requested ===\")\n",
|
|
" # Filter out messages with no text for cleaner display\n",
|
|
" messages_with_text = [\n",
|
|
" msg for msg in request.conversation if msg.text.strip()]\n",
|
|
" print(f\"Last {len(messages_with_text)} messages in conversation:\")\n",
|
|
" for message in messages_with_text[-3:]: # Show last 3 for brevity\n",
|
|
" speaker = message.author_name or message.role.value\n",
|
|
" text = message.text[:100] + \\\n",
|
|
" \"...\" if len(message.text) > 100 else message.text\n",
|
|
" print(f\" {speaker}: {text}\")\n",
|
|
" print(\"============================\")\n",
|
|
"\n",
|
|
"\n",
|
|
"print(\"Helper functions defined for event processing\")"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"id": "ff51e63c",
|
|
"metadata": {},
|
|
"source": [
|
|
"## Step 6: Test Case 1 - Flight Booking Request\n",
|
|
"\n",
|
|
"Let's test our handoff workflow with a flight booking request. The customer support agent should hand off to the booking agent.\n"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "558a73d4",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"async def test_booking_handoff():\n",
|
|
" \"\"\"Test handoff workflow for flight booking requests.\"\"\"\n",
|
|
"\n",
|
|
" display(HTML(\"\"\"\n",
|
|
" <div style='padding: 20px; background: #fff3e0; border-left: 4px solid #ff9800; border-radius: 8px; margin: 20px 0;'>\n",
|
|
" <h3 style='margin: 0 0 10px 0; color: #e65100;'>Test Case 1: Flight Booking Request</h3>\n",
|
|
" <p style='margin: 0;'><strong>Expected Flow:</strong> Customer Support → Booking Agent</p>\n",
|
|
" </div>\n",
|
|
" \"\"\"))\n",
|
|
"\n",
|
|
" # Start the workflow\n",
|
|
" print(\"[User]: I want to book a flight to Paris for next month\")\n",
|
|
" events = await drain_events(\n",
|
|
" workflow.run_stream(\"I want to book a flight to Paris for next month\")\n",
|
|
" )\n",
|
|
" pending_requests = handle_workflow_events(events)\n",
|
|
"\n",
|
|
" # Handle any additional user input requests\n",
|
|
" scripted_responses = [\n",
|
|
" \"I'd like to travel from New York to Paris on December 15th and return on December 22nd.\",\n",
|
|
" \"Yes, please confirm the booking under the name John Smith.\"\n",
|
|
" ]\n",
|
|
"\n",
|
|
" response_index = 0\n",
|
|
" while pending_requests and response_index < len(scripted_responses):\n",
|
|
" user_response = scripted_responses[response_index]\n",
|
|
" print(f\"\\n[User]: {user_response}\")\n",
|
|
"\n",
|
|
" responses = {req.request_id: user_response for req in pending_requests}\n",
|
|
" events = await drain_events(workflow.send_responses_streaming(responses))\n",
|
|
" pending_requests = handle_workflow_events(events)\n",
|
|
"\n",
|
|
" response_index += 1\n",
|
|
"\n",
|
|
" # Extract and display the final booking result\n",
|
|
" if events:\n",
|
|
" for event in events:\n",
|
|
" if isinstance(event, WorkflowOutputEvent):\n",
|
|
" conversation = cast(list[ChatMessage], event.data)\n",
|
|
" for message in conversation:\n",
|
|
" if message.author_name == \"booking_agent\" and message.text.strip():\n",
|
|
" try:\n",
|
|
" booking_data = FlightBookingResult.model_validate_json(\n",
|
|
" message.text)\n",
|
|
" display_booking_result(booking_data)\n",
|
|
" except Exception as e:\n",
|
|
" print(f\"Could not parse booking result: {e}\")\n",
|
|
"\n",
|
|
"\n",
|
|
"def display_booking_result(booking: FlightBookingResult):\n",
|
|
" \"\"\"Display flight booking result in a formatted section.\"\"\"\n",
|
|
"\n",
|
|
" display(HTML(f\"\"\"\n",
|
|
" <div style='padding: 20px; background: #e8f5e9; border-radius: 8px; margin: 15px 0; border-left: 4px solid #4caf50;'>\n",
|
|
" <h3 style='margin: 0 0 15px 0; color: #2e7d32;'>✈️ Flight Booking Confirmed</h3>\n",
|
|
" <div style='display: grid; grid-template-columns: 1fr 1fr; gap: 15px; margin-bottom: 15px;'>\n",
|
|
" <div>\n",
|
|
" <strong style='color: #333;'>Booking Reference:</strong> {booking.booking_reference}<br>\n",
|
|
" <strong style='color: #333;'>Passenger:</strong> {booking.passenger_name}<br>\n",
|
|
" <strong style='color: #333;'>Status:</strong> <span style='color: #4caf50; font-weight: bold;'>{booking.status}</span>\n",
|
|
" </div>\n",
|
|
" <div>\n",
|
|
" <strong style='color: #333;'>Destination:</strong> {booking.destination}<br>\n",
|
|
" <strong style='color: #333;'>Total Cost:</strong> {booking.total_cost}<br>\n",
|
|
" <strong style='color: #333;'>Departure:</strong> {booking.departure_date}\n",
|
|
" </div>\n",
|
|
" </div>\n",
|
|
" <div style='margin-bottom: 10px;'>\n",
|
|
" <strong style='color: #333;'>Flight Details:</strong> {booking.flight_details}\n",
|
|
" </div>\n",
|
|
" <div style='background: rgba(76,175,80,0.1); padding: 10px; border-radius: 4px; margin-top: 10px;'>\n",
|
|
" <strong style='color: #2e7d32;'>✅ Success:</strong> Flight booking completed through handoff to booking specialist\n",
|
|
" </div>\n",
|
|
" </div>\n",
|
|
" \"\"\"))\n",
|
|
"\n",
|
|
"\n",
|
|
"# Run the booking test\n",
|
|
"await test_booking_handoff()"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"id": "794c1e25",
|
|
"metadata": {},
|
|
"source": [
|
|
"## Step 7: Test Case 2 - Dispute/Refund Request\n",
|
|
"\n",
|
|
"Let's test our handoff workflow with a refund request. The customer support agent should hand off to the disputes agent."
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "d61756c0",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"async def test_dispute_handoff():\n",
|
|
" \"\"\"Test handoff workflow for dispute/refund requests.\"\"\"\n",
|
|
"\n",
|
|
" display(HTML(\"\"\"\n",
|
|
" <div style='padding: 20px; background: #fff3e0; border-left: 4px solid #ff9800; border-radius: 8px; margin: 20px 0;'>\n",
|
|
" <h3 style='margin: 0 0 10px 0; color: #e65100;'>Test Case 2: Refund Request</h3>\n",
|
|
" <p style='margin: 0;'><strong>Expected Flow:</strong> Customer Support → Disputes Agent</p>\n",
|
|
" </div>\n",
|
|
" \"\"\"))\n",
|
|
"\n",
|
|
" # Start the workflow\n",
|
|
" print(\"[User]: I need to cancel my flight and get a refund\")\n",
|
|
" events = await drain_events(\n",
|
|
" workflow.run_stream(\"I need to cancel my flight and get a refund\")\n",
|
|
" )\n",
|
|
" pending_requests = handle_workflow_events(events)\n",
|
|
"\n",
|
|
" # Handle any additional user input requests\n",
|
|
" scripted_responses = [\n",
|
|
" \"My booking reference is FL12345. I can't travel due to a family emergency.\",\n",
|
|
" \"Yes, please process the full refund back to my credit card.\"\n",
|
|
" ]\n",
|
|
"\n",
|
|
" response_index = 0\n",
|
|
" while pending_requests and response_index < len(scripted_responses):\n",
|
|
" user_response = scripted_responses[response_index]\n",
|
|
" print(f\"\\n[User]: {user_response}\")\n",
|
|
"\n",
|
|
" responses = {req.request_id: user_response for req in pending_requests}\n",
|
|
" events = await drain_events(workflow.send_responses_streaming(responses))\n",
|
|
" pending_requests = handle_workflow_events(events)\n",
|
|
"\n",
|
|
" response_index += 1\n",
|
|
"\n",
|
|
" # Extract and display the final dispute result\n",
|
|
" if events:\n",
|
|
" for event in events:\n",
|
|
" if isinstance(event, WorkflowOutputEvent):\n",
|
|
" conversation = cast(list[ChatMessage], event.data)\n",
|
|
" for message in conversation:\n",
|
|
" if message.author_name == \"disputes_agent\" and message.text.strip():\n",
|
|
" try:\n",
|
|
" dispute_data = DisputeResult.model_validate_json(\n",
|
|
" message.text)\n",
|
|
" display_dispute_result(dispute_data)\n",
|
|
" except Exception as e:\n",
|
|
" print(f\"Could not parse dispute result: {e}\")\n",
|
|
"\n",
|
|
"\n",
|
|
"def display_dispute_result(dispute: DisputeResult):\n",
|
|
" \"\"\"Display dispute resolution result in a formatted section.\"\"\"\n",
|
|
"\n",
|
|
" display(HTML(f\"\"\"\n",
|
|
" <div style='padding: 20px; background: #fff3e0; border-radius: 8px; margin: 15px 0; border-left: 4px solid #ff9800;'>\n",
|
|
" <h3 style='margin: 0 0 15px 0; color: #f57c00;'>💰 Refund Processed</h3>\n",
|
|
" <div style='display: grid; grid-template-columns: 1fr 1fr; gap: 15px; margin-bottom: 15px;'>\n",
|
|
" <div>\n",
|
|
" <strong style='color: #333;'>Reference Number:</strong> {dispute.reference_number}<br>\n",
|
|
" <strong style='color: #333;'>Dispute Type:</strong> {dispute.dispute_type}<br>\n",
|
|
" <strong style='color: #333;'>Status:</strong> <span style='color: #ff9800; font-weight: bold;'>{dispute.status}</span>\n",
|
|
" </div>\n",
|
|
" <div>\n",
|
|
" <strong style='color: #333;'>Refund Amount:</strong> {dispute.refund_amount}<br>\n",
|
|
" <strong style='color: #333;'>Refund Method:</strong> {dispute.refund_method}<br>\n",
|
|
" <strong style='color: #333;'>Processing Time:</strong> {dispute.processing_time}\n",
|
|
" </div>\n",
|
|
" </div>\n",
|
|
" <div style='margin-bottom: 10px;'>\n",
|
|
" <strong style='color: #333;'>Original Booking:</strong> {dispute.original_booking}\n",
|
|
" </div>\n",
|
|
" <div style='background: rgba(255,152,0,0.1); padding: 10px; border-radius: 4px; margin-top: 10px;'>\n",
|
|
" <strong style='color: #f57c00;'>✅ Success:</strong> Refund processed through handoff to disputes specialist\n",
|
|
" </div>\n",
|
|
" </div>\n",
|
|
" \"\"\"))\n",
|
|
"\n",
|
|
" # Run the dispute test\n",
|
|
"await test_dispute_handoff()"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"id": "9a855d8d",
|
|
"metadata": {},
|
|
"source": [
|
|
"## Step 8: Test Case 3 - Trip Confirmation Request\n",
|
|
"\n",
|
|
"Let's test our handoff workflow with a trip confirmation request. The customer support agent should hand off to the trip check agent."
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "34ea038c",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"async def test_trip_check_handoff():\n",
|
|
" \"\"\"Test handoff workflow for trip confirmation requests.\"\"\"\n",
|
|
"\n",
|
|
" display(HTML(\"\"\"\n",
|
|
" <div style='padding: 20px; background: #fff3e0; border-left: 4px solid #ff9800; border-radius: 8px; margin: 20px 0;'>\n",
|
|
" <h3 style='margin: 0 0 10px 0; color: #e65100;'>Test Case 3: Trip Confirmation</h3>\n",
|
|
" <p style='margin: 0;'><strong>Expected Flow:</strong> Customer Support → Trip Check Agent</p>\n",
|
|
" </div>\n",
|
|
" \"\"\"))\n",
|
|
"\n",
|
|
" # Start the workflow\n",
|
|
" print(\"[User]: Can you confirm my travel plans are all set?\")\n",
|
|
" events = await drain_events(\n",
|
|
" workflow.run_stream(\"Can you confirm my travel plans are all set?\")\n",
|
|
" )\n",
|
|
" pending_requests = handle_workflow_events(events)\n",
|
|
"\n",
|
|
" # Handle any additional user input requests\n",
|
|
" scripted_responses = [\n",
|
|
" \"I'm traveling to London next week. My confirmation number is TR98765.\",\n",
|
|
" \"Perfect, thank you for checking everything is ready!\"\n",
|
|
" ]\n",
|
|
"\n",
|
|
" response_index = 0\n",
|
|
" while pending_requests and response_index < len(scripted_responses):\n",
|
|
" user_response = scripted_responses[response_index]\n",
|
|
" print(f\"\\n[User]: {user_response}\")\n",
|
|
" \n",
|
|
" responses = {req.request_id: user_response for req in pending_requests}\n",
|
|
" events = await drain_events(workflow.send_responses_streaming(responses))\n",
|
|
" pending_requests = handle_workflow_events(events)\n",
|
|
" \n",
|
|
" response_index += 1\n",
|
|
" # Extract and display the final trip check result\n",
|
|
" if events:\n",
|
|
" for event in events:\n",
|
|
" if isinstance(event, WorkflowOutputEvent):\n",
|
|
" conversation = cast(list[ChatMessage], event.data)\n",
|
|
" for message in conversation:\n",
|
|
" if message.author_name == \"trip_check_agent\" and message.text.strip():\n",
|
|
" try:\n",
|
|
" trip_data = TripCheckResult.model_validate_json(\n",
|
|
" message.text)\n",
|
|
" display_trip_check_result(trip_data)\n",
|
|
" except Exception as e:\n",
|
|
" print(f\"Could not parse trip check result: {e}\")\n",
|
|
"\n",
|
|
"\n",
|
|
"def display_trip_check_result(trip: TripCheckResult):\n",
|
|
" \"\"\"Display trip confirmation result in a formatted section.\"\"\"\n",
|
|
"\n",
|
|
" display(HTML(f\"\"\"\n",
|
|
" <div style='padding: 20px; background: #f3e5f5; border-radius: 8px; margin: 15px 0; border-left: 4px solid #9c27b0;'>\n",
|
|
" <h3 style='margin: 0 0 15px 0; color: #7b1fa2;'>🎯 Trip Confirmed</h3>\n",
|
|
" <div style='display: grid; grid-template-columns: 1fr 1fr; gap: 15px; margin-bottom: 15px;'>\n",
|
|
" <div>\n",
|
|
" <strong style='color: #333;'>Trip Reference:</strong> {trip.trip_reference}<br>\n",
|
|
" <strong style='color: #333;'>Destination:</strong> {trip.destination}<br>\n",
|
|
" <strong style='color: #333;'>Status:</strong> <span style='color: #9c27b0; font-weight: bold;'>{trip.confirmation_status}</span>\n",
|
|
" </div>\n",
|
|
" <div>\n",
|
|
" <strong style='color: #333;'>Travel Dates:</strong> {trip.travel_dates}<br>\n",
|
|
" <strong style='color: #333;'>Contact Info:</strong> {trip.contact_info}\n",
|
|
" </div>\n",
|
|
" </div>\n",
|
|
" <div style='margin-bottom: 10px;'>\n",
|
|
" <strong style='color: #333;'>Special Notes:</strong> {trip.special_notes}\n",
|
|
" </div>\n",
|
|
" <div style='background: rgba(156,39,176,0.1); padding: 10px; border-radius: 4px; margin-top: 10px;'>\n",
|
|
" <strong style='color: #7b1fa2;'>✅ Success:</strong> Trip confirmed through handoff to trip check specialist\n",
|
|
" </div>\n",
|
|
" </div>\n",
|
|
" \"\"\"))\n",
|
|
"\n",
|
|
"\n",
|
|
"# Run the trip check test\n",
|
|
"await test_trip_check_handoff()\n",
|
|
"\n",
|
|
" "
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"id": "ae3cb474",
|
|
"metadata": {},
|
|
"source": [
|
|
"## Step 9: Workflow Analysis - Understanding Handoff Flow\n"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "0b21ce2b",
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"async def analyze_handoff_patterns():\n",
|
|
" \"\"\"Analyze different handoff patterns and routing decisions.\"\"\"\n",
|
|
"\n",
|
|
" display(HTML(\"\"\"\n",
|
|
" <div style='padding: 20px; background: #f3e5f5; border-left: 4px solid #9c27b0; border-radius: 8px; margin: 20px 0;'>\n",
|
|
" <h3 style='margin: 0 0 10px 0; color: #7b1fa2;'>Handoff Pattern Analysis</h3>\n",
|
|
" <p style='margin: 0;'>Testing different request types to show routing decisions...</p>\n",
|
|
" </div>\n",
|
|
" \"\"\"))\n",
|
|
"\n",
|
|
" test_requests = [\n",
|
|
" \"I want to book a round-trip flight to Tokyo\",\n",
|
|
" \"I need a refund for my cancelled flight\",\n",
|
|
" \"Please check if my travel itinerary is confirmed\",\n",
|
|
" \"Can you help me with a billing dispute?\"\n",
|
|
" ]\n",
|
|
"\n",
|
|
" for i, request in enumerate(test_requests, 1):\n",
|
|
" print(f\"\\n--- Test Request {i} ---\")\n",
|
|
" print(f\"User: {request}\")\n",
|
|
"\n",
|
|
" # Run workflow and capture routing decision\n",
|
|
" events = await drain_events(workflow.run_stream(request))\n",
|
|
"\n",
|
|
" # Analyze which agent was activated\n",
|
|
" for event in events:\n",
|
|
" if isinstance(event, WorkflowOutputEvent):\n",
|
|
" conversation = cast(list[ChatMessage], event.data)\n",
|
|
" for message in conversation:\n",
|
|
" if message.author_name == \"customer_support_agent\":\n",
|
|
" print(f\"Support Agent: {message.text[:100]}...\")\n",
|
|
" elif message.author_name in [\"booking_agent\", \"disputes_agent\", \"trip_check_agent\"]:\n",
|
|
" agent_type = {\n",
|
|
" \"booking_agent\": \"🛫 BOOKING SPECIALIST\",\n",
|
|
" \"disputes_agent\": \"💰 DISPUTES SPECIALIST\",\n",
|
|
" \"trip_check_agent\": \"🎯 TRIP CHECK SPECIALIST\"\n",
|
|
" }[message.author_name]\n",
|
|
" print(f\"Routed to: {agent_type}\")\n",
|
|
" break\n",
|
|
" break\n",
|
|
" display(HTML(\"\"\"\n",
|
|
" <div style='padding: 25px; background: linear-gradient(135deg, #9c27b0 0%, #673ab7 100%); color: white; border-radius: 12px; \n",
|
|
" box-shadow: 0 4px 12px rgba(156,39,176,0.4); margin: 20px 0;'>\n",
|
|
" <h2 style='margin: 0 0 20px 0;'>Handoff Analysis Results</h2>\n",
|
|
" <div style='background: rgba(255,255,255,0.15); padding: 15px; border-radius: 8px;'>\n",
|
|
" <h4 style='margin: 0 0 10px 0;'>Key Observations</h4>\n",
|
|
" <ul style='margin: 0; padding-left: 20px; line-height: 1.6;'>\n",
|
|
" <li><strong>Dynamic Routing:</strong> Customer support agent analyzes request intent</li>\n",
|
|
" <li><strong>Context Preservation:</strong> Full conversation history maintained</li>\n",
|
|
" <li><strong>Specialist Focus:</strong> Each agent handles their expertise area</li>\n",
|
|
" <li><strong>Seamless Handoff:</strong> Users don't need to repeat information</li>\n",
|
|
" </ul>\n",
|
|
" </div>\n",
|
|
" </div>\n",
|
|
" \"\"\"))\n",
|
|
"\n",
|
|
" # Run the analysis\n",
|
|
"await analyze_handoff_patterns()"
|
|
]
|
|
}
|
|
],
|
|
"metadata": {
|
|
"kernelspec": {
|
|
"display_name": ".venv (3.12.12)",
|
|
"language": "python",
|
|
"name": "python3"
|
|
},
|
|
"language_info": {
|
|
"codemirror_mode": {
|
|
"name": "ipython",
|
|
"version": 3
|
|
},
|
|
"file_extension": ".py",
|
|
"mimetype": "text/x-python",
|
|
"name": "python",
|
|
"nbconvert_exporter": "python",
|
|
"pygments_lexer": "ipython3",
|
|
"version": "3.12.12"
|
|
}
|
|
},
|
|
"nbformat": 4,
|
|
"nbformat_minor": 5
|
|
}
|