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:
co-authored by
kartik-mem0
parent
b819d95d18
commit
b4a50e3dc8
@@ -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: [
|
||||
|
||||
@@ -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");
|
||||
});
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user