mem0-ai-provider
v0.0.1
Published
Vercel AI Provider for providing memory to LLMs
Downloads
76
Readme
Mem0 AI SDK Provider
The Mem0 AI SDK Provider is a community-maintained 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. With Mem0, language model conversations gain memory, enabling more contextualized and personalized responses based on past interactions.
Discover more of Mem0 on GitHub.
For detailed information on using the Vercel AI SDK, refer to Vercel’s API Reference and Documentation.
Features
- 🧠 Persistent memory storage for AI conversations
- 🔄 Seamless integration with Vercel AI SDK
- 🚀 Support for multiple LLM providers
- 📝 Rich message format support
- ⚡ Streaming capabilities
- 🔍 Context-aware responses
Installation
npm install mem0-ai-provider
Before We Begin
Setting Up Mem0
Obtain your Mem0 API Key from the Mem0 dashboard.
Initialize the Mem0 Client:
import { createMem0 } from "mem0-ai-provider"; const mem0 = createMem0({ MEM0_API_KEY: "m0-xxx" // Alternatively, use the MEM0_API_KEY environment variable for better security });
Add Memories to Enhance Context:
import { LanguageModelV1Prompt } from "ai"; import { addMemories } from "mem0-ai-provider"; const messages: LanguageModelV1Prompt = [ { role: "user", content: [ { type: "text", text: "I love red cars." }, { type: "text", text: "I like Toyota Cars." }, { type: "text", text: "I prefer SUVs." }, ], } ]; await addMemories(messages, { user_id: "borat" });
These memories are now stored in your profile. You can view and manage them on the Mem0 Dashboard.
Usage Examples
1. Basic Text Generation with Memory Context
import { generateText } from "ai";
import { createMem0 } from "mem0-ai-provider";
const mem0 = createMem0();
const { text } = await generateText({
model: mem0("openai.gpt-4-turbo", {
user_id: "borat",
}),
prompt: "Suggest me a good car to buy!",
});
2. Combining OpenAI Provider with Memory Utils
import { generateText } from "ai";
import { openai } from "@ai-sdk/openai";
import { retrieveMemories } from "mem0-ai-provider";
const prompt = "Suggest me a good car to buy.";
const memories = await retrieveMemories(prompt, {user_id: "borat"});
const { text } = await generateText({
model: openai("gpt-4-turbo"),
prompt: prompt,
system: memories
});
3. Structured Message Format with Memory
import { generateText } from "ai";
import { createMem0 } from "mem0-ai-provider";
const mem0 = createMem0();
const { text } = await generateText({
model: mem0("openai.gpt-4-turbo", {
user_id: "borat",
}),
messages: [
{
role: "user",
content: [
{ 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: "Give options for every price range." },
],
}
],
});
4. Advanced Memory Integration with OpenAI
import { generateText, LanguageModelV1Prompt } from "ai";
import { openai } from "@ai-sdk/openai";
import { appendMemories, flattenMessages } from "mem0-ai-provider";
// New format using system parameter for memory context
const messages: LanguageModelV1Prompt = [
{
role: "user",
content: [
{ 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: "Give options for every price range." },
],
},
];
const memories = await retrieveMemories(messages, { user_id: "borat" });
const { text } = await generateText({
model: openai("gpt-4-turbo"),
messages: messages,
system: memories,
});
5. Streaming Responses with Memory Context
import { streamText } from "ai";
import { createMem0 } from "mem0-ai-provider";
const mem0 = createMem0();
const { textStream } = await streamText({
model: mem0("openai.gpt-4-turbo", {
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.",
);
for await (const textPart of textStream) {
process.stdout.write(textPart);
}
Core Functions
createMem0()
: Initializes a new mem0 provider instance with optional configurationretrieveMemories()
: Enriches prompts with relevant memoriesaddMemories()
: Add memories to your profile
Configuration Options
const mem0 = createMem0({
compatibility: 'strict', // Enforces strict provider compatibility
});
Best Practices
- User Identification: Always provide a unique
user_id
identifier for consistent memory retrieval - Context Management: Use appropriate context window sizes to balance performance and memory
- Error Handling: Implement proper error handling for memory operations
- Memory Cleanup: Regularly clean up unused memory contexts to optimize performance
We also have support for agent_id
, app_id
, and run_id
. Refer Docs.
Notes
- Requires proper API key configuration for underlying providers (e.g., OpenAI)
- Memory features depend on proper user identification via
user_id
- Supports both streaming and non-streaming responses
- Compatible with all Vercel AI SDK features and patterns
For detailed documentation, examples, and contribution guidelines, please visit our GitHub repository.