From 7d902682f8b2928f301e598dc7c53cb99dd64d83 Mon Sep 17 00:00:00 2001
From: ccnetcore <cc2016cc>
Date: Sat, 21 Jun 2025 21:40:51 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=AE=8C=E6=88=90=E8=B4=A6=E6=88=B7?=
 =?UTF-8?q?=E4=BF=A1=E6=81=AF=E8=BD=AC=E5=8F=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ...ramework.AiHub.Application.Contracts.csproj |  1 +
 ...FrameworkAiHubApplicationContractsModule.cs |  7 +++++--
 .../Services/AiChatService.cs                  | 18 ++++++++++++++++++
 3 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Yi.Framework.AiHub.Application.Contracts.csproj b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Yi.Framework.AiHub.Application.Contracts.csproj
index 93996947..1aeef2e7 100644
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Yi.Framework.AiHub.Application.Contracts.csproj
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Yi.Framework.AiHub.Application.Contracts.csproj
@@ -3,6 +3,7 @@
 
   <ItemGroup>
     <ProjectReference Include="..\..\..\framework\Yi.Framework.Ddd.Application.Contracts\Yi.Framework.Ddd.Application.Contracts.csproj" />
+    <ProjectReference Include="..\..\rbac\Yi.Framework.Rbac.Application.Contracts\Yi.Framework.Rbac.Application.Contracts.csproj" />
     <ProjectReference Include="..\Yi.Framework.AiHub.Domain.Shared\Yi.Framework.AiHub.Domain.Shared.csproj" />
   </ItemGroup>
   <ItemGroup>
diff --git a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/YiFrameworkAiHubApplicationContractsModule.cs b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/YiFrameworkAiHubApplicationContractsModule.cs
index 8f546800..66a83182 100644
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/YiFrameworkAiHubApplicationContractsModule.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/YiFrameworkAiHubApplicationContractsModule.cs
@@ -1,13 +1,16 @@
 using Microsoft.Extensions.DependencyInjection;
-using Yi.Framework.AiHub.Application.Contracts.Options;
 using Yi.Framework.AiHub.Domain.Shared;
 using Yi.Framework.Ddd.Application.Contracts;
+using Yi.Framework.Rbac.Application.Contracts;
 
 namespace Yi.Framework.AiHub.Application.Contracts
 {
     [DependsOn(
         typeof(YiFrameworkAiHubDomainSharedModule),
-        typeof(YiFrameworkDddApplicationContractsModule))]
+        typeof(YiFrameworkDddApplicationContractsModule),
+        
+        typeof(YiFrameworkRbacApplicationContractsModule)
+        )]
     public class YiFrameworkAiHubApplicationContractsModule : AbpModule
     {
         public override void ConfigureServices(ServiceConfigurationContext context)
diff --git a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/AiChatService.cs b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/AiChatService.cs
index cfe70772..d0b49f2f 100644
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/AiChatService.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/AiChatService.cs
@@ -1,7 +1,10 @@
 using System.Text;
+using Microsoft.AspNetCore.Authorization;
 using Microsoft.AspNetCore.Http;
+using Microsoft.AspNetCore.Mvc;
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Options;
+using Microsoft.Identity.Client;
 using Newtonsoft.Json;
 using Newtonsoft.Json.Serialization;
 using OpenAI.Chat;
@@ -10,6 +13,8 @@ using Volo.Abp.Users;
 using Yi.Framework.AiHub.Application.Contracts.Dtos;
 using Yi.Framework.AiHub.Application.Contracts.Options;
 using Yi.Framework.AiHub.Domain.Managers;
+using Yi.Framework.Rbac.Application.Contracts.IServices;
+using Yi.Framework.Rbac.Domain.Shared.Dtos;
 
 namespace Yi.Framework.AiHub.Application.Services;
 
@@ -31,6 +36,19 @@ public class AiChatService : ApplicationService
     }
 
 
+    /// <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>