added citation to streaming

langchain-ai · Jan 13, 2025 · 7d3dc5e · 7d3dc5e
1 parent e64bfb5
commit 7d3dc5e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/libs/community/langchain_community/chat_models/perplexity.py b/libs/community/langchain_community/chat_models/perplexity.py
@@ -228,10 +228,12 @@ def _stream(
                 chunk = chunk.dict()
             if len(chunk["choices"]) == 0:
                 continue
+            citations = chunk.get("citations", [])
             choice = chunk["choices"][0]
             chunk = self._convert_delta_to_message_chunk(
                 choice["delta"], default_chunk_class
             )
+            chunk.additional_kwargs["citations"] = citations
             finish_reason = choice.get("finish_reason")
             generation_info = (
                 dict(finish_reason=finish_reason) if finish_reason is not None else None