feat: init project

app/controller/add_sqlmodel_route.py

@@ -0,0 +1,52 @@
+from fastapi import FastAPI, HTTPException
+from sqlmodel import select
+
+from app.model.LlmUser import LlmUser
+from app.utils.db_utils import AsyncSessionDep
+from app.utils.next_id import next_id
+
+
+def add_sqlmodel_route(app: FastAPI):
+  @app.post("/llm_user/insert")
+  async def llm_user_insert(user: LlmUser, session: AsyncSessionDep):
+    if user.id is None:
+      user.id = await next_id()
+
+    session.add(user)
+    await session.commit()
+    await session.refresh(user)
+    return {"result": user}
+
+  @app.post("/llm_user/update")
+  async def llm_user_insert(user_dict: dict, session: AsyncSessionDep):
+
+    if user_dict.get("id") is None:
+      raise HTTPException(status_code=500, detail="Update row missing id")
+
+    update_user = (await session.exec(select(LlmUser).where(LlmUser.id == user_dict["id"]))).first()
+    if update_user is None:
+      raise HTTPException(status_code=500, detail="Update row not found")
+
+    for key, value in user_dict.items():
+      setattr(update_user, key, value)
+
+    session.add(update_user)
+    await session.commit()
+    await session.refresh(update_user)
+    return {"result": update_user}
+
+  @app.post("/llm_user/delete")
+  async def llm_user_delete(user_dict: dict, session: AsyncSessionDep):
+
+    if user_dict.get("id") is None:
+      raise HTTPException(status_code=500, detail="Update row missing id")
+
+    delete_user: LlmUser = (await session.exec(select(LlmUser).where(LlmUser.id == user_dict["id"]))).first()
+
+    if delete_user is None:
+      raise HTTPException(status_code=500, detail="Delete row not found")
+
+    await session.delete(delete_user)
+    await session.commit()
+
+    return {"result": True}