Skip to content

Commit e638039

Browse files
author
yuan.wang
committed
modify tool resp bug in multi cube
1 parent c861f61 commit e638039

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/memos/multi_mem_cube/composite_cube.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ def search_memories(self, search_req: APISearchRequest) -> dict[str, Any]:
4343
"para_mem": [],
4444
"pref_mem": [],
4545
"pref_note": "",
46+
"tool_mem": [],
4647
}
4748

4849
for view in self.cube_views:
@@ -52,6 +53,7 @@ def search_memories(self, search_req: APISearchRequest) -> dict[str, Any]:
5253
merged_results["act_mem"].extend(cube_result.get("act_mem", []))
5354
merged_results["para_mem"].extend(cube_result.get("para_mem", []))
5455
merged_results["pref_mem"].extend(cube_result.get("pref_mem", []))
56+
merged_results["tool_mem"].extend(cube_result.get("tool_mem", []))
5557

5658
note = cube_result.get("pref_note")
5759
if note:

0 commit comments

Comments
 (0)