From 195ae7baa308f3a697928dc2826360f7569b9693 Mon Sep 17 00:00:00 2001
From: Mohammad Mohtashim <45242107+keenborder786@users.noreply.github.com>
Date: Tue, 26 Nov 2024 19:45:47 +0500
Subject: [PATCH] Community: Adding citations in AIMessage for ChatPerplexity
 (#28321)

**Description**: Adding Citation in response payload of ChatPerplexity
**Issue**: #28108
---
 libs/community/langchain_community/chat_models/perplexity.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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