(conv: Conversation)
| 172 | conv.append_message(conv.roles[i % 2], ex) |
| 173 | |
| 174 | def get_prompt(conv: Conversation) -> str: |
| 175 | if conv.name == 'openchat': |
| 176 | ret = '' |
| 177 | for role, message in conv.messages: |
| 178 | if message: |
| 179 | ret += role + ": " + message + conv.sep |
| 180 | else: |
| 181 | ret += role + ":" |
| 182 | return ret |
| 183 | else: |
| 184 | return conv.get_prompt() |
| 185 | |
| 186 | # Example user input console, to play through a game. |
| 187 | def eval(args, task_num, logger): |