memory search clean up

superagentxai · Dec 23, 2024 · 997d0e6 · 997d0e6
1 parent 425eb4f
commit 997d0e6
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 4 deletions.
diff --git a/superagentx/agentxpipe.py b/superagentx/agentxpipe.py
@@ -189,7 +189,6 @@ async def retrieve_memory(
         return await self.memory.search(
             query=query_instruction,
             memory_id=self.memory_id,
-            chat_id=self.chat_id,
             limit=10,
         )
 

diff --git a/superagentx/memory/__init__.py b/superagentx/memory/__init__.py
@@ -81,7 +81,6 @@ async def search(
             self,
             query: str,
             memory_id: str,
-            chat_id: str,
             limit: int = 10,
             filters: dict | None = None
     ) -> list[dict]:

diff --git a/tests/memory/test_memory.py b/tests/memory/test_memory.py
@@ -96,8 +96,7 @@ async def test_search(self, test_memory_init: dict):
         client: Memory = test_memory_init.get("client")
         response = await client.search(
             query="agentic",
-            memory_id=datas.get("memory_id"),
-            chat_id=datas.get("chat_id")
+            memory_id=datas.get("memory_id")
         )
         logger.info(response)
         return response