feat: add SQL examples in prompt

Author: ulleo

backend/alembic/env.py

@@ -26,6 +26,7 @@
 # from apps.settings.models.setting_models import SQLModel
 # from apps.chat.models.chat_model import SQLModel
 from apps.terminology.models.terminology_model import SQLModel
+from apps.data_training.models.data_training_model import SQLModel
 # from apps.dashboard.models.dashboard_model import SQLModel
 from common.core.config import settings # noqa
 

backend/alembic/versions/042_data_training.py

@@ -0,0 +1,41 @@
+"""042_data_training
+
+Revision ID: a487d9c69341
+Revises: c4c3c36b720d
+Create Date: 2025-09-15 15:41:43.332771
+
+"""
+from alembic import op
+import sqlalchemy as sa
+import sqlmodel.sql.sqltypes
+from sqlalchemy.dialects import postgresql
+import pgvector
+
+# revision identifiers, used by Alembic.
+revision = 'a487d9c69341'
+down_revision = 'c4c3c36b720d'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('data_training',
+    sa.Column('id', sa.BigInteger(), sa.Identity(always=True), nullable=False),
+    sa.Column('oid', sa.BigInteger(), nullable=True),
+    sa.Column('datasource', sa.BigInteger(), nullable=True),
+    sa.Column('create_time', sa.DateTime(), nullable=True),
+    sa.Column('question', sqlmodel.sql.sqltypes.AutoString(length=255), nullable=True),
+    sa.Column('description', sa.Text(), nullable=True),
+    sa.Column('embedding', pgvector.sqlalchemy.vector.VECTOR(), nullable=True),
+    sa.PrimaryKeyConstraint('id')
+    )
+
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+
+    op.drop_table('data_training')
+    # ### end Alembic commands ###

backend/apps/api.py

@@ -1,6 +1,7 @@
 from fastapi import APIRouter
 
 from apps.terminology.api import terminology
+from apps.data_training.api import data_training
 from apps.chat.api import chat
 from apps.dashboard.api import dashboard_api
 from apps.datasource.api import datasource
@@ -14,6 +15,7 @@
 api_router.include_router(assistant.router)
 api_router.include_router(aimodel.router)
 api_router.include_router(terminology.router)
+api_router.include_router(data_training.router)
 api_router.include_router(datasource.router)
 api_router.include_router(chat.router)
 api_router.include_router(dashboard_api.router)

backend/apps/chat/models/chat_model.py

@@ -172,11 +172,13 @@ class AiModelQuestion(BaseModel):
     filter: str = []
     sub_query: Optional[list[dict]] = None
     terminologies: str = ""
+    data_training: str = ""
     error_msg: str = ""
 
     def sql_sys_question(self):
         return get_sql_template()['system'].format(engine=self.engine, schema=self.db_schema, question=self.question,
-                                                   lang=self.lang, terminologies=self.terminologies)
+                                                   lang=self.lang, terminologies=self.terminologies,
+                                                   data_training=self.data_training)
 
     def sql_user_question(self, current_time: str):
         return get_sql_template()['user'].format(engine=self.engine, schema=self.db_schema, question=self.question,

backend/apps/data_training/__init__.py

@@ -0,0 +1,39 @@
+from typing import Optional
+
+from fastapi import APIRouter, Query
+
+from apps.data_training.curd.data_training import page_data_training, create_training, update_training, delete_training
+from apps.data_training.models.data_training_model import DataTrainingInfo
+from common.core.deps import SessionDep, CurrentUser, Trans
+
+router = APIRouter(tags=["DataTraining"], prefix="/system/data-training")
+
+
+@router.get("/page/{current_page}/{page_size}")
+async def pager(session: SessionDep, current_user: CurrentUser, current_page: int, page_size: int,
+                question: Optional[str] = Query(None, description="搜索问题(可选)")):
+    current_page, page_size, total_count, total_pages, _list = page_data_training(session, current_page, page_size,
+                                                                                  question,
+                                                                                  current_user.oid)
+
+    return {
+        "current_page": current_page,
+        "page_size": page_size,
+        "total_count": total_count,
+        "total_pages": total_pages,
+        "data": _list
+    }
+
+
+@router.put("")
+async def create_or_update(session: SessionDep, current_user: CurrentUser, trans: Trans, info: DataTrainingInfo):
+    oid = current_user.oid
+    if info.id:
+        return update_training(session, info, oid, trans)
+    else:
+        return create_training(session, info, oid, trans)
+
+
+@router.delete("")
+async def delete(session: SessionDep, id_list: list[int]):
+    delete_training(session, id_list)