AI SDK

import { convertToModelMessages, stepCountIs, streamText } from "ai";
import { vendoTools } from "@vendoai/vendo/ai-sdk";
import { resolvePrincipal, vendo } from "@/lib/vendo";

export async function POST(req: Request) {
  const { messages } = await req.json();
  const caller = await resolvePrincipal(req);
  if (!caller) return new Response("Unauthorized", { status: 401 });
  return streamText({
    model: yourModel,            // your model, unchanged
    messages: await convertToModelMessages(messages),
    stopWhen: stepCountIs(5),    // the AI SDK stops after one step by default
    system: "…your system prompt as it is",
    tools: { ...yourTools, ...(await vendoTools(vendo, { principal: caller })) },
  }).toUIMessageStreamResponse();
}

import { isVendoToolPart, VendoToolResult } from "@vendoai/vendo/react";

if (isVendoToolPart(part)) {
  return part.state === "output-available"
    ? <VendoToolResult key={key} output={part.output} />
    : <div key={key}>Building…</div>;
}
// your own parts fall through to your own rendering