fix: pydantic2 error (#5172)

2024-06-14 03:05:04 +08:00
parent edf2047f04
commit 415022aa14
4 changed files with 5 additions and 5 deletions
--- a/api/core/model_runtime/model_providers/google/llm/llm.py
+++ b/api/core/model_runtime/model_providers/google/llm/llm.py
@@ -313,7 +313,7 @@ class GoogleLargeLanguageModel(LargeLanguageModel):
                        delta=LLMResultChunkDelta(
                            index=index,
                            message=assistant_prompt_message,
-                            finish_reason=chunk.candidates[0].finish_reason,
+                            finish_reason=str(chunk.candidates[0].finish_reason),
                            usage=usage
                        )
                    )