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

feat(sqlalchemy): some fixes

Browse Source
This commit is contained in:
Jonathan Rohde 2024-06-24 13:45:33 +02:00
parent a9b148791d
commit 8f939cf55b
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
backend/apps/webui/models/users.py
View File

@ -185,6 +185,7 @@ class UsersTable:
Session.query(User).filter_by(id=id).update( Session.query(User).filter_by(id=id).update(
{"last_active_at": int(time.time())} {"last_active_at": int(time.time())}
) )
Session.commit()
user = Session.query(User).filter_by(id=id).first() user = Session.query(User).filter_by(id=id).first()
return UserModel.model_validate(user) return UserModel.model_validate(user)

5
backend/main.py
View File

@ -794,11 +794,10 @@ app.add_middleware(
) )
@app.middleware("http") @app.middleware("http")
async def remove_session_after_request(request: Request, call_next): async def commit_session_after_request(request: Request, call_next):
response = await call_next(request) response = await call_next(request)
log.debug("Removing session after request") log.debug("Commit session after request")
Session.commit() Session.commit()
Session.remove()
return response return response

6
backend/utils/utils.py
View File

@ -113,8 +113,8 @@ def get_current_user(
) )
def get_current_user_by_api_key(db: Session, api_key: str): def get_current_user_by_api_key(api_key: str):
user = Users.get_user_by_api_key(db, api_key) user = Users.get_user_by_api_key(api_key)
if user is None: if user is None:
raise HTTPException( raise HTTPException(
@ -122,7 +122,7 @@ def get_current_user_by_api_key(db: Session, api_key: str):
detail=ERROR_MESSAGES.INVALID_TOKEN, detail=ERROR_MESSAGES.INVALID_TOKEN,
) )
else: else:
Users.update_user_last_active_by_id(db, user.id) Users.update_user_last_active_by_id(user.id)
return user return user