dify/web/app/components/workflow/utils/tool.ts

import type { ToolNodeType } from '../nodes/tool/types'
import type { InputVar, ToolWithProvider } from '../types'
import type { StructuredOutput } from '@/app/components/workflow/nodes/llm/types'
import { CollectionType } from '@/app/components/tools/types'
import { toolParametersToFormSchemas } from '@/app/components/tools/utils/to-form-schema'
import { Type } from '@/app/components/workflow/nodes/llm/types'
import { isToolAuthorizationRequired } from '@/app/components/workflow/nodes/tool/auth'
import { canFindTool } from '@/utils'

export const getToolCheckParams = (
  toolData: ToolNodeType,
  buildInTools: ToolWithProvider[],
  customTools: ToolWithProvider[],
  workflowTools: ToolWithProvider[],
  language: string,
) => {
  const { provider_id, provider_type, tool_name } = toolData
  const currentTools =
    provider_type === CollectionType.builtIn
      ? buildInTools
      : provider_type === CollectionType.custom
        ? customTools
        : workflowTools
  const currCollection = currentTools.find((item) => canFindTool(item.id, provider_id))
  const currTool = currCollection?.tools.find((tool) => tool.name === tool_name)
  const formSchemas = currTool ? toolParametersToFormSchemas(currTool.parameters) : []
  const toolInputVarSchema = formSchemas.filter((item) => item.form === 'llm')
  const toolSettingSchema = formSchemas.filter((item) => item.form !== 'llm')

  return {
    toolInputsSchema: (() => {
      const formInputs: InputVar[] = []
      toolInputVarSchema.forEach((item: any) => {
        formInputs.push({
          label: item.label[language] || item.label.en_US,
          variable: item.variable,
          type: item.type,
          required: item.required,
        })
      })
      return formInputs
    })(),
    notAuthed: isToolAuthorizationRequired(provider_type, currCollection),
    toolSettingSchema,
    language,
  }
}
export const wrapStructuredVarItem = (
  outputItem: any,
  matchedSchemaType: string,
): StructuredOutput => {
  const dataType = Type.object
  return {
    schema: {
      type: dataType,
      properties: {
        [outputItem.name]: {
          ...outputItem.value,
          schemaType: matchedSchemaType,
        },
      },
      additionalProperties: false,
    },
  }
}