Spaces:
Running
Running
File size: 2,056 Bytes
48ec4db |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
from app.backend.controllers.base_controller import engine
from sqlalchemy.orm import relationship, selectinload
from sqlalchemy import String, Column, ForeignKey
from sqlalchemy.ext.asyncio import AsyncSession
from app.backend.models.base_model import Base
from sqlalchemy.future import select
class Chat(Base):
__tablename__ = "chats"
id = Column("id", String, primary_key=True, unique=True)
title = Column("title", String, nullable=True)
user_id = Column(String, ForeignKey("users.id"))
user = relationship("User", back_populates="chats")
messages = relationship("Message", back_populates="chat", lazy="selectin")
async def add_new_chat(id: str, title: str | None, user) -> None:
async with AsyncSession(autoflush=False, bind=engine) as db:
user = await db.merge(user)
new_chat = Chat(id=id, user_id=user.id, user=user)
if title:
new_chat.title = title
db.add(new_chat)
await db.commit()
async def get_chat_by_id(id: str) -> Chat | None:
async with AsyncSession(autoflush=False, bind=engine) as db:
result = await db.execute(select(Chat).where(Chat.id == id))
return result.scalar_one_or_none()
async def get_chats_by_user_id(id: str) -> list[Chat]:
async with AsyncSession(autoflush=False, bind=engine) as db:
result = await db.execute(
select(Chat).filter(Chat.user_id == id).order_by(Chat.created_at.desc())
)
return result.scalars().all()
async def refresh_title(chat_id: str) -> bool:
async with AsyncSession(autoflush=False, bind=engine) as db:
result = await db.execute(
select(Chat)
.options(selectinload(Chat.messages))
.where(Chat.id == chat_id)
)
chat = result.scalar_one_or_none()
if not chat or not chat.messages:
return False
chat.title = chat.messages[0].content[:47]
if len(chat.messages[0].content) > 46:
chat.title += "..."
await db.commit()
return True
|