Include User in Query Export (#59)

This commit is contained in:
Yuhong Sun 2024-03-30 00:08:45 -07:00 committed by Chris Weaver
parent e3ef620094
commit 2c3ba5f021

View File

@ -89,6 +89,7 @@ class QuestionAnswerPairSnapshot(BaseModel):
retrieved_documents: list[AbridgedSearchDoc] retrieved_documents: list[AbridgedSearchDoc]
feedback: QAFeedbackType | None feedback: QAFeedbackType | None
persona_name: str persona_name: str
user_email: str | None
time_created: datetime time_created: datetime
@classmethod @classmethod
@ -112,6 +113,7 @@ class QuestionAnswerPairSnapshot(BaseModel):
retrieved_documents=ai_message.documents, retrieved_documents=ai_message.documents,
feedback=ai_message.feedback, feedback=ai_message.feedback,
persona_name=chat_session_snapshot.persona_name, persona_name=chat_session_snapshot.persona_name,
user_email=chat_session_snapshot.user_email,
time_created=user_message.time_created, time_created=user_message.time_created,
) )
for user_message, ai_message in message_pairs for user_message, ai_message in message_pairs
@ -129,6 +131,7 @@ class QuestionAnswerPairSnapshot(BaseModel):
), ),
"feedback": self.feedback.value if self.feedback else "", "feedback": self.feedback.value if self.feedback else "",
"persona_name": self.persona_name, "persona_name": self.persona_name,
"user_email": self.user_email if self.user_email else "",
"time_created": str(self.time_created), "time_created": str(self.time_created),
} }