From 4c2b6059cb1203bf7b7e90f094d51e8db70fe7ee Mon Sep 17 00:00:00 2001
From: enisn <info@enisnecipoglu.com>
Date: Thu, 11 Sep 2025 11:22:25 +0300
Subject: [PATCH] Update AbpAIModule.cs

---
 framework/src/Volo.Abp.AI/Volo/Abp/AI/AbpAIModule.cs          | 4 +++-
 .../AI/Extensions/ChatClientWithSystemMessageExtensions.cs    | 4 ----
 2 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/framework/src/Volo.Abp.AI/Volo/Abp/AI/AbpAIModule.cs b/framework/src/Volo.Abp.AI/Volo/Abp/AI/AbpAIModule.cs
index bb979de2e6..aa6f368fec 100644
--- a/framework/src/Volo.Abp.AI/Volo/Abp/AI/AbpAIModule.cs
+++ b/framework/src/Volo.Abp.AI/Volo/Abp/AI/AbpAIModule.cs
@@ -67,6 +67,7 @@ public class AbpAIModule : AbpModule
                 builderConfigurer.Action(workspaceConfig.Kernel.Builder!);
             }
 
+            // TODO: Check if we can use transient instead of singleton for Kernel
             context.Services.AddKeyedTransient<Kernel>(
                 AbpAIOptions.GetKernelServiceKeyName(workspaceConfig.Name),
                 (provider, _) => workspaceConfig.Kernel.Builder!.Build());
@@ -84,7 +85,8 @@ public class AbpAIModule : AbpModule
                 context.Services.AddKeyedTransient<IChatClient>(
                     AbpAIOptions.GetChatClientServiceKeyName(workspaceConfig.Name),
                     (sp, _) => sp.GetKeyedService<Kernel>(AbpAIOptions.GetKernelServiceKeyName(workspaceConfig.Name))?
-                        .GetRequiredService<IChatClient>() ?? throw new InvalidOperationException("Kernel or IChatClient not found with workspace name: " + workspaceConfig.Name)
+                        .GetRequiredService<IChatClient>() 
+                            ?? throw new InvalidOperationException("Kernel or IChatClient not found with workspace name: " + workspaceConfig.Name)
                 );
             }
         }
diff --git a/framework/src/Volo.Abp.AI/Volo/Abp/AI/Extensions/ChatClientWithSystemMessageExtensions.cs b/framework/src/Volo.Abp.AI/Volo/Abp/AI/Extensions/ChatClientWithSystemMessageExtensions.cs
index 2aea6ec734..59eadfd2c6 100644
--- a/framework/src/Volo.Abp.AI/Volo/Abp/AI/Extensions/ChatClientWithSystemMessageExtensions.cs
+++ b/framework/src/Volo.Abp.AI/Volo/Abp/AI/Extensions/ChatClientWithSystemMessageExtensions.cs
@@ -1,7 +1,3 @@
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Threading.Tasks;
 using Microsoft.Extensions.AI;
 using Volo.Abp.AI.Delegates;