JDSS/mem0
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files
vercel-ai-sdk/package.json
+16 -16
View File
@@ -1,6 +1,6 @@
{
"name": "@mem0/vercel-ai-provider",
"version": "2.0.6",
"version": "3.0.0",
"description": "Vercel AI Provider for providing memory to LLMs",
"main": "./dist/index.js",
"module": "./dist/index.mjs",
@@ -31,28 +31,28 @@
"author": "Saket Aryan <saketaryan2002@gmail.com>",
"license": "Apache-2.0",
"dependencies": {
"@ai-sdk/anthropic": "2.0.0",
"@ai-sdk/cohere": "2.0.0",
"@ai-sdk/google": "2.0.1",
"@ai-sdk/groq": "2.0.1",
"@ai-sdk/openai": "2.0.2",
"@ai-sdk/provider": "2.0.0",
"@ai-sdk/provider-utils": "3.0.0",
"ai": "5.0.2",
"dotenv": "^16.4.5",
"@ai-sdk/anthropic": "^3.0.82",
"@ai-sdk/cohere": "^3.0.36",
"@ai-sdk/google": "^3.0.80",
"@ai-sdk/groq": "^3.0.39",
"@ai-sdk/openai": "^3.0.69",
"@ai-sdk/provider": "^3.0.10",
"@ai-sdk/provider-utils": "^4.0.27",
"ai": "^6.0.199",
"dotenv": "^16.6.1",
"partial-json": "0.1.7",
"zod": "^3.25.0"
"zod": "^3.25.76"
},
"devDependencies": {
"@edge-runtime/vm": "^3.2.0",
"@types/jest": "^29.5.14",
"@types/node": "^18.19.46",
"@types/node": "^18.19.130",
"jest": "^29.7.0",
"nodemon": "^3.1.7",
"ts-jest": "^29.2.5",
"nodemon": "^3.1.14",
"ts-jest": "^29.4.11",
"ts-node": "^10.9.2",
"tsup": "^8.3.0",
"typescript": "^5.5.4"
"tsup": "^8.5.1",
"typescript": "^5.9.3"
},
"peerDependencies": {
"zod": "^3.0.0"
vercel-ai-sdk/pnpm-lock.yaml
Generated
+879 -981
View File
File diff suppressed because it is too large Load Diff
vercel-ai-sdk/src/mem0-facade.ts
+1 -1
View File
@@ -12,7 +12,7 @@ export class Mem0 {
provider: 'openai',
}) {
this.baseURL =
withoutTrailingSlash(options.baseURL) ?? 'http://127.0.0.1:11434/api'
withoutTrailingSlash(options.baseURL) ?? 'https://api.openai.com'
this.headers = options.headers
}
vercel-ai-sdk/src/mem0-generic-language-model.ts
+103 -160
View File
@@ -1,14 +1,14 @@
/* eslint-disable camelcase */
import {
LanguageModelV2CallOptions,
LanguageModelV2Message,
LanguageModelV2Source
LanguageModelV3,
LanguageModelV3CallOptions,
LanguageModelV3GenerateResult,
LanguageModelV3Message,
LanguageModelV3Source,
LanguageModelV3StreamResult,
} from '@ai-sdk/provider';
import { LanguageModelV2 } from '@ai-sdk/provider';
// streaming uses provider-native doStream; no middleware needed
import { Mem0ChatConfig, Mem0ChatModelId, Mem0ChatSettings, Mem0ConfigSettings, Mem0StreamResponse } from "./mem0-types";
import { Mem0ChatConfig, Mem0ChatModelId, Mem0ChatSettings, Mem0ConfigSettings } from "./mem0-types";
import { Mem0ClassSelector } from "./mem0-provider-selector";
import { Mem0ProviderSettings } from "./mem0-provider";
import { addMemories, getMemories } from "./mem0-utils";
@@ -17,12 +17,8 @@ const generateRandomId = () => {
return Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
}
export class Mem0GenericLanguageModel implements LanguageModelV2 {
readonly specificationVersion = "v2";
readonly defaultObjectGenerationMode = "json";
// We don't support images for now
readonly supportsImageUrls = false;
// Allow All Media Types for now
export class Mem0GenericLanguageModel implements LanguageModelV3 {
readonly specificationVersion = "v3";
readonly supportedUrls: Record<string, RegExp[]> = {
'*': [/.*/]
};
@@ -38,148 +34,108 @@ export class Mem0GenericLanguageModel implements LanguageModelV2 {
provider: string;
private async processMemories(messagesPrompts: LanguageModelV2Message[], mem0Config: Mem0ConfigSettings) {
private async processMemories(messagesPrompts: LanguageModelV3Message[], mem0Config: Mem0ConfigSettings) {
try {
// Add New Memories
addMemories(messagesPrompts, mem0Config).then((res) => {
return res;
}).catch((e) => {
console.error("Error while adding memories");
return { memories: [], messagesPrompts: [] };
// Add new memories — await to ensure writes complete before returning
try {
await addMemories(messagesPrompts, mem0Config);
} catch (e) {
console.error("Error while adding memories");
}
// Get memories (always returns an array since graph support is removed)
const memories: any[] = await getMemories(messagesPrompts, mem0Config) ?? [];
const mySystemPrompt = "These are the memories I have stored. Give more weightage to the question by users and try to answer that first. You have to modify your answer based on the memories I have provided. If the memories are irrelevant you can ignore them. Also don't reply to this section of the prompt, or the memories, they are only for your reference. The System prompt starts after text System Message: \n\n";
let memoriesText = "";
try {
memoriesText = memories
?.map((memory: any) => `Memory: ${memory?.memory}\n\n`)
.join("\n\n");
} catch (e) {
console.error("Error while parsing memories");
}
const memoriesPrompt = `System Message: ${mySystemPrompt} ${memoriesText} `;
// Clone the prompt array to avoid mutating the caller's reference on retries
const updatedPrompts = [...messagesPrompts];
if (memories?.length > 0) {
const systemPrompt: LanguageModelV3Message = {
role: "system",
content: memoriesPrompt
};
updatedPrompts.unshift(systemPrompt);
}
return { memories, messagesPrompts: updatedPrompts };
} catch (e) {
console.error("Error while processing memories");
return { memories: [], messagesPrompts: [...messagesPrompts] };
}
}
async doGenerate(options: LanguageModelV3CallOptions): Promise<LanguageModelV3GenerateResult> {
const provider = this.config.provider;
const mem0_api_key = this.config.mem0ApiKey;
const settings: Mem0ProviderSettings = {
provider: provider,
mem0ApiKey: mem0_api_key,
apiKey: this.config.apiKey,
modelType: this.config.modelType,
}
const mem0Config: Mem0ConfigSettings = {
mem0ApiKey: mem0_api_key,
...this.config.mem0Config,
...this.settings,
}
const selector = new Mem0ClassSelector(this.modelId, settings, this.provider_config);
const { memories, messagesPrompts: updatedPrompts } = await this.processMemories(options.prompt, mem0Config);
const model = selector.createProvider();
const ans = await model.doGenerate({
...options,
prompt: updatedPrompts,
});
// Get Memories
let memories = await getMemories(messagesPrompts, mem0Config);
const mySystemPrompt = "These are the memories I have stored. Give more weightage to the question by users and try to answer that first. You have to modify your answer based on the memories I have provided. If the memories are irrelevant you can ignore them. Also don't reply to this section of the prompt, or the memories, they are only for your reference. The System prompt starts after text System Message: \n\n";
const isGraphEnabled = mem0Config?.enable_graph;
let memoriesText = "";
let memoriesText2 = "";
try {
// @ts-ignore
if (isGraphEnabled) {
memoriesText = memories?.results?.map((memory: any) => {
return `Memory: ${memory?.memory}\n\n`;
}).join("\n\n");
memoriesText2 = memories?.relations?.map((memory: any) => {
return `Relation: ${memory?.source} -> ${memory?.relationship} -> ${memory?.target} \n\n`;
}).join("\n\n");
} else {
memoriesText = memories?.map((memory: any) => {
return `Memory: ${memory?.memory}\n\n`;
}).join("\n\n");
}
} catch(e) {
console.error("Error while parsing memories");
if (!memories || memories?.length === 0) {
return ans;
}
let graphPrompt = "";
if (isGraphEnabled) {
graphPrompt = `HERE ARE THE GRAPHS RELATIONS FOR THE PREFERENCES OF THE USER:\n\n ${memoriesText2}`;
}
const memoriesPrompt = `System Message: ${mySystemPrompt} ${memoriesText} ${graphPrompt} `;
// System Prompt - The memories go as a system prompt
const systemPrompt: LanguageModelV2Message = {
role: "system",
content: memoriesPrompt
const mem0Source: LanguageModelV3Source = {
type: "source",
sourceType: "url",
id: "mem0-" + generateRandomId(),
url: "https://app.mem0.ai?utm_source=oss&utm_medium=vercel-ai-sdk-src",
title: "Mem0 Memories",
providerMetadata: {
mem0: {
memories: memories,
memoriesText: memories
?.map((memory: any) => memory?.memory)
.join("\n\n"),
},
},
};
// Add the system prompt to the beginning of the messages if there are memories
if (memories?.length > 0) {
messagesPrompts.unshift(systemPrompt);
}
if (isGraphEnabled) {
memories = memories?.results;
}
return { memories, messagesPrompts };
} catch(e) {
console.error("Error while processing memories");
return { memories: [], messagesPrompts };
}
return {
...ans,
content: [...(ans.content ?? []), mem0Source],
};
}
async doGenerate(options: LanguageModelV2CallOptions): Promise<Awaited<ReturnType<LanguageModelV2['doGenerate']>>> {
try {
const provider = this.config.provider;
const mem0_api_key = this.config.mem0ApiKey;
const settings: Mem0ProviderSettings = {
provider: provider,
mem0ApiKey: mem0_api_key,
apiKey: this.config.apiKey,
}
const mem0Config: Mem0ConfigSettings = {
mem0ApiKey: mem0_api_key,
...this.config.mem0Config,
...this.settings,
}
const selector = new Mem0ClassSelector(this.modelId, settings, this.provider_config);
let messagesPrompts = options.prompt;
// Process memories and update prompts
const { memories, messagesPrompts: updatedPrompts } = await this.processMemories(messagesPrompts, mem0Config);
const model = selector.createProvider();
const ans = await model.doGenerate({
...options,
prompt: updatedPrompts,
});
// If there are no memories, return the original response
if (!memories || memories?.length === 0) {
return ans;
}
try {
// Create sources array with existing sources
const sources: LanguageModelV2Source[] = [
{
type: "source",
title: "Mem0 Memories",
sourceType: "url",
id: "mem0-" + generateRandomId(),
url: "https://app.mem0.ai?utm_source=oss&utm_medium=vercel-ai-sdk-src",
providerMetadata: {
mem0: {
memories: memories,
memoriesText: memories
?.map((memory: any) => memory?.memory)
.join("\n\n"),
},
},
},
];
} catch (e) {
console.error("Error while creating sources");
}
return {
...ans,
// sources
};
} catch (error) {
// Handle errors properly
console.error("Error in doGenerate:", error);
throw new Error("Failed to generate response.");
}
}
async doStream(options: LanguageModelV2CallOptions): Promise<Awaited<ReturnType<LanguageModelV2['doStream']>>> {
async doStream(options: LanguageModelV3CallOptions): Promise<LanguageModelV3StreamResult> {
try {
const provider = this.config.provider;
const mem0_api_key = this.config.mem0ApiKey;
const settings: Mem0ProviderSettings = {
provider: provider,
mem0ApiKey: mem0_api_key,
@@ -194,31 +150,18 @@ export class Mem0GenericLanguageModel implements LanguageModelV2 {
}
const selector = new Mem0ClassSelector(this.modelId, settings, this.provider_config);
let messagesPrompts = options.prompt;
// Process memories and update prompts
const { memories, messagesPrompts: updatedPrompts } = await this.processMemories(messagesPrompts, mem0Config);
const { messagesPrompts: updatedPrompts } = await this.processMemories(options.prompt, mem0Config);
const baseModel = selector.createProvider();
// Use the provider's native streaming directly to avoid buffering
const streamResponse = await baseModel.doStream({
...options,
prompt: updatedPrompts,
});
// If there are no memories, return the original stream
if (!memories || memories?.length === 0) {
return streamResponse;
}
// Return stream untouched for true streaming behavior
return {
stream: streamResponse.stream,
request: streamResponse.request,
response: streamResponse.response,
};
// Return the full stream response, preserving all V3 fields (warnings, etc.)
return streamResponse;
} catch (error) {
console.error("Error in doStream:", error);
throw new Error("Streaming failed or method not implemented.");
vercel-ai-sdk/src/mem0-provider-selector.ts
+3 -3
View File
@@ -1,13 +1,13 @@
import { Mem0ProviderSettings } from "./mem0-provider";
import Mem0AITextGenerator, { ProviderSettings } from "./provider-response-provider";
import { LanguageModelV2 } from '@ai-sdk/provider';
import { LanguageModelV3 } from '@ai-sdk/provider';
class Mem0ClassSelector {
modelId: string;
provider_wrapper: string;
config: Mem0ProviderSettings;
provider_config?: ProviderSettings;
static supportedProviders = ["openai", "anthropic", "cohere", "groq", "google"];
static supportedProviders = ["openai", "anthropic", "cohere", "groq", "google", "gemini"];
constructor(modelId: string, config: Mem0ProviderSettings, provider_config?: ProviderSettings) {
this.modelId = modelId;
@@ -24,7 +24,7 @@ class Mem0ClassSelector {
}
}
createProvider(): LanguageModelV2 {
createProvider(): LanguageModelV3 {
return new Mem0AITextGenerator(this.modelId, this.config , this.provider_config || {});
}
}
vercel-ai-sdk/src/mem0-provider.ts
+11 -10
View File
@@ -1,20 +1,20 @@
import { ProviderV2 } from '@ai-sdk/provider';
import { LanguageModelV2 } from '@ai-sdk/provider';
import { ProviderV3 } from '@ai-sdk/provider';
import { LanguageModelV3 } from '@ai-sdk/provider';
import { withoutTrailingSlash } from "@ai-sdk/provider-utils";
import { Mem0ChatModelId, Mem0ChatSettings, Mem0Config } from "./mem0-types";
import { Mem0GenericLanguageModel } from "./mem0-generic-language-model";
import { LLMProviderSettings } from "./mem0-types";
export interface Mem0Provider extends ProviderV2 {
(modelId: Mem0ChatModelId, settings?: Mem0ChatSettings): LanguageModelV2;
export interface Mem0Provider extends ProviderV3 {
(modelId: Mem0ChatModelId, settings?: Mem0ChatSettings): LanguageModelV3;
chat(modelId: Mem0ChatModelId, settings?: Mem0ChatSettings): LanguageModelV2;
completion(modelId: Mem0ChatModelId, settings?: Mem0ChatSettings): LanguageModelV2;
chat(modelId: Mem0ChatModelId, settings?: Mem0ChatSettings): LanguageModelV3;
completion(modelId: Mem0ChatModelId, settings?: Mem0ChatSettings): LanguageModelV3;
languageModel(
modelId: Mem0ChatModelId,
settings?: Mem0ChatSettings
): LanguageModelV2;
): LanguageModelV3;
}
export interface Mem0ProviderSettings {
@@ -31,7 +31,7 @@ export interface Mem0ProviderSettings {
/**
* Custom headers to include in the requests.
*/
headers?: Record<string, string>;
headers?: Record<string, string | undefined>;
name?: string;
mem0ApiKey?: string;
apiKey?: string;
@@ -51,7 +51,7 @@ export function createMem0(
}
): Mem0Provider {
const baseURL =
withoutTrailingSlash(options.baseURL) ?? "http://api.openai.com";
withoutTrailingSlash(options.baseURL) ?? "https://api.openai.com";
const getHeaders = () => ({
...options.headers,
});
@@ -113,7 +113,7 @@ export function createMem0(
mem0ApiKey: options.mem0ApiKey,
apiKey: options.apiKey,
mem0Config: options.mem0Config,
modelType: "completion",
modelType: "chat",
},
options.config
);
@@ -131,6 +131,7 @@ export function createMem0(
return createGenericModel(modelId, settings);
};
provider.specificationVersion = 'v3';
provider.languageModel = createGenericModel;
provider.completion = createCompletionModel;
provider.chat = createChatModel;
vercel-ai-sdk/src/mem0-types.ts
+2 -14
View File
@@ -1,9 +1,9 @@
import { Mem0ProviderSettings } from "./mem0-provider";
import { OpenAIProviderSettings } from "@ai-sdk/openai";
import { AnthropicProviderSettings } from "@ai-sdk/anthropic";
import { LanguageModelV2 } from '@ai-sdk/provider';
import { CohereProviderSettings } from "@ai-sdk/cohere";
import { GroqProviderSettings } from "@ai-sdk/groq";
import { GoogleGenerativeAIProviderSettings } from "@ai-sdk/google";
export type Mem0ChatModelId =
| (string & NonNullable<unknown>);
@@ -12,10 +12,6 @@ export interface Mem0ConfigSettings {
app_id?: string;
agent_id?: string;
run_id?: string;
org_name?: string;
project_name?: string;
org_id?: string;
project_id?: string;
metadata?: Record<string, any>;
filters?: Record<string, any>;
infer?: boolean;
@@ -25,20 +21,12 @@ export interface Mem0ConfigSettings {
top_k?: number;
threshold?: number;
rerank?: boolean;
enable_graph?: boolean;
host?: string;
output_format?: string;
filter_memories?: boolean;
async_mode?: boolean;
}
export interface Mem0ChatConfig extends Mem0ConfigSettings, Mem0ProviderSettings {}
export interface LLMProviderSettings extends OpenAIProviderSettings, AnthropicProviderSettings, CohereProviderSettings, GroqProviderSettings {}
export type LLMProviderSettings = OpenAIProviderSettings | AnthropicProviderSettings | CohereProviderSettings | GroqProviderSettings | GoogleGenerativeAIProviderSettings;
export interface Mem0Config extends Mem0ConfigSettings {}
export interface Mem0ChatSettings extends Mem0ConfigSettings {}
export interface Mem0StreamResponse extends Awaited<ReturnType<LanguageModelV2['doStream']>> {
memories: any;
}
vercel-ai-sdk/src/mem0-utils.ts
+57 -76
View File
@@ -1,4 +1,4 @@
import { LanguageModelV2Prompt } from '@ai-sdk/provider';
import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { Mem0ConfigSettings } from './mem0-types';
import { loadApiKey } from '@ai-sdk/provider-utils';
interface MultimodalContent {
@@ -26,7 +26,7 @@ interface Message {
content: string | MultimodalContent | Array<MultimodalContent>;
}
const flattenPrompt = (prompt: LanguageModelV2Prompt) => {
const flattenPrompt = (prompt: LanguageModelV3Prompt) => {
try {
return prompt.map((part) => {
if (part.role === "user") {
@@ -42,7 +42,6 @@ const flattenPrompt = (prompt: LanguageModelV2Prompt) => {
if (content.type === 'text' && content.text) {
return content.text;
} else if (content.type === 'file') {
// For file content, we'll include a descriptive placeholder
if (content.mediaType === 'application/pdf') {
return '[PDF document]';
} else if (content.mediaType === 'text/markdown' || content.mediaType === 'application/mdx') {
@@ -54,8 +53,6 @@ const flattenPrompt = (prompt: LanguageModelV2Prompt) => {
}
}
}
// For non-text content (images, pdfs, mdx), we'll include a placeholder
// This helps maintain context for memory search while not breaking the text flow
return "[multimodal content]";
}
return "";
@@ -66,7 +63,7 @@ const flattenPrompt = (prompt: LanguageModelV2Prompt) => {
}
}
const convertToMem0Format = (messages: LanguageModelV2Prompt) => {
const convertToMem0Format = (messages: LanguageModelV3Prompt) => {
try {
return messages.flatMap((message: any) => {
try {
@@ -85,7 +82,6 @@ const convertToMem0Format = (messages: LanguageModelV2Prompt) => {
content: obj.text,
};
} else if (obj.type === "file") {
// Handle LanguageModelV2Prompt file format
if (obj.mediaType === "application/pdf") {
return {
role: message.role,
@@ -155,7 +151,6 @@ const convertToMem0Format = (messages: LanguageModelV2Prompt) => {
}
}).filter((item: null) => item !== null);
} else {
// Handle single multimodal content object
const obj = message.content;
if (obj.type === "text") {
return {
@@ -163,7 +158,6 @@ const convertToMem0Format = (messages: LanguageModelV2Prompt) => {
content: obj.text,
};
} else if (obj.type === "file") {
// Handle LanguageModelV2Prompt file format
if (obj.mediaType === "application/pdf") {
return {
role: message.role,
@@ -239,63 +233,57 @@ const convertToMem0Format = (messages: LanguageModelV2Prompt) => {
}
}
const searchInternalMemories = async (query: string, config?: Mem0ConfigSettings, top_k: number = 5) => {
const searchInternalMemories = async (query: string, config?: Mem0ConfigSettings, top_k: number = 10) => {
try {
const filters: { OR: Array<{ [key: string]: string | undefined }> } = {
OR: [],
// v3: entity IDs go inside the filters object, not as top-level params
const filters: Record<string, any> = {
...(config?.filters ?? {}),
};
if (config?.user_id) {
filters.OR.push({
user_id: config.user_id,
});
filters.user_id = config.user_id;
}
if (config?.app_id) {
filters.OR.push({
app_id: config.app_id,
});
filters.app_id = config.app_id;
}
if (config?.agent_id) {
filters.OR.push({
agent_id: config.agent_id,
});
filters.agent_id = config.agent_id;
}
if (config?.run_id) {
filters.OR.push({
run_id: config.run_id,
});
}
const org_project_filters = {
org_id: config&&config.org_id,
project_id: config&&config.project_id,
org_name: !config?.org_id ? config&&config.org_name : undefined,
project_name: !config?.org_id ? config&&config.project_name : undefined,
filters.run_id = config.run_id;
}
const apiKey = loadApiKey({
apiKey: (config&&config.mem0ApiKey),
apiKey: config?.mem0ApiKey,
environmentVariableName: "MEM0_API_KEY",
description: "Mem0",
});
const body: Record<string, any> = {
query,
filters,
top_k: config?.top_k ?? top_k,
};
if (config?.threshold != null) {
body.threshold = config.threshold;
}
if (config?.rerank != null) {
body.rerank = config.rerank;
}
if (config?.metadata) {
body.metadata = config.metadata;
}
const options = {
method: 'POST',
headers: {
Authorization: `Token ${apiKey}`,
'Content-Type': 'application/json'
},
body: JSON.stringify({
query,
filters,
...config,
top_k: config&&config.top_k || top_k,
version: "v2",
output_format: "v1.1",
...org_project_filters
}),
body: JSON.stringify(body),
};
const baseUrl = config?.host || 'https://api.mem0.ai';
const response = await fetch(`${baseUrl}/v2/memories/search/`, options);
const response = await fetch(`${baseUrl}/v3/memories/search/`, options);
if (!response.ok) {
throw new Error(`HTTP error! status: ${response.status}`);
}
@@ -307,7 +295,7 @@ const searchInternalMemories = async (query: string, config?: Mem0ConfigSettings
}
}
const addMemories = async (messages: LanguageModelV2Prompt, config?: Mem0ConfigSettings) => {
const addMemories = async (messages: LanguageModelV3Prompt, config?: Mem0ConfigSettings) => {
try {
let finalMessages: Array<Message> = [];
if (typeof messages === "string") {
@@ -326,22 +314,30 @@ const addMemories = async (messages: LanguageModelV2Prompt, config?: Mem0ConfigS
const updateMemories = async (messages: Array<Message>, config?: Mem0ConfigSettings) => {
try {
const apiKey = loadApiKey({
apiKey: (config&&config.mem0ApiKey),
apiKey: config?.mem0ApiKey,
environmentVariableName: "MEM0_API_KEY",
description: "Mem0",
});
const body: Record<string, any> = { messages };
if (config?.user_id) body.user_id = config.user_id;
if (config?.app_id) body.app_id = config.app_id;
if (config?.agent_id) body.agent_id = config.agent_id;
if (config?.run_id) body.run_id = config.run_id;
if (config?.metadata) body.metadata = config.metadata;
if (config?.infer != null) body.infer = config.infer;
const options = {
method: 'POST',
headers: {
Authorization: `Token ${apiKey}`,
'Content-Type': 'application/json'
},
body: JSON.stringify({messages, ...config, version: "v2"}),
body: JSON.stringify(body),
};
const baseUrl = config?.host || 'https://api.mem0.ai';
const response = await fetch(`${baseUrl}/v1/memories/`, options);
const response = await fetch(`${baseUrl}/v3/memories/add/`, options);
if (!response.ok) {
throw new Error(`HTTP error! status: ${response.status}`);
}
@@ -353,58 +349,43 @@ const updateMemories = async (messages: Array<Message>, config?: Mem0ConfigSetti
}
}
const retrieveMemories = async (prompt: LanguageModelV2Prompt | string, config?: Mem0ConfigSettings) => {
const retrieveMemories = async (prompt: LanguageModelV3Prompt | string, config?: Mem0ConfigSettings) => {
try {
const message = typeof prompt === 'string' ? prompt : flattenPrompt(prompt);
const systemPrompt = "These are the memories I have stored. Give more weightage to the question by users and try to answer that first. You have to modify your answer based on the memories I have provided. If the memories are irrelevant you can ignore them. Also don't reply to this section of the prompt, or the memories, they are only for your reference. The System prompt starts after text System Message: \n\n";
const memories = await searchInternalMemories(message, config);
let memoriesText1 = "";
let memoriesText2 = "";
let graphPrompt = "";
try {
memoriesText1 = memories?.results?.map((memory: any) => {
return `Memory: ${memory.memory}\n\n`;
}).join("\n\n");
const data = await searchInternalMemories(message, config);
// The API response may be a flat array or an object with a results key
const memories: any[] = Array.isArray(data) ? data : (data?.results ?? []);
if (config?.enable_graph) {
memoriesText2 = memories?.relations?.map((memory: any) => {
return `Relation: ${memory.source} -> ${memory.relationship} -> ${memory.target} \n\n`;
}).join("\n\n");
graphPrompt = `HERE ARE THE GRAPHS RELATIONS FOR THE PREFERENCES OF THE USER:\n\n ${memoriesText2}`;
}
} catch (error) {
console.error("Error while parsing memories:", error);
}
if (!memories || memories?.length === 0) {
if (memories.length === 0) {
return "";
}
return `System Message: ${systemPrompt} ${memoriesText1} ${graphPrompt}`;
const memoriesText = memories
.map((memory: any) => `Memory: ${memory.memory}\n\n`)
.join("\n\n");
return `System Message: ${systemPrompt} ${memoriesText}`;
} catch (error) {
console.error("Error in retrieveMemories:", error);
throw error;
}
}
const getMemories = async (prompt: LanguageModelV2Prompt | string, config?: Mem0ConfigSettings) => {
const getMemories = async (prompt: LanguageModelV3Prompt | string, config?: Mem0ConfigSettings) => {
try {
const message = typeof prompt === 'string' ? prompt : flattenPrompt(prompt);
const memories = await searchInternalMemories(message, config);
if (!config?.enable_graph) {
return memories?.results;
}
return memories;
const data = await searchInternalMemories(message, config);
// Normalize: always return a flat array regardless of API response shape
return Array.isArray(data) ? data : (data?.results ?? []);
} catch (error) {
console.error("Error in getMemories:", error);
throw error;
}
}
const searchMemories = async (prompt: LanguageModelV2Prompt | string, config?: Mem0ConfigSettings) => {
const searchMemories = async (prompt: LanguageModelV3Prompt | string, config?: Mem0ConfigSettings) => {
try {
const message = typeof prompt === 'string' ? prompt : flattenPrompt(prompt);
const memories = await searchInternalMemories(message, config);
@@ -415,4 +396,4 @@ const searchMemories = async (prompt: LanguageModelV2Prompt | string, config?: M
}
}
export {addMemories, updateMemories, retrieveMemories, flattenPrompt, searchMemories, getMemories};
export {addMemories, updateMemories, retrieveMemories, flattenPrompt, searchMemories, getMemories};
vercel-ai-sdk/src/provider-response-provider.ts
+8 -15
View File
@@ -1,4 +1,4 @@
import { LanguageModelV2, LanguageModelV2CallOptions } from "@ai-sdk/provider";
import { LanguageModelV3, LanguageModelV3CallOptions, LanguageModelV3GenerateResult, LanguageModelV3StreamResult } from "@ai-sdk/provider";
import { Mem0ProviderSettings } from "./mem0-provider";
import { createOpenAI, OpenAIProviderSettings } from "@ai-sdk/openai";
import { CohereProviderSettings, createCohere } from "@ai-sdk/cohere";
@@ -7,10 +7,8 @@ import { createGoogleGenerativeAI, GoogleGenerativeAIProviderSettings } from "@a
import { createGroq, GroqProviderSettings } from "@ai-sdk/groq";
// Define a private provider field
class Mem0AITextGenerator implements LanguageModelV2 {
readonly specificationVersion = "v2";
readonly defaultObjectGenerationMode = "json";
readonly supportsImageUrls = false;
class Mem0AITextGenerator implements LanguageModelV3 {
readonly specificationVersion = "v3";
readonly modelId: string;
readonly provider = "mem0";
readonly supportedUrls: Record<string, RegExp[]> = {
@@ -59,11 +57,6 @@ class Mem0AITextGenerator implements LanguageModelV2 {
})(modelId);
break;
case "google":
this.languageModel = createGoogleGenerativeAI({
apiKey: config?.apiKey,
...provider_config as GoogleGenerativeAIProviderSettings,
})(modelId);
break;
case "gemini":
this.languageModel = createGoogleGenerativeAI({
apiKey: config?.apiKey,
@@ -74,15 +67,15 @@ class Mem0AITextGenerator implements LanguageModelV2 {
throw new Error("Invalid provider");
}
}
async doGenerate(options: LanguageModelV2CallOptions): Promise<Awaited<ReturnType<LanguageModelV2['doGenerate']>>> {
async doGenerate(options: LanguageModelV3CallOptions): Promise<LanguageModelV3GenerateResult> {
const result = await this.languageModel.doGenerate(options);
return result as Awaited<ReturnType<LanguageModelV2['doGenerate']>>;
return result as LanguageModelV3GenerateResult;
}
async doStream(options: LanguageModelV2CallOptions): Promise<Awaited<ReturnType<LanguageModelV2['doStream']>>> {
async doStream(options: LanguageModelV3CallOptions): Promise<LanguageModelV3StreamResult> {
const result = await this.languageModel.doStream(options);
return result as Awaited<ReturnType<LanguageModelV2['doStream']>>;
return result as LanguageModelV3StreamResult;
}
}
vercel-ai-sdk/tests/generate-output.test.ts
+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: [
vercel-ai-sdk/tests/mem0-provider-tests/mem0-cohere.test.ts
+2 -2
View File
@@ -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: [
vercel-ai-sdk/tests/mem0-provider-tests/mem0-google.test.ts
+2 -2
View File
@@ -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: [
vercel-ai-sdk/tests/mem0-provider-tests/mem0-groq.test.ts
+2 -2
View File
@@ -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: [
vercel-ai-sdk/tests/mem0-provider-tests/mem0-openai.test.ts
+2 -2
View File
@@ -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: [
vercel-ai-sdk/tests/mem0-provider-tests/mem0_anthropic.test.ts
+2 -2
View File
@@ -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: [
vercel-ai-sdk/tests/memory-core.test.ts
+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: [
vercel-ai-sdk/tests/utils-test/anthropic-integration.test.ts
+2 -2
View File
@@ -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: [
vercel-ai-sdk/tests/utils-test/cohere-integration.test.ts
+2 -2
View File
@@ -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: [
vercel-ai-sdk/tests/utils-test/google-integration.test.ts
+2 -2
View File
@@ -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: [
vercel-ai-sdk/tests/utils-test/groq-integration.test.ts
+2 -2
View File
@@ -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: [
vercel-ai-sdk/tests/utils-test/openai-integration.test.ts
+2 -2
View File
@@ -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: [
vercel-ai-sdk/tests/v3-provider-contract.test.ts
+486
View File
@@ -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");
});
});
});