From da560e595033fe7ca410400e94a14bce3c1c9391 Mon Sep 17 00:00:00 2001
From: Joel <iamjoel007@gmail.com>
Date: Wed, 16 Jul 2025 16:16:30 +0800
Subject: [PATCH] chore: current ui fix

---
 web/app/components/base/prompt-editor/index.tsx        |  2 +-
 .../_base/components/variable/var-reference-vars.tsx   | 10 +++++++++-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/web/app/components/base/prompt-editor/index.tsx b/web/app/components/base/prompt-editor/index.tsx
index 4288ad1e26..7d70f2b10e 100644
--- a/web/app/components/base/prompt-editor/index.tsx
+++ b/web/app/components/base/prompt-editor/index.tsx
@@ -112,7 +112,7 @@ const PromptEditor: FC<PromptEditorProps> = ({
   workflowVariableBlock,
   currentBlock = {
     show: true,
-    generatorType: GeneratorType.prompt,
+    generatorType: GeneratorType.code,
   },
   isSupportFileVar,
 }) => {
diff --git a/web/app/components/workflow/nodes/_base/components/variable/var-reference-vars.tsx b/web/app/components/workflow/nodes/_base/components/variable/var-reference-vars.tsx
index 3414903318..8575aca247 100644
--- a/web/app/components/workflow/nodes/_base/components/variable/var-reference-vars.tsx
+++ b/web/app/components/workflow/nodes/_base/components/variable/var-reference-vars.tsx
@@ -86,6 +86,14 @@ const Item: FC<ItemProps> = ({
     }
   }, [isFlat, isInCodeGeneratorInstructionEditor, itemData.variable])
 
+  const varName = useMemo(() => {
+    if(!isFlat)
+      return itemData.variable
+    if (itemData.variable === 'current')
+      return isInCodeGeneratorInstructionEditor ? 'current_code' : 'current_prompt'
+    return itemData.variable
+  }, [isFlat, isInCodeGeneratorInstructionEditor, itemData.variable])
+
   const objStructuredOutput: StructuredOutput | null = useMemo(() => {
     if (!isObj) return null
     const properties: Record<string, Field> = {};
@@ -175,7 +183,7 @@ const Item: FC<ItemProps> = ({
             {isLoopVar && <Loop className='h-3.5 w-3.5 shrink-0 text-util-colors-cyan-cyan-500' />}
             {isFlat && flatVarIcon}
             {!isEnv && !isChatVar && (
-              <div title={itemData.variable} className='system-sm-medium ml-1 w-0 grow truncate text-text-secondary'>{itemData.variable}</div>
+              <div title={itemData.variable} className='system-sm-medium ml-1 w-0 grow truncate text-text-secondary'>{varName}</div>
             )}
             {isEnv && (
               <div title={itemData.variable} className='system-sm-medium ml-1 w-0 grow truncate text-text-secondary'>{itemData.variable.replace('env.', '')}</div>