From 2cdeaa1c30c2ef2228a9834517d011211f03b718 Mon Sep 17 00:00:00 2001
From: slhafzjw <slhafzjw@slhaf.work>
Date: Sat, 21 Feb 2026 22:38:56 +0800
Subject: [PATCH] refactor(framework): make `AgentContext` capabilities
 non-null and infer capability type on registration

---
 .../partner/api/agent/factory/context/AgentContext.kt     | 8 ++++----
 1 file changed, 4 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 e2c64544..7d1dd011 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
@@ -15,9 +15,9 @@ object AgentContext {
         get() = _modules
 
     private val _capabilities =
-        mutableMapOf<Class<*>, Any?>()
+        mutableMapOf<Class<*>, Any>()
 
-    val capabilities: Map<Class<*>, Any?>
+    val capabilities: Map<Class<*>, Any>
         get() = _capabilities
 
     private val _additionalComponents = mutableSetOf<Any>()
@@ -34,8 +34,8 @@ object AgentContext {
         _modules[name] = module
     }
 
-    fun <T> addCapability(type: Class<T>, value: T) {
-        _capabilities[type] = value
+    fun addCapability(value: Any) {
+        _capabilities[value::class.java] = value
     }
 
     fun addAdditionalComponent(instance: Any): Boolean {