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

2026-06-09 10:17:00 +00:00 · 2025-12-31 18:01:41 -05:00
parent 9da28bac86
commit a7aad60989
1 changed files with 1 additions and 0 deletions
--- 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