fix: when serializing source use media type to deserialize to image or document source

This commit is contained in:
Stevan Freeborn
2025-06-14 14:01:18 -05:00
parent 5b8d88a01d
commit 74ac1ffc06
2 changed files with 11 additions and 2 deletions
+10 -1
View File
@@ -16,11 +16,20 @@ class SourceConverter : JsonConverter<Source>
{
SourceType.Text => JsonSerializer.Deserialize<TextSource>(root.GetRawText(), options)!,
SourceType.Content => JsonSerializer.Deserialize<CustomSource>(root.GetRawText(), options)!,
SourceType.Base64 => JsonSerializer.Deserialize<Base64Source>(root.GetRawText(), options)!,
SourceType.Base64 => DeserializeBase64Source(root, options),
_ => throw new JsonException($"Unknown content type: {type}")
};
}
private static Source DeserializeBase64Source(JsonElement root, JsonSerializerOptions options)
{
var mediaType = root.GetProperty("media_type").GetString() ?? throw new JsonException("Missing 'media_type' property");
var isImage = ImageType.IsValidImageType(mediaType);
return isImage
? JsonSerializer.Deserialize<ImageSource>(root.GetRawText(), options)!
: JsonSerializer.Deserialize<DocumentSource>(root.GetRawText(), options)!;
}
public override void Write(Utf8JsonWriter writer, Source value, JsonSerializerOptions options)
{
if (value is TextSource textSource)