60 lines
2.0 KiB
C#
60 lines
2.0 KiB
C#
using Microsoft.SemanticKernel;
|
|
using Microsoft.SemanticKernel.ChatCompletion;
|
|
using Microsoft.SemanticKernel.Connectors.OpenAI;
|
|
using Volo.Abp.DependencyInjection;
|
|
|
|
namespace Yi.Framework.Stock.Domain.Managers.SemanticKernel;
|
|
|
|
public class SemanticKernelClient : ITransientDependency
|
|
{
|
|
public Kernel Kernel { get; }
|
|
|
|
public SemanticKernelClient(Kernel kernel)
|
|
{
|
|
this.Kernel = kernel;
|
|
}
|
|
|
|
/// <summary>
|
|
/// 执行插件
|
|
/// </summary>
|
|
/// <param name="input"></param>
|
|
/// <param name="pluginName"></param>
|
|
/// <param name="functionName"></param>
|
|
/// <returns></returns>
|
|
public async Task<string?> InvokerFunctionAsync(string input, string pluginName, string functionName)
|
|
{
|
|
KernelFunction jsonFun = this.Kernel.Plugins.GetFunction(pluginName, functionName);
|
|
var result = await this.Kernel.InvokeAsync(function: jsonFun, new KernelArguments() { ["input"] = input });
|
|
return result.GetValue<string>();
|
|
}
|
|
|
|
/// <summary>
|
|
/// 聊天完成,FunctionCall
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
public async Task<IReadOnlyList<ChatMessageContent>> ChatCompletionAsync(string question,
|
|
params (string, string)[] functions)
|
|
{
|
|
if (functions is null)
|
|
{
|
|
throw new Exception("请选择插件");
|
|
}
|
|
|
|
var openSettings = new OpenAIPromptExecutionSettings()
|
|
{
|
|
FunctionChoiceBehavior =
|
|
FunctionChoiceBehavior.Auto(
|
|
functions.Select(x => this.Kernel.Plugins.GetFunction(x.Item1, x.Item2)).ToList(), true),
|
|
// ToolCallBehavior = ToolCallBehavior.AutoInvokeKernelFunctions,
|
|
// MaxTokens =1000
|
|
};
|
|
|
|
var chatCompletionService = this.Kernel.GetRequiredService<IChatCompletionService>();
|
|
|
|
var results = await chatCompletionService.GetChatMessageContentsAsync(
|
|
question,
|
|
executionSettings: openSettings,
|
|
kernel: Kernel);
|
|
return results;
|
|
}
|
|
} |