ReMe/memory_scope/scheme/message.py

19 lines
655 B
Python

import datetime
from typing import Dict
from pydantic import Field, BaseModel
class Message(BaseModel):
role: str = Field(..., description="The role of the message sender (user, assistant, system)")
role_name: str = Field("", description="role name")
content: str = Field(..., description="The body of the message")
time_created: int = Field(int(datetime.datetime.now().timestamp()),
description="Timestamp when the message was created")
memorized: bool = Field(False, description="indicate whether message is memorized")
meta_data: Dict[str, str] = Field({}, description="meta data for msg")