Skip to content

Commit 38257bb

Browse files
committed
perf: proxy cookies
1 parent 81feb49 commit 38257bb

2 files changed

Lines changed: 4 additions & 7 deletions

File tree

backend/open_webui/routers/configs.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -263,16 +263,13 @@ async def verify_tool_servers_config(
263263
try:
264264
client = MCPClient()
265265
headers = {}
266-
print("Cookies: ", request.cookies)
267-
cookies_str = '; '.join([f'{key}={value}' for key, value in request.cookies.items()])
268-
headers.update({'cookie': cookies_str})
266+
cookie = '; '.join([f'{key}={value}' for key, value in request.cookies.items()])
267+
headers.update({'cookie': cookie})
269268
token = None
270269
if form_data.auth_type == "bearer":
271270
token = form_data.key
272271
elif form_data.auth_type == "session":
273-
token = request.cookies.get("jms_sessionid", '')
274-
if token:
275-
token = f'jms-{token}'
272+
token = 'jms-' + request.cookies.get('jms_sessionid', '')
276273
# token = request.state.token.credentials
277274
elif form_data.auth_type == "system_oauth":
278275
try:

backend/open_webui/utils/middleware.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1336,7 +1336,7 @@ async def process_chat_payload(request, form_data, user, metadata, model):
13361336
pass
13371337
elif auth_type == "session":
13381338
headers["Authorization"] = (
1339-
f"Bearer jms-{request.cookies.get('jms_sessionid', '')}"
1339+
f"Bearer jms-{request.cookies.get('jms_sessionid', '')}"
13401340
)
13411341
elif auth_type == "system_oauth":
13421342
oauth_token = extra_params.get("__oauth_token__", None)

0 commit comments

Comments
 (0)