Merge pull request #147 from joreilly/adk

Initial ADK agent

Author: joreilly

agents/build.gradle.kts

@@ -8,6 +8,7 @@ dependencies {
     implementation(libs.koin.core)
     implementation("ai.koog:koog-agents:0.2.1")
     implementation("org.slf4j:slf4j-simple:2.0.17")
+    implementation("com.google.adk:google-adk:0.1.0")
     implementation(projects.composeApp)
 }
 

agents/src/main/kotlin/adk/adk_agent.kt

@@ -0,0 +1,73 @@
+package adk
+
+import com.google.adk.agents.BaseAgent
+import com.google.adk.agents.LlmAgent
+import com.google.adk.events.Event
+import com.google.adk.models.Gemini
+import com.google.adk.runner.InMemoryRunner
+import com.google.adk.tools.mcp.McpToolset
+import com.google.genai.Client
+import com.google.genai.types.Content
+import com.google.genai.types.Part
+import io.modelcontextprotocol.client.transport.ServerParameters
+import io.reactivex.rxjava3.functions.Consumer
+import kotlin.jvm.optionals.getOrNull
+
+
+const val USER_ID = "MainUser"
+const val NAME = "ClimateTrace Agent"
+
+val ROOT_AGENT: BaseAgent = initAgent()
+
+fun initAgent(): BaseAgent {
+    val apiKeyGoogle = ""
+
+    val mcpTools = McpToolset(
+        ServerParameters
+            .builder("java")
+            .args("-jar", "./mcp-server/build/libs/serverAll.jar", "--stdio")
+            .build()
+    ).loadTools().join()
+
+    val model = Gemini(
+        "gemini-1.5-pro",
+        Client.builder()
+            .apiKey(apiKeyGoogle)
+            .build()
+    )
+
+    return LlmAgent.builder()
+        .name(NAME)
+        .model(model)
+        .description("Agent to answer climate emissions related questions.")
+        .tools(mcpTools)
+        .build()
+}
+
+fun main() {
+    val runner = InMemoryRunner(ROOT_AGENT)
+    val session = runner
+        .sessionService()
+        .createSession(NAME, USER_ID)
+        .blockingGet()
+
+    val prompt =
+        """
+        Get emission data for France and Germany for 2023 and 2024.
+        Use units of millions for the emissions data.
+        """.trimIndent()
+
+    val userMsg = Content.fromParts(Part.fromText(prompt))
+    val events = runner.runAsync(USER_ID, session.id(), userMsg)
+
+    events.blockingForEach(Consumer { event: Event ->
+        event.content().get().parts().getOrNull()?.forEach { part ->
+            part.text().getOrNull()?.let { println(it) }
+            part.functionCall().getOrNull()?.let { println(it) }
+            part.functionResponse().getOrNull()?.let { println(it) }
+        }
+        if (event.errorCode().isPresent || event.errorMessage().isPresent) {
+            println("error: ${event.errorCode().get()}, ${event.errorMessage().get()}")
+        }
+    })
+}

agents/src/main/kotlin/ClimateTraceTool.kt renamed to agents/src/main/kotlin/koog/ClimateTraceTool.kt

@@ -1,8 +1,10 @@
+package koog
+
 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
-
+import koin
 
 
 @LLMDescription("Tools for getting climate emission information")

agents/src/main/kotlin/koog_agent.kt renamed to agents/src/main/kotlin/koog/koog_agent.kt

@@ -1,4 +1,7 @@
+package koog
+
 import ai.koog.agents.core.agent.AIAgent
+import ai.koog.agents.core.tools.Tool
 import ai.koog.agents.core.tools.ToolRegistry
 import ai.koog.agents.core.tools.reflect.asTools
 import ai.koog.agents.features.eventHandler.feature.handleEvents
@@ -16,16 +19,16 @@ val apiKeyGoogle = ""
 
 
 @OptIn(ExperimentalUuidApi::class)
-suspend fun runKoogAgent() {
+suspend fun main() {
     val agent = AIAgent(
-        executor = simpleOpenAIExecutor(openAIApiKey),
-        //executor = simpleGoogleAIExecutor(apiKeyGoogle),
-        llmModel = OpenAIModels.Chat.GPT4o,
-        //llmModel = GoogleModels.Gemini1_5Pro,
+        //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 ->
+            onToolCall { tool: Tool<*, *>, toolArgs: Tool.Args ->
                 println("Tool called: tool ${tool.name}, args $toolArgs")
             }
 
@@ -41,9 +44,8 @@ suspend fun runKoogAgent() {
 
     agent.run(
         """
-        Get emission data for France, Germany for 2023 and 2024.
+        Get emission data for France and 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()
     )
 }

agents/src/main/kotlin/main.kt

@@ -1,11 +1,3 @@
 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()
-}