outman/dify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: avoid unnecessary loading for the model selector in agent node

Browse Source
This commit is contained in:
Yi 2025-01-09 16:48:02 +08:00
parent dfe69a9c13
commit 3bdaf2dcae
1 changed files with 58 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

87
web/app/components/header/account-setting/model-provider-page/model-parameter-modal/agent-model-trigger.tsx
View File

@ -67,42 +67,71 @@ const AgentModelTrigger: FC<AgentModelTriggerProps> = ({
}
}, [modelProviders, providerName])
const [pluginInfo, setPluginInfo] = useState<PluginInfoFromMarketPlace | null>(null)
const [isPluginChecked, setIsPluginChecked] = useState(false)
const [isPluginChecked, setIsPluginChecked] = useState(!!modelProvider)
const [installed, setInstalled] = useState(false)
const [inModelList, setInModelList] = useState(false)
const invalidateInstalledPluginList = useInvalidateInstalledPluginList()
const handleOpenModal = useModelModalHandler()
const checkPluginInfo = useMemo(async () => {
if (!providerName || !modelId)
return null
const parts = providerName.split('/')
try {
const pluginInfo = await fetchPluginInfoFromMarketPlace({
org: parts[0],
name: parts[1],
})
if (pluginInfo.data.plugin.category === PluginType.model)
return pluginInfo.data.plugin
}
catch (error) {
// pass
}
return null
}, [providerName, modelId])
const checkModelList = useMemo(async () => {
if (!modelId || !currentProvider)
return false
try {
const modelsData = await fetchModelProviderModelList(
`/workspaces/current/model-providers/${currentProvider?.provider}/models`,
)
return !!modelsData.data.find(item => item.model === modelId)
}
catch (error) {
// pass
}
return false
}, [modelId, currentProvider])
useEffect(() => {
(async () => {
if (modelId && currentProvider) {
try {
const modelsData = await fetchModelProviderModelList(`/workspaces/current/model-providers/${currentProvider?.provider}/models`)
if (modelId && modelsData.data.find(item => item.model === modelId))
setInModelList(true)
}
catch (error) {
// pass
let isSubscribed = true
const initializeChecks = async () => {
if (!isPluginChecked) {
const [pluginResult, modelListResult] = await Promise.all([
checkPluginInfo,
checkModelList,
])
if (isSubscribed) {
if (pluginResult)
setPluginInfo(pluginResult)
setInModelList(modelListResult)
setIsPluginChecked(true)
}
}
if (providerName) {
const parts = providerName.split('/')
const org = parts[0]
const name = parts[1]
try {
const pluginInfo = await fetchPluginInfoFromMarketPlace({ org, name })
if (pluginInfo.data.plugin.category === PluginType.model)
setPluginInfo(pluginInfo.data.plugin)
}
catch (error) {
// pass
}
setIsPluginChecked(true)
}
else {
setIsPluginChecked(true)
}
})()
}, [providerName, modelId, currentProvider])
}
initializeChecks()
return () => {
isSubscribed = false
}
}, [checkPluginInfo, checkModelList, isPluginChecked, modelId, currentProvider])
if (modelId && !isPluginChecked)
return <Loading />