agents-core/ai.koog.agents.core.agent.entity

Package-level declarations

Types

class AIAgentEdge<IncomingOutput, OutgoingInput>

Represents a directed edge connecting two nodes in the graph of an AI agent strategy. This edge facilitates the transmission of data between a source node and a destination node, allowing transformation or filtering of output values from the source node before they reach the destination node.

AIAgentNodeBase

abstract class AIAgentNodeBase<Input, Output>

Represents an abstract node in an AI agent strategy graph, responsible for executing a specific operation and managing directed edges to other nodes.

AIAgentStateManager

class AIAgentStateManager

AIAgentStorage

class AIAgentStorage

Concurrent-safe key-value storage for an agent. You can create typed keys for your data using the createStorageKey function and set and retrieve data using it by calling set and get.

AIAgentStorageKey

data class AIAgentStorageKey<T : Any>(val name: String)

AIAgentStrategy

class AIAgentStrategy(val name: String, val nodeStart: StartAIAgentNodeBase<String>, val nodeFinish: FinishAIAgentNodeBase<String>, toolSelectionStrategy: ToolSelectionStrategy) : AIAgentSubgraph<String, String>

Represents a strategy for managing and executing AI agent workflows built as subgraphs of interconnected nodes.

AIAgentSubgraph

open class AIAgentSubgraph<Input, Output>(val name: String, val start: StartAIAgentNodeBase<Input>, val finish: FinishAIAgentNodeBase<Output>, toolSelectionStrategy: ToolSelectionStrategy) : AIAgentNodeBase<Input, Output>

AIAgentSubgraph represents a structured subgraph within an AI agent workflow. It serves as a logical segment containing a defined starting and ending point. The subgraph is responsible for executing tasks in a step-by-step manner, managing iterations, and handling tool selection strategies.

FinishAIAgentNodeBase

open class FinishAIAgentNodeBase<Output> : AIAgentNodeBase<Output, Output>

StartAIAgentNodeBase

open class StartAIAgentNodeBase<Input> : AIAgentNodeBase<Input, Input>

ToolSelectionStrategy

sealed interface ToolSelectionStrategy

Represents a strategy to select a subset of tools to be used in a subgraph during its execution.

Functions

createStorageKey

inline fun <T : Any> createStorageKey(name: String): AIAgentStorageKey<T>