feat: forward zhipu finish_reason

langgenius · Aug 23, 2024 · 8b9013b · 8b9013b
1 parent 0006c6f
commit 8b9013b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/api/core/model_runtime/model_providers/zhipuai/llm/llm.py b/api/core/model_runtime/model_providers/zhipuai/llm/llm.py
@@ -444,6 +444,7 @@ def _handle_generate_stream_response(self, model: str,
                     delta=LLMResultChunkDelta(
                         index=delta.index,
                         message=assistant_prompt_message,
+                        finish_reason=delta.finish_reason
                     )
                 )