4e87dc8b70
- Replaced `get_container_db_async_session` with `async_session_maker` for improved session handling in background tasks. - Refactored chat memory services to utilize a shared `mem0` client for better memory management. - Introduced new methods for retrieving and storing chat history, integrating with SQL and memory layers. - Enhanced error handling and response management in chat-related services. - Cleaned up unused code and improved overall structure for maintainability.
44 lines
1 KiB
Python
44 lines
1 KiB
Python
import uuid
|
|
from typing import Optional
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
|
|
class ChatMessageRequest(BaseModel):
|
|
presentation_id: uuid.UUID
|
|
message: str = Field(min_length=1, max_length=8000)
|
|
conversation_id: Optional[uuid.UUID] = None
|
|
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
|
|
class ChatMessageResponse(BaseModel):
|
|
conversation_id: uuid.UUID
|
|
response: str
|
|
tool_calls: list[str] = Field(default_factory=list)
|
|
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
|
|
class ChatHistoryMessageItem(BaseModel):
|
|
role: str
|
|
content: str
|
|
created_at: Optional[str] = None
|
|
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
|
|
class ChatHistoryResponse(BaseModel):
|
|
presentation_id: uuid.UUID
|
|
conversation_id: uuid.UUID
|
|
messages: list[ChatHistoryMessageItem]
|
|
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
|
|
class ChatConversationListItem(BaseModel):
|
|
conversation_id: uuid.UUID
|
|
updated_at: Optional[str] = None
|
|
last_message_preview: Optional[str] = None
|
|
|
|
model_config = ConfigDict(extra="forbid")
|