905 lines
36 KiB
C#
905 lines
36 KiB
C#
using System.Collections.Concurrent;
|
||
using System.Diagnostics;
|
||
using System.Runtime.CompilerServices;
|
||
using System.Text;
|
||
using System.Text.Json;
|
||
using Microsoft.AspNetCore.Http;
|
||
using Microsoft.Extensions.DependencyInjection;
|
||
using Microsoft.Extensions.Logging;
|
||
using Newtonsoft.Json;
|
||
using Newtonsoft.Json.Linq;
|
||
using Newtonsoft.Json.Serialization;
|
||
using Volo.Abp.Domain.Services;
|
||
using Yi.Framework.AiHub.Domain.AiGateWay;
|
||
using Yi.Framework.AiHub.Domain.AiGateWay.Exceptions;
|
||
using Yi.Framework.AiHub.Domain.Entities.Model;
|
||
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
||
using Yi.Framework.AiHub.Domain.Shared.Dtos;
|
||
using Yi.Framework.AiHub.Domain.Shared.Dtos.Anthropic;
|
||
using Yi.Framework.AiHub.Domain.Shared.Dtos.Gemini;
|
||
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
|
||
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi.Embeddings;
|
||
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi.Images;
|
||
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi.Responses;
|
||
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||
using Yi.Framework.AiHub.Domain.Shared.Extensions;
|
||
using Yi.Framework.Core.Extensions;
|
||
using Yi.Framework.SqlSugarCore.Abstractions;
|
||
using JsonSerializer = System.Text.Json.JsonSerializer;
|
||
using ThorJsonSerializer = Yi.Framework.AiHub.Domain.AiGateWay.ThorJsonSerializer;
|
||
|
||
namespace Yi.Framework.AiHub.Domain.Managers;
|
||
|
||
public class AiGateWayManager : DomainService
|
||
{
|
||
private readonly ISqlSugarRepository<AiAppAggregateRoot> _aiAppRepository;
|
||
private readonly ISqlSugarRepository<AiModelEntity> _aiModelRepository;
|
||
private readonly ILogger<AiGateWayManager> _logger;
|
||
private readonly AiMessageManager _aiMessageManager;
|
||
private readonly UsageStatisticsManager _usageStatisticsManager;
|
||
private readonly ISpecialCompatible _specialCompatible;
|
||
private PremiumPackageManager? _premiumPackageManager;
|
||
|
||
|
||
public AiGateWayManager(ISqlSugarRepository<AiAppAggregateRoot> aiAppRepository, ILogger<AiGateWayManager> logger,
|
||
AiMessageManager aiMessageManager, UsageStatisticsManager usageStatisticsManager,
|
||
ISpecialCompatible specialCompatible, ISqlSugarRepository<AiModelEntity> aiModelRepository)
|
||
{
|
||
_aiAppRepository = aiAppRepository;
|
||
_logger = logger;
|
||
_aiMessageManager = aiMessageManager;
|
||
_usageStatisticsManager = usageStatisticsManager;
|
||
_specialCompatible = specialCompatible;
|
||
_aiModelRepository = aiModelRepository;
|
||
}
|
||
|
||
private PremiumPackageManager PremiumPackageManager =>
|
||
_premiumPackageManager ??= LazyServiceProvider.LazyGetRequiredService<PremiumPackageManager>();
|
||
|
||
/// <summary>
|
||
/// 获取模型
|
||
/// </summary>
|
||
/// <param name="modelApiType"></param>
|
||
/// <param name="modelId"></param>
|
||
/// <returns></returns>
|
||
private async Task<AiModelDescribe> GetModelAsync(ModelApiTypeEnum modelApiType, string modelId)
|
||
{
|
||
var aiModelDescribe = await _aiModelRepository._DbQueryable
|
||
.LeftJoin<AiAppAggregateRoot>((model, app) => model.AiAppId == app.Id)
|
||
.Where((model, app) => model.ModelId == modelId)
|
||
.Where((model, app) => model.ModelApiType == modelApiType)
|
||
.Select((model, app) =>
|
||
new AiModelDescribe
|
||
{
|
||
AppId = app.Id,
|
||
AppName = app.Name,
|
||
Endpoint = app.Endpoint,
|
||
ApiKey = app.ApiKey,
|
||
OrderNum = model.OrderNum,
|
||
HandlerName = model.HandlerName,
|
||
ModelId = model.ModelId,
|
||
ModelName = model.Name,
|
||
Description = model.Description,
|
||
AppExtraUrl = app.ExtraUrl,
|
||
ModelExtraInfo = model.ExtraInfo,
|
||
Multiplier = model.Multiplier
|
||
})
|
||
.FirstAsync();
|
||
if (aiModelDescribe is null)
|
||
{
|
||
throw new UserFriendlyException($"【{modelId}】模型当前版本【{modelApiType}】格式不支持");
|
||
}
|
||
|
||
return aiModelDescribe;
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// 聊天完成-非流式
|
||
/// </summary>
|
||
/// <param name="httpContext"></param>
|
||
/// <param name="request"></param>
|
||
/// <param name="userId"></param>
|
||
/// <param name="sessionId"></param>
|
||
/// <param name="tokenId">Token Id（Web端传null或Guid.Empty）</param>
|
||
/// <param name="cancellationToken"></param>
|
||
/// <returns></returns>
|
||
public async Task CompleteChatForStatisticsAsync(HttpContext httpContext,
|
||
ThorChatCompletionsRequest request,
|
||
Guid? userId = null,
|
||
Guid? sessionId = null,
|
||
Guid? tokenId = null,
|
||
CancellationToken cancellationToken = default)
|
||
{
|
||
_specialCompatible.Compatible(request);
|
||
var response = httpContext.Response;
|
||
// 设置响应头，声明是 json
|
||
//response.ContentType = "application/json; charset=UTF-8";
|
||
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.OpenAi, request.Model);
|
||
var chatService =
|
||
LazyServiceProvider.GetRequiredKeyedService<IChatCompletionService>(modelDescribe.HandlerName);
|
||
var data = await chatService.CompleteChatAsync(modelDescribe, request, cancellationToken);
|
||
data.SupplementalMultiplier(modelDescribe.Multiplier);
|
||
if (userId is not null)
|
||
{
|
||
await _aiMessageManager.CreateUserMessageAsync(userId.Value, sessionId,
|
||
new MessageInputDto
|
||
{
|
||
Content = sessionId is null ? "不予存储" : request.Messages?.LastOrDefault().Content ?? string.Empty,
|
||
ModelId = request.Model,
|
||
TokenUsage = data.Usage,
|
||
}, tokenId);
|
||
|
||
await _aiMessageManager.CreateSystemMessageAsync(userId.Value, sessionId,
|
||
new MessageInputDto
|
||
{
|
||
Content =
|
||
sessionId is null ? "不予存储" : data.Choices?.FirstOrDefault()?.Delta.Content ?? string.Empty,
|
||
ModelId = request.Model,
|
||
TokenUsage = data.Usage
|
||
}, tokenId);
|
||
|
||
await _usageStatisticsManager.SetUsageAsync(userId.Value, request.Model, data.Usage, tokenId);
|
||
|
||
// 扣减尊享token包用量
|
||
if (PremiumPackageConst.ModeIds.Contains(request.Model))
|
||
{
|
||
var totalTokens = data.Usage?.TotalTokens ?? 0;
|
||
if (totalTokens > 0)
|
||
{
|
||
await PremiumPackageManager.TryConsumeTokensAsync(userId.Value, totalTokens);
|
||
}
|
||
}
|
||
}
|
||
|
||
await response.WriteAsJsonAsync(data, cancellationToken);
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// 聊天完成-缓存处理
|
||
/// </summary>
|
||
/// <param name="httpContext"></param>
|
||
/// <param name="request"></param>
|
||
/// <param name="userId"></param>
|
||
/// <param name="sessionId"></param>
|
||
/// <param name="tokenId">Token Id（Web端传null或Guid.Empty）</param>
|
||
/// <param name="cancellationToken"></param>
|
||
/// <returns></returns>
|
||
public async Task CompleteChatStreamForStatisticsAsync(
|
||
HttpContext httpContext,
|
||
ThorChatCompletionsRequest request,
|
||
Guid? userId = null,
|
||
Guid? sessionId = null,
|
||
Guid? tokenId = null,
|
||
CancellationToken cancellationToken = default)
|
||
{
|
||
var response = httpContext.Response;
|
||
// 设置响应头，声明是 SSE 流
|
||
response.ContentType = "text/event-stream;charset=utf-8;";
|
||
response.Headers.TryAdd("Cache-Control", "no-cache");
|
||
response.Headers.TryAdd("Connection", "keep-alive");
|
||
|
||
|
||
_specialCompatible.Compatible(request);
|
||
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.OpenAi, request.Model);
|
||
var chatService =
|
||
LazyServiceProvider.GetRequiredKeyedService<IChatCompletionService>(modelDescribe.HandlerName);
|
||
|
||
var completeChatResponse = chatService.CompleteChatStreamAsync(modelDescribe,request, cancellationToken);
|
||
var tokenUsage = new ThorUsageResponse();
|
||
|
||
//缓存队列算法
|
||
// 创建一个队列来缓存消息
|
||
var messageQueue = new ConcurrentQueue<string>();
|
||
|
||
StringBuilder backupSystemContent = new StringBuilder();
|
||
// 设置输出速率（例如每50毫秒输出一次）
|
||
var outputInterval = TimeSpan.FromMilliseconds(75);
|
||
// 标记是否完成接收
|
||
var isComplete = false;
|
||
// 启动一个后台任务来消费队列
|
||
var outputTask = Task.Run(async () =>
|
||
{
|
||
while (!(isComplete && messageQueue.IsEmpty))
|
||
{
|
||
if (messageQueue.TryDequeue(out var message))
|
||
{
|
||
await response.WriteAsync(message, Encoding.UTF8, cancellationToken).ConfigureAwait(false);
|
||
await response.Body.FlushAsync(cancellationToken).ConfigureAwait(false);
|
||
}
|
||
|
||
if (!isComplete)
|
||
{
|
||
// 如果没有完成，才等待，已完成，全部输出
|
||
await Task.Delay(outputInterval, cancellationToken).ConfigureAwait(false);
|
||
}
|
||
else
|
||
{
|
||
//已经完成了，也等待，但是速度可以放快
|
||
await Task.Delay(10, cancellationToken).ConfigureAwait(false);
|
||
}
|
||
}
|
||
}, cancellationToken);
|
||
|
||
|
||
//IAsyncEnumerable 只能在最外层捕获异常（如果你有其他办法的话...）
|
||
try
|
||
{
|
||
await foreach (var data in completeChatResponse)
|
||
{
|
||
data.SupplementalMultiplier(modelDescribe.Multiplier);
|
||
if (data.Usage is not null && (data.Usage.CompletionTokens > 0 || data.Usage.OutputTokens > 0))
|
||
{
|
||
tokenUsage = data.Usage;
|
||
}
|
||
|
||
var message = System.Text.Json.JsonSerializer.Serialize(data, ThorJsonSerializer.DefaultOptions);
|
||
backupSystemContent.Append(data.Choices.FirstOrDefault()?.Delta.Content);
|
||
// 将消息加入队列而不是直接写入
|
||
messageQueue.Enqueue($"data: {message}\n\n");
|
||
}
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
_logger.LogError(e, $"Ai对话异常");
|
||
var errorContent = $"对话Ai异常，异常信息：\n当前Ai模型：{request.Model}\n异常信息：{e.Message}\n异常堆栈：{e}";
|
||
var model = new ThorChatCompletionsResponse()
|
||
{
|
||
Choices = new List<ThorChatChoiceResponse>()
|
||
{
|
||
new ThorChatChoiceResponse()
|
||
{
|
||
Delta = new ThorChatMessage()
|
||
{
|
||
Content = errorContent
|
||
}
|
||
}
|
||
}
|
||
};
|
||
var message = JsonConvert.SerializeObject(model, new JsonSerializerSettings
|
||
{
|
||
ContractResolver = new CamelCasePropertyNamesContractResolver()
|
||
});
|
||
backupSystemContent.Append(errorContent);
|
||
messageQueue.Enqueue($"data: {message}\n\n");
|
||
}
|
||
|
||
//断开连接
|
||
messageQueue.Enqueue("data: [DONE]\n\n");
|
||
// 标记完成并发送结束标记
|
||
isComplete = true;
|
||
|
||
await outputTask;
|
||
|
||
|
||
await _aiMessageManager.CreateUserMessageAsync(userId, sessionId,
|
||
new MessageInputDto
|
||
{
|
||
Content = sessionId is null ? "不予存储" : request.Messages?.LastOrDefault()?.MessagesStore ?? string.Empty,
|
||
ModelId = request.Model,
|
||
TokenUsage = tokenUsage,
|
||
}, tokenId);
|
||
|
||
await _aiMessageManager.CreateSystemMessageAsync(userId, sessionId,
|
||
new MessageInputDto
|
||
{
|
||
Content = sessionId is null ? "不予存储" : backupSystemContent.ToString(),
|
||
ModelId = request.Model,
|
||
TokenUsage = tokenUsage
|
||
}, tokenId);
|
||
|
||
await _usageStatisticsManager.SetUsageAsync(userId, request.Model, tokenUsage, tokenId);
|
||
|
||
// 扣减尊享token包用量
|
||
if (userId is not null && PremiumPackageConst.ModeIds.Contains(request.Model))
|
||
{
|
||
var totalTokens = tokenUsage.TotalTokens ?? 0;
|
||
if (totalTokens > 0)
|
||
{
|
||
await PremiumPackageManager.TryConsumeTokensAsync(userId.Value, totalTokens);
|
||
}
|
||
}
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// 图片生成
|
||
/// </summary>
|
||
/// <param name="context"></param>
|
||
/// <param name="userId"></param>
|
||
/// <param name="sessionId"></param>
|
||
/// <param name="request"></param>
|
||
/// <param name="tokenId">Token Id（Web端传null或Guid.Empty）</param>
|
||
/// <exception cref="BusinessException"></exception>
|
||
/// <exception cref="Exception"></exception>
|
||
public async Task CreateImageForStatisticsAsync(HttpContext context, Guid? userId, Guid? sessionId,
|
||
ImageCreateRequest request, Guid? tokenId = null)
|
||
{
|
||
try
|
||
{
|
||
var model = request.Model;
|
||
if (string.IsNullOrEmpty(model)) model = "dall-e-2";
|
||
|
||
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.OpenAi, model);
|
||
|
||
// 获取渠道指定的实现类型的服务
|
||
var imageService =
|
||
LazyServiceProvider.GetRequiredKeyedService<IImageService>(modelDescribe.HandlerName);
|
||
|
||
var response = await imageService.CreateImage(request, modelDescribe);
|
||
|
||
if (response.Error != null || response.Results.Count == 0)
|
||
{
|
||
throw new BusinessException(response.Error?.Message ?? "图片生成失败", response.Error?.Code?.ToString());
|
||
}
|
||
|
||
await context.Response.WriteAsJsonAsync(response);
|
||
|
||
await _aiMessageManager.CreateUserMessageAsync(userId, sessionId,
|
||
new MessageInputDto
|
||
{
|
||
Content = sessionId is null ? "不予存储" : request.Prompt,
|
||
ModelId = model,
|
||
TokenUsage = response.Usage,
|
||
}, tokenId);
|
||
|
||
await _aiMessageManager.CreateSystemMessageAsync(userId, sessionId,
|
||
new MessageInputDto
|
||
{
|
||
Content = sessionId is null ? "不予存储" : response.Results?.FirstOrDefault()?.Url,
|
||
ModelId = model,
|
||
TokenUsage = response.Usage
|
||
}, tokenId);
|
||
|
||
await _usageStatisticsManager.SetUsageAsync(userId, model, response.Usage, tokenId);
|
||
|
||
// 扣减尊享token包用量
|
||
if (userId is not null && PremiumPackageConst.ModeIds.Contains(request.Model))
|
||
{
|
||
var totalTokens = response.Usage.TotalTokens ?? 0;
|
||
if (totalTokens > 0)
|
||
{
|
||
await PremiumPackageManager.TryConsumeTokensAsync(userId.Value, totalTokens);
|
||
}
|
||
}
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
var errorContent = $"图片生成Ai异常，异常信息：\n当前Ai模型：{request.Model}\n异常信息：{e.Message}\n异常堆栈：{e}";
|
||
throw new UserFriendlyException(errorContent);
|
||
}
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// 向量生成
|
||
/// </summary>
|
||
/// <param name="context"></param>
|
||
/// <param name="userId"></param>
|
||
/// <param name="sessionId"></param>
|
||
/// <param name="input"></param>
|
||
/// <param name="tokenId">Token Id（Web端传null或Guid.Empty）</param>
|
||
/// <exception cref="Exception"></exception>
|
||
/// <exception cref="BusinessException"></exception>
|
||
public async Task EmbeddingForStatisticsAsync(HttpContext context, Guid? userId, Guid? sessionId,
|
||
ThorEmbeddingInput input, Guid? tokenId = null)
|
||
{
|
||
try
|
||
{
|
||
if (input == null) throw new Exception("模型校验异常");
|
||
|
||
using var embedding =
|
||
Activity.Current?.Source.StartActivity("向量模型调用");
|
||
|
||
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.OpenAi, input.Model);
|
||
|
||
// 获取渠道指定的实现类型的服务
|
||
var embeddingService =
|
||
LazyServiceProvider.GetRequiredKeyedService<ITextEmbeddingService>(modelDescribe.HandlerName);
|
||
|
||
var embeddingCreateRequest = new EmbeddingCreateRequest
|
||
{
|
||
Model = input.Model,
|
||
EncodingFormat = input.EncodingFormat
|
||
};
|
||
|
||
//dto进行转换，支持多种格式
|
||
if (input.Input is JsonElement str)
|
||
{
|
||
if (str.ValueKind == JsonValueKind.String)
|
||
{
|
||
embeddingCreateRequest.Input = str.ToString();
|
||
}
|
||
else if (str.ValueKind == JsonValueKind.Array)
|
||
{
|
||
var inputString = str.EnumerateArray().Select(x => x.ToString()).ToArray();
|
||
embeddingCreateRequest.InputAsList = inputString.ToList();
|
||
}
|
||
else
|
||
{
|
||
throw new Exception("Input，输入格式错误，非string或Array类型");
|
||
}
|
||
}
|
||
else if (input.Input is string strInput)
|
||
{
|
||
embeddingCreateRequest.Input = strInput;
|
||
}
|
||
else
|
||
{
|
||
throw new Exception("Input，输入格式错误，未找到类型");
|
||
}
|
||
|
||
|
||
var stream =
|
||
await embeddingService.EmbeddingAsync(embeddingCreateRequest, modelDescribe, context.RequestAborted);
|
||
|
||
var usage = new ThorUsageResponse()
|
||
{
|
||
PromptTokens = stream.Usage?.PromptTokens ?? 0,
|
||
InputTokens = stream.Usage?.InputTokens ?? 0,
|
||
CompletionTokens = 0,
|
||
TotalTokens = stream.Usage?.InputTokens ?? 0
|
||
};
|
||
await context.Response.WriteAsJsonAsync(new
|
||
{
|
||
input.Model,
|
||
stream.Data,
|
||
stream.Error,
|
||
Object = stream.ObjectTypeName,
|
||
Usage = usage
|
||
});
|
||
|
||
//知识库暂不使用message统计
|
||
// await _aiMessageManager.CreateUserMessageAsync(userId, sessionId,
|
||
// new MessageInputDto
|
||
// {
|
||
// Content = string.Empty,
|
||
// ModelId = input.Model,
|
||
// TokenUsage = usage,
|
||
// });
|
||
//
|
||
// await _aiMessageManager.CreateSystemMessageAsync(userId, sessionId,
|
||
// new MessageInputDto
|
||
// {
|
||
// Content = string.Empty,
|
||
// ModelId = input.Model,
|
||
// TokenUsage = usage
|
||
// });
|
||
|
||
await _usageStatisticsManager.SetUsageAsync(userId, input.Model, usage, tokenId);
|
||
}
|
||
catch (ThorRateLimitException)
|
||
{
|
||
context.Response.StatusCode = 429;
|
||
}
|
||
catch (UnauthorizedAccessException e)
|
||
{
|
||
context.Response.StatusCode = 401;
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
var errorContent = $"嵌入Ai异常，异常信息：\n当前Ai模型：{input.Model}\n异常信息：{e.Message}\n异常堆栈：{e}";
|
||
throw new UserFriendlyException(errorContent);
|
||
}
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// Anthropic聊天完成-非流式
|
||
/// </summary>
|
||
/// <param name="httpContext"></param>
|
||
/// <param name="request"></param>
|
||
/// <param name="userId"></param>
|
||
/// <param name="sessionId"></param>
|
||
/// <param name="tokenId">Token Id（Web端传null或Guid.Empty）</param>
|
||
/// <param name="cancellationToken"></param>
|
||
/// <returns></returns>
|
||
public async Task AnthropicCompleteChatForStatisticsAsync(HttpContext httpContext,
|
||
AnthropicInput request,
|
||
Guid? userId = null,
|
||
Guid? sessionId = null,
|
||
Guid? tokenId = null,
|
||
CancellationToken cancellationToken = default)
|
||
{
|
||
_specialCompatible.AnthropicCompatible(request);
|
||
var response = httpContext.Response;
|
||
// 设置响应头，声明是 json
|
||
//response.ContentType = "application/json; charset=UTF-8";
|
||
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.Claude, request.Model);
|
||
var chatService =
|
||
LazyServiceProvider.GetRequiredKeyedService<IAnthropicChatCompletionService>(modelDescribe.HandlerName);
|
||
var data = await chatService.ChatCompletionsAsync(modelDescribe, request, cancellationToken);
|
||
|
||
data.SupplementalMultiplier(modelDescribe.Multiplier);
|
||
|
||
if (userId is not null)
|
||
{
|
||
await _aiMessageManager.CreateUserMessageAsync(userId.Value, sessionId,
|
||
new MessageInputDto
|
||
{
|
||
Content = "不予存储",
|
||
ModelId = request.Model,
|
||
TokenUsage = data.TokenUsage,
|
||
}, tokenId);
|
||
|
||
await _aiMessageManager.CreateSystemMessageAsync(userId.Value, sessionId,
|
||
new MessageInputDto
|
||
{
|
||
Content = "不予存储",
|
||
ModelId = request.Model,
|
||
TokenUsage = data.TokenUsage
|
||
}, tokenId);
|
||
|
||
await _usageStatisticsManager.SetUsageAsync(userId.Value, request.Model, data.TokenUsage, tokenId);
|
||
|
||
// 扣减尊享token包用量
|
||
var totalTokens = data.TokenUsage.TotalTokens ?? 0;
|
||
if (totalTokens > 0)
|
||
{
|
||
await PremiumPackageManager.TryConsumeTokensAsync(userId.Value, totalTokens);
|
||
}
|
||
}
|
||
|
||
await response.WriteAsJsonAsync(data, cancellationToken);
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// Anthropic聊天完成-缓存处理
|
||
/// </summary>
|
||
/// <param name="httpContext"></param>
|
||
/// <param name="request"></param>
|
||
/// <param name="userId"></param>
|
||
/// <param name="sessionId"></param>
|
||
/// <param name="tokenId">Token Id（Web端传null或Guid.Empty）</param>
|
||
/// <param name="cancellationToken"></param>
|
||
/// <returns></returns>
|
||
public async Task AnthropicCompleteChatStreamForStatisticsAsync(
|
||
HttpContext httpContext,
|
||
AnthropicInput request,
|
||
Guid? userId = null,
|
||
Guid? sessionId = null,
|
||
Guid? tokenId = null,
|
||
CancellationToken cancellationToken = default)
|
||
{
|
||
var response = httpContext.Response;
|
||
// 设置响应头，声明是 SSE 流
|
||
response.ContentType = "text/event-stream;charset=utf-8;";
|
||
response.Headers.TryAdd("Cache-Control", "no-cache");
|
||
response.Headers.TryAdd("Connection", "keep-alive");
|
||
|
||
_specialCompatible.AnthropicCompatible(request);
|
||
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.Claude, request.Model);
|
||
var chatService =
|
||
LazyServiceProvider.GetRequiredKeyedService<IAnthropicChatCompletionService>(modelDescribe.HandlerName);
|
||
|
||
var completeChatResponse = chatService.StreamChatCompletionsAsync(modelDescribe,request, cancellationToken);
|
||
ThorUsageResponse? tokenUsage = null;
|
||
StringBuilder backupSystemContent = new StringBuilder();
|
||
try
|
||
{
|
||
await foreach (var responseResult in completeChatResponse)
|
||
{
|
||
responseResult.Item2.SupplementalMultiplier(modelDescribe.Multiplier);
|
||
//message_start是为了保底机制
|
||
if (responseResult.Item1.Contains("message_delta") || responseResult.Item1.Contains("message_start"))
|
||
{
|
||
tokenUsage = responseResult.Item2?.TokenUsage;
|
||
}
|
||
|
||
backupSystemContent.Append(responseResult.Item2?.Delta?.Text);
|
||
await WriteAsEventStreamDataAsync(httpContext, responseResult.Item1, responseResult.Item2,
|
||
cancellationToken);
|
||
}
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
_logger.LogError(e, $"Ai对话异常");
|
||
var errorContent = $"对话Ai异常，异常信息：\n当前Ai模型：{request.Model}\n异常信息：{e.Message}\n异常堆栈：{e}";
|
||
throw new UserFriendlyException(errorContent);
|
||
}
|
||
|
||
await _aiMessageManager.CreateUserMessageAsync(userId, sessionId,
|
||
new MessageInputDto
|
||
{
|
||
Content = "不予存储",
|
||
ModelId = request.Model,
|
||
TokenUsage = tokenUsage,
|
||
}, tokenId);
|
||
|
||
await _aiMessageManager.CreateSystemMessageAsync(userId, sessionId,
|
||
new MessageInputDto
|
||
{
|
||
Content = "不予存储",
|
||
ModelId = request.Model,
|
||
TokenUsage = tokenUsage
|
||
}, tokenId);
|
||
|
||
await _usageStatisticsManager.SetUsageAsync(userId, request.Model, tokenUsage, tokenId);
|
||
|
||
// 扣减尊享token包用量
|
||
if (userId.HasValue && tokenUsage is not null)
|
||
{
|
||
var totalTokens = tokenUsage.TotalTokens ?? 0;
|
||
if (tokenUsage.TotalTokens > 0)
|
||
{
|
||
await PremiumPackageManager.TryConsumeTokensAsync(userId.Value, totalTokens);
|
||
}
|
||
}
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// OpenAi 响应-非流式-缓存处理
|
||
/// </summary>
|
||
/// <param name="httpContext"></param>
|
||
/// <param name="request"></param>
|
||
/// <param name="userId"></param>
|
||
/// <param name="sessionId"></param>
|
||
/// <param name="tokenId"></param>
|
||
/// <param name="cancellationToken"></param>
|
||
public async Task OpenAiResponsesAsyncForStatisticsAsync(HttpContext httpContext,
|
||
OpenAiResponsesInput request,
|
||
Guid? userId = null,
|
||
Guid? sessionId = null,
|
||
Guid? tokenId = null,
|
||
CancellationToken cancellationToken = default)
|
||
{
|
||
// _specialCompatible.AnthropicCompatible(request);
|
||
var response = httpContext.Response;
|
||
// 设置响应头，声明是 json
|
||
//response.ContentType = "application/json; charset=UTF-8";
|
||
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.Response, request.Model);
|
||
|
||
var chatService =
|
||
LazyServiceProvider.GetRequiredKeyedService<IOpenAiResponseService>(modelDescribe.HandlerName);
|
||
var data = await chatService.ResponsesAsync(modelDescribe, request, cancellationToken);
|
||
|
||
data.SupplementalMultiplier(modelDescribe.Multiplier);
|
||
|
||
var tokenUsage= new ThorUsageResponse
|
||
{
|
||
InputTokens = data.Usage.InputTokens,
|
||
OutputTokens = data.Usage.OutputTokens,
|
||
TotalTokens = data.Usage.InputTokens + data.Usage.OutputTokens,
|
||
};
|
||
if (userId is not null)
|
||
{
|
||
await _aiMessageManager.CreateUserMessageAsync(userId.Value, sessionId,
|
||
new MessageInputDto
|
||
{
|
||
Content = "不予存储",
|
||
ModelId = request.Model,
|
||
TokenUsage = tokenUsage,
|
||
}, tokenId);
|
||
|
||
await _aiMessageManager.CreateSystemMessageAsync(userId.Value, sessionId,
|
||
new MessageInputDto
|
||
{
|
||
Content = "不予存储",
|
||
ModelId = request.Model,
|
||
TokenUsage = tokenUsage
|
||
}, tokenId);
|
||
|
||
await _usageStatisticsManager.SetUsageAsync(userId.Value, request.Model, tokenUsage, tokenId);
|
||
|
||
// 扣减尊享token包用量
|
||
var totalTokens = tokenUsage.TotalTokens ?? 0;
|
||
if (totalTokens > 0)
|
||
{
|
||
await PremiumPackageManager.TryConsumeTokensAsync(userId.Value, totalTokens);
|
||
}
|
||
}
|
||
|
||
await response.WriteAsJsonAsync(data, cancellationToken);
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// OpenAi响应-流式-缓存处理
|
||
/// </summary>
|
||
/// <param name="httpContext"></param>
|
||
/// <param name="request"></param>
|
||
/// <param name="userId"></param>
|
||
/// <param name="sessionId"></param>
|
||
/// <param name="tokenId">Token Id（Web端传null或Guid.Empty）</param>
|
||
/// <param name="cancellationToken"></param>
|
||
/// <returns></returns>
|
||
public async Task OpenAiResponsesStreamForStatisticsAsync(
|
||
HttpContext httpContext,
|
||
OpenAiResponsesInput request,
|
||
Guid? userId = null,
|
||
Guid? sessionId = null,
|
||
Guid? tokenId = null,
|
||
CancellationToken cancellationToken = default)
|
||
{
|
||
var response = httpContext.Response;
|
||
// 设置响应头，声明是 SSE 流
|
||
response.ContentType = "text/event-stream;charset=utf-8;";
|
||
response.Headers.TryAdd("Cache-Control", "no-cache");
|
||
response.Headers.TryAdd("Connection", "keep-alive");
|
||
|
||
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.Response, request.Model);
|
||
var chatService =
|
||
LazyServiceProvider.GetRequiredKeyedService<IOpenAiResponseService>(modelDescribe.HandlerName);
|
||
|
||
var completeChatResponse = chatService.ResponsesStreamAsync(modelDescribe,request, cancellationToken);
|
||
ThorUsageResponse? tokenUsage = null;
|
||
try
|
||
{
|
||
await foreach (var responseResult in completeChatResponse)
|
||
{
|
||
//message_start是为了保底机制
|
||
if (responseResult.Item1.Contains("response.completed"))
|
||
{
|
||
var obj = responseResult.Item2!.Value;
|
||
int inputTokens = obj.GetPath("response","usage","input_tokens").GetInt();
|
||
int outputTokens = obj.GetPath("response","usage","output_tokens").GetInt();
|
||
inputTokens=Convert.ToInt32(inputTokens * modelDescribe.Multiplier);
|
||
outputTokens=Convert.ToInt32(outputTokens * modelDescribe.Multiplier);
|
||
tokenUsage = new ThorUsageResponse
|
||
{
|
||
PromptTokens =inputTokens,
|
||
InputTokens = inputTokens,
|
||
OutputTokens = outputTokens,
|
||
CompletionTokens = outputTokens,
|
||
TotalTokens = inputTokens+outputTokens,
|
||
};
|
||
}
|
||
|
||
await WriteAsEventStreamDataAsync(httpContext, responseResult.Item1, responseResult.Item2,
|
||
cancellationToken);
|
||
}
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
_logger.LogError(e, $"Ai响应异常");
|
||
var errorContent = $"响应Ai异常，异常信息：\n当前Ai模型：{request.Model}\n异常信息：{e.Message}\n异常堆栈：{e}";
|
||
throw new UserFriendlyException(errorContent);
|
||
}
|
||
|
||
await _aiMessageManager.CreateUserMessageAsync(userId, sessionId,
|
||
new MessageInputDto
|
||
{
|
||
Content = "不予存储" ,
|
||
ModelId = request.Model,
|
||
TokenUsage = tokenUsage,
|
||
}, tokenId);
|
||
|
||
await _aiMessageManager.CreateSystemMessageAsync(userId, sessionId,
|
||
new MessageInputDto
|
||
{
|
||
Content = "不予存储" ,
|
||
ModelId = request.Model,
|
||
TokenUsage = tokenUsage
|
||
}, tokenId);
|
||
|
||
await _usageStatisticsManager.SetUsageAsync(userId, request.Model, tokenUsage, tokenId);
|
||
|
||
// 扣减尊享token包用量
|
||
if (userId.HasValue && tokenUsage is not null)
|
||
{
|
||
var totalTokens = tokenUsage.TotalTokens ?? 0;
|
||
if (tokenUsage.TotalTokens > 0)
|
||
{
|
||
await PremiumPackageManager.TryConsumeTokensAsync(userId.Value, totalTokens);
|
||
}
|
||
}
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// Gemini 生成-非流式-缓存处理
|
||
/// </summary>
|
||
/// <param name="httpContext"></param>
|
||
/// <param name="modelId"></param>
|
||
/// <param name="request"></param>
|
||
/// <param name="userId"></param>
|
||
/// <param name="sessionId"></param>
|
||
/// <param name="tokenId"></param>
|
||
/// <param name="cancellationToken"></param>
|
||
public async Task GeminiGenerateContentAsyncForStatisticsAsync(HttpContext httpContext,
|
||
string modelId,
|
||
JsonElement request,
|
||
Guid? userId = null,
|
||
Guid? sessionId = null,
|
||
Guid? tokenId = null,
|
||
CancellationToken cancellationToken = default)
|
||
{
|
||
var response = httpContext.Response;
|
||
var modelDescribe = await GetModelAsync(ModelApiTypeEnum.GenerateContent, modelId);
|
||
|
||
var chatService =
|
||
LazyServiceProvider.GetRequiredKeyedService<IGeminiGenerateContentService>(modelDescribe.HandlerName);
|
||
var data = await chatService.GenerateContentAsync(modelDescribe, request, cancellationToken);
|
||
|
||
var tokenUsage= GeminiGenerateContentAcquirer.GetUsage(data);
|
||
tokenUsage.SetSupplementalMultiplier(modelDescribe.Multiplier);
|
||
|
||
if (userId is not null)
|
||
{
|
||
await _aiMessageManager.CreateUserMessageAsync(userId.Value, sessionId,
|
||
new MessageInputDto
|
||
{
|
||
Content = "不予存储",
|
||
ModelId = modelId,
|
||
TokenUsage = tokenUsage,
|
||
}, tokenId);
|
||
|
||
await _aiMessageManager.CreateSystemMessageAsync(userId.Value, sessionId,
|
||
new MessageInputDto
|
||
{
|
||
Content = "不予存储",
|
||
ModelId = modelId,
|
||
TokenUsage = tokenUsage
|
||
}, tokenId);
|
||
|
||
await _usageStatisticsManager.SetUsageAsync(userId.Value, modelId, tokenUsage, tokenId);
|
||
|
||
// 扣减尊享token包用量
|
||
var totalTokens = tokenUsage.TotalTokens ?? 0;
|
||
if (totalTokens > 0)
|
||
{
|
||
await PremiumPackageManager.TryConsumeTokensAsync(userId.Value, totalTokens);
|
||
}
|
||
}
|
||
|
||
await response.WriteAsJsonAsync(data, cancellationToken);
|
||
}
|
||
|
||
|
||
|
||
|
||
|
||
|
||
#region 流式传输格式Http响应
|
||
|
||
private static readonly byte[] EventPrefix = "event: "u8.ToArray();
|
||
private static readonly byte[] DataPrefix = "data: "u8.ToArray();
|
||
private static readonly byte[] NewLine = "\n"u8.ToArray();
|
||
private static readonly byte[] DoubleNewLine = "\n\n"u8.ToArray();
|
||
|
||
/// <summary>
|
||
/// 使用 JsonSerializer.SerializeAsync 直接序列化到响应流
|
||
/// </summary>
|
||
private static async ValueTask WriteAsEventStreamDataAsync<T>(
|
||
HttpContext context,
|
||
string @event,
|
||
T value,
|
||
CancellationToken cancellationToken = default)
|
||
{
|
||
var response = context.Response;
|
||
var bodyStream = response.Body;
|
||
// 确保 SSE Header 已经设置好
|
||
// e.g. Content-Type: text/event-stream; charset=utf-8
|
||
await response.StartAsync(cancellationToken).ConfigureAwait(false);
|
||
// 写事件类型
|
||
await bodyStream.WriteAsync(EventPrefix, cancellationToken).ConfigureAwait(false);
|
||
await WriteUtf8StringAsync(bodyStream, @event.Trim(), cancellationToken).ConfigureAwait(false);
|
||
await bodyStream.WriteAsync(NewLine, cancellationToken).ConfigureAwait(false);
|
||
// 写 data: + JSON
|
||
await bodyStream.WriteAsync(DataPrefix, cancellationToken).ConfigureAwait(false);
|
||
await JsonSerializer.SerializeAsync(
|
||
bodyStream,
|
||
value,
|
||
ThorJsonSerializer.DefaultOptions,
|
||
cancellationToken
|
||
).ConfigureAwait(false);
|
||
// 事件结束 \n\n
|
||
await bodyStream.WriteAsync(DoubleNewLine, cancellationToken).ConfigureAwait(false);
|
||
// 及时把数据发送给客户端
|
||
await bodyStream.FlushAsync(cancellationToken).ConfigureAwait(false);
|
||
}
|
||
|
||
|
||
private static async ValueTask WriteUtf8StringAsync(Stream stream, string value, CancellationToken token)
|
||
{
|
||
if (string.IsNullOrEmpty(value))
|
||
return;
|
||
var buffer = Encoding.UTF8.GetBytes(value);
|
||
await stream.WriteAsync(buffer, token).ConfigureAwait(false);
|
||
}
|
||
|
||
#endregion
|
||
} |