05 Tools and Context

import type { Handle } from "@sveltejs/kit";

export const handle: Handle = async ({ event, resolve }) => {
  const session = await auth.getSession(event.request);
  event.locals.userId = session?.user.id;
  return resolve(event);
};

declare global {
  namespace App {
    interface Locals {
      userId?: string;
    }
  }
}

import { createTool } from "@anvia/core";
import { z } from "zod";

export function createAccountTool(input: { userId: string }) {
  return createTool({
    name: "get_account_status",
    description: "Read the authenticated user's account status.",
    input: z.object({}),
    output: z.object({ plan: z.string(), openTickets: z.number() }),
    async execute() {
      return db.account.findStatus({ userId: input.userId });
    },
  });
}

import { json, type RequestHandler } from "@sveltejs/kit";
import { supportAgent } from "$lib/server/ai/support-agent";
import { createAccountTool } from "$lib/server/ai/tools";

export const POST: RequestHandler = async ({ request, locals }) => {
  if (!locals.userId) {
    return json({ error: { code: "unauthorized" } }, { status: 401 });
  }

  const { message } = await request.json();
  const response = await supportAgent
    .prompt(message)
    .tool(createAccountTool({ userId: locals.userId }))
    .context({ userId: locals.userId })
    .send();

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

const documents = await knowledge.search({
  query: message,
  filter: { userId: locals.userId },
  limit: 5,
});

const response = await supportAgent.prompt(message).documents(documents).send();