github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-24 07:35:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

langchain_mistralai: Include finish_reason in response metadata when parsing MistralAI chunks toAIMessageChunk (#31667)

Browse Source 
## Description
<!-- What does this pull request accomplish? -->
- When parsing MistralAI chunk dicts to Langchain to `AIMessageChunk`
schemas via the `_convert_chunk_to_message_chunk` utility function, the
`finish_reason` was not being included in `response_metadata` as it is
for other providers.
- This PR adds a one-liner fix to include the finish reason.

- fixes: https://github.com/langchain-ai/langchain/issues/31666
This commit is contained in:
Saran Connolly 2025-06-20 20:41:20 +01:00 committed by GitHub
parent 7ff405077d
commit 22e6d90937
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -271,7 +271,8 @@ def _convert_chunk_to_message_chunk(
if _choice.get("finish_reason") is not None and isinstance( if _choice.get("finish_reason") is not None and isinstance(
chunk.get("model"), str chunk.get("model"), str
): ):
response_metadata["model_name"] = chunk.get("model") response_metadata["model_name"] = chunk["model"]
response_metadata["finish_reason"] = _choice["finish_reason"]
return AIMessageChunk( return AIMessageChunk(
content=content, content=content,
additional_kwargs=additional_kwargs, additional_kwargs=additional_kwargs,