Method load

src/memos/mem_os/core.py:1109–1131

Dump the MemCube to a dictionary. Args: load_dir (str): The directory to load the MemCube from. user_id (str, optional): The identifier of the user to load the MemCube from. If None, the default user is used. mem_cube_id (str, optional): Th

(
        self,
        load_dir: str,
        user_id: str | None = None,
        mem_cube_id: str | None = None,
        memory_types: list[Literal["text_mem", "act_mem", "para_mem", "pref_mem"]] | None = None,
    )

        logger.info(f"MemCube {mem_cube_id} dumped to {dump_dir}")

    def load(
        self,
        load_dir: str,
        user_id: str | None = None,
        mem_cube_id: str | None = None,
        memory_types: list[Literal["text_mem", "act_mem", "para_mem", "pref_mem"]] | None = None,
    ) -> None:
        """Dump the MemCube to a dictionary.
        Args:
            load_dir (str): The directory to load the MemCube from.
            user_id (str, optional): The identifier of the user to load the MemCube from.
                If None, the default user is used.
            mem_cube_id (str, optional): The identifier of the MemCube to load.
                If None, the default MemCube for the user is used.
        """
        target_user_id = user_id if user_id is not None else self.user_id
        accessible_cubes = self.user_manager.get_user_cubes(target_user_id)
        if not mem_cube_id:
            mem_cube_id = accessible_cubes[0].cube_id
        if mem_cube_id not in self.mem_cubes:
            raise ValueError(f"MemCube with ID {mem_cube_id} does not exist. please regiester")
        self.mem_cubes[mem_cube_id].load(load_dir, memory_types=memory_types)
        logger.info(f"MemCube {mem_cube_id} loaded from {load_dir}")

    def get_user_info(self) -> dict[str, Any]:
        """Get current user information including accessible cubes.