mirror of
https://github.com/langgenius/dify.git
synced 2026-05-06 02:18:08 +08:00
@ -11,7 +11,6 @@ import type {
|
|||||||
DefaultModel,
|
DefaultModel,
|
||||||
DefaultModelResponse,
|
DefaultModelResponse,
|
||||||
Model,
|
Model,
|
||||||
ModelProvider,
|
|
||||||
ModelTypeEnum,
|
ModelTypeEnum,
|
||||||
} from './declarations'
|
} from './declarations'
|
||||||
import {
|
import {
|
||||||
@ -64,20 +63,20 @@ export const useLanguage = () => {
|
|||||||
}
|
}
|
||||||
|
|
||||||
export const useProviderCredentialsAndLoadBalancing = (
|
export const useProviderCredentialsAndLoadBalancing = (
|
||||||
provider: ModelProvider,
|
provider: string,
|
||||||
configurationMethod: ConfigurationMethodEnum,
|
configurationMethod: ConfigurationMethodEnum,
|
||||||
configured?: boolean,
|
configured?: boolean,
|
||||||
currentCustomConfigurationModelFixedFields?: CustomConfigurationModelFixedFields,
|
currentCustomConfigurationModelFixedFields?: CustomConfigurationModelFixedFields,
|
||||||
) => {
|
) => {
|
||||||
const { data: predefinedFormSchemasValue, mutate: mutatePredefined } = useSWR(
|
const { data: predefinedFormSchemasValue, mutate: mutatePredefined } = useSWR(
|
||||||
(configurationMethod === ConfigurationMethodEnum.predefinedModel && configured)
|
(configurationMethod === ConfigurationMethodEnum.predefinedModel && configured)
|
||||||
? `/workspaces/current/model-providers/${provider.plugin_id}/${provider.provider}/credentials`
|
? `/workspaces/current/model-providers/${provider}/credentials`
|
||||||
: null,
|
: null,
|
||||||
fetchModelProviderCredentials,
|
fetchModelProviderCredentials,
|
||||||
)
|
)
|
||||||
const { data: customFormSchemasValue, mutate: mutateCustomized } = useSWR(
|
const { data: customFormSchemasValue, mutate: mutateCustomized } = useSWR(
|
||||||
(configurationMethod === ConfigurationMethodEnum.customizableModel && currentCustomConfigurationModelFixedFields)
|
(configurationMethod === ConfigurationMethodEnum.customizableModel && currentCustomConfigurationModelFixedFields)
|
||||||
? `/workspaces/current/model-providers/${provider.plugin_id}/${provider.provider}/models/credentials?model=${currentCustomConfigurationModelFixedFields?.__model_name}&model_type=${currentCustomConfigurationModelFixedFields?.__model_type}`
|
? `/workspaces/current/model-providers/${provider}/models/credentials?model=${currentCustomConfigurationModelFixedFields?.__model_name}&model_type=${currentCustomConfigurationModelFixedFields?.__model_type}`
|
||||||
: null,
|
: null,
|
||||||
fetchModelProviderCredentials,
|
fetchModelProviderCredentials,
|
||||||
)
|
)
|
||||||
|
|||||||
@ -72,7 +72,7 @@ const ModelModal: FC<ModelModalProps> = ({
|
|||||||
loadBalancing: originalConfig,
|
loadBalancing: originalConfig,
|
||||||
mutate,
|
mutate,
|
||||||
} = useProviderCredentialsAndLoadBalancing(
|
} = useProviderCredentialsAndLoadBalancing(
|
||||||
provider,
|
provider.provider,
|
||||||
configurateMethod,
|
configurateMethod,
|
||||||
providerFormSchemaPredefined && provider.custom_configuration.status === CustomConfigurationStatusEnum.active,
|
providerFormSchemaPredefined && provider.custom_configuration.status === CustomConfigurationStatusEnum.active,
|
||||||
currentCustomConfigurationModelFixedFields,
|
currentCustomConfigurationModelFixedFields,
|
||||||
|
|||||||
@ -30,7 +30,7 @@ const ModelLoadBalancingModal = ({ provider, model, open = false, onClose, onSav
|
|||||||
const [loading, setLoading] = useState(false)
|
const [loading, setLoading] = useState(false)
|
||||||
|
|
||||||
const { data, mutate } = useSWR(
|
const { data, mutate } = useSWR(
|
||||||
`/workspaces/current/model-providers/${provider.plugin_id}/${provider.provider}/models/credentials?model=${model.model}&model_type=${model.model_type}`,
|
`/workspaces/current/model-providers/${provider.provider}/models/credentials?model=${model.model}&model_type=${model.model_type}`,
|
||||||
fetchModelLoadBalancingConfig,
|
fetchModelLoadBalancingConfig,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user