Fix Bug: Incorrect Type for logprobs Field in CompletionResp Class Ha…

…maWhiteGG#144
sime2408 · Dec 6, 2023 · 2a580b8 · 2a580b8
1 parent 9965a78
commit 2a580b8
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 1 deletion.
diff --git a/openai-client/src/main/java/com/hw/openai/entity/completions/Choice.java b/openai-client/src/main/java/com/hw/openai/entity/completions/Choice.java
@@ -33,7 +33,7 @@ public class Choice {
 
     private Integer index;
 
-    private Integer logprobs;
+    private LogProbs logprobs;
 
     @JsonProperty("finish_reason")
     private String finishReason;

diff --git a/openai-client/src/main/java/com/hw/openai/entity/completions/LogProbs.java b/openai-client/src/main/java/com/hw/openai/entity/completions/LogProbs.java
@@ -0,0 +1,25 @@
+package com.hw.openai.entity.completions;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.Data;
+
+import java.util.List;
+import java.util.Map;
+
+/**
+ * @author HamaWhite
+ */
+@Data
+public class LogProbs {
+
+    List<String> tokens;
+
+    @JsonProperty("token_logprobs")
+    List<Double> tokenLogprobs;
+
+    @JsonProperty("top_logprobs")
+    List<Map<String, Double>> topLogprobs;
+
+    @JsonProperty("text_offset")
+    List<Integer> textOffset;
+}