mirror of
https://github.com/getzep/graphiti.git
synced 2025-06-27 02:00:02 +00:00
29 lines
926 B
Python
29 lines
926 B
Python
|
from datetime import datetime
|
||
|
|
from typing import Literal
|
||
|
|
|
||
|
|
from pydantic import BaseModel, Field
|
||
|
|
|
||
|
|
|
||
|
|
class Result(BaseModel):
|
||
|
|
message: str
|
||
|
|
success: bool
|
||
|
|
|
||
|
|
|
||
|
|
class Message(BaseModel):
|
||
|
|
content: str = Field(..., description='The content of the message')
|
||
|
|
name: str = Field(
|
||
|
|
default='', description='The name of the episodic node for the message (message uuid)'
|
||
|
|
)
|
||
|
|
role_type: Literal['user', 'assistant', 'system'] = Field(
|
||
|
|
..., description='The role type of the message (user, assistant or system)'
|
||
|
|
)
|
||
|
|
role: str | None = Field(
|
||
|
|
description='The custom role of the message to be used alongside role_type (user name, bot name, etc.)',
|
||
|
|
)
|
||
|
|
timestamp: datetime = Field(
|
||
|
|
default_factory=datetime.now, description='The timestamp of the message'
|
||
|
|
)
|
||
|
|
source_description: str = Field(
|
||
|
|
default='', description='The description of the source of the message'
|
||
|
|
)
|