Skip to content

Commit e2dd6d3

Browse files
committed
Shutdown hook for the WS client (close)
1 parent ba378f3 commit e2dd6d3

File tree

1 file changed

+6
-1
lines changed
  • openai-client/src/main/scala/io/cequence/openaiscala/service/ws

1 file changed

+6
-1
lines changed

openai-client/src/main/scala/io/cequence/openaiscala/service/ws/WSHelper.scala

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,11 @@ trait WSHelper {
2828
.setRequestTimeout(timeouts.requestTimeout.getOrElse(DefaultTimeouts.requestTimeout))
2929
.build
3030
val asyncHttpClient = new DefaultAsyncHttpClient(asyncHttpClientConfig)
31-
new StandaloneAhcWSClient(asyncHttpClient)
31+
val client = new StandaloneAhcWSClient(asyncHttpClient)
32+
33+
// add a shutdown hook
34+
scala.sys.addShutdownHook(client.close())
35+
36+
client
3237
}
3338
}

0 commit comments

Comments
 (0)