merge main

web/app/components/workflow/block-selector/market-place-plugin/list.tsx

@@ -11,7 +11,7 @@ import { noop } from 'lodash-es'
 import { getMarketplaceUrl } from '@/utils/var'
 
 export type ListProps = {
-  wrapElemRef: React.RefObject<HTMLElement>
+  wrapElemRef: React.RefObject<HTMLElement | null>
   list: Plugin[]
   searchText: string
   tags: string[]

web/app/components/workflow/block-selector/tool-picker.tsx

@@ -178,6 +178,7 @@ const ToolPicker: FC<Props> = ({
             mcpTools={mcpTools || []}
             selectedTools={selectedTools}
             canChooseMCPTool={canChooseMCPTool}
+            onTagsChange={setTags}
           />
         </div>
       </PortalToFollowElemContent>

web/app/components/workflow/block-selector/tool/tool-list-tree-view/item.tsx

@@ -39,7 +39,6 @@ const Item: FC<Props> = ({
             key={tool.id}
             payload={tool}
             viewType={ViewType.tree}
-            isShowLetterIndex={false}
             hasSearchText={hasSearchText}
             onSelect={onSelect}
             canNotSelectMultiple={canNotSelectMultiple}

web/app/components/workflow/block-selector/types.ts

@@ -37,6 +37,7 @@ export type ToolDefaultValue = {
   paramSchemas: Record<string, any>[]
   credential_id?: string
   meta?: PluginMeta
+  output_schema?: Record<string, any>
 }
 
 export type DataSourceDefaultValue = {

web/app/components/workflow/block-selector/use-sticky-scroll.ts

@@ -8,8 +8,8 @@ export enum ScrollPosition {
 }
 
 type Params = {
-  wrapElemRef: React.RefObject<HTMLElement>
-  nextToStickyELemRef: React.RefObject<HTMLElement>
+  wrapElemRef: React.RefObject<HTMLElement | null>
+  nextToStickyELemRef: React.RefObject<HTMLElement | null>
 }
 const useStickyScroll = ({
   wrapElemRef,

web/app/components/workflow/candidate-node.tsx

@@ -1,7 +1,7 @@
 import {
   memo,
 } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import {
   useReactFlow,
   useStoreApi,

web/app/components/workflow/datasets-detail-store/provider.tsx

@@ -21,7 +21,7 @@ const DatasetsDetailProvider: FC<DatasetsDetailProviderProps> = ({
   nodes,
   children,
 }) => {
-  const storeRef = useRef<DatasetsDetailStoreApi>()
+  const storeRef = useRef<DatasetsDetailStoreApi>(undefined)
 
   if (!storeRef.current)
     storeRef.current = createDatasetsDetailStore()

web/app/components/workflow/datasets-detail-store/store.ts

@@ -2,7 +2,7 @@ import { useContext } from 'react'
 import { createStore, useStore } from 'zustand'
 import type { DataSet } from '@/models/datasets'
 import { DatasetsDetailContext } from './provider'
-import produce from 'immer'
+import { produce } from 'immer'
 
 type DatasetsDetailStore = {
   datasetsDetail: Record<string, DataSet>

web/app/components/workflow/hooks/use-config-vision.ts

@@ -1,4 +1,4 @@
-import produce from 'immer'
+import { produce } from 'immer'
 import { useCallback } from 'react'
 import { useIsChatMode } from './use-workflow'
 import type { ModelConfig, VisionSetting } from '@/app/components/workflow/types'

web/app/components/workflow/hooks/use-edges-interactions-without-sync.ts

@@ -1,5 +1,5 @@
 import { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useStoreApi } from 'reactflow'
 
 export const useEdgesInteractionsWithoutSync = () => {

web/app/components/workflow/hooks/use-edges-interactions.ts

@@ -1,5 +1,5 @@
 import { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import type {
   EdgeMouseHandler,
   OnEdgesChange,

web/app/components/workflow/hooks/use-inspect-vars-crud-common.ts

@@ -10,7 +10,7 @@ import {
   isSystemVar,
   toNodeOutputVars,
 } from '@/app/components/workflow/nodes/_base/components/variable/utils'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { Node } from '@/app/components/workflow/types'
 import { useNodesInteractionsWithoutSync } from '@/app/components/workflow/hooks/use-nodes-interactions-without-sync'
 import { useEdgesInteractionsWithoutSync } from '@/app/components/workflow/hooks/use-edges-interactions-without-sync'

web/app/components/workflow/hooks/use-node-data-update.ts

@@ -1,5 +1,5 @@
 import { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useStoreApi } from 'reactflow'
 import { useNodesSyncDraft } from './use-nodes-sync-draft'
 import { useNodesReadOnly } from './use-workflow'

web/app/components/workflow/hooks/use-nodes-interactions-without-sync.ts

@@ -1,5 +1,5 @@
 import { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useStoreApi } from 'reactflow'
 import { NodeRunningStatus } from '../types'
 

web/app/components/workflow/hooks/use-nodes-interactions.ts

@@ -1,7 +1,7 @@
 import type { MouseEvent } from 'react'
 import { useCallback, useRef, useState } from 'react'
 import { useTranslation } from 'react-i18next'
-import produce from 'immer'
+import { produce } from 'immer'
 import type {
   NodeDragHandler,
   NodeMouseHandler,
@@ -16,7 +16,7 @@ import {
   useReactFlow,
   useStoreApi,
 } from 'reactflow'
-import type { ToolDefaultValue } from '../block-selector/types'
+import type { DataSourceDefaultValue, ToolDefaultValue } from '../block-selector/types'
 import type { Edge, Node, OnNodeAdd } from '../types'
 import { BlockEnum } from '../types'
 import { useWorkflowStore } from '../store'
@@ -1286,7 +1286,7 @@ export const useNodesInteractions = () => {
       currentNodeId: string,
       nodeType: BlockEnum,
       sourceHandle: string,
-      toolDefaultValue?: ToolDefaultValue,
+      toolDefaultValue?: ToolDefaultValue | DataSourceDefaultValue,
     ) => {
       if (getNodesReadOnly()) return
 

web/app/components/workflow/hooks/use-selection-interactions.ts

@@ -2,7 +2,7 @@ import type { MouseEvent } from 'react'
 import {
   useCallback,
 } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import type {
   OnSelectionChangeFunc,
 } from 'reactflow'

web/app/components/workflow/hooks/use-workflow-interactions.ts

@@ -2,7 +2,7 @@ import {
   useCallback,
 } from 'react'
 import { useReactFlow, useStoreApi } from 'reactflow'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useStore, useWorkflowStore } from '../store'
 import {
   CUSTOM_NODE,

web/app/components/workflow/hooks/use-workflow-run-event/use-workflow-agent-log.ts

@@ -1,5 +1,5 @@
 import { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { AgentLogResponse } from '@/types/workflow'
 import { useWorkflowStore } from '@/app/components/workflow/store'
 

web/app/components/workflow/hooks/use-workflow-run-event/use-workflow-failed.ts

@@ -1,5 +1,5 @@
 import { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useWorkflowStore } from '@/app/components/workflow/store'
 import { WorkflowRunningStatus } from '@/app/components/workflow/types'
 

web/app/components/workflow/hooks/use-workflow-run-event/use-workflow-finished.ts

@@ -1,5 +1,5 @@
 import { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { WorkflowFinishedResponse } from '@/types/workflow'
 import { useWorkflowStore } from '@/app/components/workflow/store'
 import { getFilesInLogs } from '@/app/components/base/file-uploader/utils'

web/app/components/workflow/hooks/use-workflow-run-event/use-workflow-node-finished.ts

@@ -1,6 +1,6 @@
 import { useCallback } from 'react'
 import { useStoreApi } from 'reactflow'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { NodeFinishedResponse } from '@/types/workflow'
 import {
   BlockEnum,

web/app/components/workflow/hooks/use-workflow-run-event/use-workflow-node-iteration-finished.ts

@@ -1,6 +1,6 @@
 import { useCallback } from 'react'
 import { useStoreApi } from 'reactflow'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { IterationFinishedResponse } from '@/types/workflow'
 import { useWorkflowStore } from '@/app/components/workflow/store'
 import { DEFAULT_ITER_TIMES } from '@/app/components/workflow/constants'

web/app/components/workflow/hooks/use-workflow-run-event/use-workflow-node-iteration-next.ts

@@ -1,6 +1,6 @@
 import { useCallback } from 'react'
 import { useStoreApi } from 'reactflow'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { IterationNextResponse } from '@/types/workflow'
 import { useWorkflowStore } from '@/app/components/workflow/store'
 

web/app/components/workflow/hooks/use-workflow-run-event/use-workflow-node-iteration-started.ts

@@ -3,7 +3,7 @@ import {
   useReactFlow,
   useStoreApi,
 } from 'reactflow'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useWorkflowStore } from '@/app/components/workflow/store'
 import type { IterationStartedResponse } from '@/types/workflow'
 import { NodeRunningStatus } from '@/app/components/workflow/types'

web/app/components/workflow/hooks/use-workflow-run-event/use-workflow-node-loop-finished.ts

@@ -1,6 +1,6 @@
 import { useCallback } from 'react'
 import { useStoreApi } from 'reactflow'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { LoopFinishedResponse } from '@/types/workflow'
 import { useWorkflowStore } from '@/app/components/workflow/store'
 

web/app/components/workflow/hooks/use-workflow-run-event/use-workflow-node-loop-next.ts

@@ -1,6 +1,6 @@
 import { useCallback } from 'react'
 import { useStoreApi } from 'reactflow'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { LoopNextResponse } from '@/types/workflow'
 import { NodeRunningStatus } from '@/app/components/workflow/types'
 

web/app/components/workflow/hooks/use-workflow-run-event/use-workflow-node-loop-started.ts

@@ -3,7 +3,7 @@ import {
   useReactFlow,
   useStoreApi,
 } from 'reactflow'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useWorkflowStore } from '@/app/components/workflow/store'
 import type { LoopStartedResponse } from '@/types/workflow'
 import { NodeRunningStatus } from '@/app/components/workflow/types'

web/app/components/workflow/hooks/use-workflow-run-event/use-workflow-node-retry.ts

@@ -1,6 +1,6 @@
 import { useCallback } from 'react'
 import { useStoreApi } from 'reactflow'
-import produce from 'immer'
+import { produce } from 'immer'
 import type {
   NodeFinishedResponse,
 } from '@/types/workflow'

web/app/components/workflow/hooks/use-workflow-run-event/use-workflow-node-started.ts

@@ -3,7 +3,7 @@ import {
   useReactFlow,
   useStoreApi,
 } from 'reactflow'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { NodeStartedResponse } from '@/types/workflow'
 import { NodeRunningStatus } from '@/app/components/workflow/types'
 import { useWorkflowStore } from '@/app/components/workflow/store'

web/app/components/workflow/hooks/use-workflow-run-event/use-workflow-started.ts

@@ -1,6 +1,6 @@
 import { useCallback } from 'react'
 import { useStoreApi } from 'reactflow'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { WorkflowStartedResponse } from '@/types/workflow'
 import { WorkflowRunningStatus } from '@/app/components/workflow/types'
 import { useWorkflowStore } from '@/app/components/workflow/store'

web/app/components/workflow/hooks/use-workflow-run-event/use-workflow-text-chunk.ts

@@ -1,5 +1,5 @@
 import { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { TextChunkResponse } from '@/types/workflow'
 import { useWorkflowStore } from '@/app/components/workflow/store'
 

web/app/components/workflow/hooks/use-workflow-run-event/use-workflow-text-replace.ts

@@ -1,5 +1,5 @@
 import { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { TextReplaceResponse } from '@/types/workflow'
 import { useWorkflowStore } from '@/app/components/workflow/store'
 

web/app/components/workflow/nodes/_base/components/agent-strategy-selector.tsx

@@ -212,7 +212,7 @@ export const AgentStrategySelector = memo((props: AgentStrategySelectorProps) =>
                 agent_strategy_name: tool!.tool_name,
                 agent_strategy_provider_name: tool!.provider_name,
                 agent_strategy_label: tool!.tool_label,
-                agent_output_schema: tool!.output_schema,
+                agent_output_schema: tool!.output_schema || {},
                 plugin_unique_identifier: tool!.provider_id,
                 meta: tool!.meta,
               })

web/app/components/workflow/nodes/_base/components/before-run-form/form-item.tsx

@@ -2,7 +2,7 @@
 import type { FC } from 'react'
 import React, { useCallback, useMemo } from 'react'
 import { useTranslation } from 'react-i18next'
-import produce from 'immer'
+import { produce } from 'immer'
 import {
   RiDeleteBinLine,
 } from '@remixicon/react'
@@ -118,8 +118,20 @@ const FormItem: FC<Props> = ({
     <div className={cn(className)}>
       {!isArrayLikeType && !isBooleanType && (
         <div className='system-sm-semibold mb-1 flex h-6 items-center gap-1 text-text-secondary'>
-          <div className='truncate'>{typeof payload.label === 'object' ? nodeKey : payload.label}</div>
-          {!payload.required && <span className='system-xs-regular text-text-tertiary'>{t('workflow.panel.optional')}</span>}
+          <div className='truncate'>
+            {typeof payload.label === 'object' ? nodeKey : payload.label}
+          </div>
+          {payload.hide === true ? (
+            <span className='system-xs-regular text-text-tertiary'>
+              {t('workflow.panel.optional_and_hidden')}
+            </span>
+          ) : (
+            !payload.required && (
+              <span className='system-xs-regular text-text-tertiary'>
+                {t('workflow.panel.optional')}
+              </span>
+            )
+          )}
         </div>
       )}
       <div className='grow'>

web/app/components/workflow/nodes/_base/components/before-run-form/form.tsx

@@ -1,7 +1,7 @@
 'use client'
 import type { FC } from 'react'
 import React, { useCallback, useEffect, useMemo, useRef } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { InputVar } from '../../../../types'
 import FormItem from './form-item'
 import cn from '@/utils/classnames'

web/app/components/workflow/nodes/_base/components/config-vision.tsx

@@ -2,7 +2,7 @@
 import type { FC } from 'react'
 import React, { useCallback } from 'react'
 import { useTranslation } from 'react-i18next'
-import produce from 'immer'
+import { produce } from 'immer'
 import VarReferencePicker from './variable/var-reference-picker'
 import ResolutionPicker from '@/app/components/workflow/nodes/llm/components/resolution-picker'
 import Field from '@/app/components/workflow/nodes/_base/components/field'

web/app/components/workflow/nodes/_base/components/file-upload-setting.tsx

@@ -2,7 +2,7 @@
 import type { FC } from 'react'
 import React, { useCallback } from 'react'
 import useSWR from 'swr'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useTranslation } from 'react-i18next'
 import type { UploadFileSetting } from '../../../types'
 import { SupportUploadFileTypes } from '../../../types'

web/app/components/workflow/nodes/_base/components/input-var-type-icon.tsx

@@ -28,7 +28,6 @@ const getIcon = (type: InputVarType) => {
     [InputVarType.jsonObject]: RiBracesLine,
     [InputVarType.singleFile]: RiFileList2Line,
     [InputVarType.multiFiles]: RiFileCopy2Line,
-    [InputVarType.checkbox]: RiCheckboxLine,
   } as any)[type] || RiTextSnippet
 }
 

web/app/components/workflow/nodes/_base/components/memory-config.tsx

@@ -2,7 +2,7 @@
 import type { FC } from 'react'
 import React, { useCallback } from 'react'
 import { useTranslation } from 'react-i18next'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { Memory } from '../../../types'
 import { MemoryRole } from '../../../types'
 import cn from '@/utils/classnames'

web/app/components/workflow/nodes/_base/components/node-handle.tsx

@@ -16,7 +16,7 @@ import {
 } from '../../../types'
 import type { Node } from '../../../types'
 import BlockSelector from '../../../block-selector'
-import type { ToolDefaultValue } from '../../../block-selector/types'
+import type { DataSourceDefaultValue, ToolDefaultValue } from '../../../block-selector/types'
 import {
   useAvailableBlocks,
   useIsChatMode,
@@ -57,7 +57,7 @@ export const NodeTargetHandle = memo(({
     if (!connected)
       setOpen(v => !v)
   }, [connected])
-  const handleSelect = useCallback((type: BlockEnum, toolDefaultValue?: ToolDefaultValue) => {
+  const handleSelect = useCallback((type: BlockEnum, toolDefaultValue?: ToolDefaultValue | DataSourceDefaultValue) => {
     handleNodeAdd(
       {
         nodeType: type,
@@ -140,7 +140,7 @@ export const NodeSourceHandle = memo(({
     e.stopPropagation()
     setOpen(v => !v)
   }, [])
-  const handleSelect = useCallback((type: BlockEnum, toolDefaultValue?: ToolDefaultValue) => {
+  const handleSelect = useCallback((type: BlockEnum, toolDefaultValue?: ToolDefaultValue | DataSourceDefaultValue) => {
     handleNodeAdd(
       {
         nodeType: type,

web/app/components/workflow/nodes/_base/components/variable/output-var-list.tsx

@@ -1,7 +1,7 @@
 'use client'
 import type { FC } from 'react'
 import React, { useCallback, useState } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useTranslation } from 'react-i18next'
 import type { OutputVar } from '../../../code/types'
 import RemoveButton from '../remove-button'

web/app/components/workflow/nodes/_base/components/variable/utils.ts

@@ -1,4 +1,4 @@
-import produce from 'immer'
+import { produce } from 'immer'
 import { isArray, uniq } from 'lodash-es'
 import type { CodeNodeType } from '../../../code/types'
 import type { EndNodeType } from '../../../end/types'

web/app/components/workflow/nodes/_base/components/variable/var-list.tsx

@@ -2,7 +2,7 @@
 import type { FC } from 'react'
 import React, { useCallback, useMemo, useState } from 'react'
 import { useTranslation } from 'react-i18next'
-import produce from 'immer'
+import { produce } from 'immer'
 import RemoveButton from '../remove-button'
 import VarReferencePicker from './var-reference-picker'
 import Input from '@/app/components/base/input'
@@ -61,7 +61,6 @@ const VarList: FC<Props> = ({
       return
     }
     if (list.some(item => item.variable?.trim() === newKey.trim())) {
-      console.log('new key', newKey.trim())
       setToastHandle(Toast.notify({
         type: 'error',
         message: t('appDebug.varKeyError.keyAlreadyExists', { key: newKey }),

web/app/components/workflow/nodes/_base/components/variable/var-reference-picker.tsx

@@ -9,7 +9,7 @@ import {
   RiLoader4Line,
   RiMoreLine,
 } from '@remixicon/react'
-import produce from 'immer'
+import { produce } from 'immer'
 import {
   useNodes,
   useReactFlow,

web/app/components/workflow/nodes/_base/components/variable/variable-label/hooks.ts

@@ -42,17 +42,17 @@ export const useVarColor = (variables: string[], isExceptionVariable?: boolean,
       return 'text-util-colors-teal-teal-700'
 
     return 'text-text-accent'
-  }, [variables, isExceptionVariable])
+  }, [variables, isExceptionVariable, variableCategory])
 }
 
 export const useVarName = (variables: string[], notShowFullPath?: boolean) => {
-  let variableFullPathName = variables.slice(1).join('.')
-
-  if (isRagVariableVar(variables))
-    variableFullPathName = variables.slice(2).join('.')
-
-  const variablesLength = variables.length
   const varName = useMemo(() => {
+    let variableFullPathName = variables.slice(1).join('.')
+
+    if (isRagVariableVar(variables))
+      variableFullPathName = variables.slice(2).join('.')
+
+    const variablesLength = variables.length
     const isSystem = isSystemVar(variables)
     const varName = notShowFullPath ? variables[variablesLength - 1] : variableFullPathName
     return `${isSystem ? 'sys.' : ''}${varName}`

web/app/components/workflow/nodes/_base/hooks/use-one-step-run.ts

@@ -1,7 +1,7 @@
 import { useCallback, useEffect, useRef, useState } from 'react'
 import { useTranslation } from 'react-i18next'
 import { unionBy } from 'lodash-es'
-import produce from 'immer'
+import { produce } from 'immer'
 import {
   useIsChatMode,
   useNodeDataUpdate,

web/app/components/workflow/nodes/_base/hooks/use-output-var-list.ts

@@ -1,5 +1,5 @@
 import { useCallback, useRef, useState } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useBoolean, useDebounceFn } from 'ahooks'
 import type {
   CodeNodeType,

web/app/components/workflow/nodes/_base/hooks/use-var-list.ts

@@ -1,5 +1,5 @@
 import { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { Variable } from '@/app/components/workflow/types'
 
 type Params<T> = {

web/app/components/workflow/nodes/_base/node.tsx

@@ -48,8 +48,13 @@ import Tooltip from '@/app/components/base/tooltip'
 import useInspectVarsCrud from '../../hooks/use-inspect-vars-crud'
 import { ToolTypeEnum } from '../../block-selector/types'
 
+type NodeChildProps = {
+  id: string
+  data: NodeProps['data']
+}
+
 type BaseNodeProps = {
-  children: ReactElement
+  children: ReactElement<Partial<NodeChildProps>>
   id: NodeProps['id']
   data: NodeProps['data']
 }

web/app/components/workflow/nodes/agent/use-config.ts

@@ -12,7 +12,7 @@ import { useCheckInstalled, useFetchPluginsInMarketPlaceByIds } from '@/service/
 import type { Memory, Var } from '../../types'
 import { VarType as VarKindType } from '../../types'
 import useAvailableVarList from '../_base/hooks/use-available-var-list'
-import produce from 'immer'
+import { produce } from 'immer'
 import { FormTypeEnum } from '@/app/components/header/account-setting/model-provider-page/declarations'
 import { isSupportMCP } from '@/utils/plugin-version-feature'
 import { generateAgentToolValue, toolParametersToFormSchemas } from '@/app/components/tools/utils/to-form-schema'

web/app/components/workflow/nodes/answer/use-config.ts

@@ -1,5 +1,5 @@
 import { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import useVarList from '../_base/hooks/use-var-list'
 import type { Var } from '../../types'
 import { VarType } from '../../types'

web/app/components/workflow/nodes/assigner/components/var-list/index.tsx

@@ -2,7 +2,7 @@
 import type { FC } from 'react'
 import { useTranslation } from 'react-i18next'
 import React, { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { RiDeleteBinLine } from '@remixicon/react'
 import OperationSelector from '../operation-selector'
 import { AssignerNodeInputType, WriteMode } from '../../types'

web/app/components/workflow/nodes/assigner/components/var-list/use-var-list.ts

@@ -1,5 +1,5 @@
 import { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { AssignerNodeOperation, AssignerNodeType } from '../../types'
 import { AssignerNodeInputType, WriteMode } from '../../types'
 

web/app/components/workflow/nodes/assigner/use-config.ts

@@ -1,5 +1,5 @@
 import { useCallback, useMemo } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useStoreApi } from 'reactflow'
 import { VarType } from '../../types'
 import type { ValueSelector, Var } from '../../types'

web/app/components/workflow/nodes/code/types.ts

@@ -11,6 +11,11 @@ export type OutputVar = Record<string, {
   children: null // support nest in the future,
 }>
 
+export type CodeDependency = {
+  name: string
+  version?: string
+}
+
 export type CodeNodeType = CommonNodeType & {
   variables: Variable[]
   code_language: CodeLanguage

web/app/components/workflow/nodes/code/use-config.ts

@@ -1,5 +1,5 @@
 import { useCallback, useEffect, useState } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import useVarList from '../_base/hooks/use-var-list'
 import useOutputVarList from '../_base/hooks/use-output-var-list'
 import { BlockEnum, VarType } from '../../types'

web/app/components/workflow/nodes/document-extractor/use-config.ts

@@ -1,5 +1,5 @@
 import { useCallback, useMemo } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useStoreApi } from 'reactflow'
 import type { ValueSelector, Var } from '../../types'
 import { VarType } from '../../types'

web/app/components/workflow/nodes/http/components/authorization/index.tsx

@@ -2,7 +2,7 @@
 import type { FC } from 'react'
 import { useTranslation } from 'react-i18next'
 import React, { useCallback, useState } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { Authorization as AuthorizationPayloadType } from '../../types'
 import { APIType, AuthorizationType } from '../../types'
 import RadioGroup from './radio-group'

web/app/components/workflow/nodes/http/components/edit-body/index.tsx

@@ -1,7 +1,7 @@
 'use client'
 import type { FC } from 'react'
 import React, { useCallback, useMemo } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { uniqueId } from 'lodash-es'
 import type { Body, BodyPayload, KeyValue as KeyValueType } from '../../types'
 import { BodyPayloadValueType, BodyType } from '../../types'

web/app/components/workflow/nodes/http/components/key-value/key-value-edit/index.tsx

@@ -1,7 +1,7 @@
 'use client'
 import type { FC } from 'react'
 import React, { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useTranslation } from 'react-i18next'
 import type { KeyValue } from '../../../types'
 import KeyValueItem from './item'

web/app/components/workflow/nodes/http/components/key-value/key-value-edit/item.tsx

@@ -2,7 +2,7 @@
 import type { FC } from 'react'
 import React, { useCallback } from 'react'
 import { useTranslation } from 'react-i18next'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { KeyValue } from '../../../types'
 import VarReferencePicker from '../../../../_base/components/variable/var-reference-picker'
 import InputItem from './input-item'

web/app/components/workflow/nodes/http/use-config.ts

@@ -1,5 +1,5 @@
 import { useCallback, useEffect, useState } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useBoolean } from 'ahooks'
 import useVarList from '../_base/hooks/use-var-list'
 import { VarType } from '../../types'
@@ -16,7 +16,7 @@ import {
 const useConfig = (id: string, payload: HttpNodeType) => {
   const { nodesReadOnly: readOnly } = useNodesReadOnly()
 
-  const defaultConfig = useStore(s => s.nodesDefaultConfigs)[payload.type]
+  const defaultConfig = useStore(s => s.nodesDefaultConfigs?.[payload.type])
 
   const { inputs, setInputs } = useNodeCrud<HttpNodeType>(id, payload)
 

web/app/components/workflow/nodes/if-else/components/condition-list/condition-item.tsx

@@ -5,7 +5,7 @@ import {
 } from 'react'
 import { useTranslation } from 'react-i18next'
 import { RiDeleteBinLine } from '@remixicon/react'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { VarType as NumberVarType } from '../../../tool/types'
 import type {
   Condition,
@@ -209,7 +209,7 @@ const ConditionItem = ({
       onRemoveCondition?.(caseId, condition.id)
   }, [caseId, condition, conditionId, isSubVariableKey, onRemoveCondition, onRemoveSubVariableCondition])
 
-  const { getMatchedSchemaType } = useMatchSchemaType()
+  const { schemaTypeDefinitions } = useMatchSchemaType()
   const handleVarChange = useCallback((valueSelector: ValueSelector, _varItem: Var) => {
     const {
       conversationVariables,
@@ -226,7 +226,7 @@ const ConditionItem = ({
         workflowTools,
         dataSourceList: dataSourceList ?? [],
       },
-      getMatchedSchemaType,
+      schemaTypeDefinitions,
     })
 
     const newCondition = produce(condition, (draft) => {
@@ -234,11 +234,14 @@ const ConditionItem = ({
       draft.varType = resolvedVarType
       draft.value = resolvedVarType === VarType.boolean ? false : ''
       draft.comparison_operator = getOperators(resolvedVarType)[0]
+      delete draft.key
+      delete draft.sub_variable_condition
+      delete draft.numberVarType
       setTimeout(() => setControlPromptEditorRerenderKey(Date.now()))
     })
     doUpdateCondition(newCondition)
     setOpen(false)
-  }, [condition, doUpdateCondition, availableNodes, isChatMode, setControlPromptEditorRerenderKey])
+  }, [condition, doUpdateCondition, availableNodes, isChatMode, setControlPromptEditorRerenderKey, schemaTypeDefinitions])
 
   const showBooleanInput = useMemo(() => {
     if(condition.varType === VarType.boolean)

web/app/components/workflow/nodes/if-else/use-config.ts

@@ -1,5 +1,5 @@
 import { useCallback, useMemo } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { v4 as uuid4 } from 'uuid'
 import { useUpdateNodeInternals } from 'reactflow'
 import type {

web/app/components/workflow/nodes/iteration/use-config.ts

@@ -1,5 +1,5 @@
 import { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import {
   useIsChatMode,
   useNodesReadOnly,

web/app/components/workflow/nodes/iteration/use-interactions.ts

@@ -1,5 +1,5 @@
 import { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useTranslation } from 'react-i18next'
 import { useStoreApi } from 'reactflow'
 import type {

web/app/components/workflow/nodes/knowledge-base/components/retrieval-setting/top-k-and-score-threshold.tsx

@@ -1,8 +1,8 @@
-import { memo } from 'react'
+import { memo, useCallback } from 'react'
 import { useTranslation } from 'react-i18next'
 import Tooltip from '@/app/components/base/tooltip'
-import Input from '@/app/components/base/input'
 import Switch from '@/app/components/base/switch'
+import { InputNumber } from '@/app/components/base/input-number'
 
 export type TopKAndScoreThresholdProps = {
   topK: number
@@ -14,6 +14,24 @@ export type TopKAndScoreThresholdProps = {
   readonly?: boolean
   hiddenScoreThreshold?: boolean
 }
+
+const maxTopK = (() => {
+  const configValue = Number.parseInt(globalThis.document?.body?.getAttribute('data-public-top-k-max-value') || '', 10)
+  if (configValue && !isNaN(configValue))
+    return configValue
+  return 10
+})()
+const TOP_K_VALUE_LIMIT = {
+  amount: 1,
+  min: 1,
+  max: maxTopK,
+}
+const SCORE_THRESHOLD_VALUE_LIMIT = {
+  step: 0.01,
+  min: 0,
+  max: 1,
+}
+
 const TopKAndScoreThreshold = ({
   topK,
   onTopKChange,
@@ -25,18 +43,18 @@ const TopKAndScoreThreshold = ({
   hiddenScoreThreshold,
 }: TopKAndScoreThresholdProps) => {
   const { t } = useTranslation()
-  const handleTopKChange = (e: React.ChangeEvent<HTMLInputElement>) => {
-    const value = Number(e.target.value)
-    if (Number.isNaN(value))
-      return
-    onTopKChange?.(value)
-  }
+  const handleTopKChange = useCallback((value: number) => {
+    let notOutRangeValue = Number.parseInt(value.toFixed(0))
+    notOutRangeValue = Math.max(TOP_K_VALUE_LIMIT.min, notOutRangeValue)
+    notOutRangeValue = Math.min(TOP_K_VALUE_LIMIT.max, notOutRangeValue)
+    onTopKChange?.(notOutRangeValue)
+  }, [onTopKChange])
 
-  const handleScoreThresholdChange = (e: React.ChangeEvent<HTMLInputElement>) => {
-    const value = Number(e.target.value)
-    if (Number.isNaN(value))
-      return
-    onScoreThresholdChange?.(value)
+  const handleScoreThresholdChange = (value: number) => {
+    let notOutRangeValue = Number.parseFloat(value.toFixed(2))
+    notOutRangeValue = Math.max(SCORE_THRESHOLD_VALUE_LIMIT.min, notOutRangeValue)
+    notOutRangeValue = Math.min(SCORE_THRESHOLD_VALUE_LIMIT.max, notOutRangeValue)
+    onScoreThresholdChange?.(notOutRangeValue)
   }
 
   return (
@@ -49,11 +67,13 @@ const TopKAndScoreThreshold = ({
             popupContent={t('appDebug.datasetConfig.top_kTip')}
           />
         </div>
-        <Input
+        <InputNumber
+          disabled={readonly}
           type='number'
+          {...TOP_K_VALUE_LIMIT}
+          size='regular'
           value={topK}
           onChange={handleTopKChange}
-          disabled={readonly}
         />
       </div>
       {
@@ -74,11 +94,13 @@ const TopKAndScoreThreshold = ({
                 popupContent={t('appDebug.datasetConfig.score_thresholdTip')}
               />
             </div>
-            <Input
+            <InputNumber
+              disabled={readonly || !isScoreThresholdEnabled}
               type='number'
+              {...SCORE_THRESHOLD_VALUE_LIMIT}
+              size='regular'
               value={scoreThreshold}
               onChange={handleScoreThresholdChange}
-              disabled={readonly || !isScoreThresholdEnabled}
             />
           </div>
         )

web/app/components/workflow/nodes/knowledge-retrieval/components/dataset-list.tsx

@@ -1,7 +1,7 @@
 'use client'
 import type { FC } from 'react'
 import React, { useCallback, useMemo } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useTranslation } from 'react-i18next'
 import Item from './dataset-item'
 import type { DataSet } from '@/models/datasets'

web/app/components/workflow/nodes/knowledge-retrieval/components/metadata/condition-list/condition-number.tsx

@@ -18,7 +18,7 @@ type ConditionNumberProps = {
   nodesOutputVars: NodeOutPutVar[]
   availableNodes: Node[]
   isCommonVariable?: boolean
-  commonVariables: { name: string, type: string }[]
+  commonVariables: { name: string; type: string; value: string }[]
 } & ConditionValueMethodProps
 const ConditionNumber = ({
   value,

web/app/components/workflow/nodes/knowledge-retrieval/components/metadata/condition-list/condition-string.tsx

@@ -18,7 +18,7 @@ type ConditionStringProps = {
   nodesOutputVars: NodeOutPutVar[]
   availableNodes: Node[]
   isCommonVariable?: boolean
-  commonVariables: { name: string, type: string }[]
+  commonVariables: { name: string; type: string; value: string }[]
 } & ConditionValueMethodProps
 const ConditionString = ({
   value,

web/app/components/workflow/nodes/knowledge-retrieval/types.ts

@@ -128,6 +128,6 @@ export type MetadataShape = {
   availableNumberVars?: NodeOutPutVar[]
   availableNumberNodesWithParent?: Node[]
   isCommonVariable?: boolean
-  availableCommonStringVars?: { name: string; type: string; }[]
-  availableCommonNumberVars?: { name: string; type: string; }[]
+  availableCommonStringVars?: { name: string; type: string; value: string }[]
+  availableCommonNumberVars?: { name: string; type: string; value: string }[]
 }

web/app/components/workflow/nodes/knowledge-retrieval/use-config.ts

@@ -4,7 +4,7 @@ import {
   useRef,
   useState,
 } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { isEqual } from 'lodash-es'
 import { v4 as uuid4 } from 'uuid'
 import type { ValueSelector, Var } from '../../types'

web/app/components/workflow/nodes/list-operator/use-config.ts

@@ -1,5 +1,5 @@
 import { useCallback, useMemo } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useStoreApi } from 'reactflow'
 import type { ValueSelector, Var } from '../../types'
 import { VarType } from '../../types'

web/app/components/workflow/nodes/llm/components/config-prompt.tsx

@@ -2,7 +2,7 @@
 import type { FC } from 'react'
 import React, { useCallback } from 'react'
 import { useTranslation } from 'react-i18next'
-import produce from 'immer'
+import { produce } from 'immer'
 import { ReactSortable } from 'react-sortablejs'
 import { v4 as uuid4 } from 'uuid'
 import type { ModelConfig, PromptItem, ValueSelector, Var, Variable } from '../../../types'

web/app/components/workflow/nodes/llm/components/json-schema-config-modal/json-importer.tsx

@@ -24,7 +24,7 @@ const JsonImporter: FC<JsonImporterProps> = ({
   const [open, setOpen] = useState(false)
   const [json, setJson] = useState('')
   const [parseError, setParseError] = useState<any>(null)
-  const importBtnRef = useRef<HTMLButtonElement>(null)
+  const importBtnRef = useRef<HTMLElement>(null)
   const advancedEditing = useVisualEditorStore(state => state.advancedEditing)
   const isAddingNewField = useVisualEditorStore(state => state.isAddingNewField)
   const { emit } = useMittContext()

web/app/components/workflow/nodes/llm/components/json-schema-config-modal/visual-editor/context.tsx

@@ -18,7 +18,7 @@ type VisualEditorProviderProps = {
 export const VisualEditorContext = createContext<VisualEditorContextType>(null)
 
 export const VisualEditorContextProvider = ({ children }: VisualEditorProviderProps) => {
-  const storeRef = useRef<VisualEditorStore>()
+  const storeRef = useRef<VisualEditorStore | null>(null)
 
   if (!storeRef.current)
     storeRef.current = createVisualEditorStore()

web/app/components/workflow/nodes/llm/components/json-schema-config-modal/visual-editor/hooks.ts

@@ -1,4 +1,4 @@
-import produce from 'immer'
+import { produce } from 'immer'
 import type { VisualEditorProps } from '.'
 import { useMittContext } from './context'
 import { useVisualEditorStore } from './store'

web/app/components/workflow/nodes/llm/use-config.ts

@@ -1,5 +1,5 @@
 import { useCallback, useEffect, useRef, useState } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { EditionType, VarType } from '../../types'
 import type { Memory, PromptItem, ValueSelector, Var, Variable } from '../../types'
 import { useStore } from '../../store'
@@ -23,7 +23,7 @@ const useConfig = (id: string, payload: LLMNodeType) => {
   const { nodesReadOnly: readOnly } = useNodesReadOnly()
   const isChatMode = useIsChatMode()
 
-  const defaultConfig = useStore(s => s.nodesDefaultConfigs)[payload.type]
+  const defaultConfig = useStore(s => s.nodesDefaultConfigs)?.[payload.type]
   const [defaultRolePrefix, setDefaultRolePrefix] = useState<{ user: string; assistant: string }>({ user: '', assistant: '' })
   const { inputs, setInputs: doSetInputs } = useNodeCrud<LLMNodeType>(id, payload)
   const inputRef = useRef(inputs)

web/app/components/workflow/nodes/llm/utils.ts

@@ -10,7 +10,7 @@ export const checkNodeValid = (_payload: LLMNodeType) => {
 
 export const getFieldType = (field: Field) => {
   const { type, items } = field
-  if(field.schemaType === 'file') return 'file'
+  if(field.schemaType === 'file') return Type.file
   if (type !== Type.array || !items)
     return type
 

web/app/components/workflow/nodes/loop/components/condition-list/condition-item.tsx

@@ -5,7 +5,7 @@ import {
 } from 'react'
 import { useTranslation } from 'react-i18next'
 import { RiDeleteBinLine } from '@remixicon/react'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { VarType as NumberVarType } from '../../../tool/types'
 import type {
   Condition,
@@ -196,6 +196,9 @@ const ConditionItem = ({
       draft.varType = varItem.type
       draft.value = ''
       draft.comparison_operator = getOperators(varItem.type)[0]
+      delete draft.key
+      delete draft.sub_variable_condition
+      delete draft.numberVarType
     })
     doUpdateCondition(newCondition)
     setOpen(false)

web/app/components/workflow/nodes/loop/use-config.ts

@@ -2,7 +2,7 @@ import {
   useCallback,
   useRef,
 } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { v4 as uuid4 } from 'uuid'
 import {
   useIsChatMode,

web/app/components/workflow/nodes/loop/use-interactions.ts

@@ -1,5 +1,5 @@
 import { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useStoreApi } from 'reactflow'
 import type {
   BlockEnum,

web/app/components/workflow/nodes/parameter-extractor/components/extract-parameter/import-from-tool.tsx

@@ -9,7 +9,10 @@ import BlockSelector from '../../../../block-selector'
 import type { Param, ParamType } from '../../types'
 import cn from '@/utils/classnames'
 import { useStore } from '@/app/components/workflow/store'
-import type { ToolDefaultValue } from '@/app/components/workflow/block-selector/types'
+import type {
+  DataSourceDefaultValue,
+  ToolDefaultValue,
+} from '@/app/components/workflow/block-selector/types'
 import type { ToolParameter } from '@/app/components/tools/types'
 import { CollectionType } from '@/app/components/tools/types'
 import type { BlockEnum } from '@/app/components/workflow/types'
@@ -43,8 +46,11 @@ const ImportFromTool: FC<Props> = ({
   const customTools = useStore(s => s.customTools)
   const workflowTools = useStore(s => s.workflowTools)
 
-  const handleSelectTool = useCallback((_type: BlockEnum, toolInfo?: ToolDefaultValue) => {
-    const { provider_id, provider_type, tool_name } = toolInfo!
+  const handleSelectTool = useCallback((_type: BlockEnum, toolInfo?: ToolDefaultValue | DataSourceDefaultValue) => {
+    if (!toolInfo || 'datasource_name' in toolInfo)
+      return
+
+    const { provider_id, provider_type, tool_name } = toolInfo
     const currentTools = (() => {
       switch (provider_type) {
         case CollectionType.builtIn:

web/app/components/workflow/nodes/parameter-extractor/use-config.ts

@@ -1,5 +1,5 @@
 import { useCallback, useEffect, useRef, useState } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { Memory, MoreInfo, ValueSelector, Var } from '../../types'
 import { ChangeType, VarType } from '../../types'
 import { useStore } from '../../store'
@@ -27,7 +27,7 @@ const useConfig = (id: string, payload: ParameterExtractorNodeType) => {
   const { handleOutVarRenameChange } = useWorkflow()
   const isChatMode = useIsChatMode()
 
-  const defaultConfig = useStore(s => s.nodesDefaultConfigs)[payload.type]
+  const defaultConfig = useStore(s => s.nodesDefaultConfigs)?.[payload.type]
 
   const [defaultRolePrefix, setDefaultRolePrefix] = useState<{ user: string; assistant: string }>({ user: '', assistant: '' })
   const { inputs, setInputs: doSetInputs } = useNodeCrud<ParameterExtractorNodeType>(id, payload)

web/app/components/workflow/nodes/question-classifier/components/class-list.tsx

@@ -1,7 +1,7 @@
 'use client'
 import type { FC } from 'react'
 import React, { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useTranslation } from 'react-i18next'
 import { useEdgesInteractions } from '../../../hooks'
 import AddButton from '../../_base/components/add-button'

web/app/components/workflow/nodes/question-classifier/use-config.ts

@@ -1,5 +1,5 @@
 import { useCallback, useEffect, useRef, useState } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { BlockEnum, VarType } from '../../types'
 import type { Memory, ValueSelector, Var } from '../../types'
 import {
@@ -20,7 +20,7 @@ const useConfig = (id: string, payload: QuestionClassifierNodeType) => {
   const updateNodeInternals = useUpdateNodeInternals()
   const { nodesReadOnly: readOnly } = useNodesReadOnly()
   const isChatMode = useIsChatMode()
-  const defaultConfig = useStore(s => s.nodesDefaultConfigs)[payload.type]
+  const defaultConfig = useStore(s => s.nodesDefaultConfigs)?.[payload.type]
   const { getBeforeNodesInSameBranch } = useWorkflow()
   const startNode = getBeforeNodesInSameBranch(id).find(node => node.data.type === BlockEnum.Start)
   const startNodeId = startNode?.id

web/app/components/workflow/nodes/start/components/var-list.tsx

@@ -1,11 +1,10 @@
 'use client'
 import type { FC } from 'react'
 import React, { useCallback, useMemo } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useTranslation } from 'react-i18next'
 import VarItem from './var-item'
 import { ChangeType, type InputVar, type MoreInfo } from '@/app/components/workflow/types'
-import { v4 as uuid4 } from 'uuid'
 import { ReactSortable } from 'react-sortablejs'
 import { RiDraggable } from '@remixicon/react'
 import cn from '@/utils/classnames'
@@ -71,9 +70,8 @@ const VarList: FC<Props> = ({
   }, [list, onChange])
 
   const listWithIds = useMemo(() => list.map((item) => {
-    const id = uuid4()
     return {
-      id,
+      id: item.variable,
       variable: { ...item },
     }
   }), [list])
@@ -88,6 +86,8 @@ const VarList: FC<Props> = ({
     )
   }
 
+  const canDrag = !readonly && varCount > 1
+
   return (
     <ReactSortable
       className='space-y-1'
@@ -97,30 +97,23 @@ const VarList: FC<Props> = ({
       ghostClass='opacity-50'
       animation={150}
     >
-      {list.map((item, index) => {
-        const canDrag = (() => {
-          if (readonly)
-            return false
-          return varCount > 1
-        })()
-        return (
-          <div key={index} className='group relative'>
-            <VarItem
-              className={cn(canDrag && 'handle')}
-              readonly={readonly}
-              payload={item}
-              onChange={handleVarChange(index)}
-              onRemove={handleVarRemove(index)}
-              varKeys={list.map(item => item.variable)}
-              canDrag={canDrag}
-            />
-            {canDrag && <RiDraggable className={cn(
-              'handle absolute left-3 top-2.5 hidden h-3 w-3 cursor-pointer text-text-tertiary',
-              'group-hover:block',
-            )} />}
-          </div>
-        )
-      })}
+      {listWithIds.map((itemWithId, index) => (
+        <div key={itemWithId.id} className='group relative'>
+          <VarItem
+            className={cn(canDrag && 'handle')}
+            readonly={readonly}
+            payload={itemWithId.variable}
+            onChange={handleVarChange(index)}
+            onRemove={handleVarRemove(index)}
+            varKeys={list.map(item => item.variable)}
+            canDrag={canDrag}
+          />
+          {canDrag && <RiDraggable className={cn(
+            'handle absolute left-3 top-2.5 hidden h-3 w-3 cursor-pointer text-text-tertiary',
+            'group-hover:block',
+          )} />}
+        </div>
+      ))}
     </ReactSortable>
   )
 }

web/app/components/workflow/nodes/start/use-config.ts

@@ -1,5 +1,5 @@
 import { useCallback, useState } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useBoolean } from 'ahooks'
 import type { StartNodeType } from './types'
 import { ChangeType } from '@/app/components/workflow/types'

web/app/components/workflow/nodes/template-transform/use-config.ts

@@ -1,5 +1,5 @@
 import { useCallback, useEffect, useRef } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import useVarList from '../_base/hooks/use-var-list'
 import type { Var, Variable } from '../../types'
 import { VarType } from '../../types'
@@ -13,7 +13,7 @@ import useAvailableVarList from '@/app/components/workflow/nodes/_base/hooks/use
 
 const useConfig = (id: string, payload: TemplateTransformNodeType) => {
   const { nodesReadOnly: readOnly } = useNodesReadOnly()
-  const defaultConfig = useStore(s => s.nodesDefaultConfigs)[payload.type]
+  const defaultConfig = useStore(s => s.nodesDefaultConfigs)?.[payload.type]
 
   const { inputs, setInputs: doSetInputs } = useNodeCrud<TemplateTransformNodeType>(id, payload)
   const inputsRef = useRef(inputs)

web/app/components/workflow/nodes/tool/components/input-var-list.tsx

@@ -1,7 +1,7 @@
 'use client'
 import type { FC } from 'react'
 import React, { useCallback, useState } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useTranslation } from 'react-i18next'
 import type { ToolVarInputs } from '../types'
 import { VarType as VarKindType } from '../types'

web/app/components/workflow/nodes/tool/use-config.ts

@@ -1,6 +1,6 @@
 import { useCallback, useEffect, useMemo, useState } from 'react'
 import { useTranslation } from 'react-i18next'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useBoolean } from 'ahooks'
 import { useStore, useWorkflowStore } from '../../store'
 import type { ToolNodeType, ToolVarInputs } from './types'

web/app/components/workflow/nodes/tool/use-single-run-form-params.ts

@@ -5,7 +5,7 @@ import useNodeCrud from '../_base/hooks/use-node-crud'
 import { type ToolNodeType, VarType } from './types'
 import type { ValueSelector } from '@/app/components/workflow/types'
 import type { Props as FormProps } from '@/app/components/workflow/nodes/_base/components/before-run-form/form'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { NodeTracing } from '@/types/workflow'
 import { useTranslation } from 'react-i18next'
 import formatToTracingNodeList from '@/app/components/workflow/run/utils/format-log'

web/app/components/workflow/nodes/variable-assigner/components/var-group-item.tsx

@@ -2,7 +2,7 @@
 import React, { useCallback } from 'react'
 import type { ChangeEvent, FC } from 'react'
 import { useTranslation } from 'react-i18next'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useBoolean } from 'ahooks'
 import {
   RiDeleteBinLine,

web/app/components/workflow/nodes/variable-assigner/components/var-list/index.tsx

@@ -2,7 +2,7 @@
 import type { FC } from 'react'
 import { useTranslation } from 'react-i18next'
 import React, { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import RemoveButton from '../../../_base/components/remove-button'
 import ListNoDataPlaceholder from '../../../_base/components/list-no-data-placeholder'
 import VarReferencePicker from '@/app/components/workflow/nodes/_base/components/variable/var-reference-picker'

web/app/components/workflow/nodes/variable-assigner/components/var-list/use-var-list.ts

@@ -1,5 +1,5 @@
 import { useCallback } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import type { VariableAssignerNodeType } from '../../types'
 import type { ValueSelector } from '@/app/components/workflow/types'
 

web/app/components/workflow/nodes/variable-assigner/hooks.ts

@@ -4,7 +4,7 @@ import {
   useStoreApi,
 } from 'reactflow'
 import { uniqBy } from 'lodash-es'
-import produce from 'immer'
+import { produce } from 'immer'
 import {
   useIsChatMode,
   useNodeDataUpdate,

web/app/components/workflow/nodes/variable-assigner/use-config.ts

@@ -1,5 +1,5 @@
 import { useCallback, useRef, useState } from 'react'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useBoolean, useDebounceFn } from 'ahooks'
 import { v4 as uuid4 } from 'uuid'
 import type { ValueSelector, Var } from '../../types'

web/app/components/workflow/panel/chat-variable-panel/components/array-bool-list.tsx

@@ -3,7 +3,7 @@ import type { FC } from 'react'
 import React, { useCallback } from 'react'
 import { useTranslation } from 'react-i18next'
 import { RiAddLine } from '@remixicon/react'
-import produce from 'immer'
+import { produce } from 'immer'
 import RemoveButton from '@/app/components/workflow/nodes/_base/components/remove-button'
 import Button from '@/app/components/base/button'
 import BoolValue from './bool-value'

web/app/components/workflow/panel/chat-variable-panel/components/array-value-list.tsx

@@ -3,7 +3,7 @@ import type { FC } from 'react'
 import React, { useCallback } from 'react'
 import { useTranslation } from 'react-i18next'
 import { RiAddLine } from '@remixicon/react'
-import produce from 'immer'
+import { produce } from 'immer'
 import RemoveButton from '@/app/components/workflow/nodes/_base/components/remove-button'
 import Button from '@/app/components/base/button'
 import Input from '@/app/components/base/input'

web/app/components/workflow/panel/chat-variable-panel/components/object-value-item.tsx

@@ -2,7 +2,7 @@
 import type { FC } from 'react'
 import React, { useCallback, useState } from 'react'
 import { useTranslation } from 'react-i18next'
-import produce from 'immer'
+import { produce } from 'immer'
 import { useContext } from 'use-context-selector'
 import { ToastContext } from '@/app/components/base/toast'
 import VariableTypeSelector from '@/app/components/workflow/panel/chat-variable-panel/components/variable-type-select'