Source code for genai.schema._extensions

from genai.schema._api import BaseMessage, ChatRole


[docs] class HumanMessage(BaseMessage): role: ChatRole = ChatRole.USER
[docs] class SystemMessage(BaseMessage): role: ChatRole = ChatRole.SYSTEM
[docs] class AIMessage(BaseMessage): role: ChatRole = ChatRole.ASSISTANT
__all__ = ["HumanMessage", "SystemMessage", "AIMessage"]