Update chat.py

chat.py

@@ -19,14 +19,16 @@ class Message(BaseModel):
 
 
 qna_system_message = """
-You are an assistant to an insurance firm who answers user queries on policy documents.
-User input will have the context required by you to answer user questions.
+You are an assistant to an insurance firm who answers customer queries based on their insurance policy documents.
+User input will have the context required by you to answer customer questions.
 This context will begin with the word: ###Context.
-The context contains references to specific portions of a document relevant to the user query.
+The context contains references to specific portions of a document relevant to the customer query.
 
-User questions will begin with the word: ###Question.
+Customer questions will begin with the word: ###Question.
 
-Please answer user questions only using the context provided in the input.
+Information about the customer will begin with the word: ###Customer Information
+
+Please answer user questions ONLY using the context provided in the input and the customer information.
 DO NOT mention anything about the context in your final answer. 
 Your response should only contain the answer to the question AND NOTHING ELSE.
 
@@ -34,6 +36,14 @@ If the answer is not found in the context, respond "Sorry, I cannot answer your
 """
 
 qna_user_message_template = """
+###Customer Information
+Policy Number: NBHTGBP22011V012223#
+Premium Amount: $15000
+Number of premium installments: 5
+Number of installments paid: 3
+Last Premium Paid: Yes
+Last Premium Date: 2024-05-12
+
 ###Context
 Here are some documents that are relevant to the question mentioned below.
 {context}