mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-10-06 02:48:22 +00:00
[dev] add workflow name to context
This commit is contained in:
parent
68b0af8ee3
commit
3e24a00e8d
4 changed files with 8 additions and 4 deletions
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -1,3 +1,5 @@
|
|||
WORKFLOW_NAME = "workflow_name"
|
||||
|
||||
RESULT = "result"
|
||||
|
||||
CHAT_MESSAGES = "chat_messages"
|
||||
|
|
|
|||
|
|
@ -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]):
|
||||
|
|
|
|||
|
|
@ -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}")
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue