Merge pull request #510 from philippart-s/fix-npe-#489

fix: 🐛 fix NPE if chunk is null for Mistral AI
quarkiverse · Apr 25, 2024 · 924563e · 924563e
2 parents 1fcc955 + 0d8a93e
commit 924563e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/...al/runtime/src/main/java/io/quarkiverse/langchain4j/mistralai/QuarkusMistralAiClient.java b/...al/runtime/src/main/java/io/quarkiverse/langchain4j/mistralai/QuarkusMistralAiClient.java
@@ -79,7 +79,10 @@ public void accept(MistralAiChatCompletionResponse response) {
                 MistralAiChatCompletionChoice choice = response.getChoices().get(0);
                 String chunk = choice.getDelta().getContent();
                 contentBuilder.get().append(chunk);
-                handler.onNext(chunk);
+
+                if (chunk != null) {
+                    handler.onNext(chunk);
+                }
 
                 MistralAiUsage usageInfo = response.getUsage();
                 if (usageInfo != null) {