fix null values validation

czelabueno · Jan 3, 2025 · 22baa45 · 22baa45
1 parent 3a9229c
commit 22baa45
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/...flow/src/main/java/com/github/czelabueno/jai/workflow/langchain4j/node/StreamingNode.java b/...flow/src/main/java/com/github/czelabueno/jai/workflow/langchain4j/node/StreamingNode.java
@@ -121,12 +121,11 @@ private static <T extends AbstractStatefulBean> Flux<String> streamingFunction(
             StreamingChatLanguageModel streamingChatLanguageModel) {
         Sinks.Many<String> sink = Sinks.many().unicast().onBackpressureBuffer();
         CompletableFuture<AiMessage> futureResponse = new CompletableFuture<>();
-        messages = getOrDefault(
-                messages,
-                doUserMessage != null
-                        ? Arrays.asList(doUserMessage.apply(statefulBean))
-                        : Arrays.asList(UserMessage.from(statefulBean.getQuestion()))
-        );
+        if (messages == null || messages.isEmpty()) {
+            messages = doUserMessage != null ?
+                    List.of(doUserMessage.apply(statefulBean)) :
+                    List.of(UserMessage.from(getOrDefault(statefulBean.getQuestion(),"No question provided.")));
+        }
 
         streamingChatLanguageModel.generate(
                 messages,