openai/openai-dotnet
Publicmirrored from https://github.com/openai/openai-dotnetAvailable
src/Generated/Models/Embeddings/OpenAIEmbedding.Serialization.cs
161lines · modeblame
9f9f2936Jose Arriaga Maldonado2 years ago | 1 | // <auto-generated/> |
| 2 | | |
| 3 | #nullable disable | |
| 4 | | |
| 5 | using System; | |
| 6 | using System.ClientModel.Primitives; | |
| 7 | using System.Collections.Generic; | |
5dce104aJose Arriaga Maldonado1 years ago | 8 | using System.Diagnostics.CodeAnalysis; |
9f9f2936Jose Arriaga Maldonado2 years ago | 9 | using System.Text.Json; |
e0fee603Jose Arriaga Maldonado1 years ago | 10 | using OpenAI; |
9f9f2936Jose Arriaga Maldonado2 years ago | 11 | |
| 12 | namespace OpenAI.Embeddings | |
| 13 | { | |
19ceae44ShivangiReja1 years ago | 14 | public partial class OpenAIEmbedding : IJsonModel<OpenAIEmbedding> |
9f9f2936Jose Arriaga Maldonado2 years ago | 15 | { |
e0fee603Jose Arriaga Maldonado1 years ago | 16 | internal OpenAIEmbedding() |
| 17 | { | |
| 18 | } | |
| 19 | | |
19ceae44ShivangiReja1 years ago | 20 | void IJsonModel<OpenAIEmbedding>.Write(Utf8JsonWriter writer, ModelReaderWriterOptions options) |
9f9f2936Jose Arriaga Maldonado2 years ago | 21 | { |
e0fee603Jose Arriaga Maldonado1 years ago | 22 | writer.WriteStartObject(); |
| 23 | JsonModelWriteCore(writer, options); | |
| 24 | writer.WriteEndObject(); | |
| 25 | } | |
| 26 | | |
5dce104aJose Arriaga Maldonado1 years ago | 27 | [Experimental("OPENAI001")] |
e0fee603Jose Arriaga Maldonado1 years ago | 28 | protected virtual void JsonModelWriteCore(Utf8JsonWriter writer, ModelReaderWriterOptions options) |
| 29 | { | |
| 30 | string format = options.Format == "W" ? ((IPersistableModel<OpenAIEmbedding>)this).GetFormatFromOptions(options) : options.Format; | |
9f9f2936Jose Arriaga Maldonado2 years ago | 31 | if (format != "J") |
| 32 | { | |
19ceae44ShivangiReja1 years ago | 33 | throw new FormatException($"The model {nameof(OpenAIEmbedding)} does not support writing '{format}' format."); |
9f9f2936Jose Arriaga Maldonado2 years ago | 34 | } |
e0fee603Jose Arriaga Maldonado1 years ago | 35 | if (_additionalBinaryDataProperties?.ContainsKey("index") != true) |
19fc7d54Jose Arriaga Maldonado2 years ago | 36 | { |
| 37 | writer.WritePropertyName("index"u8); | |
| 38 | writer.WriteNumberValue(Index); | |
| 39 | } | |
e0fee603Jose Arriaga Maldonado1 years ago | 40 | if (_additionalBinaryDataProperties?.ContainsKey("embedding") != true) |
19fc7d54Jose Arriaga Maldonado2 years ago | 41 | { |
| 42 | writer.WritePropertyName("embedding"u8); | |
9f9f2936Jose Arriaga Maldonado2 years ago | 43 | #if NET6_0_OR_GREATER |
e0fee603Jose Arriaga Maldonado1 years ago | 44 | writer.WriteRawValue(EmbeddingProperty); |
9f9f2936Jose Arriaga Maldonado2 years ago | 45 | #else |
19fc7d54Jose Arriaga Maldonado2 years ago | 46 | using (JsonDocument document = JsonDocument.Parse(EmbeddingProperty)) |
| 47 | { | |
| 48 | JsonSerializer.Serialize(writer, document.RootElement); | |
| 49 | } | |
| 50 | #endif | |
| 51 | } | |
e0fee603Jose Arriaga Maldonado1 years ago | 52 | if (_additionalBinaryDataProperties?.ContainsKey("object") != true) |
9f9f2936Jose Arriaga Maldonado2 years ago | 53 | { |
19fc7d54Jose Arriaga Maldonado2 years ago | 54 | writer.WritePropertyName("object"u8); |
5dce104aJose Arriaga Maldonado1 years ago | 55 | writer.WriteStringValue(Object); |
9f9f2936Jose Arriaga Maldonado2 years ago | 56 | } |
5dce104aJose Arriaga Maldonado1 years ago | 57 | // Plugin customization: remove options.Format != "W" check |
0ca4c062Jose Arriaga Maldonado1 years ago | 58 | if (_additionalBinaryDataProperties != null) |
9f9f2936Jose Arriaga Maldonado2 years ago | 59 | { |
e0fee603Jose Arriaga Maldonado1 years ago | 60 | foreach (var item in _additionalBinaryDataProperties) |
9f9f2936Jose Arriaga Maldonado2 years ago | 61 | { |
19fc7d54Jose Arriaga Maldonado2 years ago | 62 | if (ModelSerializationExtensions.IsSentinelValue(item.Value)) |
| 63 | { | |
| 64 | continue; | |
| 65 | } | |
9f9f2936Jose Arriaga Maldonado2 years ago | 66 | writer.WritePropertyName(item.Key); |
| 67 | #if NET6_0_OR_GREATER | |
e0fee603Jose Arriaga Maldonado1 years ago | 68 | writer.WriteRawValue(item.Value); |
9f9f2936Jose Arriaga Maldonado2 years ago | 69 | #else |
| 70 | using (JsonDocument document = JsonDocument.Parse(item.Value)) | |
| 71 | { | |
| 72 | JsonSerializer.Serialize(writer, document.RootElement); | |
| 73 | } | |
| 74 | #endif | |
| 75 | } | |
| 76 | } | |
| 77 | } | |
| 78 | | |
e0fee603Jose Arriaga Maldonado1 years ago | 79 | OpenAIEmbedding IJsonModel<OpenAIEmbedding>.Create(ref Utf8JsonReader reader, ModelReaderWriterOptions options) => JsonModelCreateCore(ref reader, options); |
| 80 | | |
5dce104aJose Arriaga Maldonado1 years ago | 81 | [Experimental("OPENAI001")] |
e0fee603Jose Arriaga Maldonado1 years ago | 82 | protected virtual OpenAIEmbedding JsonModelCreateCore(ref Utf8JsonReader reader, ModelReaderWriterOptions options) |
9f9f2936Jose Arriaga Maldonado2 years ago | 83 | { |
e0fee603Jose Arriaga Maldonado1 years ago | 84 | string format = options.Format == "W" ? ((IPersistableModel<OpenAIEmbedding>)this).GetFormatFromOptions(options) : options.Format; |
9f9f2936Jose Arriaga Maldonado2 years ago | 85 | if (format != "J") |
| 86 | { | |
19ceae44ShivangiReja1 years ago | 87 | throw new FormatException($"The model {nameof(OpenAIEmbedding)} does not support reading '{format}' format."); |
9f9f2936Jose Arriaga Maldonado2 years ago | 88 | } |
| 89 | using JsonDocument document = JsonDocument.ParseValue(ref reader); | |
19ceae44ShivangiReja1 years ago | 90 | return DeserializeOpenAIEmbedding(document.RootElement, options); |
9f9f2936Jose Arriaga Maldonado2 years ago | 91 | } |
| 92 | | |
e0fee603Jose Arriaga Maldonado1 years ago | 93 | internal static OpenAIEmbedding DeserializeOpenAIEmbedding(JsonElement element, ModelReaderWriterOptions options) |
9f9f2936Jose Arriaga Maldonado2 years ago | 94 | { |
| 95 | if (element.ValueKind == JsonValueKind.Null) | |
| 96 | { | |
| 97 | return null; | |
| 98 | } | |
| 99 | int index = default; | |
e0fee603Jose Arriaga Maldonado1 years ago | 100 | BinaryData embeddingProperty = default; |
5dce104aJose Arriaga Maldonado1 years ago | 101 | string @object = default; |
e0fee603Jose Arriaga Maldonado1 years ago | 102 | IDictionary<string, BinaryData> additionalBinaryDataProperties = new ChangeTrackingDictionary<string, BinaryData>(); |
| 103 | foreach (var prop in element.EnumerateObject()) | |
9f9f2936Jose Arriaga Maldonado2 years ago | 104 | { |
e0fee603Jose Arriaga Maldonado1 years ago | 105 | if (prop.NameEquals("index"u8)) |
9f9f2936Jose Arriaga Maldonado2 years ago | 106 | { |
e0fee603Jose Arriaga Maldonado1 years ago | 107 | index = prop.Value.GetInt32(); |
9f9f2936Jose Arriaga Maldonado2 years ago | 108 | continue; |
| 109 | } | |
e0fee603Jose Arriaga Maldonado1 years ago | 110 | if (prop.NameEquals("embedding"u8)) |
9f9f2936Jose Arriaga Maldonado2 years ago | 111 | { |
e0fee603Jose Arriaga Maldonado1 years ago | 112 | embeddingProperty = BinaryData.FromString(prop.Value.GetRawText()); |
9f9f2936Jose Arriaga Maldonado2 years ago | 113 | continue; |
| 114 | } | |
e0fee603Jose Arriaga Maldonado1 years ago | 115 | if (prop.NameEquals("object"u8)) |
9f9f2936Jose Arriaga Maldonado2 years ago | 116 | { |
5dce104aJose Arriaga Maldonado1 years ago | 117 | @object = prop.Value.GetString(); |
9f9f2936Jose Arriaga Maldonado2 years ago | 118 | continue; |
| 119 | } | |
5dce104aJose Arriaga Maldonado1 years ago | 120 | // Plugin customization: remove options.Format != "W" check |
0ca4c062Jose Arriaga Maldonado1 years ago | 121 | additionalBinaryDataProperties.Add(prop.Name, BinaryData.FromString(prop.Value.GetRawText())); |
9f9f2936Jose Arriaga Maldonado2 years ago | 122 | } |
e0fee603Jose Arriaga Maldonado1 years ago | 123 | return new OpenAIEmbedding(index, embeddingProperty, @object, additionalBinaryDataProperties); |
9f9f2936Jose Arriaga Maldonado2 years ago | 124 | } |
| 125 | | |
e0fee603Jose Arriaga Maldonado1 years ago | 126 | BinaryData IPersistableModel<OpenAIEmbedding>.Write(ModelReaderWriterOptions options) => PersistableModelWriteCore(options); |
9f9f2936Jose Arriaga Maldonado2 years ago | 127 | |
5dce104aJose Arriaga Maldonado1 years ago | 128 | [Experimental("OPENAI001")] |
e0fee603Jose Arriaga Maldonado1 years ago | 129 | protected virtual BinaryData PersistableModelWriteCore(ModelReaderWriterOptions options) |
| 130 | { | |
| 131 | string format = options.Format == "W" ? ((IPersistableModel<OpenAIEmbedding>)this).GetFormatFromOptions(options) : options.Format; | |
9f9f2936Jose Arriaga Maldonado2 years ago | 132 | switch (format) |
| 133 | { | |
| 134 | case "J": | |
5dce104aJose Arriaga Maldonado1 years ago | 135 | return ModelReaderWriter.Write(this, options, OpenAIContext.Default); |
9f9f2936Jose Arriaga Maldonado2 years ago | 136 | default: |
19ceae44ShivangiReja1 years ago | 137 | throw new FormatException($"The model {nameof(OpenAIEmbedding)} does not support writing '{options.Format}' format."); |
9f9f2936Jose Arriaga Maldonado2 years ago | 138 | } |
| 139 | } | |
| 140 | | |
e0fee603Jose Arriaga Maldonado1 years ago | 141 | OpenAIEmbedding IPersistableModel<OpenAIEmbedding>.Create(BinaryData data, ModelReaderWriterOptions options) => PersistableModelCreateCore(data, options); |
9f9f2936Jose Arriaga Maldonado2 years ago | 142 | |
5dce104aJose Arriaga Maldonado1 years ago | 143 | [Experimental("OPENAI001")] |
e0fee603Jose Arriaga Maldonado1 years ago | 144 | protected virtual OpenAIEmbedding PersistableModelCreateCore(BinaryData data, ModelReaderWriterOptions options) |
| 145 | { | |
| 146 | string format = options.Format == "W" ? ((IPersistableModel<OpenAIEmbedding>)this).GetFormatFromOptions(options) : options.Format; | |
9f9f2936Jose Arriaga Maldonado2 years ago | 147 | switch (format) |
| 148 | { | |
| 149 | case "J": | |
e0fee603Jose Arriaga Maldonado1 years ago | 150 | using (JsonDocument document = JsonDocument.Parse(data)) |
9f9f2936Jose Arriaga Maldonado2 years ago | 151 | { |
19ceae44ShivangiReja1 years ago | 152 | return DeserializeOpenAIEmbedding(document.RootElement, options); |
9f9f2936Jose Arriaga Maldonado2 years ago | 153 | } |
| 154 | default: | |
19ceae44ShivangiReja1 years ago | 155 | throw new FormatException($"The model {nameof(OpenAIEmbedding)} does not support reading '{options.Format}' format."); |
9f9f2936Jose Arriaga Maldonado2 years ago | 156 | } |
| 157 | } | |
| 158 | | |
19ceae44ShivangiReja1 years ago | 159 | string IPersistableModel<OpenAIEmbedding>.GetFormatFromOptions(ModelReaderWriterOptions options) => "J"; |
9f9f2936Jose Arriaga Maldonado2 years ago | 160 | } |
| 161 | } |