diff --git a/libs/community/langchain_community/chat_models/perplexity.py b/libs/community/langchain_community/chat_models/perplexity.py index 88547c67d29..ce415dd59cb 100644 --- a/libs/community/langchain_community/chat_models/perplexity.py +++ b/libs/community/langchain_community/chat_models/perplexity.py @@ -261,7 +261,10 @@ class ChatPerplexity(BaseChatModel): response = self.client.chat.completions.create( model=params["model"], messages=message_dicts ) - message = AIMessage(content=response.choices[0].message.content) + message = AIMessage( + content=response.choices[0].message.content, + additional_kwargs={"citations": response.citations}, + ) return ChatResult(generations=[ChatGeneration(message=message)]) @property