Supports automatic resource release

Author: qianmoQ

src/main/java/org/devlive/sdk/openai/DefaultClient.java

@@ -1,6 +1,8 @@
 package org.devlive.sdk.openai;
 
 import lombok.extern.slf4j.Slf4j;
+import okhttp3.OkHttpClient;
+import org.apache.commons.lang3.ObjectUtils;
 import org.devlive.sdk.openai.entity.CompletionChatEntity;
 import org.devlive.sdk.openai.entity.CompletionEntity;
 import org.devlive.sdk.openai.entity.ModelEntity;
@@ -14,10 +16,11 @@
 import org.devlive.sdk.openai.utils.ProviderUtils;
 
 @Slf4j
-public abstract class DefaultClient
+public abstract class DefaultClient implements AutoCloseable
 {
     protected DefaultApi api;
     protected ProviderModel provider;
+    protected OkHttpClient client;
 
     public ModelResponse getModels()
     {
@@ -54,4 +57,13 @@ public UserKeyResponse createUserAPIKey(UserKeyEntity configure)
         return this.api.fetchCreateUserAPIKey(configure)
                 .blockingGet();
     }
+
+    public void close()
+    {
+        if (ObjectUtils.isNotEmpty(this.client)) {
+            this.client.dispatcher().cancelAll();
+            this.client.connectionPool().evictAll();
+            this.client = null;
+        }
+    }
 }

src/main/java/org/devlive/sdk/openai/OpenAiClient.java

@@ -70,6 +70,7 @@ private OpenAiClient(OpenAiClientBuilder builder)
         }
 
         super.provider = builder.provider;
+        super.client = builder.client;
         // Build a remote API client
         objectMapper.setSerializationInclusion(JsonInclude.Include.NON_NULL);
         this.api = new Retrofit.Builder()

src/test/java/org/devlive/sdk/openai/OpenAiClientTest.java

@@ -56,10 +56,11 @@ public void testClient()
     @Test
     public void testNoAuthorized()
     {
-        client = OpenAiClient.builder()
+        try (OpenAiClient client = OpenAiClient.builder()
                 .apiKey(invalidApiKey)
-                .build();
-        Assert.assertThrows(AuthorizedException.class, () -> client.getModels());
+                .build()) {
+            Assert.assertThrows(AuthorizedException.class, () -> client.getModels());
+        }
     }
 
     @Test