refac

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:

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: