1

Author: sectlying

src/main/kotlin/API.kt

@@ -4,6 +4,7 @@ import io.ktor.client.*
 import io.ktor.client.call.*
 import io.ktor.client.engine.*
 import io.ktor.client.engine.okhttp.*
+import io.ktor.client.plugins.*
 import io.ktor.client.plugins.contentnegotiation.*
 import io.ktor.client.request.*
 import io.ktor.http.*
@@ -63,6 +64,10 @@ object Client {
             install(ContentNegotiation) {
                 json()
             }
+            install(HttpTimeout) {
+                requestTimeoutMillis = Config.timeout
+                socketTimeoutMillis = Config.timeout
+            }
             engine {
                 if (Config.httpProxyUrl != "")
                     proxy = ProxyBuilder.http(Config.httpProxyUrl)
@@ -75,6 +80,10 @@ object Client {
     fun sendRequest(requestObject: RequestObject): ReceivedObject {
         return runBlocking {
             return@runBlocking client.post(Config.apiUrl) {
+                timeout {
+                    requestTimeoutMillis = Config.timeout
+                    socketTimeoutMillis = Config.timeout
+                }
                 headers {
                     headers {
                         append(HttpHeaders.Authorization, "Bearer ${Config.token}")

src/main/kotlin/Command.kt

@@ -39,6 +39,7 @@ fun chatEventHandler(idL: Long, message: String, quote: QuoteReply): MessageChai
                 val content = (response.choices[0].message.content).trim()
                 MessageChainBuilder().append(quote).append(PlainText(content)).build()
             } catch (e: Exception) {
+                BotGPT.logger.warning(e.message)
                 MessageChainBuilder().append(quote).append(PlainText("内部错误")).build()
             }
         } else

src/main/kotlin/Config.kt

@@ -10,6 +10,7 @@ object Config : AutoSavePluginConfig("config") {
     val socksProxyPort by value(0)
     val token by value("ChatGPTToken")
     val expirationTime by value(60)
+    val timeout by value<Long>(100000)
     val cooldown by value(30)
     val model by value("gpt-3.5-turbo")
     val separator by value(" ")