Skip to content

Commit 0bf3fc3

Browse files
Add @AIAgentBuilderDslMarker to the retry component
1 parent 3140900 commit 0bf3fc3

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

agents/agents-ext/src/commonMain/kotlin/ai/koog/agents/ext/agent/SubgraphWithRetry.kt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ package ai.koog.agents.ext.agent
33
import ai.koog.agents.core.agent.context.AIAgentContextBase
44
import ai.koog.agents.core.agent.entity.ToolSelectionStrategy
55
import ai.koog.agents.core.agent.entity.createStorageKey
6+
import ai.koog.agents.core.dsl.builder.AIAgentBuilderDslMarker
67
import ai.koog.agents.core.dsl.builder.AIAgentSubgraphBuilderBase
78
import ai.koog.agents.core.dsl.builder.AIAgentSubgraphDelegate
89
import ai.koog.agents.core.dsl.builder.forwardTo
@@ -79,6 +80,7 @@ public val Boolean.asConditionResult: ConditionResult
7980
* @param name The optional name of the subgraph.
8081
* @param defineAction A lambda defining the action subgraph to perform within the retry subgraph.
8182
*/
83+
@AIAgentBuilderDslMarker
8284
public inline fun <reified Input : Any, reified Output> AIAgentSubgraphBuilderBase<*, *>.subgraphWithRetry(
8385
noinline condition: suspend AIAgentContextBase.(Output) -> ConditionResult,
8486
maxRetries: Int,
@@ -205,6 +207,7 @@ public inline fun <reified Input : Any, reified Output> AIAgentSubgraphBuilderBa
205207
* edge(subgraphRetryCallLLM forwardTo nodeExecuteTool onToolCall { true })
206208
* ```
207209
*/
210+
@AIAgentBuilderDslMarker
208211
public inline fun <reified Input : Any, reified Output> AIAgentSubgraphBuilderBase<*, *>.subgraphWithRetrySimple(
209212
noinline condition: suspend AIAgentContextBase.(Output) -> ConditionResult,
210213
maxRetries: Int,

0 commit comments

Comments
 (0)