estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix: extra input for opening statement was not suitable for prompt (#143)

Browse Source
This commit is contained in:
John Wang
2023-05-22 14:32:22 +08:00
committed by GitHub
parent c43c3098a0
commit 54a6571462
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
api/core/completion.py
View File

@@ -177,13 +177,21 @@ Avoid mentioning that you obtained the information from the context.
And answer according to the language of the user's question. And answer according to the language of the user's question.
""" """
if pre_prompt: if pre_prompt:
human_inputs.update(inputs) extra_inputs = {k: inputs[k] for k in
OutLinePromptTemplate.from_template(template=pre_prompt).input_variables
if k in inputs}
if extra_inputs:
human_inputs.update(extra_inputs)
human_message_instruction += pre_prompt + "\n" human_message_instruction += pre_prompt + "\n"
human_message_prompt = human_message_instruction + "Q:{query}\nA:" human_message_prompt = human_message_instruction + "Q:{query}\nA:"
else: else:
if pre_prompt: if pre_prompt:
human_inputs.update(inputs) extra_inputs = {k: inputs[k] for k in
OutLinePromptTemplate.from_template(template=pre_prompt).input_variables
if k in inputs}
if extra_inputs:
human_inputs.update(extra_inputs)
human_message_prompt = pre_prompt + "\n" + human_message_prompt human_message_prompt = pre_prompt + "\n" + human_message_prompt
# construct main prompt # construct main prompt