Skip to content

Commit 685553f

Browse files
stainless-botStainless Bot
authored andcommitted
fix(client): ensure SSE BufferedReader gets closed
chore: unknown commit message
1 parent d86694a commit 685553f

File tree

1 file changed

+6
-2
lines changed
  • openai-java-core/src/main/kotlin/com/openai/core/handlers

1 file changed

+6
-2
lines changed

openai-java-core/src/main/kotlin/com/openai/core/handlers/SseHandler.kt

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,9 @@ internal fun sseHandler(jsonMapper: JsonMapper): Handler<StreamResponse<SseMessa
1818
object : Handler<StreamResponse<SseMessage>> {
1919

2020
override fun handle(response: HttpResponse): StreamResponse<SseMessage> {
21+
val reader = response.body().bufferedReader()
2122
val sequence = sequence {
22-
response.body().bufferedReader().useLines { lines ->
23+
reader.useLines { lines ->
2324
val state = SseState(jsonMapper)
2425
var done = false
2526
for (line in lines) {
@@ -60,7 +61,10 @@ internal fun sseHandler(jsonMapper: JsonMapper): Handler<StreamResponse<SseMessa
6061
return object : StreamResponse<SseMessage> {
6162
override fun stream(): Stream<SseMessage> = sequence.asStream()
6263

63-
override fun close() = response.close()
64+
override fun close() {
65+
reader.close()
66+
response.close()
67+
}
6468
}
6569
}
6670
}

0 commit comments

Comments
 (0)