Optimize code variables HamaWhiteGG#146

dinsy008 · Dec 7, 2023 · 7636f02 · 7636f02
1 parent b0b7463
commit 7636f02
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 12 deletions.
diff --git a/openai-client/src/main/java/com/hw/openai/OpenAiClient.java b/openai-client/src/main/java/com/hw/openai/OpenAiClient.java
@@ -30,7 +30,7 @@
 import com.hw.openai.entity.completions.CompletionResp;
 import com.hw.openai.entity.embeddings.Embedding;
 import com.hw.openai.entity.embeddings.EmbeddingResp;
-import com.hw.openai.entity.image.CreateImageRequest;
+import com.hw.openai.entity.image.CreateImageReq;
 import com.hw.openai.entity.image.ImageResp;
 import com.hw.openai.entity.models.Model;
 import com.hw.openai.entity.models.ModelResp;
@@ -321,7 +321,7 @@ public EmbeddingResp createEmbedding(Embedding embedding) {
      * @param request the create image request
      * @return the image result response
      */
-    public ImageResp createImage(CreateImageRequest request) {
+    public ImageResp createImage(CreateImageReq request) {
         return execute(service.createImage(request));
     }
 

diff --git a/...enai/entity/image/CreateImageRequest.java → ...w/openai/entity/image/CreateImageReq.java b/...enai/entity/image/CreateImageRequest.java → ...w/openai/entity/image/CreateImageReq.java
@@ -31,7 +31,7 @@
 @Data
 @Builder
 @JsonInclude(JsonInclude.Include.NON_NULL)
-public class CreateImageRequest {
+public class CreateImageReq {
 
     /**
      * A text description of the desired image(s).
@@ -64,7 +64,7 @@ public class CreateImageRequest {
      */
     @JsonProperty("response_format")
     @Builder.Default
-    private ImageResponseFormat responseFormat = ImageResponseFormat.URL;
+    private ImageRespFormat responseFormat = ImageRespFormat.URL;
 
     /**
      * The size of the generated images. Must be one of 256x256, 512x512, or 1024x1024 for dall-e-2.

diff --git a/...nai/entity/image/ImageResponseFormat.java → .../openai/entity/image/ImageRespFormat.java b/...nai/entity/image/ImageResponseFormat.java → .../openai/entity/image/ImageRespFormat.java
@@ -24,15 +24,15 @@
 /**
  * @author HamaWhite
  */
-public enum ImageResponseFormat {
+public enum ImageRespFormat {
 
     URL("url"),
 
     BASE64_JSON("b64_json");
 
     private final String value;
 
-    ImageResponseFormat(String value) {
+    ImageRespFormat(String value) {
         this.value = value;
     }
 
@@ -42,8 +42,8 @@ public String getValue() {
     }
 
     @JsonCreator
-    public static ImageResponseFormat fromValue(String value) {
-        for (ImageResponseFormat item : ImageResponseFormat.values()) {
+    public static ImageRespFormat fromValue(String value) {
+        for (ImageRespFormat item : ImageRespFormat.values()) {
             if (item.value.equalsIgnoreCase(value)) {
                 return item;
             }

diff --git a/openai-client/src/main/java/com/hw/openai/service/OpenAiService.java b/openai-client/src/main/java/com/hw/openai/service/OpenAiService.java
@@ -24,7 +24,7 @@
 import com.hw.openai.entity.completions.CompletionResp;
 import com.hw.openai.entity.embeddings.Embedding;
 import com.hw.openai.entity.embeddings.EmbeddingResp;
-import com.hw.openai.entity.image.CreateImageRequest;
+import com.hw.openai.entity.image.CreateImageReq;
 import com.hw.openai.entity.image.ImageResp;
 import com.hw.openai.entity.models.Model;
 import com.hw.openai.entity.models.ModelResp;
@@ -150,5 +150,5 @@ Single<EmbeddingResp> createEmbedding(@Path("deploymentId") String deploymentId,
      * @return A Single emitting the response containing the generated image.
      */
     @POST("images/generations")
-    Single<ImageResp> createImage(@Body CreateImageRequest request);
+    Single<ImageResp> createImage(@Body CreateImageReq request);
 }
diff --git a/openai-client/src/test/java/com/hw/openai/images/ImageTest.java b/openai-client/src/test/java/com/hw/openai/images/ImageTest.java
@@ -19,7 +19,7 @@
 package com.hw.openai.images;
 
 import com.hw.openai.OpenAiClientTest;
-import com.hw.openai.entity.image.CreateImageRequest;
+import com.hw.openai.entity.image.CreateImageReq;
 import com.hw.openai.entity.image.Image;
 
 import org.junit.jupiter.api.Disabled;
@@ -42,7 +42,7 @@ public class ImageTest extends OpenAiClientTest {
 
     @Test
     void testCreateImage() {
-        CreateImageRequest request = CreateImageRequest.builder()
+        CreateImageReq request = CreateImageReq.builder()
                 .model("dall-e-3")
                 .prompt("A cute baby sea otter")
                 .n(1)