Vercel AI SDK

import { openai } from "@ai-sdk/openai";
import { wrapLanguageModel } from "ai";
import { pandaProbeMiddleware } from "pandaprobe/integrations/vercel-ai";

const model = wrapLanguageModel({
  model: openai("gpt-5.6-terra"),
  middleware: pandaProbeMiddleware(),
});

import { generateText } from "ai";
import { flush } from "pandaprobe";

const { text } = await generateText({
  model,
  prompt: "What is the capital of France?",
});

console.log(text);
await flush();

import { streamText } from "ai";

const result = streamText({
  model,
  prompt: "Count to three.",
});

for await (const delta of result.textStream) {
  process.stdout.write(delta);
}

const model = wrapLanguageModel({
  model: openai("gpt-5.6-terra"),
  middleware: pandaProbeMiddleware({ name: "support-agent-generation" }),
});

import { withTrace } from "pandaprobe";

await withTrace(
  "support-agent",
  { input: { messages: [{ role: "user", content: "How do I reset my password?" }] } },
  async (trace) => {
    const result = await generateText({ model, prompt: "How do I reset my password?" });
    trace.setOutput({ messages: [{ role: "assistant", content: result.text }] });
  },
);