dify/web/app/components/header/account-setting/model-provider-page/model-auth/hooks/use-credential-data.ts

import type {
  Credential,
  CustomModelCredential,
  ModelProvider,
} from '@/app/components/header/account-setting/model-provider-page/declarations'
import { useMemo } from 'react'
import { useGetCredential } from './use-auth-service'

export const useCredentialData = (
  provider: ModelProvider,
  providerFormSchemaPredefined: boolean,
  isModelCredential?: boolean,
  credential?: Credential,
  model?: CustomModelCredential,
) => {
  const configFrom = useMemo(() => {
    if (providerFormSchemaPredefined) return 'predefined-model'
    return 'custom-model'
  }, [providerFormSchemaPredefined])
  const { isLoading, data: credentialData = {} } = useGetCredential(
    provider.provider,
    isModelCredential,
    credential?.credential_id,
    model,
    configFrom,
  )

  return {
    isLoading,
    credentialData,
  }
}