-
Notifications
You must be signed in to change notification settings - Fork 4.5k
/
MessageTextAnnotation.Serialization.cs
133 lines (116 loc) · 5.77 KB
/
MessageTextAnnotation.Serialization.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
// <auto-generated/>
#nullable disable
using System;
using System.ClientModel.Primitives;
using System.Text.Json;
using Azure.Core;
namespace Azure.AI.OpenAI.Assistants
{
[PersistableModelProxy(typeof(UnknownMessageTextAnnotation))]
public partial class MessageTextAnnotation : IUtf8JsonSerializable, IJsonModel<MessageTextAnnotation>
{
void IUtf8JsonSerializable.Write(Utf8JsonWriter writer) => ((IJsonModel<MessageTextAnnotation>)this).Write(writer, ModelSerializationExtensions.WireOptions);
void IJsonModel<MessageTextAnnotation>.Write(Utf8JsonWriter writer, ModelReaderWriterOptions options)
{
var format = options.Format == "W" ? ((IPersistableModel<MessageTextAnnotation>)this).GetFormatFromOptions(options) : options.Format;
if (format != "J")
{
throw new FormatException($"The model {nameof(MessageTextAnnotation)} does not support writing '{format}' format.");
}
writer.WriteStartObject();
writer.WritePropertyName("type"u8);
writer.WriteStringValue(Type);
writer.WritePropertyName("text"u8);
writer.WriteStringValue(Text);
writer.WritePropertyName("start_index"u8);
writer.WriteNumberValue(StartIndex);
writer.WritePropertyName("end_index"u8);
writer.WriteNumberValue(EndIndex);
if (options.Format != "W" && _serializedAdditionalRawData != null)
{
foreach (var item in _serializedAdditionalRawData)
{
writer.WritePropertyName(item.Key);
#if NET6_0_OR_GREATER
writer.WriteRawValue(item.Value);
#else
using (JsonDocument document = JsonDocument.Parse(item.Value))
{
JsonSerializer.Serialize(writer, document.RootElement);
}
#endif
}
}
writer.WriteEndObject();
}
MessageTextAnnotation IJsonModel<MessageTextAnnotation>.Create(ref Utf8JsonReader reader, ModelReaderWriterOptions options)
{
var format = options.Format == "W" ? ((IPersistableModel<MessageTextAnnotation>)this).GetFormatFromOptions(options) : options.Format;
if (format != "J")
{
throw new FormatException($"The model {nameof(MessageTextAnnotation)} does not support reading '{format}' format.");
}
using JsonDocument document = JsonDocument.ParseValue(ref reader);
return DeserializeMessageTextAnnotation(document.RootElement, options);
}
internal static MessageTextAnnotation DeserializeMessageTextAnnotation(JsonElement element, ModelReaderWriterOptions options = null)
{
options ??= ModelSerializationExtensions.WireOptions;
if (element.ValueKind == JsonValueKind.Null)
{
return null;
}
if (element.TryGetProperty("type", out JsonElement discriminator))
{
switch (discriminator.GetString())
{
case "file_citation": return MessageTextFileCitationAnnotation.DeserializeMessageTextFileCitationAnnotation(element, options);
case "file_path": return MessageTextFilePathAnnotation.DeserializeMessageTextFilePathAnnotation(element, options);
}
}
return UnknownMessageTextAnnotation.DeserializeUnknownMessageTextAnnotation(element, options);
}
BinaryData IPersistableModel<MessageTextAnnotation>.Write(ModelReaderWriterOptions options)
{
var format = options.Format == "W" ? ((IPersistableModel<MessageTextAnnotation>)this).GetFormatFromOptions(options) : options.Format;
switch (format)
{
case "J":
return ModelReaderWriter.Write(this, options);
default:
throw new FormatException($"The model {nameof(MessageTextAnnotation)} does not support writing '{options.Format}' format.");
}
}
MessageTextAnnotation IPersistableModel<MessageTextAnnotation>.Create(BinaryData data, ModelReaderWriterOptions options)
{
var format = options.Format == "W" ? ((IPersistableModel<MessageTextAnnotation>)this).GetFormatFromOptions(options) : options.Format;
switch (format)
{
case "J":
{
using JsonDocument document = JsonDocument.Parse(data);
return DeserializeMessageTextAnnotation(document.RootElement, options);
}
default:
throw new FormatException($"The model {nameof(MessageTextAnnotation)} does not support reading '{options.Format}' format.");
}
}
string IPersistableModel<MessageTextAnnotation>.GetFormatFromOptions(ModelReaderWriterOptions options) => "J";
/// <summary> Deserializes the model from a raw response. </summary>
/// <param name="response"> The response to deserialize the model from. </param>
internal static MessageTextAnnotation FromResponse(Response response)
{
using var document = JsonDocument.Parse(response.Content);
return DeserializeMessageTextAnnotation(document.RootElement);
}
/// <summary> Convert into a <see cref="RequestContent"/>. </summary>
internal virtual RequestContent ToRequestContent()
{
var content = new Utf8JsonRequestContent();
content.JsonWriter.WriteObjectValue(this, ModelSerializationExtensions.WireOptions);
return content;
}
}
}