Merge pull request #1110 from ionite34/backport/main/pr-1109

[dev to main] backport: update base model parsing from error response cuz civit changed response model (1109)

Author: mohnjiles

CHANGELOG.md

@@ -23,6 +23,7 @@ and this project adheres to [Semantic Versioning 2.0](https://semver.org/spec/v2
 - Fixed [#1305](https://github.com/LykosAI/StabilityMatrix/issues/1305) - FluxGym installing incorrect packages for Blackwell GPUs
 - Fixed [#1316](https://github.com/LykosAI/StabilityMatrix/issues/1316) - Errors when installing Triton & SageAttention
 - Fixed "directory is not empty" error when updating packages with symlinks
+- Fixed missing base model types in the Checkpoint Manager & Civitai Model Browser
 
 ## v2.14.2
 ### Changed

StabilityMatrix.Avalonia/Services/CivitBaseModelTypeService.cs

@@ -3,6 +3,7 @@
 using Microsoft.Extensions.Logging;
 using StabilityMatrix.Core.Api;
 using StabilityMatrix.Core.Database;
+using StabilityMatrix.Core.Extensions;
 using StabilityMatrix.Core.Models.Api;
 using StabilityMatrix.Core.Models.Database;
 
@@ -42,18 +43,18 @@ public async Task<List<string>> GetBaseModelTypes(bool forceRefresh = false, boo
                 var jsonContent = await baseModelsResponse.Content.ReadAsStringAsync();
                 var baseModels = JsonNode.Parse(jsonContent);
 
-                var jArray =
-                    baseModels?["error"]?["issues"]?[0]?["unionErrors"]?[0]?["issues"]?[0]?["options"]
-                    as JsonArray;
+                var innerJson = baseModels?["error"]?["message"]?.GetValue<string>();
+                var jArray = JsonNode.Parse(innerJson).AsArray();
+                var baseModelValues = jArray[0]?["errors"]?[0]?[0]?["values"]?.AsArray();
 
-                civitBaseModels = jArray?.GetValues<string>().ToList() ?? [];
+                civitBaseModels = baseModelValues?.GetValues<string>().ToList() ?? [];
 
                 // Cache the results
                 var cacheEntry = new CivitBaseModelTypeCacheEntry
                 {
                     Id = CacheId,
                     ModelTypes = civitBaseModels,
-                    CreatedAt = DateTimeOffset.UtcNow
+                    CreatedAt = DateTimeOffset.UtcNow,
                 };
 
                 await dbContext.UpsertCivitBaseModelTypeCacheEntry(cacheEntry);
@@ -78,7 +79,8 @@ public async Task<List<string>> GetBaseModelTypes(bool forceRefresh = false, boo
 
             // Return cached results if available, even if expired
             var expiredCache = await dbContext.GetCivitBaseModelTypeCacheEntry(CacheId);
-            return expiredCache?.ModelTypes ?? [];
+            return expiredCache?.ModelTypes
+                ?? Enum.GetValues<CivitBaseModelType>().Select(b => b.GetStringValue()).ToList();
         }
     }