Merge pull request #148 from joreilly/adk_web

support for ADK Web dev UI

Author: joreilly

agents/build.gradle.kts

@@ -1,14 +1,18 @@
 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("org.slf4j:slf4j-simple:2.0.17")
     implementation("com.google.adk:google-adk:0.1.0")
+    implementation("com.google.adk:google-adk-dev:0.1.0")
+
+    // following needed for AdkWebServer (dev UI)
+    implementation("org.apache.httpcomponents.client5:httpclient5:5.4.3")
+
     implementation(projects.composeApp)
 }
 
@@ -18,8 +22,11 @@ java {
     }
 }
 
-application {
-    mainClass = "MainKt"
+tasks.register<JavaExec>("devUi") {
+    group = "application"
+    description = "Start the ADK Dev UI server"
+    mainClass.set("com.google.adk.web.AdkWebServer")
+    classpath = sourceSets["main"].runtimeClasspath
+    args = listOf("--adk.agents.source-dir=src/main/java")
 }
 
-

agents/src/main/java/Agent.java

@@ -0,0 +1,9 @@
+import com.google.adk.agents.BaseAgent;
+
+import static adk.ClimateTraceAgent.initAgent;
+
+
+public class Agent {
+    public static BaseAgent ROOT_AGENT = initAgent();
+}
+

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

@@ -1,5 +1,6 @@
 package adk
 
+import adk.ClimateTraceAgent.Companion.initAgent
 import com.google.adk.agents.BaseAgent
 import com.google.adk.agents.LlmAgent
 import com.google.adk.events.Event
@@ -17,45 +18,52 @@ 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 = ""
+class ClimateTraceAgent {
+    companion object {
+        @JvmStatic
+        fun initAgent(): BaseAgent {
+            val apiKeyGoogle = ""
 
-    val mcpTools = McpToolset(
-        ServerParameters
-            .builder("java")
-            .args("-jar", "./mcp-server/build/libs/serverAll.jar", "--stdio")
-            .build()
-    ).loadTools().join()
+            val mcpTools = McpToolset(
+                ServerParameters
+                    .builder("java")
+                    .args("-jar", "/Users/joreilly/dev/github/ClimateTraceKMP/mcp-server/build/libs/serverAll.jar", "--stdio")
+                    .build()
+            ).loadTools().join()
 
-    val model = Gemini(
-        "gemini-1.5-pro",
-        Client.builder()
-            .apiKey(apiKeyGoogle)
-            .build()
-    )
+            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()
+            return LlmAgent.builder()
+                .name(NAME)
+                .model(model)
+                .description("Agent to answer climate emissions related questions.")
+                .instruction("You are an agent that provides climate emissions related information. Use 3 letter country codes.")
+                .tools(mcpTools)
+                .build()
+        }
+
+    }
 }
 
+
 fun main() {
-    val runner = InMemoryRunner(ROOT_AGENT)
+    val runner = InMemoryRunner(initAgent())
     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()
+            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)
@@ -70,4 +78,5 @@ fun main() {
             println("error: ${event.errorCode().get()}, ${event.errorMessage().get()}")
         }
     })
-}
+}
+