Merge pull request #146 from joreilly/ai

add basic koog ai agent to the project

Author: joreilly

agents/build.gradle.kts

@@ -0,0 +1,24 @@
+plugins {
+    alias(libs.plugins.kotlinJvm)
+    application
+}
+
+dependencies {
+    implementation(libs.mcp.kotlin)
+    implementation(libs.koin.core)
+    implementation("ai.koog:koog-agents:0.2.1")
+    implementation("org.slf4j:slf4j-simple:2.0.17")
+    implementation(projects.composeApp)
+}
+
+java {
+    toolchain {
+        languageVersion = JavaLanguageVersion.of(17)
+    }
+}
+
+application {
+    mainClass = "MainKt"
+}
+
+

agents/src/main/kotlin/ClimateTraceTool.kt

@@ -0,0 +1,30 @@
+import ai.koog.agents.core.tools.annotations.LLMDescription
+import ai.koog.agents.core.tools.annotations.Tool
+import ai.koog.agents.core.tools.reflect.ToolSet
+import dev.johnoreilly.climatetrace.data.ClimateTraceRepository
+
+
+
+@LLMDescription("Tools for getting climate emission information")
+class ClimateTraceTool : ToolSet {
+    val climateTraceRepository = koin.get<ClimateTraceRepository>()
+
+    @Tool
+    @LLMDescription("Get the list of countries")
+    suspend fun getCountries(): List<String> {
+        val countries = climateTraceRepository.fetchCountries()
+        return countries.map { "${it.name}, ${it.alpha3}" }
+    }
+
+
+    @Tool(customName = "getEmissions")
+    @LLMDescription("Get the emission data for a country for a particular year.")
+    suspend fun getEmissions(
+        @LLMDescription("country code") countryCode: String,
+        @LLMDescription("year") year: String
+    ): List<String> {
+        return climateTraceRepository.fetchCountryEmissionsInfo(countryCode, year).map {
+            it.emissions.co2.toString()
+        }
+    }
+}

agents/src/main/kotlin/koog_agent.kt

@@ -0,0 +1,64 @@
+import ai.koog.agents.core.agent.AIAgent
+import ai.koog.agents.core.tools.ToolRegistry
+import ai.koog.agents.core.tools.reflect.asTools
+import ai.koog.agents.features.eventHandler.feature.handleEvents
+import ai.koog.agents.mcp.McpToolRegistryProvider
+import ai.koog.prompt.executor.clients.google.GoogleModels
+import ai.koog.prompt.executor.clients.openai.OpenAIModels
+import ai.koog.prompt.executor.llms.all.simpleGoogleAIExecutor
+import ai.koog.prompt.executor.llms.all.simpleOpenAIExecutor
+import kotlin.uuid.ExperimentalUuidApi
+import kotlin.uuid.Uuid
+
+val openAIApiKey = ""
+val apiKeyGoogle = ""
+
+
+
+@OptIn(ExperimentalUuidApi::class)
+suspend fun runKoogAgent() {
+    val agent = AIAgent(
+        executor = simpleOpenAIExecutor(openAIApiKey),
+        //executor = simpleGoogleAIExecutor(apiKeyGoogle),
+        llmModel = OpenAIModels.Chat.GPT4o,
+        //llmModel = GoogleModels.Gemini1_5Pro,
+        toolRegistry = createToolSetRegistry()
+    ) {
+        handleEvents {
+            onToolCall { tool: ai.koog.agents.core.tools.Tool<*, *>, toolArgs: ai.koog.agents.core.tools.Tool.Args ->
+                println("Tool called: tool ${tool.name}, args $toolArgs")
+            }
+
+            onAgentRunError { strategyName: String, sessionUuid: Uuid?, throwable: Throwable ->
+                println("An error occurred: ${throwable.message}\n${throwable.stackTraceToString()}")
+            }
+
+            onAgentFinished { strategyName, result ->
+                println("Agent (strategy = $strategyName) finished with result: $result")
+            }
+        }
+    }
+
+    agent.run(
+        """
+        Get emission data for France, Germany for 2023 and 2024.
+        Use units of millions for the emissions data.
+        Also get best estimate of population for each country and show per-capita emissions.
+        """.trimIndent()
+    )
+}
+
+
+suspend fun createToolSetRegistry(): ToolRegistry {
+    val processClimateTrace = ProcessBuilder("java", "-jar",
+        "./mcp-server/build/libs/serverAll.jar", "--stdio"
+    ).start()
+    val transportClimateTrace = McpToolRegistryProvider.defaultStdioTransport(processClimateTrace)
+    val toolRegistryClimateTrace = McpToolRegistryProvider.fromTransport(transportClimateTrace)
+
+    val localToolSetRegistry = ToolRegistry { tools(ClimateTraceTool().asTools()) }
+
+    // Can use either local toolset or one based on MCP server
+    //return toolRegistryClimateTrace
+    return localToolSetRegistry
+}

agents/src/main/kotlin/main.kt

@@ -0,0 +1,11 @@
+import dev.johnoreilly.climatetrace.di.initKoin
+import kotlinx.coroutines.runBlocking
+import kotlin.uuid.ExperimentalUuidApi
+
+
+val koin = initKoin(enableNetworkLogs = true).koin
+
+@OptIn(ExperimentalUuidApi::class)
+fun main() = runBlocking{
+    runKoogAgent()
+}

settings.gradle.kts

@@ -23,3 +23,4 @@ enableFeaturePreview("TYPESAFE_PROJECT_ACCESSORS")
 rootProject.name = "ClimateTraceKMP"
 include(":composeApp")
 include(":mcp-server")
+include(":agents")