Feat/tool memory (#603)

* function call supoort

* add tool parser

* rename multi model to modal

* rename multi modal

* tool mem support

* modify multi-modal code

* pref support multi-modal messages

* modify bug in chat handle

* fix pre commit

* modify code

* add tool search

* tool search

* add split chunck in system and tool

* fix bug in plug pref search

* fix bug in pref add

---------

Co-authored-by: yuan.wang <[email protected]>

Author: Wang-Daoji

src/memos/multi_mem_cube/single_cube.py

@@ -389,6 +389,7 @@ def _search_pref(
                 top_k=search_req.pref_top_k,
                 info={
                     "user_id": search_req.user_id,
+                    "mem_cube_id": user_context.mem_cube_id,
                     "session_id": search_req.session_id,
                     "chat_history": search_req.chat_history,
                 },
@@ -566,7 +567,7 @@ def _process_pref_mem(
                 message_item_pref = ScheduleMessageItem(
                     user_id=add_req.user_id,
                     session_id=target_session_id,
-                    mem_cube_id=self.cube_id,
+                    mem_cube_id=user_context.mem_cube_id,
                     mem_cube=self.naive_mem_cube,
                     label=PREF_ADD_LABEL,
                     content=json.dumps(messages_list),
@@ -591,7 +592,7 @@ def _process_pref_mem(
                     **(add_req.info or {}),
                     "user_id": add_req.user_id,
                     "session_id": target_session_id,
-                    "mem_cube_id": self.cube_id,
+                    "mem_cube_id": user_context.mem_cube_id,
                 },
             )
             pref_ids_local: list[str] = self.naive_mem_cube.pref_mem.add(pref_memories_local)