From 6a9f5341b58b962b540401d0075fd0ad64f9f026 Mon Sep 17 00:00:00 2001 From: Dev Khant Date: Tue, 20 Aug 2024 14:37:17 +0530 Subject: [PATCH] User ID needed for .add() and .search() (#1719) --- mem0/memory/main.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/mem0/memory/main.py b/mem0/memory/main.py index 64443e95..330fc6f8 100644 --- a/mem0/memory/main.py +++ b/mem0/memory/main.py @@ -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(