From 299b222c5330d28da3eaa5b488eb1cd17192ab23 Mon Sep 17 00:00:00 2001 From: ccurme Date: Wed, 26 Mar 2025 12:30:09 -0400 Subject: [PATCH] mistral[patch]: check types in adding model_name to response_metadata (#30499) --- libs/partners/mistralai/langchain_mistralai/chat_models.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/libs/partners/mistralai/langchain_mistralai/chat_models.py b/libs/partners/mistralai/langchain_mistralai/chat_models.py index 7cdac2bcab3..6f3cac19904 100644 --- a/libs/partners/mistralai/langchain_mistralai/chat_models.py +++ b/libs/partners/mistralai/langchain_mistralai/chat_models.py @@ -274,7 +274,9 @@ def _convert_chunk_to_message_chunk( } else: usage_metadata = None - if _choice.get("finish_reason") is not None: + if _choice.get("finish_reason") is not None and isinstance( + chunk.get("model"), str + ): response_metadata["model_name"] = chunk.get("model") return AIMessageChunk( content=content,