UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-17 05:53:11 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

feat(sqlalchemy): cleanup fixes

Browse Source
This commit is contained in:
Jonathan Rohde 2024-06-24 09:56:42 +02:00
parent 070d9083d5
commit 320e658595
2 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/apps/socket/main.py
View File

@ -24,9 +24,7 @@ async def connect(sid, environ, auth):
data = decode_token(auth["token"]) data = decode_token(auth["token"])
if data is not None and "id" in data: if data is not None and "id" in data:
from apps.webui.internal.db import SessionLocal user = Users.get_user_by_id(data["id"])
user = Users.get_user_by_id(SessionLocal(), data["id"])
if user: if user:
SESSION_POOL[sid] = user.id SESSION_POOL[sid] = user.id

1
backend/main.py
View File

@ -751,7 +751,6 @@ class PipelineMiddleware(BaseHTTPMiddleware):
user = get_current_user( user = get_current_user(
request, request,
get_http_authorization_cred(request.headers.get("Authorization")), get_http_authorization_cred(request.headers.get("Authorization")),
SessionLocal(),
) )
try: try: