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

refac/fix

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-11-12 23:27:03 -05:00
parent f1c317349e
commit 0bf686396d
1 changed files with 12 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
backend/open_webui/routers/configs.py
View file

@ -277,11 +277,14 @@ async def verify_tool_servers_config(
user.id, user.id,
request.cookies.get("oauth_session_id", None), request.cookies.get("oauth_session_id", None),
) )
if oauth_token:
token = oauth_token.get("access_token", "")
except Exception as e: except Exception as e:
pass pass
if oauth_token: if token:
headers = {"Authorization": f"Bearer {oauth_token.get('access_token', '')}"} headers = {"Authorization": f"Bearer {token}"}
await client.connect(form_data.url, headers=headers) await client.connect(form_data.url, headers=headers)
specs = await client.list_tool_specs() specs = await client.list_tool_specs()
@ -307,13 +310,15 @@ async def verify_tool_servers_config(
elif form_data.auth_type == "system_oauth": elif form_data.auth_type == "system_oauth":
try: try:
if request.cookies.get("oauth_session_id", None): if request.cookies.get("oauth_session_id", None):
oauth_token = await request.app.state.oauth_manager.get_oauth_token( oauth_token = (
user.id, await request.app.state.oauth_manager.get_oauth_token(
request.cookies.get("oauth_session_id", None), user.id,
request.cookies.get("oauth_session_id", None),
)
) )
if oauth_token: if oauth_token:
token = f"{oauth_token.get('access_token', '')}" token = oauth_token.get("access_token", "")
except Exception as e: except Exception as e:
pass pass