Fix AI SDK Filters (#2625)

This commit is contained in:
Saket Aryan
2025-05-05 19:38:58 +05:30
committed by GitHub
parent c81e2efbb0
commit a0fe9ca5b2
2 changed files with 23 additions and 16 deletions

View File

@@ -1,6 +1,6 @@
{ {
"name": "@mem0/vercel-ai-provider", "name": "@mem0/vercel-ai-provider",
"version": "1.0.1", "version": "1.0.2",
"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",

View File

@@ -41,22 +41,29 @@ const convertToMem0Format = (messages: LanguageModelV1Prompt) => {
})}; })};
const searchInternalMemories = async (query: string, config?: Mem0ConfigSettings, top_k: number = 5)=> { const searchInternalMemories = async (query: string, config?: Mem0ConfigSettings, top_k: number = 5)=> {
const filters = { const filters: { AND: Array<{ [key: string]: string | undefined }> } = {
OR: [ AND: [],
{
user_id: config&&config.user_id,
},
{
app_id: config&&config.app_id,
},
{
agent_id: config&&config.agent_id,
},
{
run_id: config&&config.run_id,
},
],
}; };
if (config?.user_id) {
filters.AND.push({
user_id: config.user_id,
});
}
if (config?.app_id) {
filters.AND.push({
app_id: config.app_id,
});
}
if (config?.agent_id) {
filters.AND.push({
agent_id: config.agent_id,
});
}
if (config?.run_id) {
filters.AND.push({
run_id: config.run_id,
});
}
const org_project_filters = { const org_project_filters = {
org_id: config&&config.org_id, org_id: config&&config.org_id,
project_id: config&&config.project_id, project_id: config&&config.project_id,