From 7ee698768ce15bb9d5feae8e90bb1d454e5242a1 Mon Sep 17 00:00:00 2001
From: slhafzjw <slhafzjw@slhaf.work>
Date: Mon, 23 Feb 2026 22:59:20 +0800
Subject: [PATCH] refactor(framework): simplify `AgentContext` capability
 storage to a single implementation per capability

---
 .../partner/api/agent/factory/context/AgentContext.kt      | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/context/AgentContext.kt b/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/context/AgentContext.kt
index 45ffec1f..7ef1dfe3 100644
--- a/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/context/AgentContext.kt
+++ b/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/context/AgentContext.kt
@@ -16,9 +16,9 @@ object AgentContext {
         get() = _modules
 
     private val _capabilities =
-        mutableMapOf<String, MutableSet<CapabilityImplementation>>()
+        mutableMapOf<String, CapabilityImplementation>()
 
-    val capabilities: Map<String, Set<CapabilityImplementation>>
+    val capabilities: Map<String, CapabilityImplementation>
         get() = _capabilities
 
     private val _additionalComponents = mutableSetOf<Any>()
@@ -36,9 +36,8 @@ object AgentContext {
     }
 
     fun addCapability(capability: String, instance: Any, methods: Map<String, Method>) {
-        val capabilityImpls = _capabilities.computeIfAbsent(capability) { mutableSetOf() }
         val newImpl = CapabilityImplementation(instance.javaClass, instance, methods)
-        capabilityImpls.add(newImpl)
+        _capabilities[capability] = newImpl
     }
 
     fun addAdditionalComponent(instance: Any): Boolean {