Format chat messages.
(
user_message: str,
system_message: Optional[str] = None,
)
| 584 | |
| 585 | |
| 586 | def format_messages( |
| 587 | user_message: str, |
| 588 | system_message: Optional[str] = None, |
| 589 | ) -> List[Dict[str, str]]: |
| 590 | """Format chat messages.""" |
| 591 | messages = [] |
| 592 | if system_message: |
| 593 | messages.append({"role": "system", "content": system_message}) |
| 594 | messages.append({"role": "user", "content": user_message}) |
| 595 | return messages |
| 596 | |
| 597 | __all__ = [ |
| 598 | # Response class |