feat(vercel-ai-sdk): migrate to Vercel AI SDK v6 (#4741)

Co-authored-by: kartik-mem0 <kartik.labhshetwar@mem0.ai>
This commit is contained in:
Abhishek Chauhan
2026-06-10 18:41:15 +05:30
committed by GitHub
co-authored by kartik-mem0
parent b819d95d18
commit b4a50e3dc8
24 changed files with 1766 additions and 1418 deletions
+2 -2
View File
@@ -1,5 +1,5 @@
import { generateText, streamText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider';
import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { simulateStreamingMiddleware, wrapLanguageModel } from 'ai';
import { addMemories } from "../src";
import { testConfig } from "../config/test-config";
@@ -21,7 +21,7 @@ describe.each(testConfig.providers)('TESTS: Generate/Stream Text with model %s',
beforeAll(async () => {
// Add some test memories before all tests
const messages: LanguageModelV2Prompt = [
const messages: LanguageModelV3Prompt = [
{
role: "user",
content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { createMem0, retrieveMemories } from "../../src";
import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider';
import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config";
import { createCohere } from "@ai-sdk/cohere";
@@ -23,7 +23,7 @@ describe("COHERE MEM0 Tests", () => {
});
it("should retrieve memories and generate text using COHERE provider", async () => {
const messages: LanguageModelV2Prompt = [
const messages: LanguageModelV3Prompt = [
{
role: "user",
content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { createMem0 } from "../../src";
import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider';
import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config";
describe("GOOGLE MEM0 Tests", () => {
@@ -23,7 +23,7 @@ describe("GOOGLE MEM0 Tests", () => {
});
it("should retrieve memories and generate text using Google provider", async () => {
const messages: LanguageModelV2Prompt = [
const messages: LanguageModelV3Prompt = [
{
role: "user",
content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { createMem0, retrieveMemories } from "../../src";
import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider';
import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config";
import { createGroq } from "@ai-sdk/groq";
@@ -24,7 +24,7 @@ describe("GROQ MEM0 Tests", () => {
});
it("should retrieve memories and generate text using GROQ provider", async () => {
const messages: LanguageModelV2Prompt = [
const messages: LanguageModelV3Prompt = [
{
role: "user",
content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { createMem0 } from "../../src";
import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider';
import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config";
describe("OPENAI MEM0 Tests", () => {
@@ -22,7 +22,7 @@ describe("OPENAI MEM0 Tests", () => {
});
it("should retrieve memories and generate text using Mem0 OpenAI provider", async () => {
const messages: LanguageModelV2Prompt = [
const messages: LanguageModelV3Prompt = [
{
role: "user",
content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { createMem0, retrieveMemories } from "../../src";
import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider';
import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config";
import { createAnthropic } from "@ai-sdk/anthropic";
@@ -24,7 +24,7 @@ describe("ANTHROPIC MEM0 Tests", () => {
});
it("should retrieve memories and generate text using ANTHROPIC provider", async () => {
const messages: LanguageModelV2Prompt = [
const messages: LanguageModelV3Prompt = [
{
role: "user",
content: [
+4 -4
View File
@@ -1,5 +1,5 @@
import { addMemories, retrieveMemories } from "../src";
import { LanguageModelV2Prompt } from '@ai-sdk/provider';
import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../config/test-config";
describe("Memory Core Functions", () => {
@@ -8,7 +8,7 @@ describe("Memory Core Functions", () => {
describe("addMemories", () => {
it("should successfully add memories and return correct format", async () => {
const messages: LanguageModelV2Prompt = [
const messages: LanguageModelV3Prompt = [
{
role: "user",
content: [
@@ -34,7 +34,7 @@ describe("Memory Core Functions", () => {
describe("retrieveMemories", () => {
beforeEach(async () => {
// Add some test memories before each retrieval test
const messages: LanguageModelV2Prompt = [
const messages: LanguageModelV3Prompt = [
{
role: "user",
content: [
@@ -56,7 +56,7 @@ describe("Memory Core Functions", () => {
});
it("should retrieve memories with array of prompts", async () => {
const messages: LanguageModelV2Prompt = [
const messages: LanguageModelV3Prompt = [
{
role: "user",
content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { retrieveMemories } from "../../src";
import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider';
import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config";
import { createAnthropic } from "@ai-sdk/anthropic";
@@ -20,7 +20,7 @@ describe("ANTHROPIC Integration Tests", () => {
});
it("should retrieve memories and generate text using ANTHROPIC provider", async () => {
const messages: LanguageModelV2Prompt = [
const messages: LanguageModelV3Prompt = [
{
role: "user",
content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { retrieveMemories } from "../../src";
import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider';
import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config";
import { createCohere } from "@ai-sdk/cohere";
@@ -19,7 +19,7 @@ describe("COHERE Integration Tests", () => {
});
it("should retrieve memories and generate text using COHERE provider", async () => {
const messages: LanguageModelV2Prompt = [
const messages: LanguageModelV3Prompt = [
{
role: "user",
content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { retrieveMemories } from "../../src";
import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider';
import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config";
import { createGoogleGenerativeAI } from "@ai-sdk/google";
@@ -19,7 +19,7 @@ describe("GOOGLE Integration Tests", () => {
});
it("should retrieve memories and generate text using Google provider", async () => {
const messages: LanguageModelV2Prompt = [
const messages: LanguageModelV3Prompt = [
{
role: "user",
content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { retrieveMemories } from "../../src";
import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider';
import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config";
import { createGroq } from "@ai-sdk/groq";
@@ -20,7 +20,7 @@ describe("GROQ Integration Tests", () => {
});
it("should retrieve memories and generate text using GROQ provider", async () => {
const messages: LanguageModelV2Prompt = [
const messages: LanguageModelV3Prompt = [
{
role: "user",
content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { retrieveMemories } from "../../src";
import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider';
import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config";
import { createOpenAI } from "@ai-sdk/openai";
@@ -19,7 +19,7 @@ describe("OPENAI Integration Tests", () => {
});
it("should retrieve memories and generate text using OpenAI provider", async () => {
const messages: LanguageModelV2Prompt = [
const messages: LanguageModelV3Prompt = [
{
role: "user",
content: [
@@ -0,0 +1,486 @@
import { createMem0, Mem0Provider } from "../src";
import { Mem0GenericLanguageModel } from "../src/mem0-generic-language-model";
import { Mem0ClassSelector } from "../src/mem0-provider-selector";
import Mem0AITextGenerator from "../src/provider-response-provider";
import { Mem0ConfigSettings } from "../src/mem0-types";
// Mock fetch globally for memory API tests
const mockFetch = jest.fn();
global.fetch = mockFetch;
describe("V3 Provider Contract", () => {
beforeEach(() => {
mockFetch.mockReset();
});
describe("Mem0Provider factory", () => {
let provider: Mem0Provider;
beforeEach(() => {
provider = createMem0({
provider: "openai",
mem0ApiKey: "test-key",
apiKey: "test-openai-key",
});
});
it("should have specificationVersion v3", () => {
expect(provider.specificationVersion).toBe("v3");
});
it("should expose languageModel, chat, and completion methods", () => {
expect(typeof provider.languageModel).toBe("function");
expect(typeof provider.chat).toBe("function");
expect(typeof provider.completion).toBe("function");
});
it("should be callable as a function", () => {
const model = provider("gpt-4o");
expect(model).toBeDefined();
expect(model.specificationVersion).toBe("v3");
});
it("should throw when called with new keyword", () => {
expect(() => new (provider as any)("gpt-4o")).toThrow(
"The Mem0 model function cannot be called with the new keyword."
);
});
});
describe("Mem0GenericLanguageModel V3 interface", () => {
let model: Mem0GenericLanguageModel;
beforeEach(() => {
model = new Mem0GenericLanguageModel(
"gpt-4o",
{ user_id: "test-user" },
{
provider: "openai",
mem0ApiKey: "test-key",
apiKey: "test-openai-key",
}
);
});
it("should implement specificationVersion v3", () => {
expect(model.specificationVersion).toBe("v3");
});
it("should not have deprecated V2 properties", () => {
expect((model as any).defaultObjectGenerationMode).toBeUndefined();
expect((model as any).supportsImageUrls).toBeUndefined();
});
it("should have supportedUrls as a record", () => {
expect(model.supportedUrls).toBeDefined();
expect(typeof model.supportedUrls).toBe("object");
expect(model.supportedUrls["*"]).toBeDefined();
expect(Array.isArray(model.supportedUrls["*"])).toBe(true);
});
it("should have doGenerate and doStream methods", () => {
expect(typeof model.doGenerate).toBe("function");
expect(typeof model.doStream).toBe("function");
});
it("should have provider and modelId properties", () => {
expect(model.provider).toBe("openai");
expect(model.modelId).toBe("gpt-4o");
});
});
describe("Mem0ClassSelector", () => {
const supportedProviders = ["openai", "anthropic", "cohere", "groq", "google", "gemini"];
it.each(supportedProviders)("should accept %s as a valid provider", (providerName) => {
const selector = new Mem0ClassSelector(
"test-model",
{ provider: providerName, apiKey: "test-key" }
);
expect(selector).toBeDefined();
});
it("should throw for unsupported provider", () => {
expect(
() => new Mem0ClassSelector("test-model", { provider: "invalid-provider" })
).toThrow("Model not supported: invalid-provider");
});
it("should create a V3 provider instance", () => {
const selector = new Mem0ClassSelector(
"gpt-4o",
{ provider: "openai", apiKey: "test-key" }
);
const provider = selector.createProvider();
expect(provider.specificationVersion).toBe("v3");
expect(typeof provider.doGenerate).toBe("function");
expect(typeof provider.doStream).toBe("function");
});
});
describe("Provider modelType routing", () => {
it("should create chat model with chat modelType", () => {
const provider = createMem0({
provider: "openai",
mem0ApiKey: "test-key",
apiKey: "test-openai-key",
});
const chatModel = provider.chat("gpt-4o");
expect(chatModel).toBeDefined();
expect(chatModel.specificationVersion).toBe("v3");
expect((chatModel as any).config.modelType).toBe("chat");
});
it("should create completion model with completion modelType", () => {
const provider = createMem0({
provider: "openai",
mem0ApiKey: "test-key",
apiKey: "test-openai-key",
});
const completionModel = provider.completion("gpt-4o");
expect(completionModel).toBeDefined();
expect(completionModel.specificationVersion).toBe("v3");
expect((completionModel as any).config.modelType).toBe("completion");
});
});
describe("Default baseURL", () => {
it("should default to https, not http", () => {
const provider = createMem0({
provider: "openai",
mem0ApiKey: "test-key",
});
const model = provider("gpt-4o") as Mem0GenericLanguageModel;
expect((model as any).config.baseURL).toMatch(/^https:\/\//);
});
});
describe("google/gemini alias routing", () => {
it("should route google and gemini to the same provider", () => {
const googleModel = new Mem0AITextGenerator(
"gemini-2.0-flash",
{ provider: "google", apiKey: "test-key" },
{}
);
const geminiModel = new Mem0AITextGenerator(
"gemini-2.0-flash",
{ provider: "gemini", apiKey: "test-key" },
{}
);
expect(googleModel.specificationVersion).toBe("v3");
expect(geminiModel.specificationVersion).toBe("v3");
expect(googleModel.modelId).toBe(geminiModel.modelId);
});
});
describe("Graph memory removal", () => {
it("should not have enable_graph in Mem0ConfigSettings type", () => {
const config: Mem0ConfigSettings = {
user_id: "test-user",
mem0ApiKey: "test-key",
};
expect((config as any).enable_graph).toBeUndefined();
});
});
describe("processMemories prompt cloning", () => {
it("should not mutate the original prompt array", async () => {
const model = new Mem0GenericLanguageModel(
"gpt-4o",
{ user_id: "test-user" },
{
provider: "openai",
mem0ApiKey: "test-key",
apiKey: "test-openai-key",
}
);
const originalPrompt = [
{ role: "user" as const, content: [{ type: "text" as const, text: "Hello" }] },
];
const originalLength = originalPrompt.length;
// Mock both Mem0 API calls — addMemories (POST /v3/memories/add/) and getMemories (POST /v3/memories/search/)
mockFetch
.mockResolvedValueOnce({
ok: true,
json: async () => ({ results: [] }),
})
.mockResolvedValueOnce({
ok: true,
json: async () => [
{ memory: "User likes TypeScript" },
],
});
// Access processMemories via reflection
const processMemories = (model as any).processMemories.bind(model);
await processMemories(originalPrompt, { mem0ApiKey: "test-key", user_id: "test-user" });
// Original prompt should NOT have been mutated
expect(originalPrompt.length).toBe(originalLength);
});
});
describe("getMemories normalization", () => {
it("should return a flat array when API returns a flat array", async () => {
const { getMemories } = require("../src/mem0-utils");
mockFetch.mockResolvedValueOnce({
ok: true,
json: async () => [
{ memory: "User prefers dark mode" },
{ memory: "User likes React" },
],
});
const result = await getMemories("test query", { mem0ApiKey: "test-key", user_id: "test-user" });
expect(Array.isArray(result)).toBe(true);
expect(result).toHaveLength(2);
expect(result[0].memory).toBe("User prefers dark mode");
});
it("should return a flat array when API returns { results: [...] }", async () => {
const { getMemories } = require("../src/mem0-utils");
mockFetch.mockResolvedValueOnce({
ok: true,
json: async () => ({
results: [
{ memory: "User prefers dark mode" },
],
}),
});
const result = await getMemories("test query", { mem0ApiKey: "test-key", user_id: "test-user" });
expect(Array.isArray(result)).toBe(true);
expect(result).toHaveLength(1);
});
it("should return empty array when API returns empty results", async () => {
const { getMemories } = require("../src/mem0-utils");
mockFetch.mockResolvedValueOnce({
ok: true,
json: async () => [],
});
const result = await getMemories("test query", { mem0ApiKey: "test-key", user_id: "test-user" });
expect(Array.isArray(result)).toBe(true);
expect(result).toHaveLength(0);
});
});
describe("retrieveMemories normalization", () => {
it("should return empty string when no memories exist", async () => {
const { retrieveMemories } = require("../src/mem0-utils");
mockFetch.mockResolvedValueOnce({
ok: true,
json: async () => [],
});
const result = await retrieveMemories("test query", { mem0ApiKey: "test-key", user_id: "test-user" });
expect(result).toBe("");
});
it("should return formatted string when memories exist as flat array", async () => {
const { retrieveMemories } = require("../src/mem0-utils");
mockFetch.mockResolvedValueOnce({
ok: true,
json: async () => [
{ memory: "User likes pizza" },
],
});
const result = await retrieveMemories("test query", { mem0ApiKey: "test-key", user_id: "test-user" });
expect(result).toContain("System Message:");
expect(result).toContain("User likes pizza");
});
it("should return formatted string when memories come as { results: [...] }", async () => {
const { retrieveMemories } = require("../src/mem0-utils");
mockFetch.mockResolvedValueOnce({
ok: true,
json: async () => ({
results: [{ memory: "User likes pizza" }],
}),
});
const result = await retrieveMemories("test query", { mem0ApiKey: "test-key", user_id: "test-user" });
expect(result).toContain("System Message:");
expect(result).toContain("User likes pizza");
});
});
describe("top_k zero handling", () => {
it("should respect top_k: 0 and not fall back to default", async () => {
const { searchMemories } = require("../src/mem0-utils");
mockFetch.mockResolvedValueOnce({
ok: true,
json: async () => [],
});
await searchMemories("test query", {
mem0ApiKey: "test-key",
user_id: "test-user",
top_k: 0,
});
const body = JSON.parse(mockFetch.mock.calls[0][1].body);
expect(body.top_k).toBe(0);
});
});
describe("Mem0 v3 API migration", () => {
it("should call /v3/memories/search/ endpoint for search", async () => {
const { searchMemories } = require("../src/mem0-utils");
mockFetch.mockResolvedValueOnce({
ok: true,
json: async () => [],
});
await searchMemories("test query", { mem0ApiKey: "test-key", user_id: "test-user" });
const url = mockFetch.mock.calls[0][0];
expect(url).toBe("https://api.mem0.ai/v3/memories/search/");
});
it("should call /v3/memories/add/ endpoint for add", async () => {
const { addMemories } = require("../src/mem0-utils");
mockFetch.mockResolvedValueOnce({
ok: true,
json: async () => ({ message: "ok", status: "PENDING", event_id: "test" }),
});
await addMemories(
[{ role: "user" as const, content: [{ type: "text" as const, text: "Hello" }] }],
{ mem0ApiKey: "test-key", user_id: "test-user" }
);
const url = mockFetch.mock.calls[0][0];
expect(url).toBe("https://api.mem0.ai/v3/memories/add/");
});
it("should put entity IDs inside filters for search (not top-level)", async () => {
const { searchMemories } = require("../src/mem0-utils");
mockFetch.mockResolvedValueOnce({
ok: true,
json: async () => [],
});
await searchMemories("test query", {
mem0ApiKey: "test-key",
user_id: "user-123",
agent_id: "agent-456",
});
const body = JSON.parse(mockFetch.mock.calls[0][1].body);
expect(body.filters.user_id).toBe("user-123");
expect(body.filters.agent_id).toBe("agent-456");
// Should NOT be top-level
expect(body.user_id).toBeUndefined();
expect(body.agent_id).toBeUndefined();
});
it("should not send deprecated v2 params (version, output_format, org_id, etc.)", async () => {
const { searchMemories } = require("../src/mem0-utils");
mockFetch.mockResolvedValueOnce({
ok: true,
json: async () => [],
});
await searchMemories("test query", { mem0ApiKey: "test-key", user_id: "test-user" });
const body = JSON.parse(mockFetch.mock.calls[0][1].body);
expect(body.version).toBeUndefined();
expect(body.output_format).toBeUndefined();
expect(body.org_id).toBeUndefined();
expect(body.project_id).toBeUndefined();
expect(body.org_name).toBeUndefined();
expect(body.project_name).toBeUndefined();
});
it("should not send deprecated params in add body", async () => {
const { addMemories } = require("../src/mem0-utils");
mockFetch.mockResolvedValueOnce({
ok: true,
json: async () => ({ message: "ok", status: "PENDING", event_id: "test" }),
});
await addMemories(
[{ role: "user" as const, content: [{ type: "text" as const, text: "Hello" }] }],
{ mem0ApiKey: "test-key", user_id: "test-user" }
);
const body = JSON.parse(mockFetch.mock.calls[0][1].body);
expect(body.version).toBeUndefined();
expect(body.async_mode).toBeUndefined();
expect(body.output_format).toBeUndefined();
expect(body.enable_graph).toBeUndefined();
expect(body.filter_memories).toBeUndefined();
expect(body.org_id).toBeUndefined();
expect(body.project_id).toBeUndefined();
// Should have entity ID at top-level for add endpoint
expect(body.user_id).toBe("test-user");
});
it("should not have deprecated fields in Mem0ConfigSettings type", () => {
const config: Mem0ConfigSettings = {
user_id: "test-user",
mem0ApiKey: "test-key",
};
expect((config as any).org_id).toBeUndefined();
expect((config as any).project_id).toBeUndefined();
expect((config as any).org_name).toBeUndefined();
expect((config as any).project_name).toBeUndefined();
expect((config as any).output_format).toBeUndefined();
expect((config as any).filter_memories).toBeUndefined();
expect((config as any).async_mode).toBeUndefined();
expect((config as any).enable_graph).toBeUndefined();
});
it("should default top_k to 10 (v3 default) when not provided", async () => {
const { searchMemories } = require("../src/mem0-utils");
mockFetch.mockResolvedValueOnce({
ok: true,
json: async () => [],
});
await searchMemories("test query", { mem0ApiKey: "test-key", user_id: "test-user" });
const body = JSON.parse(mockFetch.mock.calls[0][1].body);
expect(body.top_k).toBe(10);
});
it("should merge user-provided filters with entity ID filters", async () => {
const { searchMemories } = require("../src/mem0-utils");
mockFetch.mockResolvedValueOnce({
ok: true,
json: async () => [],
});
await searchMemories("test query", {
mem0ApiKey: "test-key",
user_id: "user-123",
filters: { category: "preferences" },
});
const body = JSON.parse(mockFetch.mock.calls[0][1].body);
expect(body.filters.user_id).toBe("user-123");
expect(body.filters.category).toBe("preferences");
});
});
});