优化拆分对话实现

src/Service/FastWiki.Service/Backgrounds/ChatRecordAnalyseBackgroundService.cs

@@ -1,5 +1,8 @@
 namespace FastWiki.Service.Backgrounds;
 
+/// <summary>
+/// 聊天记录分析后台服务
+/// </summary>
 public class ChatRecordAnalyseBackgroundService : BackgroundService
 {
     protected override async Task ExecuteAsync(CancellationToken stoppingToken)

src/Service/FastWiki.Service/Backgrounds/WeChatBackgroundService.cs

@@ -0,0 +1,187 @@
+using System.Text;
+using System.Threading.Channels;
+using FastWiki.Service.Contracts.OpenAI;
+using FastWiki.Service.Contracts.WeChat;
+using FastWiki.Service.Domain.Storage.Aggregates;
+using FastWiki.Service.Infrastructure.Helper;
+using FastWiki.Service.Service;
+using Microsoft.Extensions.Caching.Memory;
+using Microsoft.SemanticKernel;
+using Microsoft.SemanticKernel.ChatCompletion;
+using MemoryService = mem0.NET.Services.MemoryService;
+
+namespace FastWiki.Service.Backgrounds;
+
+public class WeChatBackgroundService(
+    IServiceProvider serviceProvider,
+    WikiMemoryService wikiMemoryService,
+    IMemoryCache memoryCache) : BackgroundService
+{
+    public static readonly Channel<WeChatAI> Channel = System.Threading.Channels.Channel.CreateUnbounded<WeChatAI>();
+
+    protected override async Task ExecuteAsync(CancellationToken stoppingToken)
+    {
+        using var scope = serviceProvider.CreateScope();
+
+        var memoryService =
+            scope.ServiceProvider.GetRequiredService<MemoryService>();
+
+        var eventBus = scope.ServiceProvider.GetRequiredService<IEventBus>();
+        var openAIService = scope.ServiceProvider.GetRequiredService<OpenAIService>();
+        var wikiRepository = scope.ServiceProvider.GetRequiredService<IWikiRepository>();
+        var fileStorageRepository = scope.ServiceProvider.GetRequiredService<IFileStorageRepository>();
+
+        while (await Channel.Reader.WaitToReadAsync(stoppingToken))
+        {
+            var content = await Channel.Reader.ReadAsync(stoppingToken);
+
+            await SendMessageAsync(content, memoryService,eventBus,openAIService,wikiRepository,fileStorageRepository);
+        }
+    }
+
+
+    private const string OutputTemplate =
+        """
+        您好，欢迎关注FastWiki！
+        由于微信限制，我们无法立即回复您的消息，但是您的消息已经收到，我们会尽快回复您！
+        如果获取消息结果，请输入继续。
+        如果您有其他问题，可以直接回复，我们会尽快回复您！
+        """;
+
+
+    /// <summary>
+    ///     微信AI对话
+    /// </summary>
+    /// <param name="chatAi"></param>
+    /// <param name="memoryService"></param>
+    /// <param name="eventBus"></param>
+    /// <param name="openAiService"></param>
+    /// <param name="wikiRepository"></param>
+    /// <param name="fileStorageRepository"></param>
+    public async Task SendMessageAsync(WeChatAI chatAi, MemoryService memoryService, IEventBus eventBus,
+        OpenAIService openAiService, IWikiRepository wikiRepository, IFileStorageRepository fileStorageRepository)
+    {
+        var chatShareInfoQuery = new ChatShareInfoQuery(chatAi.SharedId);
+
+        await eventBus.PublishAsync(chatShareInfoQuery);
+
+        // 如果chatShareId不存在则返回让下面扣款
+        var chatShare = chatShareInfoQuery.Result;
+
+        var chatApplicationQuery = new ChatApplicationInfoQuery(chatShareInfoQuery.Result.ChatApplicationId);
+
+        await eventBus.PublishAsync(chatApplicationQuery);
+
+        var chatApplication = chatApplicationQuery?.Result;
+
+        if (chatApplication == null) return;
+
+        var requestToken = 0;
+
+        var module = new ChatCompletionDto<ChatCompletionRequestMessage>
+        {
+            messages =
+            [
+                new ChatCompletionRequestMessage
+                {
+                    content = chatAi.Content,
+                    role = "user"
+                }
+            ]
+        };
+
+        var chatHistory = new ChatHistory();
+
+        // 如果设置了Prompt，则添加
+        if (!chatApplication.Prompt.IsNullOrEmpty()) chatHistory.AddSystemMessage(chatApplication.Prompt);
+
+        // 保存对话提问
+        var createChatRecordCommand = new CreateChatRecordCommand(chatApplication.Id, chatAi.Content);
+
+        await eventBus.PublishAsync(createChatRecordCommand);
+
+        var sourceFile = new List<FileStorage>();
+        // 如果为空则不使用知识库
+        if (chatApplication.WikiIds.Count != 0)
+        {
+            var success = await OpenAIService.WikiPrompt(chatApplication, wikiMemoryService, chatAi.Content,
+                fileStorageRepository,
+                wikiRepository,
+                sourceFile, module, null, memoryService);
+
+            if (!success) return;
+        }
+
+        var output = new StringBuilder();
+
+        // 添加用户输入，并且计算请求token数量
+        module.messages.ForEach(x =>
+        {
+            if (x.content.IsNullOrEmpty()) return;
+            requestToken += TokenHelper.ComputeToken(x.content);
+
+            chatHistory.Add(new ChatMessageContent(new AuthorRole(x.role), x.content));
+        });
+
+
+        if (chatShare != null)
+        {
+            // 如果token不足则返回，使用token和当前request总和大于可用token，则返回
+            if (chatShare.AvailableToken != -1 &&
+                chatShare.UsedToken + requestToken >=
+                chatShare.AvailableToken)
+            {
+                output.Append("Token不足");
+                return;
+            }
+
+            // 如果没有过期则继续
+            if (chatShare.Expires != null &&
+                chatShare.Expires < DateTimeOffset.Now)
+            {
+                output.Append("Token已过期");
+                return;
+            }
+        }
+
+
+        try
+        {
+            await foreach (var item in openAiService.SendChatMessageAsync(chatApplication,
+                               chatHistory))
+            {
+                if (string.IsNullOrEmpty(item)) continue;
+
+                output.Append(item);
+            }
+
+            //对于对话扣款
+            if (chatShare != null)
+            {
+                var updateChatShareCommand = new DeductTokenCommand(chatShare.Id,
+                    requestToken);
+
+                await eventBus.PublishAsync(updateChatShareCommand);
+            }
+        }
+        catch (InvalidOperationException invalidOperationException)
+        {
+            output.Clear();
+            output.Append("对话异常:" + invalidOperationException.Message);
+        }
+        catch (ArgumentException argumentException)
+        {
+            output.Clear();
+            output.Append("对话异常:" + argumentException.Message);
+        }
+        catch (Exception e)
+        {
+            output.Clear();
+            output.Append("对话异常,请联系管理员");
+        }
+        finally
+        {
+            memoryCache.Set(chatAi.MessageId, output.ToString(), TimeSpan.FromMinutes(5));
+        }
+    }
+}

src/Service/FastWiki.Service/Program.cs

@@ -1,6 +1,8 @@
 using System.Text.Json;
 using FastWiki.Service;
 using FastWiki.Service.Backgrounds;
+using FastWiki.Service.Contracts.Model.Dto;
+using FastWiki.Service.Contracts.OpenAI;
 using FastWiki.Service.Options;
 using FastWiki.Service.Service;
 using Masa.Contrib.Authentication.Identity;
@@ -87,6 +89,11 @@
         Port = port,
     });
 
