Kotlin AI Agent simplification (#20)

* going away from session

* Cleanup + Removing KotlinAIAgent + renames

* Mute ToneAgentTest

* CR Fixes

Author: Rizzen

agents/agents-core/src/commonMain/kotlin/ai/grazie/code/agents/core/agent/AIAgentBase.kt

@@ -2,21 +2,20 @@ package ai.grazie.code.agents.core.agent
 
 import ai.grazie.code.agents.core.agent.AIAgentTool.AgentToolArgs
 import ai.grazie.code.agents.core.agent.AIAgentTool.AgentToolResult
-import ai.grazie.code.agents.core.model.agent.AIAgentConfig
-import ai.grazie.code.agents.core.model.agent.AIAgentStrategy
+import ai.grazie.code.agents.core.api.AIAgent
 import ai.grazie.code.agents.core.tools.*
 import kotlinx.serialization.KSerializer
 import kotlinx.serialization.Serializable
 import kotlinx.serialization.json.Json
 
-fun <TStrategy : AIAgentStrategy<TConfig>, TConfig : AIAgentConfig> AIAgentBase<TStrategy, TConfig>.asTool(
+fun AIAgent.asTool(
     agentDescription: String,
     requestDescription: String = "Input for the task"
-) = AIAgentTool<TStrategy, TConfig>(this, requestDescription, agentDescription)
+) = AIAgentTool(this, requestDescription, agentDescription)
 
 
-class AIAgentTool<TStrategy : AIAgentStrategy<TConfig>, TConfig : AIAgentConfig>(
-    val agent: AIAgentBase<TStrategy, TConfig>,
+class AIAgentTool(
+    val agent: AIAgent,
     agentName: String,
     agentDescription: String,
     requestDescription: String = "Input for the task"

agents/agents-core/src/commonMain/kotlin/ai/grazie/code/agents/core/agent/AIAgentTool.kt

@@ -1,6 +1,6 @@
 package ai.grazie.code.agents.core.api
 
-import ai.grazie.code.agents.core.agent.KotlinAIAgent
+import ai.grazie.code.agents.core.agent.AIAgentBase
 import ai.grazie.code.agents.core.agent.config.LocalAgentConfig
 import ai.grazie.code.agents.core.event.EventHandler
 import ai.grazie.code.agents.core.tools.ToolRegistry
@@ -34,7 +34,7 @@ fun simpleChatAgent(
     eventHandler: EventHandler = EventHandler.NO_HANDLER,
     toolRegistry: ToolRegistry? = null,
     maxIterations: Int = 50,
-    ): KotlinAIAgent {
+    ): AIAgentBase {
 
     val agentConfig = LocalAgentConfig(
         prompt = prompt(llmModel, "chat", params = LLMParams(temperature = temperature)) {
@@ -56,13 +56,13 @@ fun simpleChatAgent(
             }
         } with toolRegistry
 
-    return KotlinAIAgent(
-        toolRegistry = resultingToolRegistry,
+    return AIAgentBase(
+        promptExecutor = executor,
         strategy = chatAgentStrategy(),
-        eventHandler = eventHandler,
+        cs = cs,
         agentConfig = agentConfig,
-        promptExecutor = executor,
-        cs = cs
+        toolRegistry = resultingToolRegistry,
+        eventHandler = eventHandler
     )
 }
 
@@ -89,8 +89,8 @@ fun simpleSingleRunAgent(
     eventHandler: EventHandler = EventHandler.NO_HANDLER,
     toolRegistry: ToolRegistry = ToolRegistry.EMPTY,
     maxIterations: Int = 50,
-    installFeatures: suspend KotlinAIAgent.FeatureContext.() -> Unit= {}
-): KotlinAIAgent {
+    installFeatures: suspend AIAgentBase.FeatureContext.() -> Unit= {}
+): AIAgentBase {
 
     val agentConfig = LocalAgentConfig(
         prompt = prompt(llmModel, "chat", params = LLMParams(temperature = temperature)) {
@@ -99,13 +99,13 @@ fun simpleSingleRunAgent(
         maxAgentIterations = maxIterations,
     )
 
-    return KotlinAIAgent(
-        toolRegistry = toolRegistry,
-        strategy = singleRunStrategy(),
-        eventHandler = eventHandler,
-        agentConfig = agentConfig,
+    return AIAgentBase(
         promptExecutor = executor,
+        strategy = singleRunStrategy(),
         cs = cs,
+        agentConfig = agentConfig,
+        toolRegistry = toolRegistry,
+        eventHandler = eventHandler,
         installFeatures = installFeatures
     )
 }

agents/agents-core/src/commonMain/kotlin/ai/grazie/code/agents/core/agent/KotlinAIAgent.kt

@@ -0,0 +1,43 @@
+package ai.grazie.code.agents.core.environment
+
+import ai.grazie.code.agents.core.agent.AgentTerminationByClientException
+import ai.grazie.code.agents.core.engine.UnexpectedAgentMessageException
+import ai.grazie.code.agents.core.engine.UnexpectedDoubleInitializationException
+import ai.grazie.code.agents.core.model.message.EnvironmentToAgentErrorMessage
+import ai.grazie.code.agents.core.model.message.EnvironmentToAgentMessage
+import ai.grazie.code.agents.core.model.message.EnvironmentToAgentTerminationMessage
+import ai.grazie.code.agents.core.model.message.EnvironmentToolResultMultipleToAgentMessage
+import ai.grazie.code.agents.core.model.message.EnvironmentToolResultSingleToAgentMessage
+import ai.grazie.code.agents.core.model.message.LocalAgentEnvironmentToAgentInitializeMessage
+
+object AgentEnvironmentUtils {
+    fun EnvironmentToAgentMessage.mapToToolResult(): List<ReceivedToolResult> {
+        return when (this) {
+            is EnvironmentToolResultSingleToAgentMessage -> {
+                listOf(this.content.toResult())
+            }
+
+            is EnvironmentToolResultMultipleToAgentMessage -> {
+                this.content.map { it.toResult() }
+            }
+
+            is EnvironmentToAgentErrorMessage -> {
+                throw AgentTerminationByClientException(this.error.message)
+            }
+
+            is EnvironmentToAgentTerminationMessage -> {
+                throw AgentTerminationByClientException(
+                    this.content?.message ?: this.error?.message ?: ""
+                )
+            }
+
+            is LocalAgentEnvironmentToAgentInitializeMessage -> {
+                throw UnexpectedDoubleInitializationException()
+            }
+
+            else -> {
+                throw UnexpectedAgentMessageException()
+            }
+        }
+    }
+}