sdk支持流式文本

2023-12-30 19:09:30 +08:00
parent bc409042b0
commit 8a13a32615
2 changed files with 85 additions and 15 deletions
@@ -0,0 +1,48 @@
+package cn.wildfirechat.messagecontentbuilder;
+
+import cn.wildfirechat.pojos.MessagePayload;
+import cn.wildfirechat.proto.ProtoConstants;
+import com.google.gson.Gson;
+
+import java.nio.charset.StandardCharsets;
+import java.util.Base64;
+import java.util.HashMap;
+import java.util.Map;
+
+public class StreamingTextMessageContentBuilder extends MessageContentBuilder{
+    private String text;
+    private String streamId;
+    private boolean generating;
+    public static StreamingTextMessageContentBuilder newBuilder(String streamId) {
+        StreamingTextMessageContentBuilder builder = new StreamingTextMessageContentBuilder();
+        builder.generating = false;
+        builder.streamId = streamId;
+        return builder;
+    }
+    public StreamingTextMessageContentBuilder text(String text) {
+        this.text = text;
+        return this;
+    }
+
+    public StreamingTextMessageContentBuilder generating(boolean generating) {
+        this.generating = generating;
+        return this;
+    }
+
+    @Override
+    public MessagePayload build() {
+        MessagePayload payload = encodeBase();
+        payload.setSearchableContent(text);
+        payload.setContent(streamId);
+        if(generating) {
+            //正在生成消息的类型是14，存储标记位为透传为4。
+            payload.setType(14);
+            payload.setPersistFlag(4);
+        } else {
+            //已经生成消息的类型是15，存储标记位为透传为3。
+            payload.setType(15);
+            payload.setPersistFlag(3);
+        }
+        return payload;
+    }
+}