Skip to content

Commit 8ee1562

Browse files
feat(assets): refresh asset browser after downloads complete
- Add watcher in AssetBrowserModal to refresh assets when downloads finish - Fix upload wizard to only track async uploads that aren't already completed 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <[email protected]>
1 parent 422f9a0 commit 8ee1562

File tree

2 files changed

+13
-1
lines changed

2 files changed

+13
-1
lines changed

src/platform/assets/components/AssetBrowserModal.vue

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,7 @@ import { useModelUpload } from '@/platform/assets/composables/useModelUpload'
8383
import type { AssetItem } from '@/platform/assets/schemas/assetSchema'
8484
import { assetService } from '@/platform/assets/services/assetService'
8585
import { formatCategoryLabel } from '@/platform/assets/utils/categoryLabel'
86+
import { useAssetDownloadStore } from '@/stores/assetDownloadStore'
8687
import { useModelToNodeStore } from '@/stores/modelToNodeStore'
8788
import { OnCloseKey } from '@/types/widgetTypes'
8889
@@ -132,6 +133,17 @@ watch(
132133
{ immediate: true }
133134
)
134135
136+
const assetDownloadStore = useAssetDownloadStore()
137+
138+
watch(
139+
() => assetDownloadStore.hasActiveDownloads,
140+
async (hasDownloads, hadDownloads) => {
141+
if (hadDownloads && !hasDownloads) {
142+
await execute()
143+
}
144+
}
145+
)
146+
135147
const {
136148
searchQuery,
137149
selectedCategory,

src/platform/assets/composables/useUploadModelWizard.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,7 @@ export function useUploadModelWizard(modelTypes: Ref<ModelTypeOption[]>) {
233233
preview_id: previewId
234234
})
235235

236-
if (result.type === 'async') {
236+
if (result.type === 'async' && result.task.status != 'completed') {
237237
if (selectedModelType.value) {
238238
assetDownloadStore.trackDownload(
239239
result.task.task_id,

0 commit comments

Comments
 (0)