Supports automatic resource release (#13)

Author: qianmoQ

docs/docs/reference/provider/azure.md

@@ -0,0 +1,42 @@
+---
+title: Azure OpenAI
+---
+
+!!! note
+
+    Support the Open Ai service provided by Microsoft, product address: [https://azure.microsoft.com/zh-cn/products/cognitive-services/openai-service/](https://azure.microsoft.com/zh-cn/products/cognitive-services/openai-service/)
+
+### Required
+
+---
+
+!!! note
+
+    The following are some configurations that must be specified to invoke the service.
+
+|    name    | description                                                                 |
+|:----------:|-----------------------------------------------------------------------------|
+| `apiHost`  | Created zone markers in the format `${your-resource-name}.openai.azure.com` |  
+|  `apiKey`  | Azure token                                                                 |
+| `provider` | Specify `ProviderModel.azure`                                               |
+|  `model`   | Model name deployed in Azure                                                |
+| `version`  | Model version deployed in Azure                                             |
+
+### Example
+
+---
+
+```java
+// Automatic resource release
+try(OpenAiClient client=OpenAiClient.builder()
+        .apiHost("https://eus-chatgpt.openai.azure.com")
+        .apiKey(System.getProperty("azure.token"))
+        .provider(ProviderModel.azure)
+        .model("text-davinci-002")
+        .version("2022-12-01")
+        .build())
+{
+    client.createCompletion(configure).getChoices();
+}
+```
+

docs/mkdocs.yml

@@ -48,5 +48,7 @@ nav:
       - reference/models.md
       - reference/completions.md
       - reference/completions_chat.md
+      - Provider:
+          - reference/provider/azure.md
   - released.md
   - powered_by.md

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