add missing zod schema

Author: luke-mino-altherr

src/locales/en/main.json

@@ -1985,6 +1985,7 @@
     "searchAssetsPlaceholder": "Type to search...",
     "uploadModel": "Upload model",
     "uploadModelFromCivitai": "Upload a model from Civitai",
+    "uploadModelFailedToRetrieveMetadata": "Failed to retrieve metadata. Please check the link and try again.",
     "uploadModelDescription1": "Paste a Civitai model download link to add it to your library.",
     "uploadModelDescription2": "Only links from https://civitai.com are supported at the moment",
     "uploadModelDescription3": "Max file size: 1 GB",
@@ -2004,7 +2005,8 @@
     "uploadSuccess": "Model uploaded successfully!",
     "uploadFailed": "Upload failed",
     "modelAssociatedWithLink": "The model associated with the link you provided:",
-    "whatTypeOfModel": "What type of model is this?",
+    "modelTypeSelectorLabel": "What type of model is this?",
+    "modelTypeSelectorPlaceholder": "Select model type",
     "selectModelType": "Select model type",
     "notSureLeaveAsIs": "Not sure? Just leave this as is",
     "modelUploaded": "Model uploaded!",

src/platform/assets/components/UploadModelConfirmation.vue

@@ -13,11 +13,11 @@
     <!-- Model Type Selection -->
     <div class="flex flex-col gap-2">
       <label class="text-sm text-muted">
-        {{ $t('assetBrowser.whatTypeOfModel') }}
+        {{ $t('assetBrowser.modelTypeSelectorLabel') }}
       </label>
       <SingleSelect
         v-model="selectedModelType"
-        :label="$t('assetBrowser.whatTypeOfModel')"
+        :label="$t('assetBrowser.modelTypeSelectorPlaceholder')"
         :options="modelTypes"
       />
       <div class="flex items-center gap-2 text-sm text-muted">
@@ -33,30 +33,21 @@ import { computed } from 'vue'
 
 import SingleSelect from '@/components/input/SingleSelect.vue'
 import { useModelTypes } from '@/platform/assets/composables/useModelTypes'
-
-interface ModelMetadata {
-  content_length: number
-  final_url: string
-  content_type?: string
-  filename?: string
-  name?: string
-  tags?: string[]
-  preview_url?: string
-}
+import type { AssetMetadata } from '@/platform/assets/schemas/assetSchema'
 
 const props = defineProps<{
-  modelValue: string
-  metadata: ModelMetadata | null
+  modelValue: string | undefined
+  metadata: AssetMetadata | null
 }>()
 
 const emit = defineEmits<{
-  'update:modelValue': [value: string]
+  'update:modelValue': [value: string | undefined]
 }>()
 
 const { modelTypes } = useModelTypes()
 
 const selectedModelType = computed({
-  get: () => props.modelValue,
-  set: (value: string) => emit('update:modelValue', value)
+  get: () => props.modelValue ?? null,
+  set: (value: string | null) => emit('update:modelValue', value ?? undefined)
 })
 </script>

src/platform/assets/components/UploadModelDialog.vue

@@ -81,10 +81,12 @@ import { computed, onMounted, ref, watch } from 'vue'
 
 import IconTextButton from '@/components/button/IconTextButton.vue'
 import TextButton from '@/components/button/TextButton.vue'
+import { st } from '@/i18n'
 import UploadModelConfirmation from '@/platform/assets/components/UploadModelConfirmation.vue'
 import UploadModelProgress from '@/platform/assets/components/UploadModelProgress.vue'
 import UploadModelUrlInput from '@/platform/assets/components/UploadModelUrlInput.vue'
 import { useModelTypes } from '@/platform/assets/composables/useModelTypes'
+import type { AssetMetadata } from '@/platform/assets/schemas/assetSchema'
 import { assetService } from '@/platform/assets/services/assetService'
 import { useDialogStore } from '@/stores/dialogStore'
 
