42 lines
1.2 KiB
C#
42 lines
1.2 KiB
C#
using System.Collections.Generic;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.Extensions.AI;
|
|
using OllamaSharp;
|
|
|
|
namespace xAiApi.Interfaces
|
|
{
|
|
/// <summary>
|
|
/// Base Embedding Provider Service ...
|
|
/// </summary>
|
|
public interface IXAiEmbeddingServiceBase : IXAiServiceBase
|
|
{
|
|
/// <summary>
|
|
/// Embedding Specified String ...
|
|
/// </summary>
|
|
/// <param name="content"></param>
|
|
/// <param name="cancellationToken"></param>
|
|
/// <returns></returns>
|
|
Task<float[]> Embedding(
|
|
string content,
|
|
CancellationToken cancellationToken = default
|
|
);
|
|
|
|
/// <summary>
|
|
/// Embedding Specified Batch ...
|
|
/// </summary>
|
|
/// <param name="batch"></param>
|
|
/// <param name="cancellationToken"></param>
|
|
/// <returns></returns>
|
|
Task<IReadOnlyList<float[]>> Embeddings(
|
|
string[] batch,
|
|
CancellationToken cancellationToken = default
|
|
);
|
|
|
|
/// <summary>
|
|
/// Embedding Generator ...
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
IEmbeddingGenerator<string, Embedding<float>> GetEmbeddingGenerator();
|
|
}
|
|
} |