[dev] add workflow name to context

This commit is contained in:
jinli.yl 2024-06-27 11:57:35 +08:00
parent 68b0af8ee3
commit 3e24a00e8d
4 changed files with 8 additions and 4 deletions

View file

@ -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

View file

@ -1,3 +1,5 @@
WORKFLOW_NAME = "workflow_name"
RESULT = "result"
CHAT_MESSAGES = "chat_messages"

View file

@ -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]):

View file

@ -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}")