From 3e24a00e8dbe532fd52ba6c3e26e179e37c527ce Mon Sep 17 00:00:00 2001 From: "jinli.yl" Date: Thu, 27 Jun 2024 11:57:35 +0800 Subject: [PATCH] [dev] add workflow name to context --- config/config.yaml | 1 - memory_scope/constants/common_constants.py | 2 ++ memory_scope/memory/operation/base_workflow.py | 2 ++ memory_scope/memory/worker/dummy_worker.py | 7 ++++--- 4 files changed, 8 insertions(+), 4 deletions(-) diff --git a/config/config.yaml b/config/config.yaml index 7d98f49d..da00b719 100644 --- a/config/config.yaml +++ b/config/config.yaml @@ -33,7 +33,6 @@ memory_service: workflow: dummy description: "write observation memories of the user" interval_time: 60 - contextual_msg_count: 6 summary_memory: class: memory.operation.summary_memory workflow: dummy diff --git a/memory_scope/constants/common_constants.py b/memory_scope/constants/common_constants.py index d7379f91..4d35f18b 100644 --- a/memory_scope/constants/common_constants.py +++ b/memory_scope/constants/common_constants.py @@ -1,3 +1,5 @@ +WORKFLOW_NAME = "workflow_name" + RESULT = "result" CHAT_MESSAGES = "chat_messages" diff --git a/memory_scope/memory/operation/base_workflow.py b/memory_scope/memory/operation/base_workflow.py index 8b5ac67c..540f6cb8 100644 --- a/memory_scope/memory/operation/base_workflow.py +++ b/memory_scope/memory/operation/base_workflow.py @@ -5,6 +5,7 @@ from itertools import zip_longest from typing import Dict, Any, List from memory_scope.chat_v2.global_context import G_CONTEXT +from memory_scope.constants.common_constants import WORKFLOW_NAME from memory_scope.memory.worker.base_worker import BaseWorker from memory_scope.utils.logger import Logger from memory_scope.utils.timer import Timer @@ -103,6 +104,7 @@ class BaseWorkflow(object): def run_workflow(self): with Timer(f"run_workflow_{self.name}"): + self.context[WORKFLOW_NAME] = self.name for workflow_part in self.workflow_worker_list: if len(workflow_part) == 1: if not self._run_sub_workflow(workflow_part[0]): diff --git a/memory_scope/memory/worker/dummy_worker.py b/memory_scope/memory/worker/dummy_worker.py index d0cb7d93..5ae68efc 100644 --- a/memory_scope/memory/worker/dummy_worker.py +++ b/memory_scope/memory/worker/dummy_worker.py @@ -1,8 +1,9 @@ -from memory_scope.constants.common_constants import RESULT +from memory_scope.constants.common_constants import RESULT, WORKFLOW_NAME from memory_scope.memory.worker.base_worker import BaseWorker class DummyWorker(BaseWorker): def _run(self): - self.set_context(RESULT, ["test 123"]) - self.logger.info("enter dummy worker!") + workflow_name = self.get_context(WORKFLOW_NAME) + self.logger.info(f"enter workflow={workflow_name}.dummy_worker!") + self.set_context(RESULT, f"test {workflow_name}")