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