Merge branch 'fix/chore-fix' into dev/plugin-deploy

langgenius · Dec 30, 2024 · db1ad3c · db1ad3c
2 parents 7330a4a + 63665a5
commit db1ad3c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/api/core/workflow/nodes/agent/agent_node.py b/api/core/workflow/nodes/agent/agent_node.py
@@ -5,6 +5,7 @@
 from core.plugin.manager.exc import PluginDaemonClientSideError
 from core.workflow.entities.node_entities import NodeRunResult
 from core.workflow.entities.variable_pool import VariablePool
+from core.workflow.enums import SystemVariableKey
 from core.workflow.nodes.agent.entities import AgentNodeData
 from core.workflow.nodes.enums import NodeType
 from core.workflow.nodes.event.event import RunCompletedEvent
@@ -59,12 +60,15 @@ def _run(self) -> Generator:
             for_log=True,
         )
 
+        # get conversation id
+        conversation_id = self.graph_runtime_state.variable_pool.get(["sys", SystemVariableKey.CONVERSATION_ID])
+
         try:
             message_stream = strategy.invoke(
                 params=parameters,
                 user_id=self.user_id,
                 app_id=self.app_id,
-                # TODO: conversation id and message id
+                conversation_id=conversation_id.text if conversation_id else None,
             )
         except Exception as e:
             yield RunCompletedEvent(

diff --git a/api/core/workflow/nodes/tool/tool_node.py b/api/core/workflow/nodes/tool/tool_node.py
@@ -88,6 +88,9 @@ def _run(self) -> Generator:
             for_log=True,
         )
 
+        # get conversation id
+        conversation_id = self.graph_runtime_state.variable_pool.get(["sys", SystemVariableKey.CONVERSATION_ID])
+
         try:
             message_stream = ToolEngine.generic_invoke(
                 tool=tool_runtime,
@@ -97,7 +100,7 @@ def _run(self) -> Generator:
                 workflow_call_depth=self.workflow_call_depth,
                 thread_pool_id=self.thread_pool_id,
                 app_id=self.app_id,
-                # TODO: conversation id and message id
+                conversation_id=conversation_id.text if conversation_id else None,
             )
         except ToolNodeError as e:
             yield RunCompletedEvent(