Remove deprecated TokenStream API

Author: dliubarskyi

langchain4j-reactor/src/main/java/dev/langchain4j/reactor/TokenStreamToFluxAdapter.java

@@ -24,8 +24,8 @@ public boolean canAdaptTokenStreamTo(Type type) {
     @Override
     public Object adapt(TokenStream tokenStream) {
         Sinks.Many<String> sink = Sinks.many().unicast().onBackpressureBuffer();
-        tokenStream.onNext(sink::tryEmitNext)
-                .onComplete(aiMessageResponse -> sink.tryEmitComplete())
+        tokenStream.onPartialResponse(sink::tryEmitNext)
+                .onCompleteResponse(ignored -> sink.tryEmitComplete())
                 .onError(sink::tryEmitError)
                 .start();
         return sink.asFlux();

langchain4j-spring-boot-starter/src/test/java/dev/langchain4j/service/spring/mode/automatic/conflictingSyncAndStreamingModels/streaming/AiServiceWithConflictingSyncAndStreamingModelsIT.java

@@ -1,8 +1,7 @@
 package dev.langchain4j.service.spring.mode.automatic.conflictingSyncAndStreamingModels.streaming;
 
-import dev.langchain4j.data.message.AiMessage;
-import dev.langchain4j.model.chat.TestStreamingResponseHandler;
-import dev.langchain4j.model.output.Response;
+import dev.langchain4j.model.chat.TestStreamingChatResponseHandler;
+import dev.langchain4j.model.chat.response.ChatResponse;
 import dev.langchain4j.service.spring.AiServicesAutoConfig;
 import org.junit.jupiter.api.Test;
 import org.springframework.boot.autoconfigure.AutoConfigurations;
@@ -31,18 +30,18 @@ void should_create_AI_service_with_streaming_model_when_both_sync_and_streaming_
                     // given
                     AiServiceWithConflictingSyncAndStreamingModels aiService = context.getBean(AiServiceWithConflictingSyncAndStreamingModels.class);
 
-                    TestStreamingResponseHandler<AiMessage> handler = new TestStreamingResponseHandler<>();
+                    TestStreamingChatResponseHandler handler = new TestStreamingChatResponseHandler();
 
                     // when
                     aiService.chat("What is the capital of Germany?")
-                            .onNext(handler::onNext)
-                            .onComplete(handler::onComplete)
+                            .onPartialResponse(handler::onPartialResponse)
+                            .onCompleteResponse(handler::onCompleteResponse)
                             .onError(handler::onError)
                             .start();
-                    Response<AiMessage> response = handler.get();
+                    ChatResponse response = handler.get();
 
                     // then
-                    assertThat(response.content().text()).containsIgnoringCase("Berlin");
+                    assertThat(response.aiMessage().text()).containsIgnoringCase("Berlin");
                 });
     }
 }

langchain4j-spring-boot-starter/src/test/java/dev/langchain4j/service/spring/mode/automatic/streaming/StreamingAiServiceIT.java

@@ -1,8 +1,7 @@
 package dev.langchain4j.service.spring.mode.automatic.streaming;
 
-import dev.langchain4j.data.message.AiMessage;
-import dev.langchain4j.model.chat.TestStreamingResponseHandler;
-import dev.langchain4j.model.output.Response;
+import dev.langchain4j.model.chat.TestStreamingChatResponseHandler;
+import dev.langchain4j.model.chat.response.ChatResponse;
 import dev.langchain4j.service.spring.AiServicesAutoConfig;
 import org.junit.jupiter.api.Test;
 import org.springframework.boot.autoconfigure.AutoConfigurations;
@@ -31,18 +30,18 @@ void should_create_streaming_AI_service() {
                     // given
                     StreamingAiService aiService = context.getBean(StreamingAiService.class);
 
-                    TestStreamingResponseHandler<AiMessage> handler = new TestStreamingResponseHandler<>();
+                    TestStreamingChatResponseHandler handler = new TestStreamingChatResponseHandler();
 
                     // when
                     aiService.chat("What is the capital of Germany?")
-                            .onNext(handler::onNext)
-                            .onComplete(handler::onComplete)
+                            .onPartialResponse(handler::onPartialResponse)
+                            .onCompleteResponse(handler::onCompleteResponse)
                             .onError(handler::onError)
                             .start();
-                    Response<AiMessage> response = handler.get();
+                    ChatResponse response = handler.get();
 
                     // then
-                    assertThat(response.content().text()).containsIgnoringCase("Berlin");
+                    assertThat(response.aiMessage().text()).containsIgnoringCase("Berlin");
                 });
     }
 }