(self, input, worker_log, log=False)
| 9 | self.suffix = suffix |
| 10 | |
| 11 | def run(self, input, worker_log, log=False): |
| 12 | assert isinstance(input, self.input_type) |
| 13 | prompt = self.prefix + input + "\n" + worker_log + self.suffix + input + '\n' |
| 14 | response = self.call_llm(prompt, self.stop) |
| 15 | completion = response["output"] |
| 16 | if log: |
| 17 | return response |
| 18 | return completion |