From 978da7dd40cd4c9080cf07e1d9870afa8eca3071 Mon Sep 17 00:00:00 2001
From: martsforever <martsforever@aliyun.com>
Date: Wed, 1 Apr 2026 19:51:53 +0800
Subject: [PATCH] feat: LlmDemoModel

---
 app/model/LlmDemoMdel.py | 119 +++++++++++++++++++++++++++++++++++++++
 app/routes.py            |   3 +-
 2 files changed, 121 insertions(+), 1 deletion(-)
 create mode 100644 app/model/LlmDemoMdel.py

diff --git a/app/model/LlmDemoMdel.py b/app/model/LlmDemoMdel.py
new file mode 100644
index 0000000..dc9d583
--- /dev/null
+++ b/app/model/LlmDemoMdel.py
@@ -0,0 +1,119 @@
+import uuid
+from datetime import datetime, timezone, timedelta, date
+from decimal import Decimal
+from typing import List
+
+from fastapi import FastAPI, APIRouter, HTTPException
+from pydantic import BaseModel
+from sqlmodel import SQLModel, Field, select
+
+from app.utils.mysql_utils import AsyncSessionDep
+
+# 定义北京时区（UTC+8）
+beijing_timezone = timezone(timedelta(hours=8))
+
+# 定义获取当前北京时区时间的匿名函数，用于默认值生成
+current_datetime = lambda: datetime.now(beijing_timezone)
+
+
+class LlmDemoModel(SQLModel, table=True):
+  __tablename__ = "llm_demo"
+
+  id: str = Field(default=None, primary_key=True, description="唯一标识，编号")
+  created_at: datetime = Field(default_factory=current_datetime, description="创建时间")
+  updated_at: datetime = Field(default_factory=current_datetime, description="更新时间")
+  created_by: str | None = Field(default=None, description="创建人id")
+  updated_by: str | None = Field(default=None, description="更新人id")
+
+  full_name: str = Field(default=None, description="用户名称")
+  datetime_start: datetime = Field(default=None, description="开通会员时间")
+  datetime_end: datetime = Field(default=None, description="会员截止到期时间")
+  birthday: date = Field(default=None, description="生日")
+  amount: Decimal = Field(default=0, description="金额")
+
+
+def add_llm_demo_route(app: FastAPI):
+  route_prefix = "/llm_demo"
+  router = APIRouter(prefix=route_prefix, tags=[route_prefix])
+
+  @router.post('/list')
+  async def _list(
+    session: AsyncSessionDep,
+    body: ModelQuerySchema = ModelQuerySchema(),
+  ):
+    """
+    获取所有用户列表
+    """
+
+    # 构建SQL查询执行对象
+    query = select(LlmDemoModel).order_by(
+      LlmDemoModel.created_at.desc()
+    )
+    # 计算偏移量（跳过前N条），并查询比一页多1条的记录（用于判断是否有下一页）
+    query = query.offset(body.page * body.page_size).limit(body.page_size + 1)
+    result = await session.execute(query)
+    query_cls_list: List[LlmDemoModel] = result.scalars().all()
+    has_next = len(query_cls_list) == body.page_size + 1
+    # 若有下一页，移除多查询的那一条记录
+    if has_next:
+      query_cls_list.pop()
+    return {
+      "list": query_cls_list,
+      "has_next": has_next
+    }
+
+  @router.post('/insert')
+  async def _insert(body: LlmDemoModel, session: AsyncSessionDep):
+    """
+    插入用户
+    """
+    if not body.id:
+      body.id = str(uuid.uuid4())
+    session.add(body)
+    await session.commit()
+    return body
+
+  @router.post('/update')
+  async def _update(body: LlmDemoModel, session: AsyncSessionDep):
+    """
+    更新用户
+    """
+    query = select(LlmDemoModel).where(LlmDemoModel.id == body.id)
+    result = await session.execute(query)
+    query_cls = result.scalars().first()
+    if not query_cls:
+      raise HTTPException(status_code=500, detail=f"Update row with id:{body.id} not found")
+
+    update_data_exclude_true = body.model_dump(exclude_unset=True, exclude={'id'})
+    update_data_exclude_false = body.model_dump(exclude={'id'})
+
+    print(f"update_data_exclude_true: {update_data_exclude_true}")
+    print(f"update_data_exclude_false: {update_data_exclude_false}")
+
+    update_data = body.model_dump(exclude_unset=True, exclude={'id'})
+    for field, value in update_data.items():
+      setattr(query_cls, field, value)
+    session.add(query_cls)
+    await session.commit()
+    return query_cls
+
+  @router.post('/delete')
+  async def _delete(body: LlmDemoModel, session: AsyncSessionDep):
+    """
+    更新用户
+    """
+    query = select(LlmDemoModel).where(LlmDemoModel.id == body.id)
+    result = await session.execute(query)
+    query_cls = result.scalars().first()
+    if not query_cls:
+      return {"affect_row_count": 0}
+    await session.delete(query_cls)
+    await session.commit()
+    return {"affect_row_count": 1}
+
+  app.include_router(router)
+
+
+class ModelQuerySchema(BaseModel):
+  page: int = Field(default=0, description="页码")
+  page_size: int = Field(default=5, description="每页数量")
diff --git a/app/routes.py b/app/routes.py
index 1821dcc..0920fce 100644
--- a/app/routes.py
+++ b/app/routes.py
@@ -2,6 +2,7 @@ from app.controller.add_custom_stream_route import add_custom_stream_route
 from app.controller.add_docs_route import add_docs_route
 from app.controller.add_graph_proxy_route import add_graph_proxy_route
 from app.controller.add_test_route import add_test_route
+from app.model.LlmDemoMdel import add_llm_demo_route
 from app.model.LlmUserModel import add_llm_user_route
 
 # /*@formatter:off*/
@@ -10,6 +11,6 @@ routes = [
   add_test_route,               # 测试接口
   add_graph_proxy_route,        # 代理自定义工作流
   add_custom_stream_route,      # 自定义流式接口
-  add_llm_user_route,           # LlmUser 测试用户模块
+  add_llm_demo_route,           # LlmDemo 测试用户模块
 ]
 # /*@formatter:on*/