diff --git a/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/module/abstracts/AgentModule.kt b/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/module/abstracts/AgentModule.kt
index f5ee3979..68a3783a 100644
--- a/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/module/abstracts/AgentModule.kt
+++ b/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/module/abstracts/AgentModule.kt
@@ -95,7 +95,11 @@ interface ActivateModel {
      * 对应调用的模型配置名称
      */
     fun modelKey(): String {
-        return javaClass.simpleName
+        return if (this is AbstractAgentModule) {
+            this.moduleName
+        } else {
+            javaClass.simpleName
+        }
     }
 
     fun withBasicPrompt(): Boolean