xwl2023/Yi.Framework
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 完成登录校验拦截

Browse Source
This commit is contained in:
ccnetcore
2025-06-29 15:41:49 +08:00
parent 228a309545
commit 0e36f7c0b3
1 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/AiChatService.cs
View File

@@ -91,22 +91,22 @@ public class AiChatService : ApplicationService
public async Task PostSendAsync(SendMessageInput input, CancellationToken cancellationToken) public async Task PostSendAsync(SendMessageInput input, CancellationToken cancellationToken)
{ {
//除了免费模型，其他的模型都要校验 //除了免费模型，其他的模型都要校验
// if (input.Model != "DeepSeek-R1-0528") if (input.Model != "DeepSeek-R1")
// { {
// //有token，需要黑名单校验 //有token，需要黑名单校验
// if (CurrentUser.IsAuthenticated) if (CurrentUser.IsAuthenticated)
// { {
// await _aiBlacklistManager.VerifiyAiBlacklist(CurrentUser.GetId()); await _aiBlacklistManager.VerifiyAiBlacklist(CurrentUser.GetId());
// if (!CurrentUser.Roles.Contains("YiXinAi-Vip")) if (!CurrentUser.Roles.Contains("YiXinAi-Vip")&&!CurrentUser.Roles.Contains("admin"))
// { {
// throw new UserFriendlyException("该模型需要VIP用户才能使用，请购买VIP后重新登录重试"); throw new UserFriendlyException("该模型需要VIP用户才能使用，请购买VIP后重新登录重试");
// } }
// } }
// else else
// { {
// throw new UserFriendlyException("未登录用户，只能使用未加速的DeepSeek-R1，请登录后重试"); throw new UserFriendlyException("未登录用户，只能使用未加速的DeepSeek-R1，请登录后重试");
// } }
// } }
//前面都是校验，后面才是真正的调用 //前面都是校验，后面才是真正的调用
var httpContext = this._httpContextAccessor.HttpContext; var httpContext = this._httpContextAccessor.HttpContext;