get credentials of provider

pull/12372/head
JzoNg 2 years ago
parent 766ac3e255
commit 72ef04d3e4

@ -11,6 +11,7 @@ import type {
DefaultModel, DefaultModel,
DefaultModelResponse, DefaultModelResponse,
Model, Model,
ModelProvider,
ModelTypeEnum, ModelTypeEnum,
} from './declarations' } from './declarations'
import { import {
@ -63,20 +64,20 @@ export const useLanguage = () => {
} }
export const useProviderCredentialsAndLoadBalancing = ( export const useProviderCredentialsAndLoadBalancing = (
provider: string, provider: ModelProvider,
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}/credentials` ? `/workspaces/current/model-providers/${provider.plugin_id}/${provider.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}/models/credentials?model=${currentCustomConfigurationModelFixedFields?.__model_name}&model_type=${currentCustomConfigurationModelFixedFields?.__model_type}` ? `/workspaces/current/model-providers/${provider.plugin_id}/${provider.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.provider}/models/credentials?model=${model.model}&model_type=${model.model_type}`, `/workspaces/current/model-providers/${provider.plugin_id}/${provider.provider}/models/credentials?model=${model.model}&model_type=${model.model_type}`,
fetchModelLoadBalancingConfig, fetchModelLoadBalancingConfig,
) )

Loading…
Cancel
Save