From 68b0af8ee36970329abba7c30d59d353df1f8ebf Mon Sep 17 00:00:00 2001 From: "jinli.yl" Date: Thu, 27 Jun 2024 11:52:34 +0800 Subject: [PATCH] [dev] rename operation name --- config/config.yaml | 3 ++- memory_scope/memory/worker/base_worker.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) 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]