From f71834720e623761d972d4d740e9bbd90a3a86c6 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Mon, 8 Sep 2025 18:35:09 +0400 Subject: [PATCH] refac --- backend/open_webui/utils/middleware.py | 2 +- backend/open_webui/utils/tools.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py index 27b0f11290..406ee1b16f 100644 --- a/backend/open_webui/utils/middleware.py +++ b/backend/open_webui/utils/middleware.py @@ -817,7 +817,7 @@ async def process_chat_payload(request, form_data, user, metadata, model): oauth_token = None try: - oauth_token = await request.app.state.oauth_manager.get_oauth_token( + oauth_token = request.app.state.oauth_manager.get_oauth_token( user.id, request.cookies.get("oauth_session_id", None) ) except Exception as e: diff --git a/backend/open_webui/utils/tools.py b/backend/open_webui/utils/tools.py index 9d64db7d78..efdb7c85b1 100644 --- a/backend/open_webui/utils/tools.py +++ b/backend/open_webui/utils/tools.py @@ -137,7 +137,7 @@ async def get_tools( oauth_token = None try: oauth_token = ( - await request.app.state.oauth_manager.get_oauth_token( + request.app.state.oauth_manager.get_oauth_token( user.id, request.cookies.get("oauth_session_id", None), ) @@ -715,6 +715,7 @@ async def execute_tool_server( final_url, json=body_params, headers=headers, + cookies=cookies, ssl=AIOHTTP_CLIENT_SESSION_TOOL_SERVER_SSL, ) as response: if response.status >= 400: @@ -731,6 +732,7 @@ async def execute_tool_server( async with request_method( final_url, headers=headers, + cookies=cookies, ssl=AIOHTTP_CLIENT_SESSION_TOOL_SERVER_SSL, ) as response: if response.status >= 400: