From f8399d22805f2f56438017cda8a38f4e7678cdec Mon Sep 17 00:00:00 2001
From: slhafzjw <slhafzjw@slhaf.work>
Date: Sat, 21 Feb 2026 16:28:20 +0800
Subject: [PATCH] refactor(framework): add additional component storage and
 guarded registration in `AgentContext`

---
 .../api/agent/factory/context/AgentContext.kt   | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

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 c19f1516..e412cc86 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
@@ -1,6 +1,7 @@
 package work.slhaf.partner.api.agent.factory.context
 
 import com.alibaba.fastjson2.JSONArray
+import work.slhaf.partner.api.agent.factory.AgentComponent
 import work.slhaf.partner.api.agent.factory.module.abstracts.AbstractAgentModule
 import java.time.ZonedDateTime
 
@@ -18,6 +19,11 @@ object AgentContext {
     val capabilities: Map<Class<*>, Any?>
         get() = _capabilities
 
+    private val _additionalComponents = mutableSetOf<Any>()
+
+    val additionalComponents: Set<Any>
+        get() = _additionalComponents
+
     val metadata: MutableMap<String, Any> = mutableMapOf()
 
     fun addModule(name: String, module: ModuleContextData<AbstractAgentModule>) {
@@ -27,6 +33,17 @@ object AgentContext {
     fun <T> addCapability(type: Class<T>, value: T) {
         _capabilities[type] = value
     }
+
+    fun <T> addAdditionalComponent(type: Class<T>, value: T): Boolean {
+        if (type.isAnnotationPresent(AgentComponent::class.java)) {
+            return false
+        }
+        if (value == null) {
+            return false
+        }
+        _additionalComponents.add(value)
+        return true
+    }
 }
 
 sealed class ModuleContextData<out T : AbstractAgentModule> {