embeddings_3.cs

	class AmazonReviewQueryMatch
	{
	    public AmazonReview Review { get; set; }
	    public double Relatedness { get; set; }
	}
	 
	async Task<AmazonReviewQueryMatch[]> Search(string query, int topN = 5)
	{
	    float[] queryEmbeddings;
	 
	    using (var api = new OpenAIClient(apiKey))
	    {
	        var response = await api.EmbeddingsEndpoint
	            .CreateEmbeddingAsync(query, model, dimensions: 512);
	 
	        queryEmbeddings = response.Data.First().Embedding
	            .Select(e => (float)e).ToArray();
	    }
	 
	    var matches = reviews.Select(review => new AmazonReviewQueryMatch
	    {
	        Review = review,
	        Relatedness = TensorPrimitives.CosineSimilarity(
	          new ReadOnlySpan<float>(review.Embeddings),
	          new ReadOnlySpan<float>(queryEmbeddings)
	        )
	    })
	    .OrderByDescending(match => match.Relatedness)
	    .Take(topN)
	    .ToArray();
	 
	    return matches;
	}