@@ -102,15 +104,7 @@ const uploadError = ref('')
 
 const wizardData = ref<{
   url: string
-  metadata: {
-    content_length: number
-    final_url: string
-    content_type?: string
-    filename?: string
-    name?: string
-    tags?: string[]
-    preview_url?: string
-  } | null
+  metadata: AssetMetadata | null
   name: string
   tags: string[]
 }>({
@@ -120,7 +114,7 @@ const wizardData = ref<{
   tags: []
 })
 
-const selectedModelType = ref<string>('loras')
+const selectedModelType = ref<string | undefined>(undefined)
 
 const { modelTypes, fetchModelTypes } = useModelTypes()
 
@@ -186,8 +180,13 @@ async function handleFetchMetadata() {
   } catch (error) {
     console.error('Failed to retrieve metadata:', error)
     uploadError.value =
-      error instanceof Error ? error.message : 'Failed to retrieve metadata'
-    // TODO: Show error toast to user
+      error instanceof Error
+        ? error.message
+        : st(
+            'assetBrowser.uploadModelFailedToRetrieveMetadata',
+            'Failed to retrieve metadata. Please check the link and try again.'
+          )
+    currentStep.value = 1
   } finally {
     isFetchingMetadata.value = false
   }
@@ -200,7 +199,9 @@ async function handleUploadModel() {
   uploadStatus.value = 'uploading'
 
   try {
-    const tags = ['models', selectedModelType.value]
+    const tags = selectedModelType.value
+      ? ['models', selectedModelType.value]
+      : ['models']
     const filename =
       wizardData.value.metadata?.filename ||
       wizardData.value.metadata?.name ||

src/platform/assets/components/UploadModelProgress.vue

@@ -59,20 +59,12 @@
 </template>
 
 <script setup lang="ts">
-interface ModelMetadata {
-  content_length: number
-  final_url: string
-  content_type?: string
-  filename?: string
-  name?: string
-  tags?: string[]
-  preview_url?: string
-}
+import type { AssetMetadata } from '@/platform/assets/schemas/assetSchema'
 
 defineProps<{
   status: 'idle' | 'uploading' | 'success' | 'error'
   error?: string
-  metadata: ModelMetadata | null
-  modelType: string
+  metadata: AssetMetadata | null
+  modelType: string | undefined
 }>()
 </script>

src/platform/assets/schemas/assetSchema.ts

@@ -33,6 +33,17 @@ const zModelFile = z.object({
   pathIndex: z.number()
 })
 
+// Asset metadata from download URL
+const zAssetMetadata = z.object({
+  content_length: z.number(),
+  final_url: z.string(),
+  content_type: z.string().optional(),
+  filename: z.string().optional(),
+  name: z.string().optional(),
+  tags: z.array(z.string()).optional(),
+  preview_url: z.string().optional()
+})
+
 // Filename validation schema
 export const assetFilenameSchema = z
   .string()
@@ -44,10 +55,12 @@ export const assetFilenameSchema = z
 // Export schemas following repository patterns
 export const assetItemSchema = zAsset
 export const assetResponseSchema = zAssetResponse
+export const assetMetadataSchema = zAssetMetadata
 
 // Export types derived from Zod schemas
 export type AssetItem = z.infer<typeof zAsset>
 export type AssetResponse = z.infer<typeof zAssetResponse>
+export type AssetMetadata = z.infer<typeof zAssetMetadata>
 export type ModelFolder = z.infer<typeof zModelFolder>
 export type ModelFile = z.infer<typeof zModelFile>
 

src/platform/assets/services/assetService.ts

@@ -3,6 +3,7 @@ import { fromZodError } from 'zod-validation-error'
 import { assetResponseSchema } from '@/platform/assets/schemas/assetSchema'
 import type {
   AssetItem,
+  AssetMetadata,
   AssetResponse,
   ModelFile,
   ModelFolder
@@ -256,14 +257,7 @@ function createAssetService() {
    * @returns Promise with metadata including content_length, final_url, filename, etc.
    * @throws Error if metadata retrieval fails
    */
-  async function getAssetMetadata(url: string): Promise<{
-    content_length: number
-    final_url: string
-    content_type?: string
-    filename?: string
-    name?: string
-    tags?: string[]
-  }> {
+  async function getAssetMetadata(url: string): Promise<AssetMetadata> {
     const encodedUrl = encodeURIComponent(url)
     const res = await api.fetchApi(
       `${ASSETS_ENDPOINT}/metadata?url=${encodedUrl}`