Skip to content

Commit 60890ee

Browse files
committed
fix: bug in src/memos/graph_dbs/neo4j_community.py
1 parent 15142e3 commit 60890ee

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/memos/graph_dbs/neo4j_community.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -169,12 +169,13 @@ def search_by_embedding(
169169
# Return consistent format
170170
return [{"id": r.id, "score": r.score} for r in results]
171171

172-
def get_all_memory_items(self, scope: str) -> list[dict]:
172+
def get_all_memory_items(self, scope: str, include_embedding: bool = True) -> list[dict]:
173173
"""
174174
Retrieve all memory items of a specific memory_type.
175175
176176
Args:
177177
scope (str): Must be one of 'WorkingMemory', 'LongTermMemory', or 'UserMemory'.
178+
include_embedding (bool): Whether to include the large embedding field.
178179
179180
Returns:
180181
list[dict]: Full list of memory items under this scope.

0 commit comments

Comments
 (0)