From 1b52c7fdcac19cf2b8e6ad487efa210664f06c4c Mon Sep 17 00:00:00 2001 From: h Date: Sun, 6 Jul 2025 00:08:43 +0300 Subject: [PATCH] feat(solaris): add session models to db --- src/utils/db/models/session.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 src/utils/db/models/session.py diff --git a/src/utils/db/models/session.py b/src/utils/db/models/session.py new file mode 100644 index 0000000..88fc85c --- /dev/null +++ b/src/utils/db/models/session.py @@ -0,0 +1,21 @@ +from typing import Annotated, List + +from beanie import Document, Indexed +from google.genai.types import Content +from pydantic import BaseModel, Field + + +class SessionBase(BaseModel): + chat_id: Annotated[int, Indexed(unique=True)] + system_prompt: str + history: List[Content] = Field(default_factory=list) + + +class ReviewSession(SessionBase, Document): + class Settings: + name = "review_sessions" + + +class RespondSession(SessionBase, Document): + class Settings: + name = "respond_sessions"