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 a80a5496..58c9b489 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
@@ -30,11 +30,13 @@ public class AiChatService : ApplicationService
     private readonly AiBlacklistManager _aiBlacklistManager;
 
     public AiChatService(IHttpContextAccessor httpContextAccessor,
-        AiMessageManager aiMessageManager, AiBlacklistManager aiBlacklistManager)
+        AiMessageManager aiMessageManager, AiBlacklistManager aiBlacklistManager,
+        ISqlSugarRepository<AiModelEntity> aiModelRepository)
     {
         this._httpContextAccessor = httpContextAccessor;
         _aiMessageManager = aiMessageManager;
         _aiBlacklistManager = aiBlacklistManager;
+        _aiModelRepository = aiModelRepository;
     }