AI Agent

// src/agents/support-agent/config.ts
import { defineAgent } from "@/lib/agent-runtime";

export const supportAgent = defineAgent({
  name: "customer-support",
  model: "claude-sonnet-4-6",
  reasoning: "react",
  orchestration: "single",

  systemPrompt: `You are a customer support agent for Acme Corp.
    You can look up orders, process refunds, and escalate to humans.
    Always verify the customer's identity before taking actions.`,

  tools: [
    "lookup-order",
    "process-refund",
    "send-email",
    "escalate-to-human",
  ],

  memory: {
    conversation: { maxTokens: 8_000, summariseAfter: 20 },
    knowledge: { collection: "support-docs", topK: 5 },
  },

  guardrails: {
    input: ["pii-detection", "topic-filter"],
    output: ["pii-redaction", "tone-check"],
    spend: { maxTokensPerRequest: 4_000, maxTokensPerDay: 500_000 },
  },
});

// src/agents/support-agent/tools/lookup-order.ts
import { defineTool } from "@/lib/agent-runtime";
import { z } from "zod";
import { db } from "@/server/db";
import { orders } from "@/server/db/schema/orders";
import { eq } from "drizzle-orm";

export const lookupOrder = defineTool({
  name: "lookup-order",
  description: "Look up an order by ID and return its status, items, and shipping info.",
  parameters: z.object({
    orderId: z.string().describe("The order ID to look up (e.g., ORD-12345)"),
  }),
  returns: z.object({
    id: z.string(),
    status: z.enum(["pending", "shipped", "delivered", "cancelled"]),
    items: z.array(z.object({ name: z.string(), quantity: z.number() })),
    shippingAddress: z.string(),
    trackingUrl: z.string().nullable(),
  }),
  execute: async ({ orderId }) => {
    const order = await db.query.orders.findFirst({
      where: eq(orders.id, orderId),
      with: { items: true },
    });
    if (!order) throw new Error(`Order ${orderId} not found`);
    return {
      id: order.id,
      status: order.status,
      items: order.items.map((i) => ({ name: i.name, quantity: i.qty })),
      shippingAddress: order.shippingAddress,
      trackingUrl: order.trackingUrl,
    };
  },
});

// src/agents/support-agent/guardrails/pii-detection.ts
import { defineGuardrail } from "@/lib/agent-runtime";

export const piiDetection = defineGuardrail({
  name: "pii-detection",
  phase: "input",
  description: "Detect and flag PII in user messages before processing.",
  execute: async ({ message }) => {
    const patterns = [
      { label: "SSN", regex: /\b\d{3}-\d{2}-\d{4}\b/ },
      { label: "Credit Card", regex: /\b\d{4}[- ]?\d{4}[- ]?\d{4}[- ]?\d{4}\b/ },
      { label: "Email", regex: /[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}/ },
    ];

    const detected = patterns
      .filter(({ regex }) => regex.test(message))
      .map(({ label }) => label);

    if (detected.length > 0) {
      return {
        action: "flag",
        reason: `PII detected: ${detected.join(", ")}. Redacting before processing.`,
        transformedMessage: message.replace(
          /\b\d{3}-\d{2}-\d{4}\b/g,
          "[SSN REDACTED]"
        ),
      };
    }

    return { action: "pass" };
  },
});

# Invoke the agent via the tRPC-compatible REST endpoint
curl -X POST https://your-app.vercel.app/api/trpc/agent.invoke \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer <your-api-key>" \
  -d '{
    "json": {
      "agentId": "customer-support",
      "sessionId": "sess_abc123",
      "message": "What is the status of order ORD-98765?"
    }
  }'