Merge branch 'feat/workflow-backend' into deploy/dev

langgenius · Mar 17, 2024 · 4cffcc0 · 4cffcc0
2 parents e46381a + a2b3096
commit 4cffcc0
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/api/core/app/apps/advanced_chat/generate_task_pipeline.py b/api/core/app/apps/advanced_chat/generate_task_pipeline.py
@@ -160,6 +160,22 @@ def _process_blocking_response(self) -> ChatbotAppBlockingResponse:
  self._save_message()
 
  return self._to_blocking_response()
+ elif isinstance(event, QueueTextChunkEvent):
+ delta_text = event.text
+ if delta_text is None:
+ continue
+
+ if not self._is_stream_out_support(
+ event=event
+ ):
+ continue
+
+ # handle output moderation chunk
+ should_direct_answer = self._handle_output_moderation_chunk(delta_text)
+ if should_direct_answer:
+ continue
+
+ self._task_state.answer += delta_text
  else:
  continue