mirror of
https://github.com/zylon-ai/private-gpt.git
synced 2025-12-22 20:12:55 +01:00
Added chat history and chat item
This commit is contained in:
parent
542ed0ef4e
commit
4bc9dd7870
19 changed files with 409 additions and 121 deletions
21
private_gpt/users/crud/chat_crud.py
Normal file
21
private_gpt/users/crud/chat_crud.py
Normal file
|
|
@ -0,0 +1,21 @@
|
|||
from typing import Optional, List, Union, Dict, Any
|
||||
from fastapi import HTTPException, status
|
||||
from sqlalchemy.orm import Session
|
||||
from sqlalchemy.exc import IntegrityError
|
||||
|
||||
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: int) -> Optional[ChatHistory]:
|
||||
return db.query(self.model).filter(ChatHistory.conversation_id == id).first()
|
||||
|
||||
|
||||
class CRUDChatItem(CRUDBase[ChatItem, ChatItemCreate, ChatItemUpdate]):
|
||||
pass
|
||||
|
||||
|
||||
chat = CRUDChat(ChatHistory)
|
||||
chat_item = CRUDChatItem(ChatItem)
|
||||
Loading…
Add table
Add a link
Reference in a new issue