WOLBΛRG

Example — Vercel AI memory

Use @wolbarg/vercel-ai middleware with wrapLanguageModel for automatic recall and remember (AI SDK v7+).

What is it?

Production path for apps on the Vercel AI SDK: wrap any provider model with wolbargMiddleware so recall and remember happen automatically.

Runnable copy: examples/adapters/vercel-ai/ (npm start, npm run start:stream, npm run start:tools).

Requires AI SDK v7+.

Install

npm install wolbarg @wolbarg/vercel-ai ai @ai-sdk/openai

Example usage

import { generateText, wrapLanguageModel } from "ai";
import { openai } from "@ai-sdk/openai";
import { wolbarg, sqlite, openaiEmbedding } from "wolbarg";
import { wolbargMiddleware } from "@wolbarg/vercel-ai";

const memory = wolbarg({
  organization: "my-app",
  storage: sqlite("./memory.db"),
  embedding: openaiEmbedding({
    apiKey: process.env.OPENAI_API_KEY!,
    model: "text-embedding-3-small",
  }),
});
await memory.ready();

const model = wrapLanguageModel({
  model: openai("gpt-4.1-mini"),
  middleware: wolbargMiddleware({
    memory,
    agent: "assistant",
  }),
});

const { text } = await generateText({
  model,
  messages: [{ role: "user", content: "What do I prefer?" }],
});

No manual recall / remember around generateText — the middleware owns that loop.

Streaming + tools

cd examples/adapters/vercel-ai
npm run start:stream   # streamText
npm run start:tools    # multi-step tools, final-step remember