github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-14 07:07:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
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
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/partners/mistralai/langchain_mistralai/chat_models.py
View File

@ -274,7 +274,9 @@ def _convert_chunk_to_message_chunk(
} }
else: else:
usage_metadata = None 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") response_metadata["model_name"] = chunk.get("model")
return AIMessageChunk( return AIMessageChunk(
content=content, content=content,