feat(sqlalchemy): Replace peewee with sqlalchemy

2024-06-27 07:48:08 +02:00
parent 8dac2a2140
commit df09d0830a
47 changed files with 2580 additions and 1003 deletions
@@ -24,7 +24,9 @@ async def connect(sid, environ, auth):
        data = decode_token(auth["token"])

        if data is not None and "id" in data:
-            user = Users.get_user_by_id(data["id"])
+            from apps.webui.internal.db import SessionLocal
+
+            user = Users.get_user_by_id(SessionLocal(), data["id"])

        if user:
            SESSION_POOL[sid] = user.id