49 lines
1.5 KiB
C#
49 lines
1.5 KiB
C#
|
using BotSharp.Abstraction.MLTasks;
|
||
|
|
using Microsoft.SemanticKernel;
|
||
|
|
using Microsoft.SemanticKernel.AI.Embeddings;
|
||
|
|
using Microsoft.SemanticKernel.Memory;
|
||
|
|
using Microsoft.SemanticKernel.Plugins.Memory;
|
||
|
|
using System;
|
||
|
|
using System.Collections.Generic;
|
||
|
|
using System.Linq;
|
||
|
|
using System.Text;
|
||
|
|
|
||
|
|
namespace BotSharp.Plugin.SemanticKernel
|
||
|
|
{
|
||
|
|
/// <summary>
|
||
|
|
/// Use Semantic Kernel Memory as text embedding provider
|
||
|
|
/// </summary>
|
||
|
|
public class SemanticKernelTextEmbeddingProvider : ITextEmbedding
|
||
|
|
{
|
||
|
|
private readonly ITextEmbeddingGeneration _embedding;
|
||
|
|
|
||
|
|
/// <summary>
|
||
|
|
/// Constructor of <see cref="SemanticKernelTextEmbeddingProvider"/>
|
||
|
|
/// </summary>
|
||
|
|
/// <param name="kernel"></param>
|
||
|
|
public SemanticKernelTextEmbeddingProvider(ITextEmbeddingGeneration embedding, int dimension)
|
||
|
|
{
|
||
|
|
this._embedding = embedding;
|
||
|
|
Dimension = dimension;
|
||
|
|
}
|
||
|
|
|
||
|
|
public int Dimension { get; }
|
||
|
|
|
||
|
|
public float[] GetVector(string text)
|
||
|
|
{
|
||
|
|
return this._embedding.GenerateEmbeddingAsync(text)
|
||
|
|
.ConfigureAwait(false)
|
||
|
|
.GetAwaiter()
|
||
|
|
.GetResult()
|
||
|
|
.ToArray();
|
||
|
|
}
|
||
|
|
|
||
|
|
public List<float[]> GetVectors(List<string> texts)
|
||
|
|
{
|
||
|
|
return this._embedding.GenerateEmbeddingsAsync(texts).ConfigureAwait(false).GetAwaiter().GetResult()
|
||
|
|
.Select(_ => _.ToArray())
|
||
|
|
.ToList();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|