github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-06-09 18:50:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(xai): ensure citations are streamed just once (#34556)

Browse Source
This commit is contained in:
ccurme
2025-12-31 18:01:41 -05:00
committed by GitHub
parent 9da28bac86
commit a7aad60989
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
libs/partners/xai/langchain_xai/chat_models.py
View File

@@ -605,6 +605,7 @@ class ChatXAI(BaseChatOpenAI): # type: ignore[override]
(citations := chunk.get("citations")) (citations := chunk.get("citations"))
and generation_chunk and generation_chunk
and isinstance(generation_chunk.message, AIMessageChunk) and isinstance(generation_chunk.message, AIMessageChunk)
and not chunk.get("usage") # citations are repeated in final usage chunk
): ):
generation_chunk.message.additional_kwargs["citations"] = citations generation_chunk.message.additional_kwargs["citations"] = citations