ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 12:25:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refac

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-09-24 23:27:45 -05:00
parent d3b09c6a02
commit 34d16791a8
1 changed files with 2 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
backend/open_webui/main.py
View file

@ -1884,7 +1884,6 @@ async def get_current_usage(user=Depends(get_verified_user)):
# SessionMiddleware is used by authlib for oauth # SessionMiddleware is used by authlib for oauth
if len(OAUTH_PROVIDERS) > 0: if len(OAUTH_PROVIDERS) > 0:
try: try:
# Try to create Redis store for sessions
if REDIS_URL: if REDIS_URL:
redis_session_store = RedisStore( redis_session_store = RedisStore(
url=REDIS_URL, url=REDIS_URL,
@ -1893,9 +1892,7 @@ if len(OAUTH_PROVIDERS) > 0:
), ),
) )
# Add SessionAutoloadMiddleware first to handle session loading
app.add_middleware(SessionAutoloadMiddleware) app.add_middleware(SessionAutoloadMiddleware)
app.add_middleware( app.add_middleware(
StarSessionsMiddleware, StarSessionsMiddleware,
store=redis_session_store, store=redis_session_store,
@ -1903,15 +1900,10 @@ if len(OAUTH_PROVIDERS) > 0:
cookie_same_site=WEBUI_SESSION_COOKIE_SAME_SITE, cookie_same_site=WEBUI_SESSION_COOKIE_SAME_SITE,
cookie_https_only=WEBUI_SESSION_COOKIE_SECURE, cookie_https_only=WEBUI_SESSION_COOKIE_SECURE,
) )
log.info("Using StarSessions with Redis for session management") log.info("Using Redis for session")
else: else:
raise ValueError("Redis URL not configured") raise ValueError("No Redis URL provided")
except Exception as e: except Exception as e:
log.warning(
f"Failed to initialize Redis sessions, falling back to cookie based sessions: {e}"
)
# Fallback to existing SessionMiddleware
app.add_middleware( app.add_middleware(
SessionMiddleware, SessionMiddleware,
secret_key=WEBUI_SECRET_KEY, secret_key=WEBUI_SECRET_KEY,