| 123456789101112131415161718192021222324252627282930313233343536 |
- from datetime import datetime
- from typing import TYPE_CHECKING
- from core_shared import JSONValue
- from pydantic import BaseModel, Field
- if TYPE_CHECKING:
- from app.db.models import Message
- class MessageCreateRequest(BaseModel):
- session_id: str
- turn_id: str | None = None
- role: str
- content_type: str = "text"
- content_text: str | None = None
- content_json: dict[str, JSONValue] = Field(default_factory=dict)
- class MessageListRequest(BaseModel):
- session_id: str
- class MessageResponse(BaseModel):
- id: str
- session_id: str
- turn_id: str | None = None
- role: str
- content_type: str
- content_text: str | None = None
- content_json: dict[str, JSONValue] | None = None
- created_time: datetime
- @classmethod
- def from_entity(cls, entity: "Message") -> "MessageResponse":
- return cls.model_validate(entity, from_attributes=True)
|
