EmbeddingEndpointTests.cs

using NUnit.Framework;
using OpenAI_API.Embedding;
using OpenAI_API.Models;
using System;
using System.Linq;

namespace OpenAI_Tests
{
    public class EmbeddingEndpointTests
    {
        [SetUp]
        public void Setup()
        {
            OpenAI_API.APIAuthentication.Default = new OpenAI_API.APIAuthentication(Environment.GetEnvironmentVariable("TEST_OPENAI_SECRET_KEY"));
        }

        [Test]
        public void GetBasicEmbedding()
        {
            var api = new OpenAI_API.OpenAIAPI();

            Assert.IsNotNull(api.Embeddings);

            var results = api.Embeddings.CreateEmbeddingAsync(new EmbeddingRequest(Model.AdaTextEmbedding, "A test text for embedding")).Result;
            Assert.IsNotNull(results);
            if (results.CreatedUnixTime.HasValue)
            {
                Assert.NotZero(results.CreatedUnixTime.Value);
                Assert.NotNull(results.Created);
                Assert.Greater(results.Created.Value, new DateTime(2018, 1, 1));
				Assert.Less(results.Created.Value, DateTime.Now.AddDays(1));
			} else
            {
                Assert.Null(results.Created);
			}
            Assert.NotNull(results.Object);
            Assert.NotZero(results.Data.Count);
            Assert.That(results.Data.First().Embedding.Length == 1536);
        }

        [Test]
        public void ReturnedUsage()
        {
            var api = new OpenAI_API.OpenAIAPI();

            Assert.IsNotNull(api.Embeddings);

            var results = api.Embeddings.CreateEmbeddingAsync(new EmbeddingRequest(Model.AdaTextEmbedding, "A test text for embedding")).Result;
            Assert.IsNotNull(results);

			Assert.IsNotNull(results.Usage);
			Assert.GreaterOrEqual(results.Usage.PromptTokens, 5);
			Assert.GreaterOrEqual(results.Usage.TotalTokens, results.Usage.PromptTokens);
		}

        [Test]
        public void GetSimpleEmbedding()
        {
            var api = new OpenAI_API.OpenAIAPI();

            Assert.IsNotNull(api.Embeddings);

            var results = api.Embeddings.GetEmbeddingsAsync("A test text for embedding").Result;
            Assert.IsNotNull(results);
            Assert.That(results.Length == 1536);
        }

        [Test]
        public void GetSimpleEmbeddingWithDimensions()
        {
            var api = new OpenAI_API.OpenAIAPI();

            Assert.IsNotNull(api.Embeddings);

            var request = new EmbeddingRequest(Model.SmallTextEmbedding, "A test text for embedding");
            request.Dimensions = 256;
            var results = api.Embeddings.GetEmbeddingsAsync(request).Result;
            Assert.IsNotNull(results);
            Assert.That(results.Length == 256);
        }
    }
}