dify/web/features/agent-v2/agent-composer/knowledge-validation.ts

import type { AgentKnowledgeRetrievalItem } from './form-state'
import { useTranslation } from 'react-i18next'
import { MetadataFilteringModeEnum } from '@/app/components/workflow/nodes/knowledge-retrieval/types'
import { RETRIEVE_TYPE } from '@/types/app'

export type KnowledgeValidationIssueCode =
  | 'name_required'
  | 'name_duplicate'
  | 'datasets_required'
  | 'custom_query_required'
  | 'single_model_required'
  | 'metadata_model_required'
  | 'metadata_conditions_required'

type KnowledgeValidationField = 'name' | 'datasets' | 'query' | 'retrieval' | 'metadata'

type KnowledgeValidationIssue = {
  itemId: string
  code: KnowledgeValidationIssueCode
  field: KnowledgeValidationField
}

export type KnowledgeValidationResult = {
  byId: Record<string, Partial<Record<KnowledgeValidationField, KnowledgeValidationIssueCode>>>
  firstIssue?: KnowledgeValidationIssue
  isValid: boolean
}

const getKnowledgeRetrievalConcreteName = (item: AgentKnowledgeRetrievalItem) =>
  item.name?.trim() ?? ''

export const getKnowledgeRetrievalSetName = (item: AgentKnowledgeRetrievalItem) =>
  getKnowledgeRetrievalConcreteName(item) || item.id

export const useKnowledgeValidationMessage = () => {
  const { t } = useTranslation('agentV2')
  const { t: tCommon } = useTranslation('common')
  const { t: tAppDebug } = useTranslation('appDebug')
  const { t: tWorkflow } = useTranslation('workflow')

  return (issueCode?: KnowledgeValidationIssueCode) => {
    switch (issueCode) {
      case 'name_required':
        return tCommon(($) => $['errorMsg.fieldRequired'], {
          field: t(($) => $['agentDetail.configure.knowledgeRetrieval.dialog.nameLabel']),
        })
      case 'name_duplicate':
        return tAppDebug(($) => $['varKeyError.keyAlreadyExists'], {
          key: t(($) => $['agentDetail.configure.knowledgeRetrieval.dialog.nameLabel']),
        })
      case 'datasets_required':
        return tCommon(($) => $['errorMsg.fieldRequired'], {
          field: t(($) => $['agentDetail.configure.knowledgeRetrieval.dialog.knowledge.label']),
        })
      case 'custom_query_required':
        return tCommon(($) => $['errorMsg.fieldRequired'], {
          field: t(
            ($) => $['agentDetail.configure.knowledgeRetrieval.dialog.query.customInputLabel'],
          ),
        })
      case 'single_model_required':
        return tCommon(($) => $['errorMsg.fieldRequired'], {
          field: tCommon(($) => $['modelProvider.systemReasoningModel.key']),
        })
      case 'metadata_model_required':
        return t(
          ($) => $['agentDetail.configure.knowledgeRetrieval.validation.metadataModelRequired'],
        )
      case 'metadata_conditions_required':
        return tCommon(($) => $['errorMsg.fieldRequired'], {
          field: tWorkflow(($) => $['nodes.knowledgeRetrieval.metadata.panel.conditions']),
        })
      default:
        return undefined
    }
  }
}

const getKnowledgeDatasetCount = (item: AgentKnowledgeRetrievalItem) =>
  item.selectedDatasets?.length ?? item.datasetRefs?.length ?? 0

const getNormalizedKnowledgeName = (item: AgentKnowledgeRetrievalItem) =>
  getKnowledgeRetrievalConcreteName(item).toLowerCase()

export const validateKnowledgeRetrievals = (
  retrievals: AgentKnowledgeRetrievalItem[],
): KnowledgeValidationResult => {
  const byId: KnowledgeValidationResult['byId'] = {}
  const issues: KnowledgeValidationIssue[] = []
  const nameCounts = new Map<string, number>()

  retrievals.forEach((item) => {
    const normalizedName = getNormalizedKnowledgeName(item)
    if (!normalizedName) return

    nameCounts.set(normalizedName, (nameCounts.get(normalizedName) ?? 0) + 1)
  })

  const pushIssue = (issue: KnowledgeValidationIssue) => {
    byId[issue.itemId] ??= {}
    const itemIssues = byId[issue.itemId]
    if (itemIssues) itemIssues[issue.field] ??= issue.code
    issues.push(issue)
  }

  retrievals.forEach((item) => {
    const setName = getKnowledgeRetrievalConcreteName(item)
    const normalizedName = setName.toLowerCase()

    if (!setName) {
      pushIssue({
        itemId: item.id,
        code: 'name_required',
        field: 'name',
      })
    } else if ((nameCounts.get(normalizedName) ?? 0) > 1) {
      pushIssue({
        itemId: item.id,
        code: 'name_duplicate',
        field: 'name',
      })
    }

    if (!getKnowledgeDatasetCount(item)) {
      pushIssue({
        itemId: item.id,
        code: 'datasets_required',
        field: 'datasets',
      })
    }

    if (item.queryMode === 'custom' && !item.customQuery?.trim()) {
      pushIssue({
        itemId: item.id,
        code: 'custom_query_required',
        field: 'query',
      })
    }

    if (
      item.retrievalMode === RETRIEVE_TYPE.oneWay &&
      (!item.singleRetrievalConfig?.model?.provider || !item.singleRetrievalConfig.model.name)
    ) {
      pushIssue({
        itemId: item.id,
        code: 'single_model_required',
        field: 'retrieval',
      })
    }

    if (
      item.metadataFilterMode === MetadataFilteringModeEnum.automatic &&
      (!item.metadataModelConfig?.provider || !item.metadataModelConfig.name)
    ) {
      pushIssue({
        itemId: item.id,
        code: 'metadata_model_required',
        field: 'metadata',
      })
    }

    if (
      item.metadataFilterMode === MetadataFilteringModeEnum.manual &&
      !item.metadataFilteringConditions?.conditions.length
    ) {
      pushIssue({
        itemId: item.id,
        code: 'metadata_conditions_required',
        field: 'metadata',
      })
    }
  })

  return {
    byId,
    firstIssue: issues[0],
    isValid: issues.length === 0,
  }
}