mirror of
https://github.com/getzep/graphiti.git
synced 2025-06-27 02:00:02 +00:00
ba48f64492
* chore: Folder rearrangement * chore: Remove unused deps, and add mypy step in CI for graph-service * fix: Mypy errors * fix: linter * fix mypy * fix mypy * chore: Update docker setup * chore: Reduce graph service image size * chore: Install graph service deps on CI * remove cache from typecheck * chore: install graph-service deps on typecheck action * update graph service mypy direction * feat: Add release service image step * chore: Update depot configuration * chore: Update release image job to run on releases * chore: Test depot multiplatform build * update release action tag * chore: Update action to be in accordance with zep image publish * test * test * revert * chore: Update python slim image used in service docker * chore: Remove unused endpoints and dtos
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'
|
|
)
|