feat: enhance memory management with extra identifier support

- Updated MemoryManager to accept an optional extra_identifier parameter in store and query methods, allowing for more granular memory isolation.
- Modified ChatMemoryQuery and ChatMemoryStore blocks to utilize the new extra_identifier feature, improving memory management capabilities across different workflows.

Author: lss233

kirara_ai/memory/memory_manager.py

@@ -73,9 +73,11 @@ def register_decomposer(self, name: str, decomposer_class: Type[MemoryDecomposer
         """注册新的解析器"""
         self.decomposer_registry.register(name, decomposer_class)
 
-    def store(self, scope: MemoryScope, entry: MemoryEntry) -> None:
+    def store(self, scope: MemoryScope, entry: MemoryEntry, extra_identifier: Optional[str] = None) -> None:
         """存储新的记忆"""
         scope_key = scope.get_scope_key(entry.sender)
+        if extra_identifier is not None:
+            scope_key = f"{extra_identifier}-{scope_key}"
 
         if scope_key not in self.memories:
             self.memories[scope_key] = self.persistence.load(scope_key)
@@ -95,10 +97,12 @@ def store(self, scope: MemoryScope, entry: MemoryEntry) -> None:
 
         self.persistence.save(scope_key, self.memories[scope_key])
 
-    def query(self, scope: MemoryScope, sender: ChatSender) -> List[MemoryEntry]:
+    def query(self, scope: MemoryScope, sender: ChatSender, extra_identifier: Optional[str] = None) -> List[MemoryEntry]:
         """查询历史记忆"""
         relevant_memories = []
         scope_key = scope.get_scope_key(sender)
+        if extra_identifier is not None:
+            scope_key = f"{extra_identifier}-{scope_key}"
 
         if scope_key not in self.memories:
             self.memories[scope_key] = self.persistence.load(scope_key)

kirara_ai/workflow/implementations/blocks/memory/chat_memory.py

@@ -12,7 +12,7 @@
 
 
 def scope_type_options_provider(container: DependencyContainer, block: Block) -> List[str]:
-    return ["global", "member", "group"]
+    return ["global", "group", "member"]
 
 
 def decomposer_name_options_provider(container: DependencyContainer, block: Block) -> List[str]:
@@ -36,7 +36,7 @@ def __init__(
             Optional[str],
             ParamMeta(
                 label="级别",
-                description="要查询记忆的级别",
+                description="要查询记忆的级别，代表记忆可以被共享的粒度。（例如：member 级别下，同一群聊下不同用户的记忆互相隔离； group 级别下，同一群组内所有成员记忆共享，但不同群组之间记忆互相隔离）",
                 options_provider=scope_type_options_provider,
             ),
         ],
@@ -48,9 +48,17 @@ def __init__(
                 options_provider=decomposer_name_options_provider,
             ),
         ] = "default",
+        extra_identifier: Annotated[
+            Optional[str],
+            ParamMeta(
+                label="额外隔离标识符",
+                description="仅支持输入英文，可为空。对于同一用户，不同标识符之间的记忆互相隔离。可用于避免不同工作流之间记忆互相干扰。",
+            ),
+        ] = None,
     ):
         self.scope_type = scope_type
         self.decomposer_name: str = decomposer_name or "default"
+        self.extra_identifier = extra_identifier
 
     def execute(self, chat_sender: ChatSender) -> Dict[str, Any]:
         self.memory_manager = self.container.resolve(MemoryManager)
@@ -68,7 +76,7 @@ def execute(self, chat_sender: ChatSender) -> Dict[str, Any]:
         self.decomposer = decomposer_registry.get_decomposer(
             self.decomposer_name)
 
-        entries = self.memory_manager.query(self.scope, chat_sender)
+        entries = self.memory_manager.query(self.scope, chat_sender, self.extra_identifier)
         memory_content = self.decomposer.decompose(entries)
         return {"memory_content": memory_content}
 
@@ -94,13 +102,21 @@ def __init__(
             Optional[str],
             ParamMeta(
                 label="级别",
-                description="要查询记忆的级别",
+                description="要查询记忆的级别，代表记忆可以被共享的粒度。（例如：member 级别下，同一群聊下不同用户的记忆互相隔离； group 级别下，同一群组内所有成员记忆共享，但不同群组之间记忆互相隔离）",
                 options_provider=scope_type_options_provider,
             ),
         ],
+        extra_identifier: Annotated[
+            Optional[str],
+            ParamMeta(
+                label="额外隔离标识符",
+                description="仅支持输入英文，可为空。对于同一用户，不同标识符之间的记忆互相隔离。可用于避免不同工作流之间记忆互相干扰。",
+            ),
+        ] = None,
     ):
         self.scope_type = scope_type
         self.logger = get_logger("Block.ChatMemoryStore")
+        self.extra_identifier = extra_identifier
 
     def execute(
         self,
@@ -140,6 +156,6 @@ def execute(
         self.logger.debug(f"Composed messages: {composed_messages}")
         memory_entries = self.composer.compose(
             user_msg.sender if user_msg else None, composed_messages)
-        self.memory_manager.store(self.scope, memory_entries)
+        self.memory_manager.store(self.scope, memory_entries, self.extra_identifier)
 
         return {}