agents-core/ai.koog.agents.core.feature/ContextualPromptExecutor

ContextualPromptExecutor

class ContextualPromptExecutor(executor: PromptExecutor, context: AIAgentContext) : PromptExecutor(source)

A wrapper around ai.koog.prompt.executor.model.PromptExecutor that allows for adding internal functionality to the executor to catch and log events related to LLM calls.

Constructors

ContextualPromptExecutor

constructor(executor: PromptExecutor, context: AIAgentContext)

Functions

open override fun close()

open suspend override fun execute(prompt: Prompt, model: LLModel, tools: List<ToolDescriptor>): Message.Assistant

open suspend fun execute(prompt: Prompt, model: ResolvedModel, tools: List<ToolDescriptor>): Message.Assistant

executeMultipleChoices

open suspend override fun executeMultipleChoices(prompt: Prompt, model: LLModel, tools: List<ToolDescriptor>): LLMChoice

open suspend fun executeMultipleChoices(prompt: Prompt, resolvedModel: ResolvedModel, tools: List<ToolDescriptor>): LLMChoice

executeStreaming

open override fun executeStreaming(prompt: Prompt, model: LLModel, tools: List<ToolDescriptor>): Flow<StreamFrame>

Executes a streaming call to the language model with tool support.

open fun executeStreaming(prompt: Prompt, resolvedModel: ResolvedModel, tools: List<ToolDescriptor>): Flow<StreamFrame>

getBasicJsonSchemaGenerator

open override fun getBasicJsonSchemaGenerator(model: LLModel): BasicJsonSchemaGenerator

getStandardJsonSchemaGenerator

open override fun getStandardJsonSchemaGenerator(model: LLModel): StandardJsonSchemaGenerator

open suspend override fun models(): List<LLModel>

open suspend override fun moderate(prompt: Prompt, model: LLModel): ModerationResult

open suspend fun moderate(prompt: Prompt, model: ResolvedModel): ModerationResult

open suspend fun resolveModel(model: LLModel, promptExecutorOperation: PromptExecutorOperation): ResolvedModel