from sqlalchemy.sql.expression import desc, asc
from typing import Optional, List, Union, Dict, Any
from fastapi import HTTPException, status
from sqlalchemy.orm import Session
from sqlalchemy.exc import IntegrityError
import uuid
from private_gpt.users.crud.base import CRUDBase
from private_gpt.users.models.chat import ChatHistory, ChatItem
from private_gpt.users.schemas.chat import ChatHistoryCreate, ChatHistoryCreate, ChatItemCreate, ChatItemUpdate


class CRUDChat(CRUDBase[ChatHistory, ChatHistoryCreate, ChatHistoryCreate]):
    def get_by_id(self, db: Session, *, id: uuid.UUID) -> Optional[ChatHistory]:
        return db.query(self.model).filter(ChatHistory.conversation_id == id).first()

    def get_chat_history(
            self, db: Session, *, skip: int = 0, limit: int = 100
        ) -> List[ChatHistory]:
            return (
                db.query(self.model)
                .order_by(desc(getattr(ChatHistory, 'created_at')))
                .offset(skip)
                .limit(limit)
                .all()
            )

class CRUDChatItem(CRUDBase[ChatItem, ChatItemCreate, ChatItemUpdate]):
    pass


chat = CRUDChat(ChatHistory)
chat_item = CRUDChatItem(ChatItem)