Introduced AIAgentStrategy-2

Author: Rizzen

agents/agents-core/src/commonMain/kotlin/ai/koog/agents/core/agent/AIAgentLoopContext.kt

@@ -0,0 +1,48 @@
+package ai.koog.agents.core.agent
+
+import ai.koog.agents.core.agent.config.AIAgentConfigBase
+import ai.koog.agents.core.agent.context.AIAgentContext
+import ai.koog.agents.core.agent.context.AIAgentLLMContext
+import ai.koog.agents.core.agent.entity.AIAgentStateManager
+import ai.koog.agents.core.agent.entity.AIAgentStorage
+import ai.koog.agents.core.agent.entity.AIAgentStorageKey
+import ai.koog.agents.core.environment.AIAgentEnvironment
+import ai.koog.agents.core.feature.AIAgentFeature
+import ai.koog.prompt.message.Message
+
+public class AIAgentLoopContext(
+    override val environment: AIAgentEnvironment,
+    override val agentId: String,
+    override val runId: String,
+    override val agentInput: Any?,
+    override val config: AIAgentConfigBase,
+    override val llm: AIAgentLLMContext,
+    override val stateManager: AIAgentStateManager,
+    override val storage: AIAgentStorage,
+    override val strategyName: String
+) : AIAgentContext {
+    override fun store(key: AIAgentStorageKey<*>, value: Any) {
+        TODO("Not yet implemented")
+    }
+
+    override fun <T> get(key: AIAgentStorageKey<*>): T? {
+        TODO("Not yet implemented")
+    }
+
+    override fun remove(key: AIAgentStorageKey<*>): Boolean {
+        TODO("Not yet implemented")
+    }
+
+    override fun <Feature : Any> feature(key: AIAgentStorageKey<Feature>): Feature? {
+        TODO("Not yet implemented")
+    }
+
+    override fun <Feature : Any> feature(feature: AIAgentFeature<*, Feature>): Feature? {
+        TODO("Not yet implemented")
+    }
+
+    override suspend fun getHistory(): List<Message> {
+        TODO("Not yet implemented")
+    }
+
+}

agents/agents-core/src/commonMain/kotlin/ai/koog/agents/core/agent/AIAgentLoopStrategy.kt

@@ -0,0 +1,13 @@
+package ai.koog.agents.core.agent
+
+import ai.koog.agents.core.agent.entity.AIAgentStrategy
+
+/**
+ *
+ * */
+public class AIAgentLoopStrategy<Input, Output>(
+    override val name: String
+): AIAgentStrategy<Input, Output, AIAgentLoopContext> {
+
+}
+

agents/agents-core/src/commonMain/kotlin/ai/koog/agents/core/agent/entity/AIAgentGraphStrategy.kt

@@ -1,12 +1,16 @@
+@file:Suppress("MissingKDocForPublicAPI")
+
 package ai.koog.agents.core.agent.entity
 
+import ai.koog.agents.core.agent.context.AIAgentContext
 import ai.koog.agents.core.agent.context.AIAgentGraphContextBase
 import ai.koog.agents.core.annotation.InternalAgentsApi
 import ai.koog.agents.core.utils.runCatchingCancellable
 import kotlinx.serialization.json.Json
 import kotlinx.serialization.json.JsonElement
 import kotlinx.serialization.serializer
 
+
 /**
  * Represents a strategy for managing and executing AI agent workflows built as subgraphs of interconnected nodes.
  *
@@ -22,13 +26,12 @@ public class AIAgentGraphStrategy<Input, Output>(
     public val nodeFinish: FinishNode<Output>,
     toolSelectionStrategy: ToolSelectionStrategy,
     private val serializer: Json = Json { prettyPrint = true }
-) : AIAgentSubgraph<Input, Output>(
+) : AIAgentStrategy<Input, Output, AIAgentGraphContextBase>, AIAgentSubgraph<Input, Output>(
     name,
     nodeStart,
     nodeFinish,
     toolSelectionStrategy
-),
-    AIAgentStrategy {
+) {
     /**
      * Represents the metadata of the subgraph associated with the AI agent strategy.
      *

agents/agents-core/src/commonMain/kotlin/ai/koog/agents/core/agent/entity/AIAgentStrategy.kt

@@ -1,6 +1,7 @@
 package ai.koog.agents.core.agent.entity
 
 import ai.koog.agents.core.agent.context.AIAgentContext
+import ai.koog.agents.core.agent.context.AIAgentGraphContextBase
 
 /**
  * An interface representing the execution strategy of an AI agent.
@@ -14,7 +15,7 @@ import ai.koog.agents.core.agent.context.AIAgentContext
  * @param Output The type of output data that the strategy will generate.
  * @param TContext The type of context in which the strategy is executed, extending [AIAgentContext].
  */
-public interface AIAgentStrategy {
+public interface AIAgentStrategy<Input, Output, TContext: AIAgentContext> {
     /**
      * The name of the AI agent strategy.
      *
@@ -23,4 +24,22 @@ public interface AIAgentStrategy {
      * multiple strategies within the system.
      */
     public val name: String
+
+    /**
+     * Executes the AI agent's strategy using the provided context and input.
+     *
+     * This method processes the given input data within the specified context, leveraging
+     * the AI agent's internal logic and strategy to produce an output. The result of this
+     * execution may depend on the graph-based execution pipeline, decision-making processes,
+     * and other stateful operations defined in the context.
+     *
+     * @param context The execution context in which the AI agent operates. It provides access
+     * to the agent's configuration, pipeline, environment, and other components required for
+     * execution in a graph-based structure.
+     * @param input The input data to be processed by the AI agent's strategy. The type of input
+     * is defined by the strategy's implementation and is used to derive the resulting output.
+     * @return The output produced by the AI agent's strategy, or null if no output is generated.
+     * The output type is defined by the strategy's implementation.
+     */
+    public suspend fun execute(context: AIAgentGraphContextBase, input: Input): Output?
 }