diff --git a/pom.xml b/pom.xml
index cfc2d8e..599e7ca 100644
--- a/pom.xml
+++ b/pom.xml
@@ -9,9 +9,9 @@
jar
- 1.8
+ 17
- 1.0.0-beta1
+ 1.0.0-beta2
@@ -50,7 +50,7 @@
cn.hutool
hutool-all
- 5.8.0.M2
+ 5.8.24
diff --git a/src/main/java/xyz/wbsite/ai/TestStreamChat.java b/src/main/java/xyz/wbsite/ai/TestStreamChat.java
index bd75309..eed13c7 100644
--- a/src/main/java/xyz/wbsite/ai/TestStreamChat.java
+++ b/src/main/java/xyz/wbsite/ai/TestStreamChat.java
@@ -1,10 +1,13 @@
package xyz.wbsite.ai;
import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.thread.ThreadUtil;
import dev.langchain4j.data.message.AiMessage;
import dev.langchain4j.data.message.ChatMessage;
import dev.langchain4j.data.message.UserMessage;
import dev.langchain4j.model.StreamingResponseHandler;
+import dev.langchain4j.model.chat.response.ChatResponse;
+import dev.langchain4j.model.chat.response.StreamingChatResponseHandler;
import dev.langchain4j.model.openai.OpenAiStreamingChatModel;
import dev.langchain4j.model.output.Response;
@@ -26,20 +29,25 @@ public class TestStreamChat {
UserMessage.from("假如树上有10只鸟,10分钟前飞走了2只,5分钟前又飞回了1只,刚刚又来了3只,那现在树上有几只鸟?")
);
- model.generate(messages, new StreamingResponseHandler() {
+ ThreadUtil.execAsync(new Runnable() {
@Override
- public void onNext(String s) {
- System.out.print(s);
- }
-
- @Override
- public void onError(Throwable throwable) {
- System.err.println(throwable.getMessage());
- }
-
- @Override
- public void onComplete(Response response) {
- System.out.println("onComplete");
+ public void run() {
+ model.chat(messages, new StreamingChatResponseHandler(){
+ @Override
+ public void onPartialResponse(String s) {
+ System.out.print(s);
+ }
+
+ @Override
+ public void onCompleteResponse(ChatResponse chatResponse) {
+ System.out.println("onComplete");
+ }
+
+ @Override
+ public void onError(Throwable throwable) {
+ System.err.println(throwable.getMessage());
+ }
+ });
}
});
}