fix: Refresh model dropdowns after upload

src/platform/assets/composables/useUploadModelWizard.ts

@@ -4,6 +4,7 @@ import { computed, ref, watch } from 'vue'
 import { st } from '@/i18n'
 import type { AssetMetadata } from '@/platform/assets/schemas/assetSchema'
 import { assetService } from '@/platform/assets/services/assetService'
+import { useAssetsStore } from '@/stores/assetsStore'
 
 interface WizardData {
   url: string
@@ -18,6 +19,7 @@ interface ModelTypeOption {
 }
 
 export function useUploadModelWizard(modelTypes: Ref<ModelTypeOption[]>) {
+  const assetsStore = useAssetsStore()
   const currentStep = ref(1)
   const isFetchingMetadata = ref(false)
   const isUploading = ref(false)
@@ -143,6 +145,12 @@ export function useUploadModelWizard(modelTypes: Ref<ModelTypeOption[]>) {
 
       uploadStatus.value = 'success'
       currentStep.value = 3
+
+      // Refresh model caches for the uploaded model type
+      if (selectedModelType.value) {
+        await assetsStore.refreshModelsByType(selectedModelType.value)
+      }
+
       return true
     } catch (error) {
       console.error('Failed to upload asset:', error)

src/stores/assetsStore.ts

@@ -322,27 +322,50 @@ export const useAssetsStore = defineStore('assets', () => {
         }
       }
 
+      /**
+       * Invalidate and refresh model assets for all node types that use a specific model category
+       * @param modelType The model category (e.g., 'checkpoints', 'loras')
+       */
+      async function refreshModelsByType(modelType: string): Promise<void> {
+        const { useModelToNodeStore } = await import('./modelToNodeStore')
+        const modelToNodeStore = useModelToNodeStore()
+
+        // Get all node types that use this model category
+        const providers = modelToNodeStore.getAllNodeProviders(modelType)
+        if (!providers.length) return
+
+        // Refresh each node type's cache
+        await Promise.all(
+          providers.map((provider) =>
+            updateModelsForNodeType(provider.nodeDef.name)
+          )
+        )
+      }
+
       return {
         modelAssetsByNodeType,
         modelLoadingByNodeType,
         modelErrorByNodeType,
-        updateModelsForNodeType
+        updateModelsForNodeType,
+        refreshModelsByType
       }
     }
 
     return {
       modelAssetsByNodeType: shallowReactive(new Map<string, AssetItem[]>()),
       modelLoadingByNodeType: shallowReactive(new Map<string, boolean>()),
       modelErrorByNodeType: shallowReactive(new Map<string, Error | null>()),
-      updateModelsForNodeType: async () => []
+      updateModelsForNodeType: async () => [],
+      refreshModelsByType: async () => {}
     }
   }
 
   const {
     modelAssetsByNodeType,
     modelLoadingByNodeType,
     modelErrorByNodeType,
-    updateModelsForNodeType
+    updateModelsForNodeType,
+    refreshModelsByType
   } = getModelState()
 
   return {
@@ -369,6 +392,7 @@ export const useAssetsStore = defineStore('assets', () => {
     modelAssetsByNodeType,
     modelLoadingByNodeType,
     modelErrorByNodeType,
-    updateModelsForNodeType
+    updateModelsForNodeType,
+    refreshModelsByType
   }
 })