(self, prompt)
| 115 | |
| 116 | |
| 117 | def build_messages(self, prompt): |
| 118 | messages = [ |
| 119 | {"role": "system", "content": "You are an excellent code programmer."}, |
| 120 | {"role": "user", "content": prompt} |
| 121 | ] |
| 122 | |
| 123 | return messages |
| 124 | |
| 125 | @timeout_decorator.timeout(60) |
| 126 | def generate(self, prompt, n = 20, temperature = 0.7): |