fix(answerStream): handle end of stream

coveo · Aug 13, 2024 · 7f353c3 · 7f353c3
1 parent 87ef80c
commit 7f353c3
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/packages/headless/src/api/knowledge/stream-answer-api.ts b/packages/headless/src/api/knowledge/stream-answer-api.ts
@@ -149,7 +149,7 @@ const updateCacheWithEvent = (
       }
       break;
     case 'genqa.endOfStreamType':
-      if (parsedPayload.answerGenerated) {
+      if (draft.answer?.length || parsedPayload.answerGenerated) {
         handleEndOfStream(draft, parsedPayload);
       }
       break;
@@ -236,9 +236,7 @@ const constructAnswerQueryParams = (state: StateNeededByAnswerAPI) => {
     q,
     pipelineRuleParameters: {
       mlGenerativeQuestionAnswering: {
-        responseFormat: {
-          answerStyle: state.generatedAnswer.responseFormat.answerStyle,
-        },
+        responseFormat: state.generatedAnswer.responseFormat,
         citationsFieldToInclude,
       },
     },