Skip to content

Commit 687b04a

Browse files
committed
feat: add timeout for handler
1 parent 743bbb7 commit 687b04a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/main/kotlin/cc/unitmesh/devti/llms/azure/AzureOpenAIProvider.kt

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ import okhttp3.MediaType.Companion.toMediaTypeOrNull
2525
import okhttp3.OkHttpClient
2626
import okhttp3.Request
2727
import okhttp3.RequestBody
28+
import java.time.Duration
2829

2930

3031
@Serializable
@@ -46,7 +47,8 @@ class AzureOpenAIProvider(val project: Project) : LLMProvider {
4647
private val autoDevSettingsState = AutoDevSettingsState.getInstance()
4748
private val url get() = autoDevSettingsState.customOpenAiHost
4849
private var customPromptConfig: CustomPromptConfig? = null
49-
private var client = OkHttpClient()
50+
private val timeout = Duration.ofSeconds(600)
51+
private var client = OkHttpClient().newBuilder().readTimeout(timeout).build()
5052
private val openAiVersion: String
5153

5254
init {

0 commit comments

Comments
 (0)