Skip to content

Commit 0b77b76

Browse files
committed
adding memories with async mode
1 parent c02ba62 commit 0b77b76

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

veadk/memory/long_term_memory_backends/mem0_backend.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,7 @@ def save_memory(self, event_strings: list[str], **kwargs) -> bool:
8989
[{"role": "user", "content": event_string}],
9090
user_id=user_id,
9191
output_format="v1.1",
92+
async_mode=True,
9293
)
9394
logger.debug(f"Saved memory result: {result}")
9495

@@ -121,7 +122,16 @@ def search_memory(self, query: str, top_k: int, **kwargs) -> list[str]:
121122
query, user_id=user_id, output_format="v1.1", top_k=top_k
122123
)
123124

125+
logger.debug(f"return relevant memories: {memories}")
126+
124127
memory_list = []
128+
# 如果 memories 是列表,直接返回
129+
if isinstance(memories, list):
130+
for mem in memories:
131+
if "memory" in mem:
132+
memory_list.append(mem["memory"])
133+
return memory_list
134+
125135
if memories.get("results", []):
126136
for mem in memories["results"]:
127137
if "memory" in mem:

0 commit comments

Comments
 (0)