Fix streaming final result (#611)

xebia-functional · Dec 28, 2023 · 46aed9e · 46aed9e
1 parent bdc12b6
commit 46aed9e
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/StreamedFunction.kt b/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/StreamedFunction.kt
@@ -112,11 +112,12 @@ sealed class StreamedFunction<out A> {
                 // because the previous chunk may had a partial property whose body
                 // may had not been fully streamed
                 streamProperty(path, currentProperty, functionCall.arguments, streamedProperties)
-
-                // we stream the result
-                streamResult(functionCall, messages, serializer)
               }
             }
+            if (finishReason != null) {
+              // we stream the result
+              streamResult(functionCall, messages, serializer)
+            }
           }
         }
     }