From 6e15d7f777014cf7f22cc455ec3dc9b3d4e5a7ea Mon Sep 17 00:00:00 2001
From: Joel <iamjoel007@gmail.com>
Date: Fri, 26 Jul 2024 15:00:29 +0800
Subject: [PATCH] feat: doc extract inputs

---
 .../workflow/nodes/doc-extractor/panel.tsx           | 12 +++++++++++-
 web/i18n/en-US/workflow.ts                           |  4 ++++
 web/i18n/zh-Hans/workflow.ts                         |  6 ++++++
 3 files changed, 21 insertions(+), 1 deletion(-)

diff --git a/web/app/components/workflow/nodes/doc-extractor/panel.tsx b/web/app/components/workflow/nodes/doc-extractor/panel.tsx
index 612220b811..b5177306c9 100644
--- a/web/app/components/workflow/nodes/doc-extractor/panel.tsx
+++ b/web/app/components/workflow/nodes/doc-extractor/panel.tsx
@@ -2,6 +2,7 @@ import type { FC } from 'react'
 import React from 'react'
 import { useTranslation } from 'react-i18next'
 import VarReferencePicker from '../_base/components/variable/var-reference-picker'
+import OutputVars, { VarItem } from '../_base/components/output-vars'
 import useConfig from './use-config'
 import type { DocExtractorNodeType } from './types'
 import Field from '@/app/components/workflow/nodes/_base/components/field'
@@ -26,7 +27,7 @@ const Panel: FC<NodePanelProps<DocExtractorNodeType>> = ({
     <div className='mt-2'>
       <div className='px-4 pb-4 space-y-4'>
         <Field
-          title={t(`${i18nPrefix}.assignedVariable`)}
+          title={t(`${i18nPrefix}.inputVars`)}
         >
           <VarReferencePicker
             readonly={readOnly}
@@ -38,6 +39,15 @@ const Panel: FC<NodePanelProps<DocExtractorNodeType>> = ({
           />
         </Field>
       </div>
+      <div className='px-4 pt-4 pb-2'>
+        <OutputVars>
+          <VarItem
+            name='text'
+            type='string'
+            description={t(`${i18nPrefix}.outputVars.text`)}
+          />
+        </OutputVars>
+      </div>
     </div>
   )
 }
diff --git a/web/i18n/en-US/workflow.ts b/web/i18n/en-US/workflow.ts
index 9b6f538380..180e4fb8ec 100644
--- a/web/i18n/en-US/workflow.ts
+++ b/web/i18n/en-US/workflow.ts
@@ -494,6 +494,10 @@ const translation = {
       },
     },
     docExtractor: {
+      inputVars: 'Input Variables',
+      outputVars: {
+        text: 'Extracted text',
+      },
     },
   },
   tracing: {
diff --git a/web/i18n/zh-Hans/workflow.ts b/web/i18n/zh-Hans/workflow.ts
index 5b1574ac6c..d5296f9f10 100644
--- a/web/i18n/zh-Hans/workflow.ts
+++ b/web/i18n/zh-Hans/workflow.ts
@@ -493,6 +493,12 @@ const translation = {
         showAuthor: '显示作者',
       },
     },
+    docExtractor: {
+      inputVars: '输入变量',
+      outputVars: {
+        text: '提取的文本',
+      },
+    },
   },
   tracing: {
     stopBy: '由{{user}}终止',