69 lines
2.6 KiB
C#
69 lines
2.6 KiB
C#
|
using System;
|
||
|
|
using System.Collections.Generic;
|
||
|
|
using System.Threading.Tasks;
|
||
|
|
using BotSharp.Abstraction.Agents;
|
||
|
|
using BotSharp.Abstraction.Conversations.Models;
|
||
|
|
using BotSharp.Abstraction.MLTasks;
|
||
|
|
using BotSharp.Plugin.SemanticKernel;
|
||
|
|
using Microsoft.Extensions.DependencyInjection;
|
||
|
|
using Microsoft.SemanticKernel;
|
||
|
|
using Moq;
|
||
|
|
using Xunit;
|
||
|
|
using BotSharp.Abstraction.Agents.Enums;
|
||
|
|
using BotSharp.Abstraction.Agents.Models;
|
||
|
|
using System.Linq;
|
||
|
|
using System.Runtime;
|
||
|
|
using BotSharp.Abstraction.Conversations;
|
||
|
|
using BotSharp.Abstraction.Models;
|
||
|
|
using Microsoft.SemanticKernel.AI.ChatCompletion;
|
||
|
|
using Microsoft.SemanticKernel.AI;
|
||
|
|
using Microsoft.SemanticKernel.Connectors.AI.OpenAI.AzureSdk;
|
||
|
|
|
||
|
|
namespace BotSharp.Plugin.SemanticKernel.Tests
|
||
|
|
{
|
||
|
|
public class SemanticKernelChatCompletionProviderTests
|
||
|
|
{
|
||
|
|
private readonly Mock<IKernel> _kernelMock;
|
||
|
|
private readonly Mock<IServiceProvider> _servicesMock;
|
||
|
|
private readonly Mock<ITokenStatistics> _tokenStatisticsMock;
|
||
|
|
private readonly SemanticKernelChatCompletionProvider _provider;
|
||
|
|
|
||
|
|
public SemanticKernelChatCompletionProviderTests()
|
||
|
|
{
|
||
|
|
_kernelMock = new Mock<IKernel>();
|
||
|
|
_servicesMock = new Mock<IServiceProvider>();
|
||
|
|
_tokenStatisticsMock = new Mock<ITokenStatistics>();
|
||
|
|
_provider = new SemanticKernelChatCompletionProvider(_kernelMock.Object, _servicesMock.Object, _tokenStatisticsMock.Object);
|
||
|
|
}
|
||
|
|
|
||
|
|
[Fact]
|
||
|
|
public void GetChatCompletions_Returns_RoleDialogModel()
|
||
|
|
{
|
||
|
|
// Arrange
|
||
|
|
var agent = new Agent();
|
||
|
|
var conversations = new List<RoleDialogModel>
|
||
|
|
{
|
||
|
|
new RoleDialogModel(AgentRole.User, "Hello")
|
||
|
|
};
|
||
|
|
|
||
|
|
var chatHistoryMock = new Mock<ChatHistory>();
|
||
|
|
var chatCompletionMock = new Mock<Microsoft.SemanticKernel.AI.ChatCompletion.IChatCompletion>();
|
||
|
|
chatCompletionMock.Setup(x => x.CreateNewChat(It.IsAny<string>())).Returns(chatHistoryMock.Object);
|
||
|
|
chatCompletionMock.Setup(x => x.GetChatCompletionsAsync(chatHistoryMock.Object, It.IsAny<AIRequestSettings>(), It.IsAny<CancellationToken>()))
|
||
|
|
.ReturnsAsync(new List<IChatResult>
|
||
|
|
{
|
||
|
|
new MockChatResult("How can I help you?")
|
||
|
|
});
|
||
|
|
|
||
|
|
_kernelMock.Setup(x => x.GetService<Microsoft.SemanticKernel.AI.ChatCompletion.IChatCompletion>(null)).Returns(chatCompletionMock.Object);
|
||
|
|
|
||
|
|
// Act
|
||
|
|
var result = _provider.GetChatCompletions(agent, conversations);
|
||
|
|
|
||
|
|
// Assert
|
||
|
|
Assert.IsType<RoleDialogModel>(result);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|