From e2ef92ce430e16bdee7ae652f30dc86984255dd3 Mon Sep 17 00:00:00 2001
From: slhafzjw <slhafzjw@slhaf.work>
Date: Fri, 20 Feb 2026 15:29:38 +0800
Subject: [PATCH] refactor(framework): use `moduleName` as `modelKey` for
 `AbstractAgentModule` instances in ActivateModel

---
 .../api/agent/factory/module/abstracts/AgentModule.kt       | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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