New example: basic code-agent

examples/code-agent/step-01-basic-agent/Module.md

@@ -0,0 +1,3 @@
+# Module Code Agent (Step 1)
+
+Code Agent example.

examples/code-agent/step-01-basic-agent/build.gradle.kts

@@ -0,0 +1,26 @@
+plugins {
+    id("ai.kotlin.jvm")
+    alias(libs.plugins.shadow)
+    application
+}
+
+application.mainClass.set("ai.koog.agents.examples.codeagent.step01.MainKt")
+
+dependencies {
+    implementation(project(":agents:agents-core"))
+    implementation(project(":agents:agents-ext"))
+    implementation(project(":prompt:prompt-executor:prompt-executor-llms-all"))
+
+    implementation(libs.kotlinx.coroutines.core)
+
+    implementation(libs.logback.classic)
+}
+
+tasks.test {
+    useJUnitPlatform()
+}
+
+tasks.shadowJar {
+    archiveBaseName.set("code-agent")
+    mergeServiceFiles()
+}

examples/code-agent/step-01-basic-agent/src/main/kotlin/Main.kt

@@ -0,0 +1,51 @@
+package ai.koog.agents.examples.codeagent.step01
+
+import ai.koog.agents.core.agent.AIAgent
+import ai.koog.agents.core.agent.singleRunStrategy
+import ai.koog.agents.core.tools.ToolRegistry
+import ai.koog.agents.ext.tool.file.EditFileTool
+import ai.koog.agents.ext.tool.file.ListDirectoryTool
+import ai.koog.agents.ext.tool.file.ReadFileTool
+import ai.koog.agents.ext.tool.file.WriteFileTool
+import ai.koog.agents.features.eventHandler.feature.handleEvents
+import ai.koog.prompt.executor.clients.openai.OpenAIModels
+import ai.koog.prompt.executor.llms.all.simpleOpenAIExecutor
+import ai.koog.rag.base.files.JVMFileSystemProvider
+import kotlinx.coroutines.runBlocking
+
+val agent = AIAgent(
+    promptExecutor = simpleOpenAIExecutor(System.getenv("OPENAI_API_KEY")),
+    strategy = singleRunStrategy(),
+    systemPrompt = """
+        You are a highly skilled programmer tasked with updating the provided codebase according to the given task.
+        Your goal is to deliver production-ready code changes that integrate seamlessly with the existing codebase and solve given task.
+    """.trimIndent(),
+    llmModel = OpenAIModels.Chat.GPT4_1,
+    temperature = 0.0,
+    toolRegistry = ToolRegistry {
+        tool(ListDirectoryTool(JVMFileSystemProvider.ReadOnly))
+        tool(ReadFileTool(JVMFileSystemProvider.ReadOnly))
+        tool(WriteFileTool(JVMFileSystemProvider.ReadWrite))
+        tool(EditFileTool(JVMFileSystemProvider.ReadWrite))
+    },
+    maxIterations = 100
+) {
+    handleEvents {
+        onToolCall { ctx ->
+            println("Tool called: ${ctx.tool.name}, args=${ctx.toolArgs}")
+        }
+    }
+}
+
+fun main(args: Array<String>) = runBlocking {
+    if (args.size < 2) {
+        println("Error: Please provide the project absolute path and a task as arguments")
+        println("Usage: <absolute_path> <task>")
+        return@runBlocking
+    }
+
+    val (path, task) = args
+    val input = "Project path: $path\n\n$task"
+    val result = agent.run(input)
+    println(result)
+}

examples/code-agent/step-01-basic-agent/src/main/resources/logback.xml

@@ -0,0 +1,11 @@
+<configuration>
+    <appender name="STDOUT" class="ch.qos.logback.core.ConsoleAppender">
+        <encoder>
+            <pattern>%d{HH:mm:ss.SSS} [%thread] %-5level %logger{36} - %msg%n</pattern>
+        </encoder>
+    </appender>
+
+    <root level="ERROR">
+        <appender-ref ref="STDOUT" />
+    </root>
+</configuration>

gradle/libs.versions.toml

@@ -22,7 +22,7 @@ mockito="5.19.0"
 mockk = "1.13.8"
 opentelemetry = "1.51.0"
 oshai-logging = "7.0.7"
-shadow = "8.1.1"
+shadow = "9.1.0"
 slf4j = "2.0.17"
 testcontainers = "1.19.7"
 exposed = "0.58.0"
@@ -120,5 +120,5 @@ kotlin-serialization = { id = "org.jetbrains.kotlin.plugin.serialization", versi
 kotlin-spring = { id = "org.jetbrains.kotlin.plugin.spring", version.ref = "kotlin" }
 kotlinx-kover = { id = "org.jetbrains.kotlinx.kover", version.ref = "kover" }
 ktlint = { id = "org.jlleitschuh.gradle.ktlint", version.ref = "ktlint" }
-shadow = { id = "com.github.johnrengelman.shadow", version.ref = "shadow" }
+shadow = { id = "com.gradleup.shadow", version.ref = "shadow" }
 spring-management = { id = "io.spring.dependency-management", version.ref = "spring-management" }

koog-agents/build.gradle.kts

@@ -13,6 +13,7 @@ val excluded = setOf(
     ":agents:agents-features:agents-features-sql", // Optional SQL persistence provider
     ":agents:agents-mcp-server",
     ":examples",
+    ":examples:code-agent:step-01-basic-agent",
     ":integration-tests",
     ":test-utils",
     ":koog-spring-boot-starter",

settings.gradle.kts

@@ -25,6 +25,7 @@ include(":agents:agents-tools")
 include(":agents:agents-utils")
 
 include(":examples")
+include(":examples:code-agent:step-01-basic-agent")
 
 include(":integration-tests")