Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix verify mcp connection with oauth type (#19149)

Browse Source
This commit is contained in:
Oleg Yermolenko
2025-11-12 15:05:17 -05:00
committed by GitHub
parent 6eea0d40ab
commit 8dde493e8e
1 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/routers/configs.py
+9 -4
View File
@@ -270,17 +270,18 @@ async def verify_tool_servers_config(
elif form_data.auth_type == "session": elif form_data.auth_type == "session":
token = request.state.token.credentials token = request.state.token.credentials
elif form_data.auth_type == "system_oauth": elif form_data.auth_type == "system_oauth":
oauth_token = None
try: try:
if request.cookies.get("oauth_session_id", None): if request.cookies.get("oauth_session_id", None):
token = await request.app.state.oauth_manager.get_oauth_token( oauth_token = await request.app.state.oauth_manager.get_oauth_token(
user.id, user.id,
request.cookies.get("oauth_session_id", None), request.cookies.get("oauth_session_id", None),
) )
except Exception as e: except Exception as e:
pass pass
if token: if oauth_token:
headers = {"Authorization": f"Bearer {token}"} headers = {"Authorization": f"Bearer {oauth_token.get('access_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()
@@ -306,10 +307,14 @@ 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):
token = await request.app.state.oauth_manager.get_oauth_token( oauth_token = await request.app.state.oauth_manager.get_oauth_token(
user.id, user.id,
request.cookies.get("oauth_session_id", None), request.cookies.get("oauth_session_id", None),
) )
if oauth_token:
token = f"{oauth_token.get('access_token', '')}"
except Exception as e: except Exception as e:
pass pass