Agents

secret llm_creds {
  vars: [ANTHROPIC_API_KEY]
}

agent support_agent {
  label: "Support Agent"
  description: "Triages support tickets and drafts responses."
  provider: anthropic
  model: "claude-3-5-sonnet-20241022"
  secrets: llm_creds
  temperature: 0.3
  maxTokens: 2048
  maxSteps: 15
  system: @ts {
    return "You are a support agent. Use the available tools to investigate and resolve tickets."
  }
  tools: [lookup_ticket, post_reply]

  profile escalation {
    description: "Handles escalated tickets with expanded tool access"
    system: @ts {
      return "You are a senior support agent. Be thorough. Document every action you take."
    }
    tools: [lookup_ticket, post_reply]
  }
}

secret vendor_keys {
  vars: [OPENROUTER_API_KEY]
}

agent researcher {
  label: "Researcher"
  secrets: vendor_keys
  model: "openai/gpt-4o-mini"
  tools: [search_kb]
  system: @ts {
    return "Research the question and return concise findings with sources."
  }
}

agent writer {
  label: "Writer"
  secrets: vendor_keys
  model: "openai/gpt-4o-mini"
  system: @ts {
    return "Turn findings into clear, well-structured prose."
  }
}

agent orchestrator {
  label: "Orchestrator"
  secrets: vendor_keys
  model: "google/gemini-3.1-flash-lite"
  maxSteps: 16
  team: [researcher, writer]
  system: @ts {
    return [
      "You coordinate specialists.",
      "Call a team tool with a clear task describing what to do.",
      "Relay the specialist's answer plainly.",
    ].join("\n")
  }
}

agent data_agent {
  model: "gpt-4o"
  secrets: oai_creds

  sandbox: {
    cpus: 2
    memoryMiB: 2048
    diskGiB: 10
    autoStopMinutes: 30
    autoArchiveMinutes: 1440
    autoDeleteMinutes: -1
    ephemeral: false
  }
}

node triage {
  type: agent
  label: "Triage ticket"
  agent: support_agent
  profile: escalation
  prompt: @ts {
    return "Investigate and resolve this ticket: " + context.nodes.root.output.description
  }
  schema: @json {
    {
      "type": "object",
      "required": ["resolution", "category"],
      "properties": {
        "resolution": { "type": "string" },
        "category": { "type": "string" }
      },
      "additionalProperties": false
    }
  }
}

workflow lookup_ticket {
  label: "Look up ticket"
  description: "Retrieve a support ticket by its ID. Returns the ticket title, description, and current status."

  root {
    type: code
    label: "Entry"
    inputSchema: @json {
      {
        "type": "object",
        "required": ["ticket_id"],
        "properties": { "ticket_id": { "type": "string" } },
        "additionalProperties": false
      }
    }
    outputSchema: @json {
      {
        "type": "object",
        "required": ["ticket_id"],
        "properties": { "ticket_id": { "type": "string" } },
        "additionalProperties": false
      }
    }
    code: @ts {
      return { ticket_id: context.nodes.root.input.ticket_id }
    }
  }

  node fetch {
    type: postgres
    label: "Fetch from DB"
    postgres: main_db
    select: @sql {
      SELECT title, description, status FROM tickets WHERE id = {{ticket_id}}
    }
    params: @ts {
      return { ticket_id: context.nodes.root.output.ticket_id }
    }
    schema: @json {
      {
        "type": "array",
        "items": {
          "type": "object",
          "properties": {
            "title": { "type": "string" },
            "description": { "type": "string" },
            "status": { "type": "string" }
          }
        }
      }
    }
  }

  flow {
    root -> fetch
  }
}

swirls chat start support_agent

secret llm_creds {
  vars: [ANTHROPIC_API_KEY]
}

agent triage_agent {
  label: "Triage Agent"
  description: "Classifies and routes support tickets."
  provider: anthropic
  model: "claude-3-5-sonnet-20241022"
  secrets: llm_creds
  maxSteps: 10
  system: @ts {
    return "You are a support triage agent. Classify the ticket and determine routing."
  }
  tools: [fetch_kb]

  profile critical {
    description: "For tickets marked urgent"
    system: @ts {
      return "You are handling a critical ticket. Escalate quickly. Be concise."
    }
    tools: [fetch_kb]
  }
}

