Upload app.py

app.py

@@ -334,18 +334,23 @@ query_generator_template_llama2 = f"<s>[INST] <<SYS>>{query_generator_message}<<
 
 
 # 2. 生成された質問文を用いて、参考文献を要約するchain のプロンプト
+# question_prompt_message = """
+# From the following references, extract key information relevant to the question
+# and summarize it in a natural English sentence with clear subject, verb, object,
+# and complement. If there is no information in the reference that answers the question,
+# do not summarize and simply answer "NO INFO"
+# """.replace("\n", "")
+
 question_prompt_message = """
-From the following references, extract key information relevant to the question
-and summarize it in a natural English sentence with clear subject, verb, object,
-and complement. If there is no information in the reference that answers the question,
-do not summarize and simply answer "NO INFO"
-""".replace("\n", "")
+1. Determine if any of the following references provide information that answers the Question, and if there is no information, answer "NO INFO" and stop.
+2. From the following references, extract key information relevant to the question and summarize it in a natural English sentence with clear subject, verb, object,　and complement.
+""".strip()
 
 question_prompt_common_format = """
 ===
 [Question] {query}
 [references] {context}
-[Summary] """
+[Answer]"""
 
 question_prompt_template_std = f"{question_prompt_message}{question_prompt_common_format}"
 question_prompt_template_llama2 = f"<s>[INST] <<SYS>>{question_prompt_message}<</SYS>>{question_prompt_common_format}[/INST]"
@@ -570,7 +575,7 @@ def set_chains(ss: SessionState, summarization_mode) -> SessionState:
     ss.conversation_chain = ConversationChain(
       llm         = ss.llm,
       prompt      = chat_prompt,
-      memory      = ss.memory,
+      # memory      = ss.memory,
       input_key   = "query",
       output_key  = "output_text",
       verbose     = True,