Multi-Agent Architecture: Coordinator and Specialist Agents

from google import genai
from google.genai import types

client = genai.Client()
model = "gemini-2.5-flash"

# Helper to execute a specialist subagent
def run_subagent(name: str, system_prompt: str, task: str, tools: list[types.Tool] | None = None) -> str:
    """Executes a specialist agent with isolated context and restricted tools."""
    print(f"\n🤖 [Specialist: {name}] Activated with task:\n   -> \"{task}\"")
    
    config = types.GenerateContentConfig(
        system_instruction=system_prompt,
        tools=tools,
        temperature=0.0,
        max_output_tokens=600
    )
    
    response = client.models.generate_content(
        model=model,
        contents=task,
        config=config
    )
    
    finding = response.text or "Completed without text output."
    print(f"📋 [Specialist: {name}] Findings:\n   {finding.strip()}")
    return finding

# Specialist 1: Logistics & Orders
order_specialist_system = """
You are the Order Logistics Specialist.
Your job is to analyze delivery tracking, shipping carriers, and package condition reports.
Be factual, concise, and report the order state.
"""

# Specialist 2: Billing & Payments
billing_specialist_system = """
You are the Billing & Accounts Specialist.
Your job is to audit payment ledgers, detect duplicate charges, and confirm credit card transaction states.
"""

# Specialist 3: Store Policy & Compliance
policy_specialist_system = """
You are the Store Policy Compliance Officer.
Analyze return requests against company guidelines (30-day window, damage waivers).
State clearly whether the claim is approved or rejected under policy.
"""

def handle_customer_inquiry(customer_message: str) -> str:
    print(f"📥 Received Customer Case: \"{customer_message}\"")
    
    # 1. Step A: Dispatch to Logistics Specialist
    order_findings = run_subagent(
        name="LogisticsAgent",
        system_prompt=order_specialist_system,
        task=f"Customer message: '{customer_message}'. Summarize shipment and delivery facts."
    )
    
    # 2. Step B: Dispatch to Billing Specialist
    billing_findings = run_subagent(
        name="BillingAgent",
        system_prompt=billing_specialist_system,
        task=f"Customer message: '{customer_message}'. Check if there is an apparent duplicate charge or invoice discrepancy."
    )
    
    # 3. Step C: Dispatch to Policy Specialist
    policy_findings = run_subagent(
        name="PolicyAgent",
        system_prompt=policy_specialist_system,
        task=f"Customer message: '{customer_message}'. Findings from logistics: {order_findings}. Does this qualify for replacement/refund?"
    )
    
    # 4. Coordinator Synthesizes Final Customer Response
    coordinator_prompt = f"""
    You are the Customer Support Coordinator.
    Write a helpful, empathetic, and unified response to the customer based on these specialist findings:

    Customer Message: "{customer_message}"
    Logistics Findings: {order_findings}
    Billing Findings: {billing_findings}
    Policy Findings: {policy_findings}

    Provide a clear, single cohesive response addressing all their concerns.
    """
    
    response = client.models.generate_content(
        model=model,
        contents=coordinator_prompt,
        config=types.GenerateContentConfig(temperature=0.2, max_output_tokens=500)
    )
    
    return response.text

case = "Hi, my jacket arrived yesterday with a torn pocket (ORD-77819), and I also see two charges of $149 on my credit card. I want this resolved!"
final_reply = handle_customer_inquiry(case)

print(f"\n================ FINAL CUSTOMER REPLY ================\n{final_reply}")

# --- Execution Output ---
# 🤖 [Specialist: LogisticsAgent] Activated with task:
#    -> "Customer message: 'Hi, my jacket arrived yesterday...'. Summarize shipment and delivery facts."
# 📋 [Specialist: LogisticsAgent] Findings:
#    Order ORD-77819 was delivered yesterday. Item: Blue Winter Jacket ($149.00). Torn pocket reported.
#
# 🤖 [Specialist: BillingAgent] Activated with task:
#    -> "Customer message: 'Hi, my jacket arrived yesterday...'. Check if there is an apparent duplicate charge."
# 📋 [Specialist: BillingAgent] Findings:
#    Two charges of $149.00 detected. The second transaction appears to be a temporary authorization hold.
#
# 🤖 [Specialist: PolicyAgent] Activated with task:
#    -> "Customer message: 'Hi, my jacket arrived yesterday...'. Does this qualify for replacement/refund?"
# 📋 [Specialist: PolicyAgent] Findings:
#    Approved. Damaged items reported within 7 days of delivery qualify for full refund or replacement.
#
# ================ FINAL CUSTOMER REPLY ================
# Dear Customer,
#
# I have investigated both issues regarding your order ORD-77819:
# 1. Torn Pocket (Blue Winter Jacket): Your jacket arrived damaged within the return window,
#    making you eligible for an immediate replacement or full refund.
# 2. Duplicate $149 Charge: The second charge is a pending authorization hold that will
#    expire and drop off your statement within 48 hours.
#
# Would you like me to process a replacement jacket or issue a return label for a full refund?