mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-16 10:02:30 +03:00
d0b8b797f6
Co-authored-by: Zewed <dewez.antoine2@gmail.com> Co-authored-by: Antoine Dewez <44063631+Zewed@users.noreply.github.com>
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
from dataclasses import asdict, dataclass
|
|
from typing import Optional
|
|
from uuid import UUID
|
|
|
|
|
|
@dataclass
|
|
class Chat:
|
|
chat_id: str
|
|
user_id: str
|
|
creation_time: str
|
|
chat_name: str
|
|
|
|
def __init__(self, chat_dict: dict):
|
|
self.chat_id = chat_dict.get("chat_id", "")
|
|
self.user_id = chat_dict.get("user_id", "")
|
|
self.creation_time = chat_dict.get("creation_time", "")
|
|
self.chat_name = chat_dict.get("chat_name", "")
|
|
|
|
|
|
@dataclass
|
|
class ChatHistory:
|
|
chat_id: str
|
|
message_id: str
|
|
user_message: str
|
|
assistant: str
|
|
message_time: str
|
|
prompt_id: Optional[UUID]
|
|
brain_id: Optional[UUID]
|
|
metadata: Optional[dict] = None
|
|
|
|
def __init__(self, chat_dict: dict):
|
|
self.chat_id = chat_dict.get("chat_id", "")
|
|
self.message_id = chat_dict.get("message_id", "")
|
|
self.user_message = chat_dict.get("user_message", "")
|
|
self.assistant = chat_dict.get("assistant", "")
|
|
self.message_time = chat_dict.get("message_time", "")
|
|
|
|
self.prompt_id = chat_dict.get("prompt_id")
|
|
self.brain_id = chat_dict.get("brain_id")
|
|
self.metadata = chat_dict.get("metadata")
|
|
|
|
def to_dict(self):
|
|
return asdict(self)
|