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"]