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

fix(memory): honor prompt param in vector store extraction (#4914)

Browse Source
This commit is contained in:
Andrew Halpern
2026-04-23 22:36:54 +05:30
committed by GitHub
parent 5be2630f5b
commit cca7551192
3 changed files with 44 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mem0/memory/main.py
+6 -5
View File
@@ -656,10 +656,10 @@ class Memory(MemoryBase):
else:
messages = parse_vision_messages(messages)
vector_store_result = self._add_to_vector_store(messages, processed_metadata, effective_filters, infer)
vector_store_result = self._add_to_vector_store(messages, processed_metadata, effective_filters, infer, prompt=prompt)
return {"results": vector_store_result}
def _add_to_vector_store(self, messages, metadata, filters, infer):
def _add_to_vector_store(self, messages, metadata, filters, infer, prompt=None):
if not infer:
returned_memories = []
for message_dict in messages:
@@ -726,7 +726,7 @@ class Memory(MemoryBase):
if is_agent_scoped:
system_prompt += AGENT_CONTEXT_SUFFIX
custom_instr = self.custom_instructions
custom_instr = prompt or self.custom_instructions
user_prompt = generate_additive_extraction_prompt(
existing_memories=existing_memories,
@@ -2064,7 +2064,7 @@ class AsyncMemory(MemoryBase):
else:
messages = parse_vision_messages(messages)
vector_store_result = await self._add_to_vector_store(messages, processed_metadata, effective_filters, infer)
vector_store_result = await self._add_to_vector_store(messages, processed_metadata, effective_filters, infer, prompt=prompt)
return {"results": vector_store_result}
async def _add_to_vector_store(
@@ -2073,6 +2073,7 @@ class AsyncMemory(MemoryBase):
metadata: dict,
effective_filters: dict,
infer: bool,
prompt: Optional[str] = None,
):
if not infer:
returned_memories = []
@@ -2141,7 +2142,7 @@ class AsyncMemory(MemoryBase):
if is_agent_scoped:
system_prompt += AGENT_CONTEXT_SUFFIX
custom_instr = self.custom_instructions
custom_instr = prompt or self.custom_instructions
user_prompt = generate_additive_extraction_prompt(
existing_memories=existing_memories,