feat: 支持尊享包用量统计

2025-11-10 15:18:05 +08:00
parent b7756e2112
commit 1b4c3cbb8d
7 changed files with 75 additions and 1 deletions
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/AiChatService.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/AiChatService.cs
@@ -0,0 +1,176 @@
+using System.Collections.Concurrent;
+using System.Text;
+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 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.Domain.Entities;
+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 ISqlSugarRepository<AiModelEntity> _aiModelRepository;
+    private readonly AiBlacklistManager _aiBlacklistManager;
+    private readonly ILogger<AiChatService> _logger;
+    private readonly AiGateWayManager _aiGateWayManager;
+    private readonly PremiumPackageManager _premiumPackageManager;
+
+    public AiChatService(IHttpContextAccessor httpContextAccessor,
+        AiBlacklistManager aiBlacklistManager,
+        ISqlSugarRepository<AiModelEntity> aiModelRepository,
+        ILogger<AiChatService> logger, AiGateWayManager aiGateWayManager, PremiumPackageManager premiumPackageManager)
+    {
+        _httpContextAccessor = httpContextAccessor;
+        _aiBlacklistManager = aiBlacklistManager;
+        _aiModelRepository = aiModelRepository;
+        _logger = logger;
+        _aiGateWayManager = aiGateWayManager;
+        _premiumPackageManager = premiumPackageManager;
+    }
+
+
+    /// <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.ModelType == ModelTypeEnum.Chat)
+            .OrderByDescending(x => x.OrderNum)
+            .Select(x => new ModelGetListOutput
+            {
+                Id = x.Id,
+                Category = "chat",
+                ModelId = x.ModelId,
+                ModelName = x.Name,
+                ModelDescribe = x.Description,
+                ModelPrice = 0,
+                ModelType = "1",
+                ModelShow = "0",
+                SystemPrompt = null,
+                ApiHost = null,
+                ApiKey = null,
+                Remark = x.Description
+            }).ToListAsync();
+        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.Contains("DeepSeek-R1"))
+        {
+            //有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 && PremiumPackageConst.ModeIds.Contains(input.Model))
+        {
+            // 检查尊享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, cancellationToken);
+    }
+
+
+    /// <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("当前接口不支持第三方使用");
+        }
+
+        if (CurrentUser.IsAuthenticated)
+        {
+            await _aiBlacklistManager.VerifiyAiBlacklist(CurrentUser.GetId());
+            if (CurrentUser.IsAiVip())
+            {
+                input.Model = "gpt-5-chat";
+            }
+            else
+            {
+                input.Model = "gpt-4.1-mini";
+            }
+        }
+        else
+        {
+            input.Model = "DeepSeek-R1-0528";
+        }
+
+        //ai网关代理httpcontext
+        await _aiGateWayManager.CompleteChatStreamForStatisticsAsync(_httpContextAccessor.HttpContext, input,
+            CurrentUser.Id, null, cancellationToken);
+    }
+}
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/MessageService.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/MessageService.cs
@@ -0,0 +1,42 @@
+using Mapster;
+using Microsoft.AspNetCore.Authorization;
+using Microsoft.AspNetCore.Mvc;
+using SqlSugar;
+using Volo.Abp.Application.Dtos;
+using Volo.Abp.Application.Services;
+using Volo.Abp.Users;
+using Yi.Framework.AiHub.Application.Contracts.Dtos;
+using Yi.Framework.AiHub.Domain.Entities;
+using Yi.Framework.AiHub.Domain.Entities.Chat;
+using Yi.Framework.SqlSugarCore.Abstractions;
+
+namespace Yi.Framework.AiHub.Application.Services;
+
+public class MessageService : ApplicationService
+{
+    private readonly ISqlSugarRepository<MessageAggregateRoot> _repository;
+
+    public MessageService(ISqlSugarRepository<MessageAggregateRoot> repository)
+    {
+        _repository = repository;
+    }
+
+    /// <summary>
+    /// 查询消息
+    /// 需要会话id
+    /// </summary>
+    /// <param name="input"></param>
+    /// <returns></returns>
+    [Authorize]
+    public async Task<PagedResultDto<MessageDto>> GetListAsync([FromQuery]MessageGetListInput input)
+    {
+        RefAsync<int> total = 0;
+        var userId = CurrentUser.GetId();
+        var entities = await _repository._DbQueryable
+            .Where(x => x.SessionId == input.SessionId)
+            .Where(x=>x.UserId == userId)
+            .OrderBy(x => x.Id)
+            .ToPageListAsync(input.SkipCount, input.MaxResultCount, total);
+        return new PagedResultDto<MessageDto>(total, entities.Adapt<List<MessageDto>>());
+    }
+}
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/SessionService.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/SessionService.cs
@@ -0,0 +1,92 @@
+using Mapster;
+using Microsoft.AspNetCore.Authorization;
+using Microsoft.AspNetCore.Mvc;
+using SqlSugar;
+using Volo.Abp.Application.Dtos;
+using Volo.Abp.Application.Services;
+using Volo.Abp.Domain.Repositories;
+using Volo.Abp.Users;
+using Yi.Framework.AiHub.Application.Contracts.Dtos;
+using Yi.Framework.AiHub.Domain.Entities;
+using Yi.Framework.AiHub.Domain.Entities.Chat;
+using Yi.Framework.SqlSugarCore.Abstractions;
+
+namespace Yi.Framework.AiHub.Application.Services;
+
+public class SessionService : CrudAppService<SessionAggregateRoot, SessionDto, Guid,SessionGetListInput,SessionCreateAndUpdateInput>
+{
+    private readonly ISqlSugarRepository<SessionAggregateRoot, Guid> _repository;
+    public readonly ISqlSugarRepository<MessageAggregateRoot, Guid> _messageRepository;
+    public SessionService(ISqlSugarRepository<SessionAggregateRoot, Guid> repository, ISqlSugarRepository<MessageAggregateRoot, Guid> messageRepository) : base(repository)
+    {
+        _repository = repository;
+        _messageRepository = messageRepository;
+    }
+
+    /// <summary>
+    /// 创建会话
+    /// </summary>
+    /// <param name="input"></param>
+    /// <returns></returns>
+    [Authorize]
+    public override async Task<SessionDto> CreateAsync(SessionCreateAndUpdateInput input)
+    {
+        var entity = await MapToEntityAsync(input);
+        entity.UserId = CurrentUser.GetId();
+        await _repository.InsertAsync(entity);
+        return entity.Adapt<SessionDto>();
+    }
+
+    /// <summary>
+    /// 详情会话
+    /// </summary>
+    /// <param name="id"></param>
+    /// <returns></returns>
+    [Authorize]
+    public override Task<SessionDto> GetAsync(Guid id)
+    {
+        return base.GetAsync(id);
+    }
+
+    /// <summary>
+    /// 编辑会话
+    /// </summary>
+    /// <param name="id"></param>
+    /// <param name="input"></param>
+    /// <returns></returns>
+    [Authorize]
+    public override Task<SessionDto> UpdateAsync(Guid id, SessionCreateAndUpdateInput input)
+    {
+        return base.UpdateAsync(id, input);
+    }
+
+    /// <summary>
+    /// 删除会话
+    /// </summary>
+    /// <param name="id"></param>
+    /// <returns></returns>
+    [Authorize]
+    public override async Task DeleteAsync(Guid id)
+    {
+        await base.DeleteAsync(id);
+        //对应的消息一起删除
+        await _messageRepository.DeleteAsync(x => x.SessionId == id);
+    }
+
+    /// <summary>
+    /// 查询会话
+    /// </summary>
+    /// <param name="input"></param>
+    /// <returns></returns>
+    [Authorize]
+    public override async Task<PagedResultDto<SessionDto>> GetListAsync(SessionGetListInput input)
+    {
+        RefAsync<int> total = 0;
+        var userId = CurrentUser.GetId();
+        var entities = await _repository._DbQueryable
+            .Where(x=>x.UserId == userId)
+            .OrderByDescending(x => x.Id)
+            .ToPageListAsync(input.SkipCount, input.MaxResultCount, total);
+        return new PagedResultDto<SessionDto>(total, entities.Adapt<List<SessionDto>>());
+    }
+}
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/TokenService.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/Chat/TokenService.cs
@@ -0,0 +1,55 @@
+using Microsoft.AspNetCore.Authorization;
+using Volo.Abp.Application.Services;
+using Volo.Abp.Users;
+using Yi.Framework.AiHub.Application.Contracts.Dtos.Token;
+using Yi.Framework.AiHub.Domain.Entities.OpenApi;
+using Yi.Framework.AiHub.Domain.Extensions;
+using Yi.Framework.AiHub.Domain.Managers;
+using Yi.Framework.SqlSugarCore.Abstractions;
+
+namespace Yi.Framework.AiHub.Application.Services;
+
+public class TokenService : ApplicationService
+{
+    private readonly ISqlSugarRepository<TokenAggregateRoot> _tokenRepository;
+    private readonly TokenManager _tokenManager;
+
+    /// <summary>
+    /// 构造函数
+    /// </summary>
+    /// <param name="tokenRepository"></param>
+    /// <param name="tokenManager"></param>
+    public TokenService(ISqlSugarRepository<TokenAggregateRoot> tokenRepository, TokenManager tokenManager)
+    {
+        _tokenRepository = tokenRepository;
+        _tokenManager = tokenManager;
+    }
+
+    /// <summary>
+    /// 获取token
+    /// </summary>
+    /// <returns></returns>
+    [Authorize]
+    public async Task<TokenOutput> GetAsync()
+    {
+        return new TokenOutput
+        {
+            ApiKey = await _tokenManager.GetAsync(CurrentUser.GetId())
+        };
+    }
+
+    /// <summary>
+    /// 创建token
+    /// </summary>
+    /// <exception cref="UserFriendlyException"></exception>
+    [Authorize]
+    public async Task CreateAsync()
+    {
+        if (!CurrentUser.IsAiVip())
+        {
+            throw new UserFriendlyException("充值成为Vip，畅享第三方token服务");
+        }
+
+        await _tokenManager.CreateAsync(CurrentUser.GetId());
+    }
+}