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
+23
View File
@@ -1489,6 +1489,29 @@ A full-featured command-line interface for Mem0, available in both Python and No
</Update> </Update>
<Update label="2026-06-10" description="Vercel AI SDK v3.0.0">
**Major Release** — Migrated to Vercel AI SDK v6 (`LanguageModelV3` / `ProviderV3`) and Mem0 v3 API.
**Breaking Changes:**
- **AI SDK v6:** Upgraded from AI SDK v5 (`LanguageModelV2`) to v6 (`LanguageModelV3`). Users must upgrade `ai` to `^6.0.199` and all `@ai-sdk/*` provider packages to `^3.x` ([#4741](https://github.com/mem0ai/mem0/pull/4741))
- **Mem0 v3 API:** Memory endpoints migrated from `/v1/memories/` and `/v2/memories/search/` to `/v3/memories/add/` and `/v3/memories/search/`. Entity IDs (`user_id`, `agent_id`, `run_id`) now go inside the `filters` object for search requests ([#4741](https://github.com/mem0ai/mem0/pull/4741))
- **Graph memory removed:** All `enable_graph`, graph prompts, and relation-extraction code removed. Graph memory is now a project-level setting on the Platform ([#4741](https://github.com/mem0ai/mem0/pull/4741))
- **Deprecated params removed:** `org_id`, `project_id`, `org_name`, `project_name`, `output_format`, `filter_memories`, `async_mode`, `enable_graph`, `version`, `api_version` removed from `Mem0ConfigSettings` ([#4741](https://github.com/mem0ai/mem0/pull/4741))
**New Features:**
- **V3 provider contract:** `specificationVersion: 'v3'`, `supportedUrls` property, V3 content array in `doGenerate`, V3 stream lifecycle events in `doStream` ([#4741](https://github.com/mem0ai/mem0/pull/4741))
- **Mem0 source in responses:** Memories are attached as a `source` in `generateText`/`streamText` responses with `providerMetadata.mem0.memories` for programmatic access ([#4741](https://github.com/mem0ai/mem0/pull/4741))
**Bug Fixes:**
- **Async memory storage:** `addMemories` is now properly `await`ed — memories no longer silently fail to store ([#4741](https://github.com/mem0ai/mem0/pull/4741))
- **Prompt mutation:** Prompt array is now cloned before injecting memory context, preventing side effects on the caller's array ([#4741](https://github.com/mem0ai/mem0/pull/4741))
- **Null guard on content:** `doGenerate` guards against null `content` from upstream providers ([#4741](https://github.com/mem0ai/mem0/pull/4741))
- **Stream response:** `doStream` now returns the full `LanguageModelV3StreamResult` object preserving all V3 fields ([#4741](https://github.com/mem0ai/mem0/pull/4741))
- **Response normalization:** `getMemories` and `retrieveMemories` now handle both array and `{results: [...]}` envelope responses from the v3 API ([#4741](https://github.com/mem0ai/mem0/pull/4741))
</Update>
<Update label="2026-06-01" description="Vercel AI SDK v2.0.6"> <Update label="2026-06-01" description="Vercel AI SDK v2.0.6">
**Security:** **Security:**
+151 -116
View File
@@ -6,25 +6,33 @@ description: "Use the Mem0 AI SDK Provider with Vercel AI SDK for persistent mem
The [**Mem0 AI SDK Provider**](https://www.npmjs.com/package/@mem0/vercel-ai-provider) is a library developed by **Mem0** to integrate with the Vercel AI SDK. This library brings enhanced AI interaction capabilities to your applications by introducing persistent memory functionality. The [**Mem0 AI SDK Provider**](https://www.npmjs.com/package/@mem0/vercel-ai-provider) is a library developed by **Mem0** to integrate with the Vercel AI SDK. This library brings enhanced AI interaction capabilities to your applications by introducing persistent memory functionality.
<Note type="info"> <Note type="info">
Mem0 AI SDK now supports <strong>Vercel AI SDK V5</strong>. Mem0 AI SDK Provider v3.0.0 supports <strong>Vercel AI SDK v6</strong> (<code>LanguageModelV3</code> / <code>ProviderV3</code>). If you are upgrading from v2.x, see the <a href="https://ai-sdk.dev/docs/migration-guides/migration-guide-6-0">AI SDK v6 migration guide</a>.
</Note> </Note>
## Overview ## Overview
1. Offers persistent memory storage for conversational AI 1. Offers persistent memory storage for conversational AI
2. Enables smooth integration with the Vercel AI SDK 2. Enables smooth integration with the Vercel AI SDK v6
3. Ensures compatibility with multiple LLM providers 3. Ensures compatibility with multiple LLM providers (OpenAI, Anthropic, Google, Groq, Cohere)
4. Supports structured message formats for clarity 4. Supports structured message formats for clarity
5. Facilitates streaming response capabilities 5. Facilitates streaming response capabilities
6. Attaches Mem0 memories as sources in responses for programmatic access
## Setup and Configuration ## Setup and Configuration
Install the SDK provider using npm: Install the SDK provider and AI SDK:
```bash ```bash
npm install @mem0/vercel-ai-provider npm install @mem0/vercel-ai-provider ai@^6
``` ```
### Peer Dependencies
`@mem0/vercel-ai-provider` v3.0.0 requires:
- `ai` v6+ (`^6.0.199`)
- `@ai-sdk/provider` v3+ (`^3.0.10`)
- Provider packages at v3+: `@ai-sdk/openai@^3`, `@ai-sdk/anthropic@^3`, `@ai-sdk/google@^3`, `@ai-sdk/groq@^3`, `@ai-sdk/cohere@^3`
## Getting Started ## Getting Started
### Setting Up Mem0 ### Setting Up Mem0
@@ -41,7 +49,7 @@ npm install @mem0/vercel-ai-provider
mem0ApiKey: "m0-xxx", mem0ApiKey: "m0-xxx",
apiKey: "provider-api-key", apiKey: "provider-api-key",
config: { config: {
// Options for LLM Provider // Options for the upstream LLM provider (e.g. baseURL)
}, },
// Optional Mem0 Global Config // Optional Mem0 Global Config
mem0Config: { mem0Config: {
@@ -57,154 +65,153 @@ npm install @mem0/vercel-ai-provider
3. Add Memories to Enhance Context: 3. Add Memories to Enhance Context:
```typescript ```typescript
import { LanguageModelV2Prompt } from "@ai-sdk/provider";
import { addMemories } from "@mem0/vercel-ai-provider"; import { addMemories } from "@mem0/vercel-ai-provider";
const messages: LanguageModelV2Prompt = [ const messages = [
{ role: "user", content: [{ type: "text", text: "I love red cars." }] }, { role: "user", content: [{ type: "text", text: "I love red cars." }] },
]; ];
await addMemories(messages, { user_id: "borat" }); await addMemories(messages, { user_id: "borat" });
``` ```
### Standalone Features: ### Standalone Features
```typescript ```typescript
await addMemories(messages, { user_id: "borat", mem0ApiKey: "m0-xxx" }); await addMemories(messages, { user_id: "borat", mem0ApiKey: "m0-xxx" });
await retrieveMemories(prompt, { user_id: "borat", mem0ApiKey: "m0-xxx" }); await retrieveMemories(prompt, { user_id: "borat", mem0ApiKey: "m0-xxx" });
await getMemories(prompt, { user_id: "borat", mem0ApiKey: "m0-xxx" }); await getMemories(prompt, { user_id: "borat", mem0ApiKey: "m0-xxx" });
``` ```
> For standalone features, such as `addMemories`, `retrieveMemories`, and `getMemories`, you must either set `MEM0_API_KEY` as an environment variable or pass it directly in the function call.
> `getMemories` will return raw memories in the form of an array of objects, while `retrieveMemories` will return a response in string format with a system prompt ingested with the retrieved memories. > For standalone features, such as `addMemories`, `retrieveMemories`, and `getMemories`, you must either set `MEM0_API_KEY` as an environment variable or pass it directly in the function call.
> `getMemories` returns an array of memory objects. > `getMemories` will return raw memories in the form of an array of objects, while `retrieveMemories` will return a response in string format with a system prompt ingested with the retrieved memories.
### 1. Basic Text Generation with Memory Context ### 1. Basic Text Generation with Memory Context
```typescript ```typescript
import { generateText } from "ai"; import { generateText } from "ai";
import { createMem0 } from "@mem0/vercel-ai-provider"; import { createMem0 } from "@mem0/vercel-ai-provider";
const mem0 = createMem0(); const mem0 = createMem0();
const { text } = await generateText({ const { text } = await generateText({
model: mem0("gpt-4-turbo", { user_id: "borat" }), model: mem0("gpt-5-mini", { user_id: "borat" }),
prompt: "Suggest me a good car to buy!", prompt: "Suggest me a good car to buy!",
}); });
``` ```
### 2. Combining OpenAI Provider with Memory Utils ### 2. Combining OpenAI Provider with Memory Utils
```typescript ```typescript
import { generateText } from "ai"; import { generateText } from "ai";
import { openai } from "@ai-sdk/openai"; import { openai } from "@ai-sdk/openai";
import { retrieveMemories } from "@mem0/vercel-ai-provider"; import { retrieveMemories } from "@mem0/vercel-ai-provider";
const prompt = "Suggest me a good car to buy."; const prompt = "Suggest me a good car to buy.";
const memories = await retrieveMemories(prompt, { user_id: "borat" }); const memories = await retrieveMemories(prompt, { user_id: "borat" });
const { text } = await generateText({ const { text } = await generateText({
model: openai("gpt-4-turbo"), model: openai("gpt-5-mini"),
prompt: prompt, prompt: prompt,
system: memories, system: memories,
}); });
``` ```
### 3. Structured Message Format with Memory ### 3. Structured Message Format with Memory
```typescript ```typescript
import { generateText } from "ai"; import { generateText } from "ai";
import { createMem0 } from "@mem0/vercel-ai-provider"; import { createMem0 } from "@mem0/vercel-ai-provider";
const mem0 = createMem0(); const mem0 = createMem0();
const { text } = await generateText({ const { text } = await generateText({
model: mem0("gpt-4-turbo", { user_id: "borat" }), model: mem0("gpt-5-mini", { user_id: "borat" }),
messages: [ messages: [
{ {
role: "user", role: "user",
content: [ content: [
{ type: "text", text: "Suggest me a good car to buy." }, { type: "text", text: "Suggest me a good car to buy." },
{ type: "text", text: "Why is it better than the other cars for me?" }, { type: "text", text: "Why is it better than the other cars for me?" },
],
},
], ],
}); },
``` ],
});
```
### 3. Streaming Responses with Memory Context ### 4. Streaming Responses with Memory Context
```typescript ```typescript
import { streamText } from "ai"; import { streamText } from "ai";
import { createMem0 } from "@mem0/vercel-ai-provider"; import { createMem0 } from "@mem0/vercel-ai-provider";
const mem0 = createMem0(); const mem0 = createMem0();
const { textStream } = streamText({ const { textStream } = streamText({
model: mem0("gpt-4-turbo", { model: mem0("gpt-5-mini", {
user_id: "borat", user_id: "borat",
}), }),
prompt: "Suggest me a good car to buy! Why is it better than the other cars for me? Give options for every price range.", prompt: "Suggest me a good car to buy! Why is it better than the other cars for me? Give options for every price range.",
}); });
for await (const textPart of textStream) { for await (const textPart of textStream) {
process.stdout.write(textPart); process.stdout.write(textPart);
} }
``` ```
### 4. Generate Responses with Tools Call ### 5. Generate Responses with Tools Call
```typescript ```typescript
import { generateText } from "ai"; import { generateText, tool } from "ai";
import { createMem0 } from "@mem0/vercel-ai-provider"; import { createMem0 } from "@mem0/vercel-ai-provider";
import { z } from "zod"; import { z } from "zod";
const mem0 = createMem0({ const mem0 = createMem0({
provider: "anthropic", provider: "anthropic",
apiKey: "anthropic-api-key", apiKey: "anthropic-api-key",
mem0Config: { mem0Config: {
// Global User ID user_id: "borat"
user_id: "borat" }
} });
});
const prompt = "What the temperature in the city that I live in?" const result = await generateText({
model: mem0('claude-sonnet-4-20250514'),
tools: {
weather: tool({
description: 'Get the weather in a location',
parameters: z.object({
location: z.string().describe('The location to get the weather for'),
}),
execute: async ({ location }) => ({
location,
temperature: 72 + Math.floor(Math.random() * 21) - 10,
}),
}),
},
prompt: "What the temperature in the city that I live in?",
});
const result = await generateText({ console.log(result);
model: mem0('claude-3-5-sonnet-20240620'), ```
tools: {
weather: tool({
description: 'Get the weather in a location',
parameters: z.object({
location: z.string().describe('The location to get the weather for'),
}),
execute: async ({ location }) => ({
location,
temperature: 72 + Math.floor(Math.random() * 21) - 10,
}),
}),
},
prompt: prompt,
});
console.log(result); ### 6. Get Sources from Memory
```
### 5. Get sources from memory `generateText` and `streamText` responses include Mem0 memories as a source, giving you programmatic access to the memories that influenced the response:
```typescript ```typescript
const { text, sources } = await generateText({ const { text, sources } = await generateText({
model: mem0("gpt-4-turbo"), model: mem0("gpt-5-mini", { user_id: "borat" }),
prompt: "Suggest me a good car to buy!", prompt: "Suggest me a good car to buy!",
}); });
// sources[0].title === "Mem0 Memories"
// sources[0].providerMetadata.mem0.memories — array of memory objects
console.log(sources); console.log(sources);
``` ```
The same can be done for `streamText` as well. The same can be done for `streamText` as well.
### 6. File Support with Memory Context ### 7. File Support with Memory Context
Mem0 AI SDK supports file processing with memory context. Here's an example of analyzing a PDF file: Mem0 AI SDK supports file processing with memory context. Here's an example of analyzing a PDF file:
@@ -226,15 +233,11 @@ const mem0 = createMem0({
}); });
async function main() { async function main() {
// Read the PDF file
const filePath = join(process.cwd(), 'my_pdf.pdf'); const filePath = join(process.cwd(), 'my_pdf.pdf');
const fileBuffer = readFileSync(filePath); const fileBuffer = readFileSync(filePath);
// Convert the file's arrayBuffer to a Base64 data URL
const arrayBuffer = fileBuffer.buffer.slice(fileBuffer.byteOffset, fileBuffer.byteOffset + fileBuffer.byteLength); const arrayBuffer = fileBuffer.buffer.slice(fileBuffer.byteOffset, fileBuffer.byteOffset + fileBuffer.byteLength);
const uint8Array = new Uint8Array(arrayBuffer); const uint8Array = new Uint8Array(arrayBuffer);
// Convert Uint8Array to an array of characters
const charArray = Array.from(uint8Array, byte => String.fromCharCode(byte)); const charArray = Array.from(uint8Array, byte => String.fromCharCode(byte));
const binaryString = charArray.join(''); const binaryString = charArray.join('');
const base64Data = Buffer.from(binaryString, 'binary').toString('base64'); const base64Data = Buffer.from(binaryString, 'binary').toString('base64');
@@ -274,24 +277,56 @@ main();
| Provider | Configuration Value | | Provider | Configuration Value |
|----------|-------------------| |----------|-------------------|
| OpenAI | openai | | OpenAI | `openai` |
| Anthropic | anthropic | | Anthropic | `anthropic` |
| Google | google | | Google / Gemini | `google` or `gemini` |
| Groq | groq | | Groq | `groq` |
| Cohere | `cohere` |
> **Note**: You can use `google` as provider for Gemini (Google) models. They are same and internally they use `@ai-sdk/google` package. > **Note**: You can use either `google` or `gemini` as the provider value for Google Gemini models. Both map to the `@ai-sdk/google` package internally.
## Configuration Options
### Mem0ConfigSettings
These options can be passed per-request when creating a model instance:
| Option | Type | Description |
|--------|------|-------------|
| `user_id` | `string` | User identifier for memory scoping |
| `agent_id` | `string` | Agent identifier |
| `app_id` | `string` | Application identifier |
| `run_id` | `string` | Run/session identifier |
| `metadata` | `object` | Custom metadata for memories |
| `filters` | `object` | Filters for memory search |
| `infer` | `boolean` | Enable inference-based retrieval |
| `top_k` | `number` | Number of memories to retrieve (default: 10) |
| `threshold` | `number` | Relevance threshold for search |
| `rerank` | `boolean` | Enable reranking of results |
| `page` | `number` | Page number for pagination |
| `page_size` | `number` | Results per page |
## Key Features ## Key Features
- `createMem0()`: Initializes a new Mem0 provider instance. - `createMem0()`: Initializes a new Mem0 provider instance implementing `ProviderV3`.
- `retrieveMemories()`: Retrieves memory context for prompts. - `retrieveMemories()`: Retrieves memory context for prompts as a formatted system prompt string.
- `getMemories()`: Get memories from your profile in array format. - `getMemories()`: Get memories from your profile in array format.
- `addMemories()`: Adds user memories to enhance contextual responses. - `addMemories()`: Adds user memories to enhance contextual responses.
## Migrating from v2.x
If you're upgrading from `@mem0/vercel-ai-provider` v2.x:
1. **Upgrade AI SDK**: `npm install ai@^6` and update all `@ai-sdk/*` provider packages to `^3.x`
2. **Remove deprecated params**: Remove `org_id`, `project_id`, `output_format`, `filter_memories`, `async_mode`, `enable_graph` from your config
3. **Remove graph memory**: All graph-related options (`enable_graph`, graph prompts) have been removed. Graph memory is now a project-level setting on the Mem0 Platform
4. **Update imports**: `LanguageModelV2Prompt` is now `LanguageModelV3Prompt` if you import types directly
## Best Practices ## Best Practices
1. **User Identification**: Use a unique `user_id` for consistent memory retrieval. 1. **User Identification**: Use a unique `user_id` for consistent memory retrieval.
2. **Memory Cleanup**: Regularly clean up unused memory data. 2. **Memory Cleanup**: Regularly clean up unused memory data.
3. **Sources**: Access `result.sources` to inspect which memories influenced the response.
> **Note**: We also have support for `agent_id`, `app_id`, and `run_id`. Refer [Docs](/api-reference/memory/add-memories). > **Note**: We also have support for `agent_id`, `app_id`, and `run_id`. Refer [Docs](/api-reference/memory/add-memories).
+16 -16
View File
@@ -1,6 +1,6 @@
{ {
"name": "@mem0/vercel-ai-provider", "name": "@mem0/vercel-ai-provider",
"version": "2.0.6", "version": "3.0.0",
"description": "Vercel AI Provider for providing memory to LLMs", "description": "Vercel AI Provider for providing memory to LLMs",
"main": "./dist/index.js", "main": "./dist/index.js",
"module": "./dist/index.mjs", "module": "./dist/index.mjs",
@@ -31,28 +31,28 @@
"author": "Saket Aryan <saketaryan2002@gmail.com>", "author": "Saket Aryan <saketaryan2002@gmail.com>",
"license": "Apache-2.0", "license": "Apache-2.0",
"dependencies": { "dependencies": {
"@ai-sdk/anthropic": "2.0.0", "@ai-sdk/anthropic": "^3.0.82",
"@ai-sdk/cohere": "2.0.0", "@ai-sdk/cohere": "^3.0.36",
"@ai-sdk/google": "2.0.1", "@ai-sdk/google": "^3.0.80",
"@ai-sdk/groq": "2.0.1", "@ai-sdk/groq": "^3.0.39",
"@ai-sdk/openai": "2.0.2", "@ai-sdk/openai": "^3.0.69",
"@ai-sdk/provider": "2.0.0", "@ai-sdk/provider": "^3.0.10",
"@ai-sdk/provider-utils": "3.0.0", "@ai-sdk/provider-utils": "^4.0.27",
"ai": "5.0.2", "ai": "^6.0.199",
"dotenv": "^16.4.5", "dotenv": "^16.6.1",
"partial-json": "0.1.7", "partial-json": "0.1.7",
"zod": "^3.25.0" "zod": "^3.25.76"
}, },
"devDependencies": { "devDependencies": {
"@edge-runtime/vm": "^3.2.0", "@edge-runtime/vm": "^3.2.0",
"@types/jest": "^29.5.14", "@types/jest": "^29.5.14",
"@types/node": "^18.19.46", "@types/node": "^18.19.130",
"jest": "^29.7.0", "jest": "^29.7.0",
"nodemon": "^3.1.7", "nodemon": "^3.1.14",
"ts-jest": "^29.2.5", "ts-jest": "^29.4.11",
"ts-node": "^10.9.2", "ts-node": "^10.9.2",
"tsup": "^8.3.0", "tsup": "^8.5.1",
"typescript": "^5.5.4" "typescript": "^5.9.3"
}, },
"peerDependencies": { "peerDependencies": {
"zod": "^3.0.0" "zod": "^3.0.0"
+879 -981
View File
File diff suppressed because it is too large Load Diff
+1 -1
View File
@@ -12,7 +12,7 @@ export class Mem0 {
provider: 'openai', provider: 'openai',
}) { }) {
this.baseURL = this.baseURL =
withoutTrailingSlash(options.baseURL) ?? 'http://127.0.0.1:11434/api' withoutTrailingSlash(options.baseURL) ?? 'https://api.openai.com'
this.headers = options.headers this.headers = options.headers
} }
+103 -160
View File
@@ -1,14 +1,14 @@
/* eslint-disable camelcase */ /* eslint-disable camelcase */
import { import {
LanguageModelV2CallOptions, LanguageModelV3,
LanguageModelV2Message, LanguageModelV3CallOptions,
LanguageModelV2Source LanguageModelV3GenerateResult,
LanguageModelV3Message,
LanguageModelV3Source,
LanguageModelV3StreamResult,
} from '@ai-sdk/provider'; } from '@ai-sdk/provider';
import { LanguageModelV2 } from '@ai-sdk/provider'; import { Mem0ChatConfig, Mem0ChatModelId, Mem0ChatSettings, Mem0ConfigSettings } from "./mem0-types";
// streaming uses provider-native doStream; no middleware needed
import { Mem0ChatConfig, Mem0ChatModelId, Mem0ChatSettings, Mem0ConfigSettings, Mem0StreamResponse } from "./mem0-types";
import { Mem0ClassSelector } from "./mem0-provider-selector"; import { Mem0ClassSelector } from "./mem0-provider-selector";
import { Mem0ProviderSettings } from "./mem0-provider"; import { Mem0ProviderSettings } from "./mem0-provider";
import { addMemories, getMemories } from "./mem0-utils"; 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); return Math.random().toString(36).substring(2, 15) + Math.random().toString(36).substring(2, 15);
} }
export class Mem0GenericLanguageModel implements LanguageModelV2 { export class Mem0GenericLanguageModel implements LanguageModelV3 {
readonly specificationVersion = "v2"; readonly specificationVersion = "v3";
readonly defaultObjectGenerationMode = "json";
// We don't support images for now
readonly supportsImageUrls = false;
// Allow All Media Types for now
readonly supportedUrls: Record<string, RegExp[]> = { readonly supportedUrls: Record<string, RegExp[]> = {
'*': [/.*/] '*': [/.*/]
}; };
@@ -38,148 +34,108 @@ export class Mem0GenericLanguageModel implements LanguageModelV2 {
provider: string; provider: string;
private async processMemories(messagesPrompts: LanguageModelV2Message[], mem0Config: Mem0ConfigSettings) { private async processMemories(messagesPrompts: LanguageModelV3Message[], mem0Config: Mem0ConfigSettings) {
try { try {
// Add New Memories // Add new memories — await to ensure writes complete before returning
addMemories(messagesPrompts, mem0Config).then((res) => { try {
return res; await addMemories(messagesPrompts, mem0Config);
}).catch((e) => { } catch (e) {
console.error("Error while adding memories"); console.error("Error while adding memories");
return { memories: [], messagesPrompts: [] }; }
// 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 if (!memories || memories?.length === 0) {
let memories = await getMemories(messagesPrompts, mem0Config); return ans;
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");
} }
let graphPrompt = ""; const mem0Source: LanguageModelV3Source = {
if (isGraphEnabled) { type: "source",
graphPrompt = `HERE ARE THE GRAPHS RELATIONS FOR THE PREFERENCES OF THE USER:\n\n ${memoriesText2}`; sourceType: "url",
} id: "mem0-" + generateRandomId(),
url: "https://app.mem0.ai?utm_source=oss&utm_medium=vercel-ai-sdk-src",
const memoriesPrompt = `System Message: ${mySystemPrompt} ${memoriesText} ${graphPrompt} `; title: "Mem0 Memories",
providerMetadata: {
// System Prompt - The memories go as a system prompt mem0: {
const systemPrompt: LanguageModelV2Message = { memories: memories,
role: "system", memoriesText: memories
content: memoriesPrompt ?.map((memory: any) => memory?.memory)
.join("\n\n"),
},
},
}; };
// Add the system prompt to the beginning of the messages if there are memories return {
if (memories?.length > 0) { ...ans,
messagesPrompts.unshift(systemPrompt); content: [...(ans.content ?? []), mem0Source],
} };
if (isGraphEnabled) {
memories = memories?.results;
}
return { memories, messagesPrompts };
} catch(e) {
console.error("Error while processing memories");
return { memories: [], messagesPrompts };
}
} }
async doGenerate(options: LanguageModelV2CallOptions): Promise<Awaited<ReturnType<LanguageModelV2['doGenerate']>>> { 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,
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']>>> {
try { try {
const provider = this.config.provider; const provider = this.config.provider;
const mem0_api_key = this.config.mem0ApiKey; const mem0_api_key = this.config.mem0ApiKey;
const settings: Mem0ProviderSettings = { const settings: Mem0ProviderSettings = {
provider: provider, provider: provider,
mem0ApiKey: mem0_api_key, mem0ApiKey: mem0_api_key,
@@ -194,31 +150,18 @@ export class Mem0GenericLanguageModel implements LanguageModelV2 {
} }
const selector = new Mem0ClassSelector(this.modelId, settings, this.provider_config); const selector = new Mem0ClassSelector(this.modelId, settings, this.provider_config);
let messagesPrompts = options.prompt; const { messagesPrompts: updatedPrompts } = await this.processMemories(options.prompt, mem0Config);
// Process memories and update prompts
const { memories, messagesPrompts: updatedPrompts } = await this.processMemories(messagesPrompts, mem0Config);
const baseModel = selector.createProvider(); const baseModel = selector.createProvider();
// Use the provider's native streaming directly to avoid buffering
const streamResponse = await baseModel.doStream({ const streamResponse = await baseModel.doStream({
...options, ...options,
prompt: updatedPrompts, prompt: updatedPrompts,
}); });
// If there are no memories, return the original stream // Return the full stream response, preserving all V3 fields (warnings, etc.)
if (!memories || memories?.length === 0) { return streamResponse;
return streamResponse;
}
// Return stream untouched for true streaming behavior
return {
stream: streamResponse.stream,
request: streamResponse.request,
response: streamResponse.response,
};
} catch (error) { } catch (error) {
console.error("Error in doStream:", error); console.error("Error in doStream:", error);
throw new Error("Streaming failed or method not implemented."); throw new Error("Streaming failed or method not implemented.");
+3 -3
View File
@@ -1,13 +1,13 @@
import { Mem0ProviderSettings } from "./mem0-provider"; import { Mem0ProviderSettings } from "./mem0-provider";
import Mem0AITextGenerator, { ProviderSettings } from "./provider-response-provider"; import Mem0AITextGenerator, { ProviderSettings } from "./provider-response-provider";
import { LanguageModelV2 } from '@ai-sdk/provider'; import { LanguageModelV3 } from '@ai-sdk/provider';
class Mem0ClassSelector { class Mem0ClassSelector {
modelId: string; modelId: string;
provider_wrapper: string; provider_wrapper: string;
config: Mem0ProviderSettings; config: Mem0ProviderSettings;
provider_config?: ProviderSettings; 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) { constructor(modelId: string, config: Mem0ProviderSettings, provider_config?: ProviderSettings) {
this.modelId = modelId; this.modelId = modelId;
@@ -24,7 +24,7 @@ class Mem0ClassSelector {
} }
} }
createProvider(): LanguageModelV2 { createProvider(): LanguageModelV3 {
return new Mem0AITextGenerator(this.modelId, this.config , this.provider_config || {}); return new Mem0AITextGenerator(this.modelId, this.config , this.provider_config || {});
} }
} }
+11 -10
View File
@@ -1,20 +1,20 @@
import { ProviderV2 } from '@ai-sdk/provider'; import { ProviderV3 } from '@ai-sdk/provider';
import { LanguageModelV2 } from '@ai-sdk/provider'; import { LanguageModelV3 } from '@ai-sdk/provider';
import { withoutTrailingSlash } from "@ai-sdk/provider-utils"; import { withoutTrailingSlash } from "@ai-sdk/provider-utils";
import { Mem0ChatModelId, Mem0ChatSettings, Mem0Config } from "./mem0-types"; import { Mem0ChatModelId, Mem0ChatSettings, Mem0Config } from "./mem0-types";
import { Mem0GenericLanguageModel } from "./mem0-generic-language-model"; import { Mem0GenericLanguageModel } from "./mem0-generic-language-model";
import { LLMProviderSettings } from "./mem0-types"; import { LLMProviderSettings } from "./mem0-types";
export interface Mem0Provider extends ProviderV2 { export interface Mem0Provider extends ProviderV3 {
(modelId: Mem0ChatModelId, settings?: Mem0ChatSettings): LanguageModelV2; (modelId: Mem0ChatModelId, settings?: Mem0ChatSettings): LanguageModelV3;
chat(modelId: Mem0ChatModelId, settings?: Mem0ChatSettings): LanguageModelV2; chat(modelId: Mem0ChatModelId, settings?: Mem0ChatSettings): LanguageModelV3;
completion(modelId: Mem0ChatModelId, settings?: Mem0ChatSettings): LanguageModelV2; completion(modelId: Mem0ChatModelId, settings?: Mem0ChatSettings): LanguageModelV3;
languageModel( languageModel(
modelId: Mem0ChatModelId, modelId: Mem0ChatModelId,
settings?: Mem0ChatSettings settings?: Mem0ChatSettings
): LanguageModelV2; ): LanguageModelV3;
} }
export interface Mem0ProviderSettings { export interface Mem0ProviderSettings {
@@ -31,7 +31,7 @@ export interface Mem0ProviderSettings {
/** /**
* Custom headers to include in the requests. * Custom headers to include in the requests.
*/ */
headers?: Record<string, string>; headers?: Record<string, string | undefined>;
name?: string; name?: string;
mem0ApiKey?: string; mem0ApiKey?: string;
apiKey?: string; apiKey?: string;
@@ -51,7 +51,7 @@ export function createMem0(
} }
): Mem0Provider { ): Mem0Provider {
const baseURL = const baseURL =
withoutTrailingSlash(options.baseURL) ?? "http://api.openai.com"; withoutTrailingSlash(options.baseURL) ?? "https://api.openai.com";
const getHeaders = () => ({ const getHeaders = () => ({
...options.headers, ...options.headers,
}); });
@@ -113,7 +113,7 @@ export function createMem0(
mem0ApiKey: options.mem0ApiKey, mem0ApiKey: options.mem0ApiKey,
apiKey: options.apiKey, apiKey: options.apiKey,
mem0Config: options.mem0Config, mem0Config: options.mem0Config,
modelType: "completion", modelType: "chat",
}, },
options.config options.config
); );
@@ -131,6 +131,7 @@ export function createMem0(
return createGenericModel(modelId, settings); return createGenericModel(modelId, settings);
}; };
provider.specificationVersion = 'v3';
provider.languageModel = createGenericModel; provider.languageModel = createGenericModel;
provider.completion = createCompletionModel; provider.completion = createCompletionModel;
provider.chat = createChatModel; provider.chat = createChatModel;
+2 -14
View File
@@ -1,9 +1,9 @@
import { Mem0ProviderSettings } from "./mem0-provider"; import { Mem0ProviderSettings } from "./mem0-provider";
import { OpenAIProviderSettings } from "@ai-sdk/openai"; import { OpenAIProviderSettings } from "@ai-sdk/openai";
import { AnthropicProviderSettings } from "@ai-sdk/anthropic"; import { AnthropicProviderSettings } from "@ai-sdk/anthropic";
import { LanguageModelV2 } from '@ai-sdk/provider';
import { CohereProviderSettings } from "@ai-sdk/cohere"; import { CohereProviderSettings } from "@ai-sdk/cohere";
import { GroqProviderSettings } from "@ai-sdk/groq"; import { GroqProviderSettings } from "@ai-sdk/groq";
import { GoogleGenerativeAIProviderSettings } from "@ai-sdk/google";
export type Mem0ChatModelId = export type Mem0ChatModelId =
| (string & NonNullable<unknown>); | (string & NonNullable<unknown>);
@@ -12,10 +12,6 @@ export interface Mem0ConfigSettings {
app_id?: string; app_id?: string;
agent_id?: string; agent_id?: string;
run_id?: string; run_id?: string;
org_name?: string;
project_name?: string;
org_id?: string;
project_id?: string;
metadata?: Record<string, any>; metadata?: Record<string, any>;
filters?: Record<string, any>; filters?: Record<string, any>;
infer?: boolean; infer?: boolean;
@@ -25,20 +21,12 @@ export interface Mem0ConfigSettings {
top_k?: number; top_k?: number;
threshold?: number; threshold?: number;
rerank?: boolean; rerank?: boolean;
enable_graph?: boolean;
host?: string; host?: string;
output_format?: string;
filter_memories?: boolean;
async_mode?: boolean;
} }
export interface Mem0ChatConfig extends Mem0ConfigSettings, Mem0ProviderSettings {} 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 Mem0Config extends Mem0ConfigSettings {}
export interface Mem0ChatSettings extends Mem0ConfigSettings {} export interface Mem0ChatSettings extends Mem0ConfigSettings {}
export interface Mem0StreamResponse extends Awaited<ReturnType<LanguageModelV2['doStream']>> {
memories: any;
}
+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 { Mem0ConfigSettings } from './mem0-types';
import { loadApiKey } from '@ai-sdk/provider-utils'; import { loadApiKey } from '@ai-sdk/provider-utils';
interface MultimodalContent { interface MultimodalContent {
@@ -26,7 +26,7 @@ interface Message {
content: string | MultimodalContent | Array<MultimodalContent>; content: string | MultimodalContent | Array<MultimodalContent>;
} }
const flattenPrompt = (prompt: LanguageModelV2Prompt) => { const flattenPrompt = (prompt: LanguageModelV3Prompt) => {
try { try {
return prompt.map((part) => { return prompt.map((part) => {
if (part.role === "user") { if (part.role === "user") {
@@ -42,7 +42,6 @@ const flattenPrompt = (prompt: LanguageModelV2Prompt) => {
if (content.type === 'text' && content.text) { if (content.type === 'text' && content.text) {
return content.text; return content.text;
} else if (content.type === 'file') { } else if (content.type === 'file') {
// For file content, we'll include a descriptive placeholder
if (content.mediaType === 'application/pdf') { if (content.mediaType === 'application/pdf') {
return '[PDF document]'; return '[PDF document]';
} else if (content.mediaType === 'text/markdown' || content.mediaType === 'application/mdx') { } 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 "[multimodal content]";
} }
return ""; return "";
@@ -66,7 +63,7 @@ const flattenPrompt = (prompt: LanguageModelV2Prompt) => {
} }
} }
const convertToMem0Format = (messages: LanguageModelV2Prompt) => { const convertToMem0Format = (messages: LanguageModelV3Prompt) => {
try { try {
return messages.flatMap((message: any) => { return messages.flatMap((message: any) => {
try { try {
@@ -85,7 +82,6 @@ const convertToMem0Format = (messages: LanguageModelV2Prompt) => {
content: obj.text, content: obj.text,
}; };
} else if (obj.type === "file") { } else if (obj.type === "file") {
// Handle LanguageModelV2Prompt file format
if (obj.mediaType === "application/pdf") { if (obj.mediaType === "application/pdf") {
return { return {
role: message.role, role: message.role,
@@ -155,7 +151,6 @@ const convertToMem0Format = (messages: LanguageModelV2Prompt) => {
} }
}).filter((item: null) => item !== null); }).filter((item: null) => item !== null);
} else { } else {
// Handle single multimodal content object
const obj = message.content; const obj = message.content;
if (obj.type === "text") { if (obj.type === "text") {
return { return {
@@ -163,7 +158,6 @@ const convertToMem0Format = (messages: LanguageModelV2Prompt) => {
content: obj.text, content: obj.text,
}; };
} else if (obj.type === "file") { } else if (obj.type === "file") {
// Handle LanguageModelV2Prompt file format
if (obj.mediaType === "application/pdf") { if (obj.mediaType === "application/pdf") {
return { return {
role: message.role, 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 { try {
const filters: { OR: Array<{ [key: string]: string | undefined }> } = { // v3: entity IDs go inside the filters object, not as top-level params
OR: [], const filters: Record<string, any> = {
...(config?.filters ?? {}),
}; };
if (config?.user_id) { if (config?.user_id) {
filters.OR.push({ filters.user_id = config.user_id;
user_id: config.user_id,
});
} }
if (config?.app_id) { if (config?.app_id) {
filters.OR.push({ filters.app_id = config.app_id;
app_id: config.app_id,
});
} }
if (config?.agent_id) { if (config?.agent_id) {
filters.OR.push({ filters.agent_id = config.agent_id;
agent_id: config.agent_id,
});
} }
if (config?.run_id) { if (config?.run_id) {
filters.OR.push({ filters.run_id = config.run_id;
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,
} }
const apiKey = loadApiKey({ const apiKey = loadApiKey({
apiKey: (config&&config.mem0ApiKey), apiKey: config?.mem0ApiKey,
environmentVariableName: "MEM0_API_KEY", environmentVariableName: "MEM0_API_KEY",
description: "Mem0", 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 = { const options = {
method: 'POST', method: 'POST',
headers: { headers: {
Authorization: `Token ${apiKey}`, Authorization: `Token ${apiKey}`,
'Content-Type': 'application/json' 'Content-Type': 'application/json'
}, },
body: JSON.stringify({ body: JSON.stringify(body),
query,
filters,
...config,
top_k: config&&config.top_k || top_k,
version: "v2",
output_format: "v1.1",
...org_project_filters
}),
}; };
const baseUrl = config?.host || 'https://api.mem0.ai'; 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) { if (!response.ok) {
throw new Error(`HTTP error! status: ${response.status}`); 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 { try {
let finalMessages: Array<Message> = []; let finalMessages: Array<Message> = [];
if (typeof messages === "string") { if (typeof messages === "string") {
@@ -326,22 +314,30 @@ const addMemories = async (messages: LanguageModelV2Prompt, config?: Mem0ConfigS
const updateMemories = async (messages: Array<Message>, config?: Mem0ConfigSettings) => { const updateMemories = async (messages: Array<Message>, config?: Mem0ConfigSettings) => {
try { try {
const apiKey = loadApiKey({ const apiKey = loadApiKey({
apiKey: (config&&config.mem0ApiKey), apiKey: config?.mem0ApiKey,
environmentVariableName: "MEM0_API_KEY", environmentVariableName: "MEM0_API_KEY",
description: "Mem0", 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 = { const options = {
method: 'POST', method: 'POST',
headers: { headers: {
Authorization: `Token ${apiKey}`, Authorization: `Token ${apiKey}`,
'Content-Type': 'application/json' 'Content-Type': 'application/json'
}, },
body: JSON.stringify({messages, ...config, version: "v2"}), body: JSON.stringify(body),
}; };
const baseUrl = config?.host || 'https://api.mem0.ai'; 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) { if (!response.ok) {
throw new Error(`HTTP error! status: ${response.status}`); 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 { try {
const message = typeof prompt === 'string' ? prompt : flattenPrompt(prompt); 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 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 { const data = await searchInternalMemories(message, config);
memoriesText1 = memories?.results?.map((memory: any) => { // The API response may be a flat array or an object with a results key
return `Memory: ${memory.memory}\n\n`; const memories: any[] = Array.isArray(data) ? data : (data?.results ?? []);
}).join("\n\n");
if (config?.enable_graph) { if (memories.length === 0) {
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) {
return ""; 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) { } catch (error) {
console.error("Error in retrieveMemories:", error); console.error("Error in retrieveMemories:", error);
throw error; throw error;
} }
} }
const getMemories = async (prompt: LanguageModelV2Prompt | string, config?: Mem0ConfigSettings) => { const getMemories = async (prompt: LanguageModelV3Prompt | string, config?: Mem0ConfigSettings) => {
try { try {
const message = typeof prompt === 'string' ? prompt : flattenPrompt(prompt); const message = typeof prompt === 'string' ? prompt : flattenPrompt(prompt);
const memories = await searchInternalMemories(message, config); const data = await searchInternalMemories(message, config);
// Normalize: always return a flat array regardless of API response shape
if (!config?.enable_graph) { return Array.isArray(data) ? data : (data?.results ?? []);
return memories?.results;
}
return memories;
} catch (error) { } catch (error) {
console.error("Error in getMemories:", error); console.error("Error in getMemories:", error);
throw error; throw error;
} }
} }
const searchMemories = async (prompt: LanguageModelV2Prompt | string, config?: Mem0ConfigSettings) => { const searchMemories = async (prompt: LanguageModelV3Prompt | string, config?: Mem0ConfigSettings) => {
try { try {
const message = typeof prompt === 'string' ? prompt : flattenPrompt(prompt); const message = typeof prompt === 'string' ? prompt : flattenPrompt(prompt);
const memories = await searchInternalMemories(message, config); 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};
@@ -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 { Mem0ProviderSettings } from "./mem0-provider";
import { createOpenAI, OpenAIProviderSettings } from "@ai-sdk/openai"; import { createOpenAI, OpenAIProviderSettings } from "@ai-sdk/openai";
import { CohereProviderSettings, createCohere } from "@ai-sdk/cohere"; import { CohereProviderSettings, createCohere } from "@ai-sdk/cohere";
@@ -7,10 +7,8 @@ import { createGoogleGenerativeAI, GoogleGenerativeAIProviderSettings } from "@a
import { createGroq, GroqProviderSettings } from "@ai-sdk/groq"; import { createGroq, GroqProviderSettings } from "@ai-sdk/groq";
// Define a private provider field // Define a private provider field
class Mem0AITextGenerator implements LanguageModelV2 { class Mem0AITextGenerator implements LanguageModelV3 {
readonly specificationVersion = "v2"; readonly specificationVersion = "v3";
readonly defaultObjectGenerationMode = "json";
readonly supportsImageUrls = false;
readonly modelId: string; readonly modelId: string;
readonly provider = "mem0"; readonly provider = "mem0";
readonly supportedUrls: Record<string, RegExp[]> = { readonly supportedUrls: Record<string, RegExp[]> = {
@@ -59,11 +57,6 @@ class Mem0AITextGenerator implements LanguageModelV2 {
})(modelId); })(modelId);
break; break;
case "google": case "google":
this.languageModel = createGoogleGenerativeAI({
apiKey: config?.apiKey,
...provider_config as GoogleGenerativeAIProviderSettings,
})(modelId);
break;
case "gemini": case "gemini":
this.languageModel = createGoogleGenerativeAI({ this.languageModel = createGoogleGenerativeAI({
apiKey: config?.apiKey, apiKey: config?.apiKey,
@@ -74,15 +67,15 @@ class Mem0AITextGenerator implements LanguageModelV2 {
throw new Error("Invalid provider"); 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); 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); const result = await this.languageModel.doStream(options);
return result as Awaited<ReturnType<LanguageModelV2['doStream']>>; return result as LanguageModelV3StreamResult;
} }
} }
+2 -2
View File
@@ -1,5 +1,5 @@
import { generateText, streamText } from "ai"; import { generateText, streamText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider'; import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { simulateStreamingMiddleware, wrapLanguageModel } from 'ai'; import { simulateStreamingMiddleware, wrapLanguageModel } from 'ai';
import { addMemories } from "../src"; import { addMemories } from "../src";
import { testConfig } from "../config/test-config"; import { testConfig } from "../config/test-config";
@@ -21,7 +21,7 @@ describe.each(testConfig.providers)('TESTS: Generate/Stream Text with model %s',
beforeAll(async () => { beforeAll(async () => {
// Add some test memories before all tests // Add some test memories before all tests
const messages: LanguageModelV2Prompt = [ const messages: LanguageModelV3Prompt = [
{ {
role: "user", role: "user",
content: [ content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { createMem0, retrieveMemories } from "../../src"; import { createMem0, retrieveMemories } from "../../src";
import { generateText } from "ai"; import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider'; import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config"; import { testConfig } from "../../config/test-config";
import { createCohere } from "@ai-sdk/cohere"; 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 () => { it("should retrieve memories and generate text using COHERE provider", async () => {
const messages: LanguageModelV2Prompt = [ const messages: LanguageModelV3Prompt = [
{ {
role: "user", role: "user",
content: [ content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { createMem0 } from "../../src"; import { createMem0 } from "../../src";
import { generateText } from "ai"; import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider'; import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config"; import { testConfig } from "../../config/test-config";
describe("GOOGLE MEM0 Tests", () => { describe("GOOGLE MEM0 Tests", () => {
@@ -23,7 +23,7 @@ describe("GOOGLE MEM0 Tests", () => {
}); });
it("should retrieve memories and generate text using Google provider", async () => { it("should retrieve memories and generate text using Google provider", async () => {
const messages: LanguageModelV2Prompt = [ const messages: LanguageModelV3Prompt = [
{ {
role: "user", role: "user",
content: [ content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { createMem0, retrieveMemories } from "../../src"; import { createMem0, retrieveMemories } from "../../src";
import { generateText } from "ai"; import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider'; import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config"; import { testConfig } from "../../config/test-config";
import { createGroq } from "@ai-sdk/groq"; 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 () => { it("should retrieve memories and generate text using GROQ provider", async () => {
const messages: LanguageModelV2Prompt = [ const messages: LanguageModelV3Prompt = [
{ {
role: "user", role: "user",
content: [ content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { createMem0 } from "../../src"; import { createMem0 } from "../../src";
import { generateText } from "ai"; import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider'; import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config"; import { testConfig } from "../../config/test-config";
describe("OPENAI MEM0 Tests", () => { describe("OPENAI MEM0 Tests", () => {
@@ -22,7 +22,7 @@ describe("OPENAI MEM0 Tests", () => {
}); });
it("should retrieve memories and generate text using Mem0 OpenAI provider", async () => { it("should retrieve memories and generate text using Mem0 OpenAI provider", async () => {
const messages: LanguageModelV2Prompt = [ const messages: LanguageModelV3Prompt = [
{ {
role: "user", role: "user",
content: [ content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { createMem0, retrieveMemories } from "../../src"; import { createMem0, retrieveMemories } from "../../src";
import { generateText } from "ai"; import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider'; import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config"; import { testConfig } from "../../config/test-config";
import { createAnthropic } from "@ai-sdk/anthropic"; 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 () => { it("should retrieve memories and generate text using ANTHROPIC provider", async () => {
const messages: LanguageModelV2Prompt = [ const messages: LanguageModelV3Prompt = [
{ {
role: "user", role: "user",
content: [ content: [
+4 -4
View File
@@ -1,5 +1,5 @@
import { addMemories, retrieveMemories } from "../src"; import { addMemories, retrieveMemories } from "../src";
import { LanguageModelV2Prompt } from '@ai-sdk/provider'; import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../config/test-config"; import { testConfig } from "../config/test-config";
describe("Memory Core Functions", () => { describe("Memory Core Functions", () => {
@@ -8,7 +8,7 @@ describe("Memory Core Functions", () => {
describe("addMemories", () => { describe("addMemories", () => {
it("should successfully add memories and return correct format", async () => { it("should successfully add memories and return correct format", async () => {
const messages: LanguageModelV2Prompt = [ const messages: LanguageModelV3Prompt = [
{ {
role: "user", role: "user",
content: [ content: [
@@ -34,7 +34,7 @@ describe("Memory Core Functions", () => {
describe("retrieveMemories", () => { describe("retrieveMemories", () => {
beforeEach(async () => { beforeEach(async () => {
// Add some test memories before each retrieval test // Add some test memories before each retrieval test
const messages: LanguageModelV2Prompt = [ const messages: LanguageModelV3Prompt = [
{ {
role: "user", role: "user",
content: [ content: [
@@ -56,7 +56,7 @@ describe("Memory Core Functions", () => {
}); });
it("should retrieve memories with array of prompts", async () => { it("should retrieve memories with array of prompts", async () => {
const messages: LanguageModelV2Prompt = [ const messages: LanguageModelV3Prompt = [
{ {
role: "user", role: "user",
content: [ content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { retrieveMemories } from "../../src"; import { retrieveMemories } from "../../src";
import { generateText } from "ai"; import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider'; import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config"; import { testConfig } from "../../config/test-config";
import { createAnthropic } from "@ai-sdk/anthropic"; 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 () => { it("should retrieve memories and generate text using ANTHROPIC provider", async () => {
const messages: LanguageModelV2Prompt = [ const messages: LanguageModelV3Prompt = [
{ {
role: "user", role: "user",
content: [ content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { retrieveMemories } from "../../src"; import { retrieveMemories } from "../../src";
import { generateText } from "ai"; import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider'; import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config"; import { testConfig } from "../../config/test-config";
import { createCohere } from "@ai-sdk/cohere"; 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 () => { it("should retrieve memories and generate text using COHERE provider", async () => {
const messages: LanguageModelV2Prompt = [ const messages: LanguageModelV3Prompt = [
{ {
role: "user", role: "user",
content: [ content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { retrieveMemories } from "../../src"; import { retrieveMemories } from "../../src";
import { generateText } from "ai"; import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider'; import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config"; import { testConfig } from "../../config/test-config";
import { createGoogleGenerativeAI } from "@ai-sdk/google"; 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 () => { it("should retrieve memories and generate text using Google provider", async () => {
const messages: LanguageModelV2Prompt = [ const messages: LanguageModelV3Prompt = [
{ {
role: "user", role: "user",
content: [ content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { retrieveMemories } from "../../src"; import { retrieveMemories } from "../../src";
import { generateText } from "ai"; import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider'; import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config"; import { testConfig } from "../../config/test-config";
import { createGroq } from "@ai-sdk/groq"; 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 () => { it("should retrieve memories and generate text using GROQ provider", async () => {
const messages: LanguageModelV2Prompt = [ const messages: LanguageModelV3Prompt = [
{ {
role: "user", role: "user",
content: [ content: [
@@ -3,7 +3,7 @@ dotenv.config();
import { retrieveMemories } from "../../src"; import { retrieveMemories } from "../../src";
import { generateText } from "ai"; import { generateText } from "ai";
import { LanguageModelV2Prompt } from '@ai-sdk/provider'; import { LanguageModelV3Prompt } from '@ai-sdk/provider';
import { testConfig } from "../../config/test-config"; import { testConfig } from "../../config/test-config";
import { createOpenAI } from "@ai-sdk/openai"; 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 () => { it("should retrieve memories and generate text using OpenAI provider", async () => {
const messages: LanguageModelV2Prompt = [ const messages: LanguageModelV3Prompt = [
{ {
role: "user", role: "user",
content: [ 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");
});
});
});