koog-agents-optimization/ai.koog.agents.optimization.utils.agentic

Package-level declarations

Properties

Hint message instructing the LLM that it must call a tool to continue, used with nodeCallLLMWithHint.

Functions

fun Prompt.appendToSystemMessage(text: String): Prompt

Creates a new Prompt with text appended to the first system message. If no system message exists, the prompt is returned unchanged.

generateContextOverflowMessage

fun generateContextOverflowMessage(contextApproxSize: Int, overflowCoef: Double = 2.0): String

Util function for investigating the behaviour of the system on model context overflow. It is very approximate and should be used mainly for preliminary experiments.

nodeCallLLMWithHint

inline fun <Input, Output> AIAgentSubgraphBuilderBase<Input, Output>.nodeCallLLMWithHint(hint: String): AIAgentNodeDelegate<Any?, Message.Assistant>

A single LLM call node with an additional user message. Intended for guiding the LLM towards with a hint.

onAssistantMessage

infix fun <IncomingOutput, IntermediateOutput, OutgoingInput> AIAgentEdgeBuilderIntermediate<IncomingOutput, IntermediateOutput, OutgoingInput>.onAssistantMessage(block: suspend (Message.Assistant) -> Boolean): AIAgentEdgeBuilderIntermediate<IncomingOutput, String, OutgoingInput>

Matches any assistant turn and forwards its text — the empty string when the turn carries none.

prettyPrint

fun Prompt.prettyPrint(): String

Renders the prompt's messages into a human-readable, newline-separated string, each prefixed with its index and message role. Intended for logging and debugging.

replaceSystemMessage

fun Prompt.replaceSystemMessage(id: String = this.id, systemMessageEditor: (String) -> String): Prompt

Create a new Prompt object with the same messages, except for the system prompt.

toolResults

fun PromptBuilder.toolResults(results: ReceivedToolResults)

Appends every result of a tool batch as one user message, then asks the LLM to continue.