Skip to content

Commit 82959ce

Browse files
authored
Update oauth_sessions.py (open-webui#21794)
1 parent 9478c5e commit 82959ce

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

backend/open_webui/models/oauth_sessions.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,6 +152,7 @@ def get_session_by_id(
152152
with get_db_context(db) as db:
153153
session = db.query(OAuthSession).filter_by(id=session_id).first()
154154
if session:
155+
db.expunge(session)
155156
session.token = self._decrypt_token(session.token)
156157
return OAuthSessionModel.model_validate(session)
157158

@@ -172,6 +173,7 @@ def get_session_by_id_and_user_id(
172173
.first()
173174
)
174175
if session:
176+
db.expunge(session)
175177
session.token = self._decrypt_token(session.token)
176178
return OAuthSessionModel.model_validate(session)
177179

@@ -193,6 +195,7 @@ def get_session_by_provider_and_user_id(
193195
.first()
194196
)
195197
if session:
198+
db.expunge(session)
196199
session.token = self._decrypt_token(session.token)
197200
return OAuthSessionModel.model_validate(session)
198201

@@ -212,6 +215,7 @@ def get_sessions_by_user_id(
212215
results = []
213216
for session in sessions:
214217
try:
218+
db.expunge(session)
215219
session.token = self._decrypt_token(session.token)
216220
results.append(OAuthSessionModel.model_validate(session))
217221
except Exception as e:
@@ -246,6 +250,7 @@ def update_session_by_id(
246250
session = db.query(OAuthSession).filter_by(id=session_id).first()
247251

248252
if session:
253+
db.expunge(session)
249254
session.token = self._decrypt_token(session.token)
250255
return OAuthSessionModel.model_validate(session)
251256

0 commit comments

Comments
 (0)