bossanyit/workouttest_server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

v1.2.6.1 openai timeout

Browse Source
This commit is contained in:
bossanyit 2023-05-04 08:05:09 +02:00
parent 63c0dc5595
commit a8d54fa78f
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.gitignore vendored
View File

@ -30,4 +30,4 @@ out/
### VS Code ### ### VS Code ###
.vscode/ .vscode/
logs/aitrainer.log logs/

6
src/main/kotlin/com/aitrainer/api/openai/OpenAIService.kt
View File

@ -7,6 +7,7 @@ import com.aallam.openai.api.chat.ChatMessage
import com.aallam.openai.client.OpenAI import com.aallam.openai.client.OpenAI
import com.aallam.openai.api.completion.CompletionRequest import com.aallam.openai.api.completion.CompletionRequest
import com.aallam.openai.api.completion.TextCompletion import com.aallam.openai.api.completion.TextCompletion
import com.aallam.openai.api.http.Timeout
import com.aallam.openai.api.logging.LogLevel import com.aallam.openai.api.logging.LogLevel
import com.aallam.openai.api.model.Model import com.aallam.openai.api.model.Model
import com.aallam.openai.api.model.ModelId import com.aallam.openai.api.model.ModelId
@ -16,6 +17,8 @@ import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.withContext import kotlinx.coroutines.withContext
import org.springframework.beans.factory.annotation.Value import org.springframework.beans.factory.annotation.Value
import org.springframework.stereotype.Service import org.springframework.stereotype.Service
import kotlin.time.Duration.Companion.seconds
@Service @Service
class OpenAIService(@Value("\${openai.key}") private val openaiKey: String, private val modelName: String?, private val temperature: Double?) { class OpenAIService(@Value("\${openai.key}") private val openaiKey: String, private val modelName: String?, private val temperature: Double?) {
@ -26,7 +29,8 @@ class OpenAIService(@Value("\${openai.key}") private val openaiKey: String, priv
private suspend fun connect(modelName: String) { private suspend fun connect(modelName: String) {
val config = OpenAIConfig( val config = OpenAIConfig(
token = openaiKey, token = openaiKey,
logLevel = LogLevel.All logLevel = LogLevel.All,
timeout = Timeout(socket = 300.seconds)
) )
openAI = OpenAI(config) openAI = OpenAI(config)
modelId = ModelId(modelName) modelId = ModelId(modelName)