feat(agent): add enabled flag to CustomAgentConfig #195

Add an `enabled` flag to CustomAgentConfig and filter enabled agents.
unit-mesh · May 29, 2024 · 0dfaa32 · 0dfaa32
1 parent efb06f3
commit 0dfaa32
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/main/kotlin/cc/unitmesh/devti/agent/model/CustomAgentConfig.kt b/src/main/kotlin/cc/unitmesh/devti/agent/model/CustomAgentConfig.kt
@@ -56,6 +56,7 @@ data class CustomAgentConfig(
     val interactive: InteractionType = InteractionType.ChatPanel,
     val auth: CustomAgentAuth? = null,
     val defaultTimeout: Long = 10,
+    val enabled: Boolean = true,
 ) {
     var state: CustomAgentState = CustomAgentState.START
 
@@ -67,7 +68,11 @@ data class CustomAgentConfig(
             } catch (e: Exception) {
                 emptyList()
             }
-            return configs
+
+            /**
+             * Only return enabled agents
+             */
+            return configs.filter { it.enabled }
         }
     }
 }

diff --git a/src/main/resources/cc/unitmesh/devti/custom/schema/autodev-custom-agent.json b/src/main/resources/cc/unitmesh/devti/custom/schema/autodev-custom-agent.json
@@ -56,6 +56,9 @@
       },
       "defaultTimeout": {
         "type": "number"
+      },
+      "enabled": {
+        "type": "boolean"
       }
     },
     "required": [