feat: forward zhipu finish_reason (#7560)

langgenius · Aug 23, 2024 · a247177 · a247177
1 parent a40073a
commit a247177
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
                     )
                 )