05 Tools and Context

import { AgentBuilder, createTool } from "@anvia/core";
import { z } from "zod";
import { model } from "@/app/ai/support-agent";
import { orders } from "@/app/db/orders";

type SupportScope = {
  userId: string;
};

export function createSupportAgent(scope: SupportScope) {
  const lookupOrder = createTool({
    name: "lookup_order",
    description: "Look up one order owned by the current user.",
    input: z.object({
      orderId: z.string(),
    }),
    output: z.object({
      status: z.string(),
    }),
    async execute({ orderId }) {
      return orders.findForUser(scope.userId, orderId);
    },
  });

  return new AgentBuilder("support", model)
    .instructions("Use tools for account-specific data.")
    .tool(lookupOrder)
    .defaultMaxTurns(3)
    .build();
}

import { createSupportAgent } from "@/app/ai/create-support-agent";
import { requireUser } from "@/app/auth";

export async function POST(request: Request): Promise<Response> {
  const user = await requireUser(request);
  const { message } = (await request.json()) as { message?: string };

  if (!message?.trim()) {
    return Response.json({ error: "message is required" }, { status: 400 });
  }

  const agent = createSupportAgent({ userId: user.id });
  const response = await agent.prompt(message).send();

  return Response.json({ output: response.output });
}

const agent = new AgentBuilder("support", model)
  .instructions("Use retrieved support docs when relevant.")
  .dynamicContext(supportDocsIndex, {
    topK: 3,
    threshold: 0.7,
  })
  .tool(lookupOrder)
  .build();