fix: work in progess: models & migrations & repos

src/assistant/lib/agent/chat_repository.py

@@ -15,7 +15,7 @@ class ChatHistoryRepository:
         self.logger = logging.getLogger(__name__)
 
     async def get_last_session_id(self, request: models.RequestLastSessionId) -> uuid.UUID | None:
-        """Get a new session ID."""
+        """Get a current session ID if exists."""
 
         try:
             async with self.pg_async_session() as session:
@@ -23,10 +23,7 @@ class ChatHistoryRepository:
                     sa.select(orm_models.ChatHistory)
                     .filter_by(channel=request.channel, user_id=request.user_id)
                     .filter(
-                        (
-                            sa.func.extract("epoch", orm_models.ChatHistory.created)
-                            - sa.func.extract("epoch", orm_models.ChatHistory.modified) / 60
-                        )
+                        (sa.text("NOW()") - sa.func.extract("epoch", orm_models.ChatHistory.created)) / 60
                         <= request.minutes_ago
                     )
                     .order_by(orm_models.ChatHistory.created.desc())
@@ -39,3 +36,48 @@ class ChatHistoryRepository:
                     return chat_session.id
         except sqlalchemy.exc.SQLAlchemyError as error:
             self.logger.exception("Error: %s", error)
+
+    async def get_messages_by_sid(self, request: models.RequestChatHistory):
+        """Get all messages of a chat by session ID."""
+
+        try:
+            async with self.pg_async_session() as session:
+                statement = (
+                    sa.select(orm_models.ChatHistory)
+                    .filter_by(id=request.session_id)
+                    .order_by(orm_models.ChatHistory.created.desc())
+                )
+                result = await session.execute(statement)
+                for row in result.scalars().all():
+                    print("Row: ", row)
+        except sqlalchemy.exc.SQLAlchemyError as error:
+            self.logger.exception("Error: %s", error)
+
+    # async def get_all_by_session_id(self, request: models.RequestChatHistory) -> list[models.ChatHistory]:
+    #     try:
+    #         async with self.pg_async_session() as session:
+    #             statement = (
+    #                 sa.select(orm_models.ChatHistory)
+    #                 .filter_by(id=request.session_id)
+    #                 .order_by(orm_models.ChatHistory.created.desc())
+    #             )
+    #             result = await session.execute(statement)
+
+    #             return [models.ChatHistory.from_orm(chat_history) for chat_history in result.scalars().all()]
+
+    async def add_message(self, request: models.ChatMessage) -> None:
+        """Add a message to the chat history."""
+        try:
+            async with self.pg_async_session() as session:
+                chat_history = orm_models.ChatHistory(
+                    id=uuid.uuid4(),
+                    session_id=request.session_id,
+                    user_id=request.user_id,
+                    channel=request.channel,
+                    content=request.message,
+                )
+                session.add(chat_history)
+                await session.commit()
+                # TODO: Add refresh to session and return added object
+        except sqlalchemy.exc.SQLAlchemyError as error:
+            self.logger.exception("Error: %s", error)