From c5b0f69b51f1d34cbfb2f8f9f9ce2e2addfb0817 Mon Sep 17 00:00:00 2001
From: ccnetcore <cc2016cc>
Date: Thu, 17 Jul 2025 23:16:16 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E9=87=8D=E6=9E=84=E5=AE=8C=E6=88=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../Dtos/AiModelDescribe.cs                              | 3 +--
 .../Entities/Model/AiAppAggregateRoot.cs                 | 5 +++++
 .../Entities/Model/AiModelEntity.cs                      | 1 +
 .../Managers/AiGateWayManager.cs                         | 9 ++++++---
 .../src/Yi.Abp.Application/Services/TestService.cs       | 4 +---
 5 files changed, 14 insertions(+), 8 deletions(-)

diff --git a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain.Shared/Dtos/AiModelDescribe.cs b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain.Shared/Dtos/AiModelDescribe.cs
index ca226e95..560a3991 100644
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain.Shared/Dtos/AiModelDescribe.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain.Shared/Dtos/AiModelDescribe.cs
@@ -46,8 +46,7 @@ public class AiModelDescribe
     /// 模型描述
     /// </summary>
     public string? Description { get; set; }
-
-
+    
     /// <summary>
     /// 额外url
     /// </summary>
diff --git a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/Model/AiAppAggregateRoot.cs b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/Model/AiAppAggregateRoot.cs
index 6a698f4f..981ab3e9 100644
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/Model/AiAppAggregateRoot.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/Model/AiAppAggregateRoot.cs
@@ -20,6 +20,11 @@ public class AiAppAggregateRoot : FullAuditedAggregateRoot<Guid>, IOrderNum
     /// </summary>
     public string Endpoint { get; set; }
 
+    /// <summary>
+    /// 额外url
+    /// </summary>
+    public string? ExtraUrl { get; set; }
+
     /// <summary>
     /// 应用key
     /// </summary>
diff --git a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/Model/AiModelEntity.cs b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/Model/AiModelEntity.cs
index d9dcd656..d77967f9 100644
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/Model/AiModelEntity.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/Model/AiModelEntity.cs
@@ -44,4 +44,5 @@ public class AiModelEntity : Entity<Guid>, IOrderNum,ISoftDelete
     /// ai应用id
     /// </summary>
     public Guid AiAppId { get; set; }
+
 }
\ No newline at end of file
diff --git a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/AiGateWayManager.cs b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/AiGateWayManager.cs
index e9d32e09..7e8c05d2 100644
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/AiGateWayManager.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/AiGateWayManager.cs
@@ -55,7 +55,8 @@ public class AiGateWayManager : DomainService
                     HandlerName = model.HandlerName,
                     ModelId = model.ModelId,
                     ModelName = model.Name,
-                    Description = model.Description
+                    Description = model.Description,
+                    ExtraUrl = app.ExtraUrl
                 };
             }
         }
@@ -75,7 +76,8 @@ public class AiGateWayManager : DomainService
         [EnumeratorCancellation] CancellationToken cancellationToken)
     {
         var modelDescribe = await GetModelAsync(request.Model);
-        var chatService = LazyServiceProvider.GetRequiredKeyedService<IChatCompletionService>(modelDescribe.HandlerName);
+        var chatService =
+            LazyServiceProvider.GetRequiredKeyedService<IChatCompletionService>(modelDescribe.HandlerName);
         await foreach (var result in chatService.CompleteChatStreamAsync(modelDescribe, request, cancellationToken))
         {
             yield return result;
@@ -103,7 +105,8 @@ public class AiGateWayManager : DomainService
         response.ContentType = "application/json; charset=UTF-8";
         await using var writer = new StreamWriter(response.Body, Encoding.UTF8, leaveOpen: true);
         var modelDescribe = await GetModelAsync(request.Model);
-        var chatService = LazyServiceProvider.GetRequiredKeyedService<IChatCompletionService>(modelDescribe.HandlerName);
+        var chatService =
+            LazyServiceProvider.GetRequiredKeyedService<IChatCompletionService>(modelDescribe.HandlerName);
         var data = await chatService.CompleteChatAsync(modelDescribe, request, cancellationToken);
         if (userId is not null)
         {
diff --git a/Yi.Abp.Net8/src/Yi.Abp.Application/Services/TestService.cs b/Yi.Abp.Net8/src/Yi.Abp.Application/Services/TestService.cs
index 2f709936..a2528cdc 100644
--- a/Yi.Abp.Net8/src/Yi.Abp.Application/Services/TestService.cs
+++ b/Yi.Abp.Net8/src/Yi.Abp.Application/Services/TestService.cs
@@ -1,8 +1,6 @@
-using System.Xml.Linq;
-using Mapster;
+using Mapster;
 using Medallion.Threading;
 using Microsoft.AspNetCore.Mvc;
-using Microsoft.AspNetCore.RateLimiting;
 using Volo.Abp.Application.Services;
 using Volo.Abp.DistributedLocking;
 using Volo.Abp.Domain.Repositories;