From a7aad60989581d80283ed2a13e19b41706daaec3 Mon Sep 17 00:00:00 2001 From: ccurme Date: Wed, 31 Dec 2025 18:01:41 -0500 Subject: [PATCH] fix(xai): ensure citations are streamed just once (#34556) --- libs/partners/xai/langchain_xai/chat_models.py | 1 + 1 file changed, 1 insertion(+) diff --git a/libs/partners/xai/langchain_xai/chat_models.py b/libs/partners/xai/langchain_xai/chat_models.py index 6c9bfc433bb..8e5d5c7b948 100644 --- a/libs/partners/xai/langchain_xai/chat_models.py +++ b/libs/partners/xai/langchain_xai/chat_models.py @@ -605,6 +605,7 @@ class ChatXAI(BaseChatOpenAI): # type: ignore[override] (citations := chunk.get("citations")) and generation_chunk 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