a

ape/llm.py

@@ -64,7 +64,7 @@ class LLMGPT(object):
         """
         prompt = ''
         for shot in few_shot:
-            prompt += MyTemplate['few_shot_prompt'].format(shot[0], shot[1])
+            prompt += MyTemplate['few_shot_prompt'].format(input=shot[0], output=shot[1])
         print(prompt)
         result = self.gen_chain.generate(prompt)
         return result

ape/prompt.py

@@ -1,7 +1,5 @@
 # -*-coding:utf-8 -*-
-from langchain import PromptTemplate
-
-few_shot_prompt = "Input: {input}\nOutput: {output}"
+few_shot_prompt = "Input: {input}\nOutput: {output}\n"
 
 gen_user_prompt = '{few_shot}'
 
@@ -15,7 +13,7 @@ Everything between the ``` must be valid json. """
 
 eval_prompt = "Instruction: {prompt}\nInput: {input}\nOutput: {output}"
 
-test_sys_prompt = """I want you to act as an AI assisted doctor. You are capable of answering anything related to medical." \
+test_sys_prompt = """I want you to act as an AI assisted doctor. You are capable of answering anything related to medical.
 Given an instruction and input, you must come up with corresponding output, and always respond in chinese.
 """