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

Add limit in get_all and search for Graph (#1920)

Browse Source
This commit is contained in:
Dev Khant
2024-09-28 01:51:38 +05:30
committed by GitHub
parent aaf8e6e7ff
commit 68c7355f47
3 changed files with 18 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mem0/memory/main.py
View File

@@ -286,7 +286,7 @@ class Memory(MemoryBase):
with concurrent.futures.ThreadPoolExecutor() as executor:
future_memories = executor.submit(self._get_all_from_vector_store, filters, limit)
future_graph_entities = (
executor.submit(self.graph.get_all, filters) if self.version == "v1.1" and self.enable_graph else None
executor.submit(self.graph.get_all, filters, limit) if self.version == "v1.1" and self.enable_graph else None
)
all_memories = future_memories.result()
@@ -374,7 +374,7 @@ class Memory(MemoryBase):
with concurrent.futures.ThreadPoolExecutor() as executor:
future_memories = executor.submit(self._search_vector_store, query, filters, limit)
future_graph_entities = (
executor.submit(self.graph.search, query, filters)
executor.submit(self.graph.search, query, filters, limit)
if self.version == "v1.1" and self.enable_graph
else None
)