diff --git a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/AiChatService.cs b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/AiChatService.cs index 64c8ce4d..950ee61d 100644 --- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/AiChatService.cs +++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/AiChatService.cs @@ -100,23 +100,23 @@ public class AiChatService : ApplicationService } var gateWay = LazyServiceProvider.GetRequiredService(); - // var completeChatResponse = gateWay.CompleteChatAsync(input.Model, history, cancellationToken); - // await using var writer = new StreamWriter(response.Body, Encoding.UTF8, leaveOpen: true); - // await foreach (var data in completeChatResponse) - // { - // var model = MapToMessage(input.Model, data); - // var message = JsonConvert.SerializeObject(model, new JsonSerializerSettings - // { - // ContractResolver = new CamelCasePropertyNamesContractResolver() - // }); - // - // await writer.WriteLineAsync($"data: {message}\n"); - // await writer.FlushAsync(cancellationToken); // 确保立即推送数据 - // } + var completeChatResponse = gateWay.CompleteChatAsync(input.Model, history, cancellationToken); + await using var writer = new StreamWriter(response.Body, Encoding.UTF8, leaveOpen: true); + await foreach (var data in completeChatResponse) + { + var model = MapToMessage(input.Model, data); + var message = JsonConvert.SerializeObject(model, new JsonSerializerSettings + { + ContractResolver = new CamelCasePropertyNamesContractResolver() + }); + + await writer.WriteLineAsync($"data: {message}\n"); + await writer.FlushAsync(cancellationToken); // 确保立即推送数据 + } //断开连接 - // await writer.WriteLineAsync("data: done\n"); - // await writer.FlushAsync(cancellationToken); // 确保立即推送数据 + await writer.WriteLineAsync("data: done\n"); + await writer.FlushAsync(cancellationToken); // 确保立即推送数据 if (CurrentUser.IsAuthenticated && input.SessionId.HasValue) {