Update query_data.py

query_data.py

@@ -15,7 +15,7 @@ from langchain.schema import (
     SystemMessage
 )
 
-system_template = """Use the following pieces of context to answer the users question. 
+system_template = """Use the following pieces of context to answer the user's question. Use only the context to answer the question.
 If you don't know the answer, just say that you don't know, don't try to make up an answer.
 ----------------
 {context}"""
@@ -27,7 +27,7 @@ messages = [
 prompt = ChatPromptTemplate.from_messages(messages)
 
 _template = """Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question.
-You can assume the question about the syllabus of the H2 Economics, H2 History and H2 Geography A-Level Examinations in Singapore.
+You can assume the question about the law case between Review Publishing Company Limited and Lee Hsien Loong in Singapore.
 
 Chat History:
 {chat_history}
@@ -35,10 +35,10 @@ Follow Up Input: {question}
 Standalone question:"""
 CONDENSE_QUESTION_PROMPT = PromptTemplate.from_template(_template)
 
-#template = """You are an AI assistant for answering questions about history, geography or economics for the H2 A-Levels.
+#template = """You are an AI assistant for answering questions about the law case.
 #You are given the following extracted parts of a long document and a question. Provide a conversational answer.
 #If you don't know the answer, just say "Hmm, I'm not sure." Don't try to make up an answer.
-#If the question is not about history, geography or economics, politely inform them that you are tuned to only answer questions about it.
+#If the question is not about law, politely inform them that you are tuned to only answer questions about it.
 #Question: {question}
 #=========
 #{context}