From 191ed70bb434276e07c0b61a1e262f4252cb760f Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <shaohu.zhang@fit2cloud.com>
Date: Wed, 18 Sep 2024 13:53:53 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=B7=A5=E4=BD=9C=E6=B5=81=E5=BC=80?=
 =?UTF-8?q?=E5=A7=8B=E8=8A=82=E7=82=B9=E6=B7=BB=E5=8A=A0=E4=B8=8A=E4=B8=8B?=
 =?UTF-8?q?=E6=96=87,=E4=BC=9A=E8=AF=9Did=E5=8F=82=E6=95=B0=20#1094?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../flow/step_node/start_node/i_start_node.py |  3 -
 .../start_node/impl/base_start_node.py        |  7 ++-
 ui/src/workflow/nodes/start-node/index.vue    | 61 ++++++++-----------
 3 files changed, 30 insertions(+), 41 deletions(-)

diff --git a/apps/application/flow/step_node/start_node/i_start_node.py b/apps/application/flow/step_node/start_node/i_start_node.py
index 4c1ecfd2a12..bb23ad3f53e 100644
--- a/apps/application/flow/step_node/start_node/i_start_node.py
+++ b/apps/application/flow/step_node/start_node/i_start_node.py
@@ -16,9 +16,6 @@
 class IStarNode(INode):
     type = 'start-node'
 
-    def get_node_params_serializer_class(self) -> Type[serializers.Serializer] | None:
-        return None
-
     def _run(self):
         return self.execute(**self.flow_params_serializer.data)
 
diff --git a/apps/application/flow/step_node/start_node/impl/base_start_node.py b/apps/application/flow/step_node/start_node/impl/base_start_node.py
index dc4fb541c97..f6528660d76 100644
--- a/apps/application/flow/step_node/start_node/impl/base_start_node.py
+++ b/apps/application/flow/step_node/start_node/impl/base_start_node.py
@@ -15,11 +15,16 @@
 
 class BaseStartStepNode(IStarNode):
     def execute(self, question, **kwargs) -> NodeResult:
+        history_chat_record = self.flow_params_serializer.data.get('history_chat_record', [])
+        history_context = [{'question': chat_record.problem_text, 'answer': chat_record.answer_text} for chat_record in
+                           history_chat_record]
+        chat_id = self.flow_params_serializer.data.get('chat_id')
         """
         开始节点 初始化全局变量
         """
         return NodeResult({'question': question},
-                          {'time': datetime.now().strftime('%Y-%m-%d %H:%M:%S'), 'start_time': time.time()})
+                          {'time': datetime.now().strftime('%Y-%m-%d %H:%M:%S'), 'start_time': time.time(),
+                           'history_context': history_context, 'chat_id': str(chat_id)})
 
     def get_details(self, index: int, **kwargs):
         global_fields = []
diff --git a/ui/src/workflow/nodes/start-node/index.vue b/ui/src/workflow/nodes/start-node/index.vue
index a91e98cbbac..e50a109a4c5 100644
--- a/ui/src/workflow/nodes/start-node/index.vue
+++ b/ui/src/workflow/nodes/start-node/index.vue
@@ -2,25 +2,18 @@
   <NodeContainer :nodeModel="nodeModel">
     <h5 class="title-decoration-1 mb-8">全局变量</h5>
     <div
+      v-for="item in nodeModel.properties.config.globalFields"
       class="flex-between border-r-4 p-8-12 mb-8 layout-bg lighter"
       @mouseenter="showicon = true"
       @mouseleave="showicon = false"
     >
-      <span>当前时间 {time}</span>
+      <span>{{ item.label }} {{ '{' + item.value + '}' }}</span>
       <el-tooltip effect="dark" content="复制参数" placement="top" v-if="showicon === true">
-        <el-button link @click="copyClick(globeLabel)" style="padding: 0">
-          <AppIcon iconName="app-copy"></AppIcon>
-        </el-button>
-      </el-tooltip>
-    </div>
-    <div v-for="(item, index) in inputFieldList" :key="index"
-         class="flex-between border-r-4 p-8-12 mb-8 layout-bg lighter"
-         @mouseenter="showicon = true"
-         @mouseleave="showicon = false"
-    >
-      <span>{{ item.name }} {{ '{' + item.variable + '}' }}</span>
-      <el-tooltip effect="dark" content="复制参数" placement="top" v-if="showicon === true">
-        <el-button link @click="copyClick('{{' + '全局变量.' + item.variable + '}}')" style="padding: 0">
+        <el-button
+          link
+          @click="copyClick('{{' + '全局变量.' + item.value + '}}')"
+          style="padding: 0"
+        >
           <AppIcon iconName="app-copy"></AppIcon>
         </el-button>
       </el-tooltip>
@@ -28,42 +21,36 @@
   </NodeContainer>
 </template>
 <script setup lang="ts">
-import { set } from 'lodash'
+import { cloneDeep, set } from 'lodash'
 import NodeContainer from '@/workflow/common/NodeContainer.vue'
 import { copyClick } from '@/utils/clipboard'
 import { ref, computed, onMounted } from 'vue'
 
 const props = defineProps<{ nodeModel: any }>()
 
-const globeLabel = '{{全局变量.time}}'
-
 const showicon = ref(false)
-
+const globalFields = [
+  { label: '当前时间', value: 'time' },
+  { label: '历史聊天记录', value: 'history_context' },
+  { label: '对话id', value: 'chat_id' }
+]
 const inputFieldList = ref<any[]>([])
 
-function handleRefreshFieldList(data: any[]) {
-  props.nodeModel.graphModel.nodes
+const getRefreshFieldList = () => {
+  return props.nodeModel.graphModel.nodes
     .filter((v: any) => v.id === 'base-node')
-    .map((v: any) => {
-      // eslint-disable-next-line vue/no-mutating-props
-      props.nodeModel.properties.config.globalFields = [
-        {
-          label: '当前时间',
-          value: 'time'
-        }, ...v.properties.input_field_list.map((i: any) => {
-          return { label: i.name, value: i.variable }
-        })
-      ]
-      inputFieldList.value = v.properties.input_field_list
-    })
+    .map((v: any) => cloneDeep(v.properties.input_field_list))
+    .reduce((x: any, y: any) => [...x, ...y], [])
+    .map((i: any) => ({ label: i.name, value: i.variable }))
 }
-
-props.nodeModel.graphModel.eventCenter.on('refreshFieldList', (data: any) => {
-  handleRefreshFieldList(data)
-})
+const refreshFieldList = () => {
+  const refreshFieldList = getRefreshFieldList()
+  set(props.nodeModel.properties.config, 'globalFields', [...globalFields, ...refreshFieldList])
+}
+props.nodeModel.graphModel.eventCenter.on('refreshFieldList', refreshFieldList)
 
 onMounted(() => {
-  handleRefreshFieldList([])
+  refreshFieldList()
 })
 </script>
 <style lang="scss" scoped></style>