diff --git a/config/config.yaml b/config/config.yaml index 696c39cb..7d98f49d 100644 --- a/config/config.yaml +++ b/config/config.yaml @@ -13,8 +13,9 @@ memory_service: class: memory.service.chat_memory_service history_msg_count: 32 contextual_msg_count: 6 + read_memory_key: read_memory memory_operations: - read_user_message: + contextual_message: class: memory.operation.read_memory workflow: dummy description: "read session messages of the user" diff --git a/memory_scope/memory/worker/base_worker.py b/memory_scope/memory/worker/base_worker.py index 9f9ed979..85426cc3 100644 --- a/memory_scope/memory/worker/base_worker.py +++ b/memory_scope/memory/worker/base_worker.py @@ -52,5 +52,5 @@ class BaseWorker(metaclass=ABCMeta): else: self.context[key] = value - def __getattr__(self, key): + def __getattr__(self, key: str): return self.kwargs[key]