Merge branch 'feat/mcp-06-18' into deploy/dev

2026-05-04 01:18:05 +08:00 · 2025-10-20 10:58:27 +08:00
parent 1705c07967 8cf4a0d3ad
commit 40e3d4dc7d
463 changed files with 10928 additions and 7216 deletions
--- a/web/app/components/workflow/block-selector/tool-picker.tsx
+++ b/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>
--- a/web/app/components/workflow/block-selector/tool/tool-list-tree-view/item.tsx
+++ b/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}
--- a/web/app/components/workflow/block-selector/types.ts
+++ b/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 = {
--- a/web/app/components/workflow/block-selector/use-sticky-scroll.ts
+++ b/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,
--- a/web/app/components/workflow/datasets-detail-store/provider.tsx
+++ b/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()
--- a/web/app/components/workflow/hooks/use-nodes-interactions.ts
+++ b/web/app/components/workflow/hooks/use-nodes-interactions.ts
@ -15,7 +15,7 @@ import {
  getOutgoers,
  useReactFlow,
 } 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'
@ -1264,7 +1264,7 @@ export const useNodesInteractions = () => {
      currentNodeId: string,
      nodeType: BlockEnum,
      sourceHandle: string,
-      toolDefaultValue?: ToolDefaultValue,
+      toolDefaultValue?: ToolDefaultValue | DataSourceDefaultValue,
    ) => {
      if (getNodesReadOnly()) return

--- a/web/app/components/workflow/nodes/_base/components/agent-strategy-selector.tsx
+++ b/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,
              })
--- a/web/app/components/workflow/nodes/_base/components/input-var-type-icon.tsx
+++ b/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
 }

--- a/web/app/components/workflow/nodes/_base/components/node-handle.tsx
+++ b/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,
--- a/web/app/components/workflow/nodes/_base/components/variable/variable-label/hooks.ts
+++ b/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}`
--- a/web/app/components/workflow/nodes/_base/node.tsx
+++ b/web/app/components/workflow/nodes/_base/node.tsx
@ -53,8 +53,13 @@ import { useAppContext } from '@/context/app-context'
 import { useStore } from '@/app/components/workflow/store'
 import { useCollaboration } from '@/app/components/workflow/collaboration/hooks/use-collaboration'

+type NodeChildProps = {
+  id: string
+  data: NodeProps['data']
+}
+
 type BaseNodeProps = {
-  children: ReactElement
+  children: ReactElement<Partial<NodeChildProps>>
  id: NodeProps['id']
  data: NodeProps['data']
 }
--- a/web/app/components/workflow/nodes/code/types.ts
+++ b/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
--- a/web/app/components/workflow/nodes/http/use-config.ts
+++ b/web/app/components/workflow/nodes/http/use-config.ts
@ -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)

--- a/web/app/components/workflow/nodes/if-else/components/condition-list/condition-item.tsx
+++ b/web/app/components/workflow/nodes/if-else/components/condition-list/condition-item.tsx
@ -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)
--- a/web/app/components/workflow/nodes/knowledge-base/components/retrieval-setting/top-k-and-score-threshold.tsx
+++ b/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>
        )
--- a/web/app/components/workflow/nodes/knowledge-retrieval/components/metadata/condition-list/condition-number.tsx
+++ b/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,
--- a/web/app/components/workflow/nodes/knowledge-retrieval/components/metadata/condition-list/condition-string.tsx
+++ b/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,
--- a/web/app/components/workflow/nodes/knowledge-retrieval/types.ts
+++ b/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 }[]
 }
--- a/web/app/components/workflow/nodes/llm/components/json-schema-config-modal/json-importer.tsx
+++ b/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()
--- a/web/app/components/workflow/nodes/llm/components/json-schema-config-modal/visual-editor/context.tsx
+++ b/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()
--- a/web/app/components/workflow/nodes/llm/use-config.ts
+++ b/web/app/components/workflow/nodes/llm/use-config.ts
@ -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)
--- a/web/app/components/workflow/nodes/llm/utils.ts
+++ b/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

--- a/web/app/components/workflow/nodes/loop/components/condition-list/condition-item.tsx
+++ b/web/app/components/workflow/nodes/loop/components/condition-list/condition-item.tsx
@ -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)
--- a/web/app/components/workflow/nodes/parameter-extractor/components/extract-parameter/import-from-tool.tsx
+++ b/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:
--- a/web/app/components/workflow/nodes/parameter-extractor/use-config.ts
+++ b/web/app/components/workflow/nodes/parameter-extractor/use-config.ts
@ -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)
--- a/web/app/components/workflow/nodes/question-classifier/use-config.ts
+++ b/web/app/components/workflow/nodes/question-classifier/use-config.ts
@ -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
--- a/web/app/components/workflow/nodes/template-transform/use-config.ts
+++ b/web/app/components/workflow/nodes/template-transform/use-config.ts
@ -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)
--- a/web/app/components/workflow/run/agent-log/agent-log-nav-more.tsx
+++ b/web/app/components/workflow/run/agent-log/agent-log-nav-more.tsx
@ -9,7 +9,7 @@ import Button from '@/app/components/base/button'
 import type { AgentLogItemWithChildren } from '@/types/workflow'

 type AgentLogNavMoreProps = {
-  options: { id: string; label: string }[]
+  options: AgentLogItemWithChildren[]
  onShowAgentOrToolLog: (detail?: AgentLogItemWithChildren) => void
 }
 const AgentLogNavMore = ({
@ -41,10 +41,10 @@ const AgentLogNavMore = ({
          {
            options.map(option => (
              <div
-                key={option.id}
+                key={option.message_id}
                className='system-md-regular flex h-8 cursor-pointer items-center rounded-lg px-2 text-text-secondary hover:bg-state-base-hover'
                onClick={() => {
-                  onShowAgentOrToolLog(option as AgentLogItemWithChildren)
+                  onShowAgentOrToolLog(option)
                  setOpen(false)
                }}
              >
--- a/web/app/components/workflow/store/workflow/form-slice.ts
+++ b/web/app/components/workflow/store/workflow/form-slice.ts
@ -4,8 +4,8 @@ import type {
 } from '@/app/components/workflow/types'

 export type FormSliceShape = {
-  inputs: Record<string, string>
-  setInputs: (inputs: Record<string, string>) => void
+  inputs: Record<string, string | number | boolean>
+  setInputs: (inputs: Record<string, string | number | boolean>) => void
  files: RunFile[]
  setFiles: (files: RunFile[]) => void
 }
--- a/web/app/components/workflow/workflow-preview/components/nodes/base.tsx
+++ b/web/app/components/workflow/workflow-preview/components/nodes/base.tsx
@ -23,8 +23,10 @@ import {
 } from '../node-handle'
 import ErrorHandleOnNode from '../error-handle-on-node'

+type NodeChildElement = ReactElement<Partial<NodeProps>>
+
 type NodeCardProps = NodeProps & {
-  children?: ReactElement
+  children?: NodeChildElement
 }

 const BaseCard = ({