Skip to content

Commit 2d36feb

Browse files
added a test
1 parent 3ba9e02 commit 2d36feb

File tree

1 file changed

+31
-0
lines changed

1 file changed

+31
-0
lines changed

tests/Embeddings/EmbeddingsTests.cs

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
using OpenAI.Tests.Utility;
44
using System;
55
using System.ClientModel;
6+
using System.ClientModel.Primitives;
67
using System.Collections.Generic;
78
using System.Threading.Tasks;
89
using static OpenAI.Tests.TestHelpers;
@@ -205,4 +206,34 @@ public void SerializeEmbeddingCollection()
205206
{
206207
// TODO: Add this test.
207208
}
209+
210+
[Test]
211+
public void JsonArraySupport()
212+
{
213+
string json = """
214+
{
215+
"object":"list",
216+
"data":[
217+
{
218+
"object":"embedding",
219+
"embedding":[-0.011229509,0.107915245,-0.15163477]
220+
}
221+
]
222+
}
223+
""";
224+
225+
BinaryData binaryData = BinaryData.FromString(json);
226+
227+
OpenAIEmbeddingCollection embeddings = ModelReaderWriter.Read<OpenAIEmbeddingCollection>(binaryData);
228+
229+
Assert.That(embeddings, Is.Not.Null);
230+
Assert.That(embeddings.Count, Is.EqualTo(1));
231+
var embedding = embeddings[0];
232+
Assert.That(embedding, Is.Not.Null);
233+
ReadOnlySpan<float> vector = embedding.ToFloats().Span;
234+
Assert.That(vector.Length, Is.EqualTo(3));
235+
Assert.That(vector[0], Is.EqualTo(-0.011229509f));
236+
Assert.That(vector[1], Is.EqualTo(0.107915245f));
237+
Assert.That(vector[2], Is.EqualTo(-0.15163477f));
238+
}
208239
}

0 commit comments

Comments
 (0)