mistral[patch]: check types in adding model_name to response_metadata (#30499)

This commit is contained in:
ccurme 2025-03-26 12:30:09 -04:00 committed by GitHub
parent 22d1a7d7b6
commit 299b222c53
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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,