(self)
| 27 | skip_next: bool = False |
| 28 | |
| 29 | def get_prompt(self): |
| 30 | if self.sep_style == SeparatorStyle.SINGLE: |
| 31 | ret = self.system + self.sep + '\n' |
| 32 | for role, message in self.messages: |
| 33 | if message: |
| 34 | if type(message) is tuple: |
| 35 | message, _, _ = message |
| 36 | ret += role + ": " + message + self.sep |
| 37 | else: |
| 38 | ret += role + ":" |
| 39 | return ret |
| 40 | elif self.sep_style == SeparatorStyle.TWO: |
| 41 | seps = [self.sep, self.sep2] |
| 42 | ret = self.system + seps[0] |
| 43 | for i, (role, message) in enumerate(self.messages): |
| 44 | if message: |
| 45 | if type(message) is tuple: |
| 46 | message, _, _ = message |
| 47 | ret += role + ": " + message + seps[i % 2] |
| 48 | else: |
| 49 | ret += role + ":" |
| 50 | return ret |
| 51 | if self.sep_style == SeparatorStyle.MPT: |
| 52 | if self.system: |
| 53 | ret = self.system + self.sep |
| 54 | else: |
| 55 | ret = '' |
| 56 | for role, message in self.messages: |
| 57 | if message: |
| 58 | if type(message) is tuple: |
| 59 | message, _, _ = message |
| 60 | ret += role + message + self.sep |
| 61 | else: |
| 62 | ret += role |
| 63 | return ret |
| 64 | else: |
| 65 | raise ValueError(f"Invalid style: {self.sep_style}") |
| 66 | |
| 67 | |
| 68 | def append_message(self, role, message): |
no outgoing calls
no test coverage detected