Support create new api key

Author: qianmoQ

docs/docs/reference/users.md

@@ -43,3 +43,32 @@ Returns:
 }
 ```
 
+### Create a key
+
+---
+
+Create a new key.
+
+```java
+UserKeyEntity configure=UserKeyEntity.builder()
+        .name("Create first key")
+        .action("create")
+        .build();
+client.createUserAPIKey(configure)
+```
+
+Return:
+
+```json
+{
+  "result": "success",
+  "key": {
+    "sensitive_id": "sk-xxx",
+    "object": "api_key",
+    "name": "Test",
+    "created": 1688525108,
+    "last_use": null,
+    "publishable": false
+  }
+}
+```

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

@@ -4,6 +4,7 @@
 import org.devlive.sdk.openai.entity.CompletionChatEntity;
 import org.devlive.sdk.openai.entity.CompletionEntity;
 import org.devlive.sdk.openai.entity.ModelEntity;
+import org.devlive.sdk.openai.entity.UserKeyEntity;
 import org.devlive.sdk.openai.response.CompleteChatResponse;
 import org.devlive.sdk.openai.response.CompleteResponse;
 import org.devlive.sdk.openai.response.ModelResponse;
@@ -46,4 +47,10 @@ public interface DefaultApi
      */
     @GET(value = "dashboard/user/api_keys")
     Single<UserKeyResponse> fetchUserAPIKeys();
+
+    /**
+     * Create a key for the given
+     */
+    @POST(value = "dashboard/user/api_keys")
+    Single<UserKeyResponse> fetchCreateUserAPIKey(@Body UserKeyEntity configure);
 }

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

@@ -3,6 +3,7 @@
 import org.devlive.sdk.openai.entity.CompletionChatEntity;
 import org.devlive.sdk.openai.entity.CompletionEntity;
 import org.devlive.sdk.openai.entity.ModelEntity;
+import org.devlive.sdk.openai.entity.UserKeyEntity;
 import org.devlive.sdk.openai.response.CompleteChatResponse;
 import org.devlive.sdk.openai.response.CompleteResponse;
 import org.devlive.sdk.openai.response.ModelResponse;
@@ -41,4 +42,10 @@ public UserKeyResponse getKeys()
         return this.api.fetchUserAPIKeys()
                 .blockingGet();
     }
+
+    public UserKeyResponse createUserAPIKey(UserKeyEntity configure)
+    {
+        return this.api.fetchCreateUserAPIKey(configure)
+                .blockingGet();
+    }
 }

src/main/java/org/devlive/sdk/openai/entity/UserKeyEntity.java

@@ -35,5 +35,11 @@ public class UserKeyEntity
     private String lastUseTime;
 
     @JsonProperty(value = "publishable")
-    private boolean publishable;
+    private Boolean publishable;
+
+    /**
+     * When creating a key use
+     */
+    @JsonProperty(value = "action")
+    private String action;
 }

src/main/java/org/devlive/sdk/openai/interceptor/DefaultInterceptor.java

@@ -8,6 +8,7 @@
 import okhttp3.Request;
 import okhttp3.Response;
 import okhttp3.ResponseBody;
+import okio.Buffer;
 import org.apache.commons.lang3.ObjectUtils;
 import org.apache.commons.lang3.StringUtils;
 import org.devlive.sdk.openai.exception.AuthorizedException;
@@ -50,6 +51,13 @@ public Response intercept(Chain chain) throws IOException
 
         Request original = chain.request();
         Request request = this.headers(original);
+
+        if (ObjectUtils.isNotEmpty(request.body())) {
+            Buffer buffer = new Buffer();
+            request.body().writeTo(buffer);
+            log.debug("Request body {}", buffer.readUtf8());
+        }
+
         Response response = chain.proceed(request);
         if (!response.isSuccessful()) {
             log.error("Failed to intercept request");
@@ -72,7 +80,7 @@ public Response intercept(Chain chain) throws IOException
         }
 
         // Has error
-        if (response.code() == 404 || response.code() == 400) {
+        if (response.code() == 404 || response.code() == 400 || response.code() == 403) {
             ResponseBody body = response.body();
             if (ObjectUtils.isEmpty(body)) {
                 throw new NullPointerException("Failed to intercept request because no body");

src/main/java/org/devlive/sdk/openai/response/UserKeyResponse.java

@@ -22,4 +22,10 @@ public class UserKeyResponse
 
     @JsonProperty(value = "data")
     private List<UserKeyEntity> keys;
+
+    @JsonProperty(value = "result")
+    private String result;
+
+    @JsonProperty(value = "key")
+    private UserKeyEntity key;
 }

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

@@ -5,7 +5,9 @@
 import org.devlive.sdk.openai.entity.CompletionChatEntity;
 import org.devlive.sdk.openai.entity.CompletionEntity;
 import org.devlive.sdk.openai.entity.CompletionMessageEntity;
+import org.devlive.sdk.openai.entity.UserKeyEntity;
 import org.devlive.sdk.openai.exception.AuthorizedException;
+import org.devlive.sdk.openai.exception.RequestException;
 import org.devlive.sdk.openai.model.CompletionModel;
 import org.junit.Assert;
 import org.junit.Before;
@@ -118,4 +120,14 @@ public void testGetKeys()
     {
         Assert.assertNotNull(client.getKeys());
     }
+
+    @Test
+    public void testCreateUserAPIKey()
+    {
+        UserKeyEntity configure = UserKeyEntity.builder()
+                .name("Create first key")
+                .action("create")
+                .build();
+        Assert.assertThrows(RequestException.class, () -> client.createUserAPIKey(configure));
+    }
 }