From 8dff735a568d723efaacd942a2af7b2015d1943d Mon Sep 17 00:00:00 2001
From: TaBuE <840545133@qq.com>
Date: Fri, 14 Nov 2025 21:28:10 +0800
Subject: [PATCH] [KTOR-8881] Determine whether the timeout is configured in
 application.yaml; if it is, read the configuration instead of exiting
 directly.(#807 OR #906)

---
 .../kotlin/ai/koog/ktor/utils/EnvConfigLoader.kt       | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/koog-ktor/src/commonMain/kotlin/ai/koog/ktor/utils/EnvConfigLoader.kt b/koog-ktor/src/commonMain/kotlin/ai/koog/ktor/utils/EnvConfigLoader.kt
index 597f882714..ad67b6abf0 100644
--- a/koog-ktor/src/commonMain/kotlin/ai/koog/ktor/utils/EnvConfigLoader.kt
+++ b/koog-ktor/src/commonMain/kotlin/ai/koog/ktor/utils/EnvConfigLoader.kt
@@ -145,17 +145,19 @@ private inline fun KoogAgentsConfig.config(
 }
 
 private fun KoogAgentsConfig.TimeoutConfiguration.configure(key: String, config: ApplicationConfig) {
-    if (config.propertyOrNull(key) == null) {
-        config.propertyOrNull("requestTimeoutMillis")
+    if (config.propertyOrNull(key) != null) {
+        val configKey = config.config(key)
+
+        configKey.propertyOrNull("requestTimeoutMillis")
             ?.getString()
             ?.toLongOrNull()
             ?.let { requestTimeout = it.milliseconds }
 
-        config.propertyOrNull("connectTimeoutMillis")
+        configKey.propertyOrNull("connectTimeoutMillis")
             ?.getString()
             ?.toLongOrNull()
             ?.let { connectTimeout = it.milliseconds }
-        config.propertyOrNull("socketTimeoutMillis")
+        configKey.propertyOrNull("socketTimeoutMillis")
             ?.getString()
             ?.toLongOrNull()
             ?.let { socketTimeout = it.milliseconds }