Skip to content

Commit 90c4f87

Browse files
committed
Fix java example
1 parent 35ea853 commit 90c4f87

2 files changed

Lines changed: 6 additions & 5 deletions

File tree

  • agents/agents-core/src/jvmCommonMain/kotlin/ai/koog/agents/core/agent/entity
  • examples/simple-examples-java/src/main/java/ai/koog/agents/example/calculator

agents/agents-core/src/jvmCommonMain/kotlin/ai/koog/agents/core/agent/entity/AgentEdgeBuilder.kt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -272,6 +272,7 @@ public open class FullAgentEdgeBuilder<IncomingOutput, IntermediateOutput, Outgo
272272
*
273273
* @param transformation A function that converts the intermediate output to a String for the user message.
274274
*/
275+
@JvmOverloads
275276
public fun asUserMessage(
276277
transformation: SimpleTransformation<IntermediateOutput, String> = { it.toString() }
277278
): FullAgentEdgeBuilder<IncomingOutput, Message.User, OutgoingInput> =

examples/simple-examples-java/src/main/java/ai/koog/agents/example/calculator/Calculator.java

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -95,21 +95,21 @@ private static void runCalculatorExample(RunMode runMode) {
9595
// start → callLLM (wrap the agent's String input as a user message)
9696
graph.edge(
9797
AIAgentEdge.builder().from(graph.nodeStart).to(nodeCallLLM)
98-
.asUserMessage(s -> s)
98+
.asUserMessage()
9999
.build()
100100
);
101101

102102
// callLLM → finish (when LLM returns a text response)
103103
graph.edge(
104104
AIAgentEdge.builder().from(nodeCallLLM).to(graph.nodeFinish)
105-
.onTextParts()
105+
.onTextMessage()
106106
.build()
107107
);
108108

109109
// callLLM → executeTools (when LLM returns tool calls)
110110
graph.edge(
111111
AIAgentEdge.builder().from(nodeCallLLM).to(nodeExecuteTools)
112-
.onToolCalls(call -> true)
112+
.onToolCalls()
113113
.build()
114114
);
115115

@@ -137,14 +137,14 @@ private static void runCalculatorExample(RunMode runMode) {
137137
// sendResults → executeTools (when LLM requests more tool calls)
138138
graph.edge(
139139
AIAgentEdge.builder().from(nodeSendResults).to(nodeExecuteTools)
140-
.onToolCalls(call -> true)
140+
.onToolCalls()
141141
.build()
142142
);
143143

144144
// sendResults → finish (when LLM returns a final text response)
145145
graph.edge(
146146
AIAgentEdge.builder().from(nodeSendResults).to(graph.nodeFinish)
147-
.onTextParts()
147+
.onTextMessage()
148148
.build()
149149
);
150150

0 commit comments

Comments
 (0)