klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

User ID needed for .add() and .search() (#1719)

Browse Source
This commit is contained in:
Dev Khant
2024-08-20 14:37:17 +05:30
committed by GitHub
parent e31ca239a0
commit 6a9f5341b5
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
mem0/memory/main.py
View File

@@ -86,6 +86,11 @@ class Memory(MemoryBase):
if run_id: if run_id:
filters["run_id"] = metadata["run_id"] = run_id filters["run_id"] = metadata["run_id"] = run_id
if not any(key in filters for key in ("user_id", "agent_id", "run_id")):
raise ValueError(
"One of the filters: user_id, agent_id or run_id is required!"
)
if not prompt: if not prompt:
prompt = MEMORY_DEDUCTION_PROMPT.format(user_input=data, metadata=metadata) prompt = MEMORY_DEDUCTION_PROMPT.format(user_input=data, metadata=metadata)
extracted_memories = self.llm.generate_response( extracted_memories = self.llm.generate_response(
@@ -292,6 +297,11 @@ class Memory(MemoryBase):
if run_id: if run_id:
filters["run_id"] = run_id filters["run_id"] = run_id
if not any(key in filters for key in ("user_id", "agent_id", "run_id")):
raise ValueError(
"One of the filters: user_id, agent_id or run_id is required!"
)
capture_event("mem0.search", self, {"filters": len(filters), "limit": limit}) capture_event("mem0.search", self, {"filters": len(filters), "limit": limit})
embeddings = self.embedding_model.embed(query) embeddings = self.embedding_model.embed(query)
memories = self.vector_store.search( memories = self.vector_store.search(