71 lines
2.1 KiB
C#
71 lines
2.1 KiB
C#
using System.Collections.Generic;
|
|
using Microsoft.Extensions.Options;
|
|
using OpenAI;
|
|
using OpenAI.Managers;
|
|
using OpenAI.ObjectModels;
|
|
using OpenAI.ObjectModels.RequestModels;
|
|
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(List<AiChatContextDto> aiChatContextDtos)
|
|
{
|
|
var temp = "站长正在接入ChatGpt,敬请期待~";
|
|
|
|
for (var i = 0; i < temp.Length; i++)
|
|
{
|
|
await Task.Delay(200);
|
|
yield return temp[i].ToString();
|
|
}
|
|
|
|
|
|
|
|
|
|
//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 = Models.Gpt_3_5_Turbo
|
|
//});
|
|
|
|
//await foreach (var result in completionResult)
|
|
//{
|
|
// if (result.Successful)
|
|
// {
|
|
// yield return result.Choices.FirstOrDefault()?.Message.Content ?? string.Empty;
|
|
// }
|
|
//}
|
|
|
|
}
|
|
}
|
|
}
|