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

This commit is contained in:
Dev Khant
2024-08-20 14:37:17 +05:30
committed by GitHub
parent e31ca239a0
commit 6a9f5341b5

View File

@@ -86,6 +86,11 @@ class Memory(MemoryBase):
if 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:
prompt = MEMORY_DEDUCTION_PROMPT.format(user_input=data, metadata=metadata)
extracted_memories = self.llm.generate_response(
@@ -292,6 +297,11 @@ class Memory(MemoryBase):
if 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})
embeddings = self.embedding_model.embed(query)
memories = self.vector_store.search(