From 2c3ba5f0216fe154d7854b46822487e9867a0cbc Mon Sep 17 00:00:00 2001
From: Yuhong Sun <yuhongsun96@gmail.com>
Date: Sat, 30 Mar 2024 00:08:45 -0700
Subject: [PATCH] Include User in Query Export (#59)

---
 backend/ee/danswer/server/query_history/api.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/backend/ee/danswer/server/query_history/api.py b/backend/ee/danswer/server/query_history/api.py
index afcffee00..e4cf7f578 100644
--- a/backend/ee/danswer/server/query_history/api.py
+++ b/backend/ee/danswer/server/query_history/api.py
@@ -89,6 +89,7 @@ class QuestionAnswerPairSnapshot(BaseModel):
     retrieved_documents: list[AbridgedSearchDoc]
     feedback: QAFeedbackType | None
     persona_name: str
+    user_email: str | None
     time_created: datetime
 
     @classmethod
@@ -112,6 +113,7 @@ class QuestionAnswerPairSnapshot(BaseModel):
                 retrieved_documents=ai_message.documents,
                 feedback=ai_message.feedback,
                 persona_name=chat_session_snapshot.persona_name,
+                user_email=chat_session_snapshot.user_email,
                 time_created=user_message.time_created,
             )
             for user_message, ai_message in message_pairs
@@ -129,6 +131,7 @@ class QuestionAnswerPairSnapshot(BaseModel):
             ),
             "feedback": self.feedback.value if self.feedback else "",
             "persona_name": self.persona_name,
+            "user_email": self.user_email if self.user_email else "",
             "time_created": str(self.time_created),
         }