openai/openai-dotnet
Publicmirrored from https://github.com/openai/openai-dotnetAvailable
examples/Embeddings/Example03_MultipleEmbeddingsAsync.cs
38lines · modeblame
9f9f2936Jose Arriaga Maldonado2 years ago | 1 | using NUnit.Framework; |
| 2 | using OpenAI.Embeddings; | |
| 3 | using System; | |
| 4 | using System.Collections.Generic; | |
| 5 | using System.Threading.Tasks; | |
| 6 | | |
8cc6643fJose Arriaga Maldonado2 years ago | 7 | namespace OpenAI.Examples; |
74848f8dJose Arriaga Maldonado2 years ago | 8 | |
8cc6643fJose Arriaga Maldonado2 years ago | 9 | public partial class EmbeddingExamples |
9f9f2936Jose Arriaga Maldonado2 years ago | 10 | { |
74848f8dJose Arriaga Maldonado2 years ago | 11 | [Test] |
8cc6643fJose Arriaga Maldonado2 years ago | 12 | public async Task Example03_MultipleEmbeddingsAsync() |
9f9f2936Jose Arriaga Maldonado2 years ago | 13 | { |
74848f8dJose Arriaga Maldonado2 years ago | 14 | EmbeddingClient client = new("text-embedding-3-small", Environment.GetEnvironmentVariable("OPENAI_API_KEY")); |
9f9f2936Jose Arriaga Maldonado2 years ago | 15 | |
74848f8dJose Arriaga Maldonado2 years ago | 16 | string category = "Luxury"; |
| 17 | string description = "Best hotel in town if you like luxury hotels. They have an amazing infinity pool, a spa," | |
| 18 | + " and a really helpful concierge. The location is perfect -- right downtown, close to all the tourist" | |
| 19 | + " attractions. We highly recommend this hotel."; | |
| 20 | List<string> inputs = [category, description]; | |
9f9f2936Jose Arriaga Maldonado2 years ago | 21 | |
19ceae44ShivangiReja1 years ago | 22 | OpenAIEmbeddingCollection collection = await client.GenerateEmbeddingsAsync(inputs); |
9f9f2936Jose Arriaga Maldonado2 years ago | 23 | |
19ceae44ShivangiReja1 years ago | 24 | foreach (OpenAIEmbedding embedding in collection) |
74848f8dJose Arriaga Maldonado2 years ago | 25 | { |
2ab1a942Jose Arriaga Maldonado1 years ago | 26 | ReadOnlyMemory<float> vector = embedding.ToFloats(); |
9f9f2936Jose Arriaga Maldonado2 years ago | 27 | |
74848f8dJose Arriaga Maldonado2 years ago | 28 | Console.WriteLine($"Dimension: {vector.Length}"); |
| 29 | Console.WriteLine($"Floats: "); | |
| 30 | for (int i = 0; i < vector.Length; i++) | |
| 31 | { | |
665fb062Jose Arriaga Maldonado2 years ago | 32 | Console.WriteLine($" [{i,4}] = {vector.Span[i]}"); |
9f9f2936Jose Arriaga Maldonado2 years ago | 33 | } |
74848f8dJose Arriaga Maldonado2 years ago | 34 | |
| 35 | Console.WriteLine(); | |
9f9f2936Jose Arriaga Maldonado2 years ago | 36 | } |
| 37 | } | |
| 38 | } |