74 lines
2.3 KiB
C#
74 lines
2.3 KiB
C#
using System.Collections.Generic;
|
|
using System.Net;
|
|
using Microsoft.Extensions.Options;
|
|
using OpenAI;
|
|
using OpenAI.Managers;
|
|
using OpenAI.ObjectModels;
|
|
using OpenAI.ObjectModels.RequestModels;
|
|
using OpenAI.ObjectModels.ResponseModels;
|
|
using Volo.Abp.DependencyInjection;
|
|
using Volo.Abp.Domain.Services;
|
|
using Yi.Framework.ChatHub.Domain.Shared.Dtos;
|
|
using Yi.Framework.ChatHub.Domain.Shared.Options;
|
|
namespace Yi.Framework.ChatHub.Domain.Managers
|
|
{
|
|
public class AiManager : ISingletonDependency
|
|
{
|
|
public AiManager(IOptions<AiOptions> options)
|
|
{
|
|
this.OpenAIService = new OpenAIService(new OpenAiOptions()
|
|
{
|
|
ApiKey = options.Value.ApiKey,
|
|
BaseDomain = options.Value.BaseDomain
|
|
});
|
|
}
|
|
private OpenAIService OpenAIService { get; }
|
|
|
|
public async IAsyncEnumerable<string> ChatAsStreamAsync(string model, List<AiChatContextDto> aiChatContextDtos)
|
|
{
|
|
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 ?? string.Empty;
|
|
}
|
|
else
|
|
{
|
|
error = result.HttpStatusCode;
|
|
break;
|
|
}
|
|
|
|
}
|
|
if (error == HttpStatusCode.PaymentRequired)
|
|
{
|
|
yield return "余额不足,请联系站长充值";
|
|
|
|
}
|
|
|
|
}
|
|
}
|
|
}
|