import { AgentBuilder, OpenAIAdapter } from 'Weave';

const adapter = new OpenAIAdapter({ apiKey: process.env.OPENAI_API_KEY!, model: 'gpt-4o' });

// 1. Specialized Sub-Agents
const billingAgent = new AgentBuilder()
  .name('BillingAgent')
  .instructions('Handle subscription upgrades, invoices, and payment refunds.')
  .llm(adapter)
  .build();

const techAgent = new AgentBuilder()
  .name('TechSupportAgent')
  .instructions('Debug API integration errors and code bugs.')
  .llm(adapter)
  .build();

// 2. Primary Triage Agent
const triageAgent = new AgentBuilder()
  .name('TriageAgent')
  .instructions('Route customer inquiries to BillingAgent or TechSupportAgent based on intent.')
  .llm(adapter)
  .subAgents([billingAgent, techAgent])
  .maxHandoffDepth(5) // Customize max delegation depth (defaults to 3)
  .build();

async function run() {
  const result = await triageAgent.run('I need a copy of my invoice for last month.');

  console.log(`Active Responding Agent: ${result.activeAgentName}`);
  console.log(`Response: ${result.output}`);
  console.log(`Handoff Path: ${result.handoffHistory?.join(' -> ')}`);
}

run().catch(console.error);