a

ape/llm.py

@@ -24,7 +24,7 @@ class LLMGPT(object):
     def __init__(self, openai_key):
         self.gen_llm = ChatOpenAI(openai_api_key=openai_key, max_tokens=2000, temperature=0.7, verbose=True)
         self.eval_llm = OpenAI(openai_api_key=openai_key, max_tokens=2000, temperature=0.7, logprobs=1, verbose=True)
-        self.test_llm = ChatOpenAI(openai_api_key=openai_key, max_tokens=2000, temperature=0.7, verbose=True)
+        self.test_llm = OpenAI(openai_api_key=openai_key, max_tokens=2000, temperature=0.7,  verbose=True)
         self.gen_chain = None
         self.eval_chain = None
 
@@ -61,13 +61,8 @@ class LLMGPT(object):
 
     def generate_output(self, test_prompt, instruction, input):
         if not test_prompt:
-            test_prompt = MyTemplate['test_user_prompt']
-        prompt = ChatPromptTemplate.from_messages(
-            [
-                SystemMessagePromptTemplate.from_template(MyTemplate['test_sys_prompt']),
-                HumanMessagePromptTemplate.from_template(test_prompt),
-            ]
-        )
+            test_prompt = MyTemplate['test_prompt']
+        prompt = PromptTemplate.from_template(test_prompt)
         test_chain = LLMChain(llm=self.test_llm, prompt=prompt)
         output = test_chain({'input': input, 'instruction': instruction})
         return output

ape/prompt.py

@@ -7,15 +7,12 @@ gen_sys_prompt = ""
 
 eval_prompt = "任务描述:{instruction}\n输入:{input}\n输出:{output}"
 
-test_sys_prompt = ""
-
-test_user_prompt = "任务描述:{instruction}\n输入:{input}\n输出:"
+test_prompt = "任务描述:{instruction}\n输入:{input}\n输出:"
 
 MyTemplate = {
     'gen_user_prompt': gen_user_prompt,
     'gen_sys_prompt': gen_sys_prompt,
     'eval_prompt': eval_prompt,
     'few_shot_prompt': few_shot_prompt,
-    'test_sys_prompt': test_sys_prompt,
-    'test_user_prompt': test_user_prompt
+    'test_prompt': test_prompt
 }

app.py

@@ -49,7 +49,7 @@ with gr.Blocks(title="Automatic Prompt Engineer", theme=gr.themes.Glass()) as de
                                      placeholder=MyTemplate['eval_prompt'],
                                      value='', label="Prompt for Evaluation")
             test_prompt = gr.Textbox(max_lines=100, lines=3, interative=True,
-                                     placeholder=MyTemplate['test_user_prompt'],
+                                     placeholder=MyTemplate['test_prompt'],
                                      value='', label="Prompt for Single Test")
 
             with gr.Row():