a

ape/ape.py

@@ -45,6 +45,7 @@ def generate(instance, openai_key, n_instruct):
     for few_shot in train_iter:
         instruction = LLM.generate_instruction(few_shot)
         print(instruction)
+        instructions.append(instruction)
     return '\n'.join(instructions)
 
 

ape/llm.py

@@ -65,13 +65,11 @@ class LLMGPT(object):
         prompt = ''
         for shot in few_shot:
             prompt += MyTemplate['few_shot_prompt'].format(input=shot[0], output=shot[1])
-        print(prompt)
-        result = self.gen_chain({'few_shot':prompt})
+        result = self.gen_chain({'few_shot': prompt})
         return result
 
     def generate_output(self, instruction, input):
         output = self.test_chain({'input': input, 'instruction': instruction})
-        print(output)
         return output
 
     def generate_logprobs(self, ):

ape/prompt.py

@@ -1,23 +1,17 @@
 # -*-coding:utf-8 -*-
 few_shot_prompt = "Input: {input}\nOutput: {output}\n"
 
-gen_user_prompt = '{few_shot}'
+gen_user_prompt = """I gave a friend who is medical school teacher an instruction. 
+Based on the instruction they produced the following input-output pairs:\n\n{few_shot}\n\nThe instruction was to """
 
-gen_sys_prompt = """I want you to act as an AI assisted doctor. You are capable of answering anything related to medical. Given 
-a list of input-output pairs, you must come up with the correct instruction in medical-related area.
-You must respond in the following format, and always respond in chinese.
-```
-{{'instruction':"$YOUR_INSTRUCTION"}}
-```
-Everything between the ``` must be valid json. """
+gen_sys_prompt = ""
 
 eval_prompt = "Instruction: {instruction}\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.
-Given an instruction and input, you must come up with corresponding output, and always respond in chinese.
-"""
+test_sys_prompt = ""
 
-test_user_prompt = "Instruction: {instruction}\nInput: {input}\nOutput: "
+test_user_prompt = """I gave a friend who is medical school teacher an instruction. Base on this instruction, they will
+transform the input to output.\n Instruction: {instruction}Input: {input}\nOutput: """
 
 MyTemplate = {
     'gen_user_prompt': gen_user_prompt,