Migrate a Custom-Strategy Agent With a Shared HTTP Client to 1.0

plugins {
    kotlin("jvm") version "2.0.21"
    application
}

dependencies {
    implementation("ai.koog:koog-agents:0.7.1")
    implementation("io.ktor:ktor-client-cio:2.3.12")
}

kotlin {
    jvmToolchain(11)
}

import ai.koog.agents.core.dsl.builder.strategy
import ai.koog.agents.core.dsl.extension.*

val weatherChat = strategy<String, String>("weather-chat") {
    val callModel by nodeLLMRequest()
    val runTools by nodeExecuteTools()   // tool results are written back to the prompt automatically

    edge(nodeStart forwardTo callModel)
    edge(callModel forwardTo runTools onToolCall { true })
    edge(callModel forwardTo nodeFinish onAssistantMessage { true })
    edge(runTools forwardTo callModel)   // loop back to the model after the tools have run
}

import ai.koog.agents.core.agent.AIAgent
import ai.koog.prompt.executor.clients.openai.OpenAILLMClient
import ai.koog.prompt.executor.clients.openai.OpenAIModels
import ai.koog.prompt.executor.llms.SingleLLMPromptExecutor
import io.ktor.client.HttpClient
import io.ktor.client.engine.cio.CIO
import kotlinx.coroutines.runBlocking
import kotlin.time.Clock

fun main() = runBlocking {
    val sharedHttp = HttpClient(CIO) {
        // tuned timeouts + corporate proxy shared across the app
    }

    val client = OpenAILLMClient(System.getenv("OPENAI_API_KEY"), httpClient = sharedHttp)
    val executor = SingleLLMPromptExecutor(client)

    val agent = AIAgent.invoke(
        promptExecutor = executor,
        strategy = weatherChat,
        llmModel = OpenAIModels.Chat.GPT4o,
        systemPrompt = "You are a weather assistant.",
        toolRegistry = weatherTools,
        clock = Clock.System,
    )

    println(agent.run("What's the weather in Paris?"))
}