From 6b06c5b9574daf85302e15e3ca60e36557d4f544 Mon Sep 17 00:00:00 2001
From: takatost <takatost@gmail.com>
Date: Tue, 2 Apr 2024 22:36:07 +0800
Subject: [PATCH] optimize workflow inputs

---
 api/core/app/task_pipeline/workflow_cycle_manage.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/api/core/app/task_pipeline/workflow_cycle_manage.py b/api/core/app/task_pipeline/workflow_cycle_manage.py
index 42ea3ca780..49c79a5c05 100644
--- a/api/core/app/task_pipeline/workflow_cycle_manage.py
+++ b/api/core/app/task_pipeline/workflow_cycle_manage.py
@@ -71,6 +71,9 @@ class WorkflowCycleManage:
 
         inputs = {**user_inputs}
         for key, value in (system_inputs or {}).items():
+            if key.value == 'conversation':
+                continue
+
             inputs[f'sys.{key.value}'] = value
         inputs = WorkflowEngineManager.handle_special_values(inputs)
 
@@ -267,8 +270,6 @@ class WorkflowCycleManage:
         :param workflow_run: workflow run
         :return:
         """
-        inputs = {variable: value for variable, value in workflow_run.inputs_dict.items()
-                  if variable != 'sys.conversation'}
         return WorkflowStartStreamResponse(
             task_id=task_id,
             workflow_run_id=workflow_run.id,
@@ -276,7 +277,7 @@ class WorkflowCycleManage:
                 id=workflow_run.id,
                 workflow_id=workflow_run.workflow_id,
                 sequence_number=workflow_run.sequence_number,
-                inputs=inputs,
+                inputs=workflow_run.inputs_dict,
                 created_at=int(workflow_run.created_at.timestamp())
             )
         )