54 lines
1.6 KiB
C#
54 lines
1.6 KiB
C#
using BotSharp.Abstraction.Conversations.Models;
|
|
using BotSharp.Plugin.SemanticKernel;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.SemanticKernel;
|
|
using Microsoft.SemanticKernel.AI.TextCompletion;
|
|
using Moq;
|
|
using System.Collections.Generic;
|
|
using System.Threading.Tasks;
|
|
using Xunit;
|
|
using BotSharp.Abstraction.Agents.Enums;
|
|
using BotSharp.Abstraction.Agents.Models;
|
|
using BotSharp.Abstraction.Conversations;
|
|
using System;
|
|
using System.Linq;
|
|
using Microsoft;
|
|
using Microsoft.SemanticKernel.AI;
|
|
using BotSharp.Plugin.SemanticKernel.UnitTests.Helpers;
|
|
|
|
namespace BotSharp.Plugin.SemanticKernel.Tests
|
|
{
|
|
public class SemanticKernelTextCompletionProviderTests
|
|
{
|
|
private readonly IServiceProvider _services;
|
|
private readonly ITokenStatistics _tokenStatistics;
|
|
|
|
public SemanticKernelTextCompletionProviderTests()
|
|
{
|
|
|
|
_services = new ServiceCollection().BuildServiceProvider();
|
|
_tokenStatistics = Mock.Of<ITokenStatistics>();
|
|
}
|
|
|
|
[Fact]
|
|
public async Task GetCompletion_ReturnsExpectedResult()
|
|
{
|
|
// Arrange
|
|
|
|
var text = "Hello";
|
|
var expected = "Hello, world!";
|
|
var _kernel = new KernelBuilder()
|
|
.WithAIService<ITextCompletion>("", new SemanticKernelHelper(expected))
|
|
.Build();
|
|
var provider = new SemanticKernelTextCompletionProvider(_kernel, _services, _tokenStatistics);
|
|
|
|
// Act
|
|
var result = await provider.GetCompletion(text, "agent1", "message1");
|
|
|
|
// Assert
|
|
Assert.Equal(expected, result);
|
|
|
|
}
|
|
}
|
|
}
|