| 394 | |
| 395 | |
| 396 | class HumanPromptMessage(HumanMessage): |
| 397 | sqlbot_system: bool = True |
| 398 | |
| 399 | def __init__( |
| 400 | self, content: Union[str, list[Union[str, dict]]], **kwargs: Any |
| 401 | ) -> None: |
| 402 | super().__init__(content=content, **kwargs) |
| 403 | |
| 404 | |
| 405 | class AIPromptMessage(AIMessage): |