From c34fc071e06df201988c738ce53d48a1f561b9e2 Mon Sep 17 00:00:00 2001
From: Joel <iamjoel007@gmail.com>
Date: Wed, 28 Aug 2024 10:57:26 +0800
Subject: [PATCH] feat: vison file to api define

---
 web/app/components/workflow/hooks/use-config-vision.ts      | 2 +-
 .../workflow/nodes/_base/components/config-vision.tsx       | 6 +++---
 web/app/components/workflow/nodes/llm/default.ts            | 2 +-
 .../workflow/nodes/parameter-extractor/default.ts           | 2 +-
 .../workflow/nodes/question-classifier/default.ts           | 2 +-
 web/app/components/workflow/types.ts                        | 2 +-
 6 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/web/app/components/workflow/hooks/use-config-vision.ts b/web/app/components/workflow/hooks/use-config-vision.ts
index bc0f45a9e9..7c36ed939a 100644
--- a/web/app/components/workflow/hooks/use-config-vision.ts
+++ b/web/app/components/workflow/hooks/use-config-vision.ts
@@ -62,7 +62,7 @@ const useConfigVision = (model: ModelConfig, {
         enabled: true,
         configs: {
           detail: Resolution.high,
-          valueSelector: [],
+          variable_selector: [],
         },
       })
     }
diff --git a/web/app/components/workflow/nodes/_base/components/config-vision.tsx b/web/app/components/workflow/nodes/_base/components/config-vision.tsx
index 2fc0bac7de..ac7a9b7563 100644
--- a/web/app/components/workflow/nodes/_base/components/config-vision.tsx
+++ b/web/app/components/workflow/nodes/_base/components/config-vision.tsx
@@ -30,7 +30,7 @@ const ConfigVision: FC<Props> = ({
   nodeId,
   config = {
     detail: Resolution.high,
-    valueSelector: [],
+    variable_selector: [],
   },
   onConfigChange,
 }) => {
@@ -48,7 +48,7 @@ const ConfigVision: FC<Props> = ({
 
   const handleVarSelectorChange = useCallback((valueSelector: ValueSelector | string) => {
     const newConfig = produce(config, (draft) => {
-      draft.valueSelector = valueSelector as ValueSelector
+      draft.variable_selector = valueSelector as ValueSelector
     })
     onConfigChange(newConfig)
   }, [config, onConfigChange])
@@ -74,7 +74,7 @@ const ConfigVision: FC<Props> = ({
               className='mb-4'
               filterVar={filterVar}
               nodeId={nodeId}
-              value={config.valueSelector || []}
+              value={config.variable_selector || []}
               onChange={handleVarSelectorChange}
               readonly={readOnly}
             />
diff --git a/web/app/components/workflow/nodes/llm/default.ts b/web/app/components/workflow/nodes/llm/default.ts
index 1cfb5c977a..e6f3e1f56d 100644
--- a/web/app/components/workflow/nodes/llm/default.ts
+++ b/web/app/components/workflow/nodes/llm/default.ts
@@ -79,7 +79,7 @@ const nodeDefault: NodeDefault<LLMNodeType> = {
         })
       }
     }
-    if (!errorMessages && payload.vision?.enabled && !payload.vision.configs?.valueSelector?.length)
+    if (!errorMessages && payload.vision?.enabled && !payload.vision.configs?.variable_selector?.length)
       errorMessages = t(`${i18nPrefix}.fieldRequired`, { field: t(`${i18nPrefix}.fields.visionVariable`) })
     return {
       isValid: !errorMessages,
diff --git a/web/app/components/workflow/nodes/parameter-extractor/default.ts b/web/app/components/workflow/nodes/parameter-extractor/default.ts
index e767300353..69bb67eb9b 100644
--- a/web/app/components/workflow/nodes/parameter-extractor/default.ts
+++ b/web/app/components/workflow/nodes/parameter-extractor/default.ts
@@ -57,7 +57,7 @@ const nodeDefault: NodeDefault<ParameterExtractorNodeType> = {
           errorMessages = t(`${i18nPrefix}.errorMsg.fieldRequired`, { field: t(`${i18nPrefix}.nodes.parameterExtractor.addExtractParameterContent.descriptionPlaceholder`) })
       })
     }
-    if (!errorMessages && payload.vision?.enabled && !payload.vision.configs?.valueSelector?.length)
+    if (!errorMessages && payload.vision?.enabled && !payload.vision.configs?.variable_selector?.length)
       errorMessages = t(`${i18nPrefix}.errorMsg.fieldRequired`, { field: t(`${i18nPrefix}.errorMsg.fields.visionVariable`) })
     return {
       isValid: !errorMessages,
diff --git a/web/app/components/workflow/nodes/question-classifier/default.ts b/web/app/components/workflow/nodes/question-classifier/default.ts
index bd99dbadac..a0936b66e3 100644
--- a/web/app/components/workflow/nodes/question-classifier/default.ts
+++ b/web/app/components/workflow/nodes/question-classifier/default.ts
@@ -54,7 +54,7 @@ const nodeDefault: NodeDefault<QuestionClassifierNodeType> = {
     if (!errorMessages && (payload.classes.some(item => !item.name)))
       errorMessages = t(`${i18nPrefix}.errorMsg.fieldRequired`, { field: t(`${i18nPrefix}.nodes.questionClassifiers.topicName`) })
 
-    if (!errorMessages && payload.vision?.enabled && !payload.vision.configs?.valueSelector?.length)
+    if (!errorMessages && payload.vision?.enabled && !payload.vision.configs?.variable_selector?.length)
       errorMessages = t(`${i18nPrefix}.errorMsg.fieldRequired`, { field: t(`${i18nPrefix}.errorMsg.fields.visionVariable`) })
     return {
       isValid: !errorMessages,
diff --git a/web/app/components/workflow/types.ts b/web/app/components/workflow/types.ts
index e7df40e2b1..7fe8093bba 100644
--- a/web/app/components/workflow/types.ts
+++ b/web/app/components/workflow/types.ts
@@ -369,6 +369,6 @@ export type UploadFileSetting = {
 }
 
 export type VisionSetting = {
-  valueSelector: ValueSelector
+  variable_selector: ValueSelector
   detail: Resolution
 }