@@ -3,6 +3,7 @@ package ai.koog.agents.ext.agent
3
3
import ai.koog.agents.core.agent.context.AIAgentContextBase
4
4
import ai.koog.agents.core.agent.entity.ToolSelectionStrategy
5
5
import ai.koog.agents.core.agent.entity.createStorageKey
6
+ import ai.koog.agents.core.dsl.builder.AIAgentBuilderDslMarker
6
7
import ai.koog.agents.core.dsl.builder.AIAgentSubgraphBuilderBase
7
8
import ai.koog.agents.core.dsl.builder.AIAgentSubgraphDelegate
8
9
import ai.koog.agents.core.dsl.builder.forwardTo
@@ -79,6 +80,7 @@ public val Boolean.asConditionResult: ConditionResult
79
80
* @param name The optional name of the subgraph.
80
81
* @param defineAction A lambda defining the action subgraph to perform within the retry subgraph.
81
82
*/
83
+ @AIAgentBuilderDslMarker
82
84
public inline fun <reified Input : Any , reified Output > AIAgentSubgraphBuilderBase <* , * >.subgraphWithRetry (
83
85
noinline condition : suspend AIAgentContextBase .(Output ) -> ConditionResult ,
84
86
maxRetries : Int ,
@@ -205,6 +207,7 @@ public inline fun <reified Input : Any, reified Output> AIAgentSubgraphBuilderBa
205
207
* edge(subgraphRetryCallLLM forwardTo nodeExecuteTool onToolCall { true })
206
208
* ```
207
209
*/
210
+ @AIAgentBuilderDslMarker
208
211
public inline fun <reified Input : Any , reified Output > AIAgentSubgraphBuilderBase <* , * >.subgraphWithRetrySimple (
209
212
noinline condition : suspend AIAgentContextBase .(Output ) -> ConditionResult ,
210
213
maxRetries : Int ,
0 commit comments