|
@@ -157,11 +157,6 @@ And answer according to the language of the user's question.
|
|
|
else:
|
|
|
messages: List[BaseMessage] = []
|
|
|
|
|
|
- if pre_prompt:
|
|
|
- # append pre prompt as system message
|
|
|
- system_message = PromptBuilder.to_system_message(pre_prompt, inputs)
|
|
|
- messages.append(system_message)
|
|
|
-
|
|
|
human_inputs = {
|
|
|
"query": query
|
|
|
}
|
|
@@ -170,7 +165,7 @@ And answer according to the language of the user's question.
|
|
|
|
|
|
if chain_output:
|
|
|
human_inputs['context'] = chain_output
|
|
|
- human_message_prompt = """Use the following CONTEXT as your learned knowledge.
|
|
|
+ human_message_instruction = """Use the following CONTEXT as your learned knowledge.
|
|
|
[CONTEXT]
|
|
|
{context}
|
|
|
[END CONTEXT]
|
|
@@ -180,9 +175,16 @@ When answer to user:
|
|
|
- If you don't know when you are not sure, ask for clarification.
|
|
|
Avoid mentioning that you obtained the information from the context.
|
|
|
And answer according to the language of the user's question.
|
|
|
+"""
|
|
|
+ if pre_prompt:
|
|
|
+ human_inputs.update(inputs)
|
|
|
+ human_message_instruction += pre_prompt + "\n"
|
|
|
|
|
|
-Q:{query}
|
|
|
-A:"""
|
|
|
+ human_message_prompt = human_message_instruction + "Q:{query}\nA:"
|
|
|
+ else:
|
|
|
+ if pre_prompt:
|
|
|
+ human_inputs.update(inputs)
|
|
|
+ human_message_prompt = pre_prompt + "\n" + human_message_prompt
|
|
|
|
|
|
# construct main prompt
|
|
|
human_message = PromptBuilder.to_human_message(
|