feat: 重构聊天室语义内核

This commit is contained in:
chenchun
2025-03-21 18:24:59 +08:00
parent 0b111852ec
commit cbb3510d94
7 changed files with 78 additions and 67 deletions

View File

@@ -2,7 +2,7 @@ namespace Yi.Framework.Stock.Domain.Managers.SemanticKernel
{
public class SemanticKernelOptions
{
public string ModelId { get; set; }
public List<string> ModelIds { get; set; }
public string Endpoint { get; set; }
public string ApiKey { get; set; }
}

View File

@@ -32,9 +32,10 @@ namespace Yi.Framework.Stock.Domain
#pragma warning disable SKEXP0010
// 从配置中获取值
var options = semanticKernelSection.Get<SemanticKernelOptions>();
//股市优先使用第一个ai模型
services.AddKernel()
.AddOpenAIChatCompletion(
modelId: options.ModelId,
modelId: options.ModelIds.FirstOrDefault(),
endpoint: new Uri(options.Endpoint),
apiKey: options.ApiKey);
#pragma warning restore SKEXP0010

View File

@@ -1,6 +1,9 @@
using System.Collections.Generic;
using System.Net;
using Microsoft.Extensions.Options;
using Microsoft.SemanticKernel;
using Microsoft.SemanticKernel.ChatCompletion;
using Microsoft.SemanticKernel.Connectors.OpenAI;
// using OpenAI;
// using OpenAI.Managers;
// using OpenAI.ObjectModels;
@@ -14,61 +17,46 @@ namespace Yi.Framework.ChatHub.Domain.Managers
{
public class AiManager : ISingletonDependency
{
public AiManager(IOptions<AiOptions> options)
private readonly Kernel _kernel;
public AiManager(Kernel kernel)
{
// this.OpenAIService = new OpenAIService(new OpenAiOptions()
// {
// ApiKey = options.Value.ApiKey,
// BaseDomain = options.Value.BaseDomain
// });
_kernel = kernel;
}
// private OpenAIService OpenAIService { get; }
public async IAsyncEnumerable<string> ChatAsStreamAsync(string model, List<AiChatContextDto> aiChatContextDtos)
public async IAsyncEnumerable<string?> ChatAsStreamAsync(string model, List<AiChatContextDto> aiChatContextDtos)
{
throw new NotImplementedException("准备sk重构");
yield break;
// if (aiChatContextDtos.Count == 0)
// {
// yield return null;
// }
//
// List<ChatMessage> messages = aiChatContextDtos.Select(x =>
// {
// if (x.AnswererType == AnswererTypeEnum.Ai)
// {
// return ChatMessage.FromSystem(x.Message);
// }
// else
// {
// return ChatMessage.FromUser(x.Message);
// }
// }).ToList();
// var completionResult = OpenAIService.ChatCompletion.CreateCompletionAsStream(new ChatCompletionCreateRequest
// {
// Messages = messages,
// Model =model
// });
//
// HttpStatusCode? error = null;
// await foreach (var result in completionResult)
// {
// if (result.Successful)
// {
// yield return result.Choices.FirstOrDefault()?.Message.Content ?? null;
// }
// else
// {
// error = result.HttpStatusCode;
// break;
// }
//
// }
// if (error == HttpStatusCode.PaymentRequired)
// {
// yield return "余额不足,请联系站长充值";
//
// }
if (aiChatContextDtos.Count == 0)
{
yield return null;
}
var openSettings = new OpenAIPromptExecutionSettings()
{
MaxTokens =1000
};
var chatCompletionService = this._kernel.GetRequiredService<IChatCompletionService>(model);
var history =new ChatHistory();
foreach (var aiChatContextDto in aiChatContextDtos)
{
if (aiChatContextDto.AnswererType==AnswererTypeEnum.Ai)
{
history.AddSystemMessage(aiChatContextDto.Message);
}
else if(aiChatContextDto.AnswererType==AnswererTypeEnum.User)
{
history.AddUserMessage(aiChatContextDto.Message);
}
}
var results = chatCompletionService.GetStreamingChatMessageContentsAsync(
chatHistory: history,
executionSettings: openSettings,
kernel: _kernel);
await foreach (var result in results)
{
yield return result.Content;
}
}
}
}

View File

@@ -8,7 +8,7 @@
<ItemGroup>
<PackageReference Include="Volo.Abp.AspNetCore.SignalR" Version="$(AbpVersion)" />
<PackageReference Include="Microsoft.SemanticKernel" Version="1.40.0" />
<PackageReference Include="Volo.Abp.Ddd.Domain" Version="$(AbpVersion)" />
<PackageReference Include="Volo.Abp.Caching" Version="$(AbpVersion)" />
<ProjectReference Include="..\..\..\framework\Yi.Framework.Caching.FreeRedis\Yi.Framework.Caching.FreeRedis.csproj" />

View File

@@ -1,6 +1,10 @@
using Volo.Abp.Domain;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.SemanticKernel;
using Volo.Abp.Domain;
using Yi.Framework.Caching.FreeRedis;
using Yi.Framework.ChatHub.Domain.Shared;
using Yi.Framework.Core.Options;
namespace Yi.Framework.ChatHub.Domain
@@ -13,9 +17,27 @@ namespace Yi.Framework.ChatHub.Domain
)]
public class YiFrameworkChatHubDomainModule : AbpModule
{
public override async Task OnPostApplicationInitializationAsync(ApplicationInitializationContext context)
public override void ConfigureServices(ServiceConfigurationContext context)
{
var configuration = context.Services.GetConfiguration();
var services = context.Services;
// 配置绑定
var semanticKernelSection = configuration.GetSection("SemanticKernel");
services.Configure<SemanticKernelOptions>(configuration.GetSection("SemanticKernel"));
#pragma warning disable SKEXP0010
// 从配置中获取值
var options = semanticKernelSection.Get<SemanticKernelOptions>();
foreach (var optionsModelId in options.ModelIds)
{
services.AddKernel()
.AddOpenAIChatCompletion(
serviceId: optionsModelId,
modelId: optionsModelId,
endpoint: new Uri(options.Endpoint),
apiKey: options.ApiKey);
}
#pragma warning restore SKEXP0010
}
}
}