workflow fetch_kb {
  label: "Search knowledge base"
  description: "Search the knowledge base for relevant articles. Pass a keyword query."

  root {
    type: code
    label: "Entry"
    inputSchema: @json {
      {
        "type": "object",
        "required": ["query"],
        "properties": { "query": { "type": "string" } },
        "additionalProperties": false
      }
    }
    outputSchema: @json {
      {
        "type": "object",
        "required": ["articles"],
        "properties": {
          "articles": { "type": "array", "items": { "type": "string" } }
        },
        "additionalProperties": false
      }
    }
    code: @ts {
      return { articles: ["Article A", "Article B"] }
    }
  }
}

workflow triage_ticket {
  label: "Triage Ticket"

  root {
    type: code
    label: "Entry"
    inputSchema: @json {
      {
        "type": "object",
        "required": ["subject", "body", "priority"],
        "properties": {
          "subject": { "type": "string" },
          "body": { "type": "string" },
          "priority": { "type": "string" }
        },
        "additionalProperties": false
      }
    }
    outputSchema: @json {
      {
        "type": "object",
        "required": ["subject", "body", "priority"],
        "properties": {
          "subject": { "type": "string" },
          "body": { "type": "string" },
          "priority": { "type": "string" }
        },
        "additionalProperties": false
      }
    }
    code: @ts {
      return {
        subject: context.nodes.root.input.subject,
        body: context.nodes.root.input.body,
        priority: context.nodes.root.input.priority
      }
    }
  }

  node run_agent {
    type: agent
    label: "Triage with agent"
    agent: triage_agent
    profile: critical
    prompt: @ts {
      const { subject, body } = context.nodes.root.output
      return `Triage this ticket.\nSubject: ${subject}\nBody: ${body}`
    }
    schema: @json {
      {
        "type": "object",
        "required": ["category", "suggested_response"],
        "properties": {
          "category": { "type": "string" },
          "suggested_response": { "type": "string" }
        },
        "additionalProperties": false
      }
    }
  }

  flow {
    root -> run_agent
  }
}

// Incorrect
node triage {
  type: agent
  agent: triage_agent
  prompt: @ts { return "Triage this ticket" }
  outputSchema: @json { { "type": "object" } }
}

// Correct
node triage {
  type: agent
  agent: triage_agent
  prompt: @ts { return "Triage this ticket" }
  schema: @json { { "type": "object" } }
}

// Incorrect
agent bad_agent {
  model: "gpt-4o"
  secrets: "llm_creds"
}

// Correct
agent good_agent {
  model: "gpt-4o"
  secrets: llm_creds
}

// Incorrect: provider is anthropic but secret declares the wrong key
secret bad_creds {
  vars: [OPENROUTER_API_KEY]
}

agent bad_agent {
  provider: anthropic
  model: "claude-3-5-sonnet-20241022"
  secrets: bad_creds
}

// Correct
secret llm_creds {
  vars: [ANTHROPIC_API_KEY]
}

agent good_agent {
  provider: anthropic
  model: "claude-3-5-sonnet-20241022"
  secrets: llm_creds
}

// Incorrect
agent bad_agent {
  model: "gpt-4o"
  secrets: llm_creds
  system: "You are a helpful assistant."
}

// Correct
agent good_agent {
  model: "gpt-4o"
  secrets: llm_creds
  system: @ts {
    return "You are a helpful assistant."
  }
}

// Incorrect: post_reply is not in the agent's tools list
agent bad_agent {
  model: "gpt-4o"
  secrets: llm_creds
  tools: [lookup_ticket]

  profile escalation {
    tools: [lookup_ticket, post_reply]
  }
}

// Correct
agent good_agent {
  model: "gpt-4o"
  secrets: llm_creds
  tools: [lookup_ticket, post_reply]

  profile escalation {
    tools: [lookup_ticket, post_reply]
  }
}