feat: Rename & Delete Chat

Author: ulleo

backend/alembic/versions/ff653d5df198_007_add_chat.py

@@ -0,0 +1,52 @@
+"""006_add_chat
+
+Revision ID: ff653d5df198
+Revises: 0a6f11be9be4
+Create Date: 2025-05-21 15:44:29.763091
+
+"""
+from alembic import op
+import sqlalchemy as sa
+import sqlmodel.sql.sqltypes
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision = 'ff653d5df198'
+down_revision = 'e6276ddab06e'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('chat',
+    sa.Column('id', sa.Integer(), sa.Identity(always=True), nullable=False),
+    sa.Column('create_time', sa.DateTime(timezone=True), nullable=True),
+    sa.Column('create_by', sa.BigInteger(), nullable=True),
+    sa.Column('brief', sqlmodel.sql.sqltypes.AutoString(length=64), nullable=True),
+    sa.Column('chat_type', sqlmodel.sql.sqltypes.AutoString(length=20), nullable=False),
+    sa.Column('datasource', sa.Integer(), nullable=False),
+    sa.Column('engine_type', sqlmodel.sql.sqltypes.AutoString(length=64), nullable=False),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_table('chat_record',
+    sa.Column('id', sa.Integer(), sa.Identity(always=True), nullable=False),
+    sa.Column('chat_id', sa.Integer(), nullable=True),
+    sa.Column('create_time', sa.DateTime(timezone=True), nullable=True),
+    sa.Column('create_by', sa.BigInteger(), nullable=True),
+    sa.Column('datasource', sa.Integer(), nullable=False),
+    sa.Column('engine_type', sqlmodel.sql.sqltypes.AutoString(length=64), nullable=False),
+    sa.Column('question', sa.Text(), nullable=True),
+    sa.Column('full_question', sa.Text(), nullable=True),
+    sa.Column('answer', sa.Text(), nullable=True),
+    sa.Column('run_time', sa.Float(), nullable=False),
+    sa.PrimaryKeyConstraint('id')
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table('chat_record')
+    op.drop_table('chat')
+    # ### end Alembic commands ###

backend/apps/chat/api/chat.py

@@ -2,8 +2,9 @@
 from fastapi.responses import StreamingResponse
 from sqlmodel import select
 
-from apps.chat.curd.chat import list_chats, get_chat_with_records, create_chat, save_question, save_answer
-from apps.chat.models.chat_model import CreateChat, ChatRecord
+from apps.chat.curd.chat import list_chats, get_chat_with_records, create_chat, save_question, save_answer, rename_chat, \
+    delete_chat
+from apps.chat.models.chat_model import CreateChat, ChatRecord, RenameChat
 from apps.chat.schemas.chat_base_schema import LLMConfig
 from apps.chat.schemas.chat_schema import ChatQuestion
 from apps.chat.schemas.llm import AgentService
@@ -32,6 +33,28 @@ async def list_chat(session: SessionDep, current_user: CurrentUser, chart_id: in
         )
 
 
+@router.post("/rename")
+async def rename(session: SessionDep, chat: RenameChat):
+    try:
+        return rename_chat(session=session, rename_object=chat)
+    except Exception as e:
+        raise HTTPException(
+            status_code=500,
+            detail=str(e)
+        )
+
+
+@router.get("/delete/{chart_id}")
+async def delete(session: SessionDep, chart_id: int):
+    try:
+        return delete_chat(session=session, chart_id=chart_id)
+    except Exception as e:
+        raise HTTPException(
+            status_code=500,
+            detail=str(e)
+        )
+
+
 @router.post("/start")
 async def start_chat(session: SessionDep, current_user: CurrentUser, create_chat_obj: CreateChat):
     try:

backend/apps/chat/curd/chat.py

@@ -5,7 +5,7 @@
 from sqlalchemy import text, and_
 from sqlmodel import select
 
-from apps.chat.models.chat_model import Chat, ChatRecord, CreateChat, ChatInfo
+from apps.chat.models.chat_model import Chat, ChatRecord, CreateChat, ChatInfo, RenameChat
 from apps.chat.schemas.chat_schema import ChatQuestion
 from apps.datasource.models.datasource import CoreDatasource
 from common.core.deps import SessionDep, CurrentUser
@@ -17,6 +17,32 @@ def list_chats(session: SessionDep, current_user: CurrentUser) -> List[Chat]:
     return chart_list
 
 
+def rename_chat(session: SessionDep, rename_object: RenameChat) -> str:
+    chat = session.query(Chat).filter(Chat.id == rename_object.id).first()
+    if not chat:
+        raise Exception(f"Chat with id {rename_object.id} not found")
+
+    chat.brief = rename_object.brief.strip()[:20]
+    session.add(chat)
+    session.flush()
+    session.refresh(chat)
+
+    brief = chat.brief
+    session.commit()
+    return brief
+
+
+def delete_chat(session, chart_id) -> str:
+    chat = session.query(Chat).filter(Chat.id == chart_id).first()
+    if not chat:
+        return f'Chat with id {chart_id} has been deleted'
+
+    session.delete(chat)
+    session.commit()
+
+    return f'Chat with id {chart_id} has been deleted'
+
+
 def get_chat_with_records(session: SessionDep, chart_id: int, current_user: CurrentUser) -> ChatInfo:
     chat = session.query(Chat).filter(Chat.id == chart_id).first()
     if not chat:
@@ -98,6 +124,7 @@ def save_question(session: SessionDep, current_user: CurrentUser, question: Chat
 
     return result
 
+
 def save_full_question(session: SessionDep, id: int, full_question: str) -> ChatRecord:
     if not id:
         raise Exception("Record id cannot be None")
@@ -114,6 +141,7 @@ def save_full_question(session: SessionDep, id: int, full_question: str) -> Chat
 
     return result
 
+
 def save_answer(session: SessionDep, id: int, answer: str) -> ChatRecord:
     if not id:
         raise Exception("Record id cannot be None")

backend/apps/chat/models/chat_model.py

@@ -36,6 +36,11 @@ class CreateChat(BaseModel):
     datasource: int = None
 
 
+class RenameChat(BaseModel):
+    id: int = None
+    brief: str = ''
+
+
 class ChatInfo(BaseModel):
     id: Optional[int] = None
     create_time: datetime = None

frontend/components.d.ts

@@ -147,12 +147,22 @@ function startChat(data: any): Promise<ChatInfo> {
     return request.post('/chat/start', data)
 }
 
+function renameChat(chat_id: number, brief: string): Promise<string> {
+    return request.post('/chat/rename', {id: chat_id, brief: brief})
+}
+
+function deleteChat(id: number): Promise<string> {
+    return request.get(`/chat/delete/${id}`)
+}
+
 export const chatApi = {
     toChatRecord,
     toChatRecordList,
     toChatInfo,
     toChatInfoList,
     list,
     get,
-    startChat
+    startChat,
+    renameChat,
+    deleteChat
 }