+builder.Services.AddScoped<OpenAIService>();
+builder.Services.AddScoped<FeishuService>();
+
+builder.Services.AddHostedService<WeChatBackgroundService>();
+
 builder.Services
     .AddCors(options =>
     {
@@ -108,9 +115,9 @@
         options.Role = "role";
     })
     .AddHttpContextAccessor()
-    .AddSwaggerGen(options =>
+    .AddSwaggerGen(swaggerGenOptions =>
     {
-        options.AddSecurityDefinition("Bearer", new OpenApiSecurityScheme
+        swaggerGenOptions.AddSecurityDefinition("Bearer", new OpenApiSecurityScheme
         {
             Name = "Authorization",
             Type = SecuritySchemeType.ApiKey,
@@ -120,7 +127,7 @@
             Description =
                 "JWT Authorization header using the Bearer scheme. \r\n\r\n Enter 'Bearer' [space] and then your token in the text input below.\r\n\r\nExample: \"Bearer xxxxxxxxxxxxxxx\""
         });
-        options.AddSecurityRequirement(new OpenApiSecurityRequirement
+        swaggerGenOptions.AddSecurityRequirement(new OpenApiSecurityRequirement
         {
             {
                 new OpenApiSecurityScheme
@@ -131,20 +138,20 @@
                         Id = "Bearer"
                     }
                 },
-                new string[] { }
+                []
             }
         });
 
-        options.SwaggerDoc("v1",
+        swaggerGenOptions.SwaggerDoc("v1",
             new OpenApiInfo
             {
                 Title = "FastWiki.ServiceApp",
                 Version = "v1",
                 Contact = new OpenApiContact { Name = "FastWiki.ServiceApp" }
             });
         foreach (var item in Directory.GetFiles(Directory.GetCurrentDirectory(), "*.xml"))
-            options.IncludeXmlComments(item, true);
-        options.DocInclusionPredicate((_, _) => true);
+            swaggerGenOptions.IncludeXmlComments(item, true);
+        swaggerGenOptions.DocInclusionPredicate((_, _) => true);
     })
     .AddScoped<IHistoryService, HistoryService>()
     .AddMasaDbContext<WikiDbContext>(opt =>
@@ -230,13 +237,18 @@
     return Results.Text($"window.thor = {JsonSerializer.Serialize(webEnv)};", "application/javascript");
 });
 
-app.MapPost("/v1/chat/completions", OpenAIService.Completions)
+app.MapPost("/v1/chat/completions", (OpenAIService openAiService, HttpContext context,
+            ChatCompletionDto<ChatCompletionRequestMessage> input,
+            ChatApplicationService chatApplicationService) =>
+        openAiService.Completions(context, input, chatApplicationService))
     .WithTags("OpenAI")
     .WithGroupName("OpenAI")
     .WithDescription("OpenAI Completions")
     .WithOpenApi();
 
-app.MapPost("/v1/feishu/completions/{id}", FeishuService.Completions)
+app.MapPost("/v1/feishu/completions/{id}", (FeishuService feiShuService, HttpContext context,
+            FeishuChatInput input, string id) =>
+        feiShuService.Completions(id, context, input))
     .WithTags("Feishu")
     .WithGroupName("Feishu")
     .WithDescription("飞书对话接入处理")