273 lines
9.7 KiB
C#
273 lines
9.7 KiB
C#
using System.Collections.Concurrent;
|
||
using System.Reflection;
|
||
using System.Text;
|
||
using System.Text.Encodings.Web;
|
||
using Dm.util;
|
||
using Microsoft.AspNetCore.Authorization;
|
||
using Microsoft.AspNetCore.Http;
|
||
using Microsoft.AspNetCore.Mvc;
|
||
using Microsoft.Extensions.DependencyInjection;
|
||
using Microsoft.Extensions.Logging;
|
||
using Microsoft.Extensions.Options;
|
||
using ModelContextProtocol;
|
||
using ModelContextProtocol.Server;
|
||
using Newtonsoft.Json;
|
||
using Newtonsoft.Json.Serialization;
|
||
using OpenAI.Chat;
|
||
using Volo.Abp.Application.Services;
|
||
using Volo.Abp.Users;
|
||
using Yi.Framework.AiHub.Application.Contracts.Dtos;
|
||
using Yi.Framework.AiHub.Application.Contracts.Dtos.Chat;
|
||
using Yi.Framework.AiHub.Domain;
|
||
using Yi.Framework.AiHub.Domain.Entities;
|
||
using Yi.Framework.AiHub.Domain.Entities.Chat;
|
||
using Yi.Framework.AiHub.Domain.Entities.Model;
|
||
using Yi.Framework.AiHub.Domain.Extensions;
|
||
using Yi.Framework.AiHub.Domain.Managers;
|
||
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
||
using Yi.Framework.AiHub.Domain.Shared.Dtos;
|
||
using Yi.Framework.AiHub.Domain.Shared.Dtos.OpenAi;
|
||
using Yi.Framework.AiHub.Domain.Shared.Enums;
|
||
using Yi.Framework.Rbac.Application.Contracts.IServices;
|
||
using Yi.Framework.Rbac.Domain.Shared.Dtos;
|
||
using Yi.Framework.SqlSugarCore.Abstractions;
|
||
|
||
namespace Yi.Framework.AiHub.Application.Services;
|
||
|
||
/// <summary>
|
||
/// ai服务
|
||
/// </summary>
|
||
public class AiChatService : ApplicationService
|
||
{
|
||
private readonly IHttpContextAccessor _httpContextAccessor;
|
||
private readonly AiBlacklistManager _aiBlacklistManager;
|
||
private readonly ILogger<AiChatService> _logger;
|
||
private readonly AiGateWayManager _aiGateWayManager;
|
||
private readonly ModelManager _modelManager;
|
||
private readonly PremiumPackageManager _premiumPackageManager;
|
||
private readonly ChatManager _chatManager;
|
||
private readonly TokenManager _tokenManager;
|
||
private readonly IAccountService _accountService;
|
||
private readonly ISqlSugarRepository<AgentStoreAggregateRoot> _agentStoreRepository;
|
||
private readonly ISqlSugarRepository<AiModelEntity> _aiModelRepository;
|
||
private const string FreeModelId = "DeepSeek-V3-0324";
|
||
|
||
public AiChatService(IHttpContextAccessor httpContextAccessor,
|
||
AiBlacklistManager aiBlacklistManager,
|
||
ILogger<AiChatService> logger,
|
||
AiGateWayManager aiGateWayManager,
|
||
ModelManager modelManager,
|
||
PremiumPackageManager premiumPackageManager,
|
||
ChatManager chatManager, TokenManager tokenManager, IAccountService accountService,
|
||
ISqlSugarRepository<AgentStoreAggregateRoot> agentStoreRepository,
|
||
ISqlSugarRepository<AiModelEntity> aiModelRepository)
|
||
{
|
||
_httpContextAccessor = httpContextAccessor;
|
||
_aiBlacklistManager = aiBlacklistManager;
|
||
_logger = logger;
|
||
_aiGateWayManager = aiGateWayManager;
|
||
_modelManager = modelManager;
|
||
_premiumPackageManager = premiumPackageManager;
|
||
_chatManager = chatManager;
|
||
_tokenManager = tokenManager;
|
||
_accountService = accountService;
|
||
_agentStoreRepository = agentStoreRepository;
|
||
_aiModelRepository = aiModelRepository;
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// 查询已登录的账户信息
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
[Route("ai-chat/account")]
|
||
[Authorize]
|
||
public async Task<UserRoleMenuDto> GetAsync()
|
||
{
|
||
var accountService = LazyServiceProvider.GetRequiredService<IAccountService>();
|
||
var output = await accountService.GetAsync();
|
||
return output;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取对话模型列表
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
public async Task<List<ModelGetListOutput>> GetModelAsync()
|
||
{
|
||
var output = await _aiModelRepository._DbQueryable
|
||
.Where(x => x.IsEnabled == true)
|
||
.Where(x => x.ModelType == ModelTypeEnum.Chat)
|
||
.Where(x => x.ModelApiType == ModelApiTypeEnum.Completions)
|
||
.OrderByDescending(x => x.OrderNum)
|
||
.Select(x => new ModelGetListOutput
|
||
{
|
||
Id = x.Id,
|
||
ModelId = x.ModelId,
|
||
ModelName = x.Name,
|
||
ModelDescribe = x.Description,
|
||
Remark = x.Description,
|
||
IsPremiumPackage = x.IsPremium,
|
||
ModelApiType = x.ModelApiType,
|
||
}).ToListAsync();
|
||
|
||
output.ForEach(x =>
|
||
{
|
||
if (x.ModelId == FreeModelId)
|
||
{
|
||
x.IsPremiumPackage = false;
|
||
x.IsFree = true;
|
||
}
|
||
});
|
||
|
||
return output;
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// 发送消息
|
||
/// </summary>
|
||
/// <param name="input"></param>
|
||
/// <param name="sessionId"></param>
|
||
/// <param name="cancellationToken"></param>
|
||
[HttpPost("ai-chat/send")]
|
||
public async Task PostSendAsync([FromBody] ThorChatCompletionsRequest input, [FromQuery] Guid? sessionId,
|
||
CancellationToken cancellationToken)
|
||
{
|
||
//除了免费模型，其他的模型都要校验
|
||
if (input.Model!=FreeModelId)
|
||
{
|
||
//有token，需要黑名单校验
|
||
if (CurrentUser.IsAuthenticated)
|
||
{
|
||
await _aiBlacklistManager.VerifiyAiBlacklist(CurrentUser.GetId());
|
||
if (!CurrentUser.IsAiVip())
|
||
{
|
||
throw new UserFriendlyException("该模型需要VIP用户才能使用，请购买VIP后重新登录重试");
|
||
}
|
||
}
|
||
else
|
||
{
|
||
throw new UserFriendlyException("未登录用户，只能使用未加速的DeepSeek-R1，请登录后重试");
|
||
}
|
||
}
|
||
|
||
//如果是尊享包服务，需要校验是是否尊享包足够
|
||
if (CurrentUser.IsAuthenticated)
|
||
{
|
||
var isPremium = await _modelManager.IsPremiumModelAsync(input.Model);
|
||
|
||
if (isPremium)
|
||
{
|
||
// 检查尊享token包用量
|
||
var availableTokens = await _premiumPackageManager.GetAvailableTokensAsync(CurrentUser.GetId());
|
||
if (availableTokens <= 0)
|
||
{
|
||
throw new UserFriendlyException("尊享token包用量不足，请先购买尊享token包");
|
||
}
|
||
}
|
||
}
|
||
|
||
//ai网关代理httpcontext
|
||
await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
||
CurrentUser.Id, sessionId, null, CancellationToken.None);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 发送消息
|
||
/// </summary>
|
||
/// <param name="input"></param>
|
||
/// <param name="cancellationToken"></param>
|
||
[HttpPost("ai-chat/FileMaster/send")]
|
||
public async Task PostFileMasterSendAsync([FromBody] ThorChatCompletionsRequest input,
|
||
CancellationToken cancellationToken)
|
||
{
|
||
if (!string.IsNullOrWhiteSpace(input.Model))
|
||
{
|
||
throw new BusinessException("当前接口不支持第三方使用");
|
||
}
|
||
input.Model = "gpt-5-chat";
|
||
if (CurrentUser.IsAuthenticated)
|
||
{
|
||
await _aiBlacklistManager.VerifiyAiBlacklist(CurrentUser.GetId());
|
||
}
|
||
|
||
//ai网关代理httpcontext
|
||
await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
|
||
CurrentUser.Id, null, null, CancellationToken.None);
|
||
}
|
||
|
||
|
||
/// <summary>
|
||
/// Agent 发送消息
|
||
/// </summary>
|
||
[HttpPost("ai-chat/agent/send")]
|
||
public async Task PostAgentSendAsync([FromBody] AgentSendInput input, CancellationToken cancellationToken)
|
||
{
|
||
var tokenValidation = await _tokenManager.ValidateTokenAsync(input.TokenId, input.ModelId);
|
||
|
||
await _aiBlacklistManager.VerifiyAiBlacklist(tokenValidation.UserId);
|
||
// 验证用户是否为VIP
|
||
var userInfo = await _accountService.GetAsync(null, null, tokenValidation.UserId);
|
||
if (userInfo == null)
|
||
{
|
||
throw new UserFriendlyException("用户信息不存在");
|
||
}
|
||
|
||
// 检查是否为VIP（使用RoleCodes判断）
|
||
if (!userInfo.RoleCodes.Contains(AiHubConst.VipRole) && userInfo.User.UserName != "cc")
|
||
{
|
||
throw new UserFriendlyException("该接口为尊享服务专用，需要VIP权限才能使用");
|
||
}
|
||
|
||
//如果是尊享包服务，需要校验是是否尊享包足够
|
||
var isPremium = await _modelManager.IsPremiumModelAsync(input.ModelId);
|
||
|
||
if (isPremium)
|
||
{
|
||
// 检查尊享token包用量
|
||
var availableTokens = await _premiumPackageManager.GetAvailableTokensAsync(tokenValidation.UserId);
|
||
if (availableTokens <= 0)
|
||
{
|
||
throw new UserFriendlyException("尊享token包用量不足，请先购买尊享token包");
|
||
}
|
||
}
|
||
|
||
await _chatManager.AgentCompleteChatStreamAsync(_httpContextAccessor.HttpContext,
|
||
input.SessionId,
|
||
input.Content,
|
||
tokenValidation.Token,
|
||
tokenValidation.TokenId,
|
||
input.ModelId,
|
||
tokenValidation.UserId,
|
||
input.Tools,
|
||
CancellationToken.None);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取 Agent 工具
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
[HttpPost("ai-chat/agent/tool")]
|
||
public List<AgentToolOutput> GetAgentToolAsync()
|
||
{
|
||
var agentTools = _chatManager.GetTools().Select(x => new AgentToolOutput
|
||
{
|
||
Code = x.Code,
|
||
Name = x.Name
|
||
}).ToList();
|
||
return agentTools;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取 Agent 上下文
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
[HttpPost("ai-chat/agent/context/{sessionId}")]
|
||
[Authorize]
|
||
public async Task<string?> GetAgentContextAsync([FromRoute] Guid sessionId)
|
||
{
|
||
var data = await _agentStoreRepository.GetFirstAsync(x => x.SessionId == sessionId);
|
||
return data?.Store;
|
||
}
|
||
} |