Method run

src/memos/mem_chat/simple.py:37–138

Run the MemChat.

(self)

        self._mem_cube = value

    def run(self) -> None:
        """Run the MemChat."""

        # Start MemChat

        print(
            "\n📢 [System] " + "Simple MemChat is running.\n"
            "Commands: 'bye' to quit, 'clear' to clear chat history, 'mem' to show all memories, 'export' to export chat history\n",
        )

        messages = []
        while True:
            # Get user input

            user_input = input("👤 [You] ").strip()
            print()

            if user_input.lower() == "bye":
                break
            elif user_input.lower() == "clear":
                messages = []
                print("📢 [System] Chat history cleared.")
                continue
            elif user_input.lower() == "mem":
                if self.config.enable_textual_memory:
                    all_memories = self.mem_cube.text_mem.get_all()
                    print(f"🧠 [Memory] \n{self._str_memories(all_memories)}\n")
                else:
                    print("📢 [System] Textual memory is not enabled.\n")
                continue
            elif user_input.lower() == "export":
                if messages:
                    filepath = self._export_chat_history(messages)
                    print(f"📢 [System] Chat history exported to: {filepath}\n")
                else:
                    print("📢 [System] No chat history to export.\n")
                continue
            elif user_input == "":
                continue

            # Get memories

            if self.config.enable_textual_memory:
                memories = self.mem_cube.text_mem.search(user_input, top_k=self.config.top_k)
                print(
                    f"🧠 [Memory] Searched memories:\n{self._str_memories(memories, mode='concise')}\n"
                )
                system_prompt = self._build_system_prompt(memories)
            else:
                system_prompt = self._build_system_prompt()
            current_messages = [
                {"role": "system", "content": system_prompt},
                *messages,
                {"role": "user", "content": user_input},
            ]

            if self.config.enable_activation_memory:
                past_key_values = None