File tree Expand file tree Collapse file tree 2 files changed +3
-2
lines changed
src/main/kotlin/cc/unitmesh/devti/llms Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -173,6 +173,7 @@ class AzureOpenAIProvider(val project: Project) : LLMProvider {
173
173
ObjectMapper ().readValue(sse!! .data, ChatCompletionResult ::class .java)
174
174
val completion = result.choices[0 ].message
175
175
if (completion != null && completion.content != null ) {
176
+ output + = completion.content
176
177
trySend(completion.content)
177
178
}
178
179
}
Original file line number Diff line number Diff line change @@ -52,8 +52,7 @@ class CustomLLMProvider(val project: Project) : LLMProvider {
52
52
53
53
private var client = OkHttpClient ()
54
54
private val timeout = Duration .ofSeconds(600 )
55
- private val messages: MutableList <Message > = ArrayList ()
56
-
55
+ private val messages: MutableList <Message > = mutableListOf ()
57
56
private val logger = logger<CustomLLMProvider >()
58
57
59
58
override fun clearMessage () {
@@ -139,6 +138,7 @@ class CustomLLMProvider(val project: Project) : LLMProvider {
139
138
}
140
139
}
141
140
}
141
+
142
142
close()
143
143
}
144
144
awaitClose()
You can’t perform that action at this time.
0 commit comments