Spaces:
Sleeping
Sleeping
File size: 645 Bytes
24bec11 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
from langchain.schema import ChatMessage
from enum import Enum
class ChatProfileRoleEnum(str, Enum):
User = "user"
Assistant = "assistant"
class ChatProfile:
def __init__(self, role: str, message: str):
self.role = role
self.message = message
def build_message(self) -> ChatMessage:
return ChatMessage(role=self.role, content=self.message)
class Assistant(ChatProfile):
def __init__(self, message: str):
super().__init__(ChatProfileRoleEnum.Assistant, message)
class User(ChatProfile):
def __init__(self, message: str):
super().__init__(ChatProfileRoleEnum.User, message)
|