| 123 | |
| 124 | @dataclass |
| 125 | class AssistantMessage(Message): |
| 126 | audio_codes_list: List[Union[str, torch.Tensor]] |
| 127 | content: str = AUDIO_PLACEHOLDER |
| 128 | |
| 129 | def to_dict(self): |
| 130 | return { |
| 131 | "role": "assistant", |
| 132 | "content": self.content, |
| 133 | "audio_codes_list": self.audio_codes_list, |
| 134 | } |
| 135 | |
| 136 | |
| 137 | USER_MESSAGE_FIELDS = ( |
no outgoing calls
no test coverage detected