refactor(framework): make AgentContext capabilities non-null and infer capability type on registration

This commit is contained in:
2026-02-21 22:38:56 +08:00
parent 833fe4deb3
commit 2cdeaa1c30

View File

@@ -15,9 +15,9 @@ object AgentContext {
get() = _modules get() = _modules
private val _capabilities = private val _capabilities =
mutableMapOf<Class<*>, Any?>() mutableMapOf<Class<*>, Any>()
val capabilities: Map<Class<*>, Any?> val capabilities: Map<Class<*>, Any>
get() = _capabilities get() = _capabilities
private val _additionalComponents = mutableSetOf<Any>() private val _additionalComponents = mutableSetOf<Any>()
@@ -34,8 +34,8 @@ object AgentContext {
_modules[name] = module _modules[name] = module
} }
fun <T> addCapability(type: Class<T>, value: T) { fun addCapability(value: Any) {
_capabilities[type] = value _capabilities[value::class.java] = value
} }
fun addAdditionalComponent(instance: Any): Boolean { fun addAdditionalComponent(instance: Any): Boolean {