adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(mcp market): Handling 401 error

Browse Source
This commit is contained in:
Timebomb2018
2026-03-25 18:46:43 +08:00
parent 504e5ba61e
commit 2df615eca0
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
api/app/controllers/mcp_market_config_controller.py
View File

@@ -91,9 +91,11 @@ async def get_mcp_servers(
try: try:
cookies = api.get_cookies(token) cookies = api.get_cookies(token)
headers=api.builder_headers(api.headers)
headers['Authorization'] = f'Bearer {token}'
r = api.session.put( r = api.session.put(
url=api.mcp_base_url, url=api.mcp_base_url,
headers=api.builder_headers(api.headers), headers=headers,
json=body, json=body,
cookies=cookies) cookies=cookies)
raise_for_http_status(r) raise_for_http_status(r)
@@ -173,6 +175,7 @@ async def get_operational_mcp_servers(
url = f'{api.mcp_base_url}/operational' url = f'{api.mcp_base_url}/operational'
headers = api.builder_headers(api.headers) headers = api.builder_headers(api.headers)
headers['Authorization'] = f'Bearer {token}'
try: try:
cookies = api.get_cookies(access_token=token, cookies_required=True) cookies = api.get_cookies(access_token=token, cookies_required=True)
@@ -260,7 +263,9 @@ async def create_mcp_market_config(
api.login(create_data.token) api.login(create_data.token)
body = {'filter': {}, 'page_number': 1, 'page_size': 1, 'search': None} body = {'filter': {}, 'page_number': 1, 'page_size': 1, 'search': None}
cookies = api.get_cookies(create_data.token) cookies = api.get_cookies(create_data.token)
r = api.session.put(url=api.mcp_base_url, headers=api.builder_headers(api.headers), json=body, cookies=cookies) headers = api.builder_headers(api.headers)
headers['Authorization'] = f'Bearer {create_data.token}'
r = api.session.put(url=api.mcp_base_url, headers=headers, json=body, cookies=cookies)
raise_for_http_status(r) raise_for_http_status(r)
except Exception as e: except Exception as e:
api_logger.warning(f"Token validation failed for ModelScope MCP market: {str(e)}") api_logger.warning(f"Token validation failed for ModelScope MCP market: {str(e)}")
@@ -290,9 +295,11 @@ async def create_mcp_market_config(
'search': "" 'search': ""
} }
cookies = api.get_cookies(token) cookies = api.get_cookies(token)
headers = api.builder_headers(api.headers)
headers['Authorization'] = f'Bearer {token}'
r = api.session.put( r = api.session.put(
url=api.mcp_base_url, url=api.mcp_base_url,
headers=api.builder_headers(api.headers), headers=headers,
json=body, json=body,
cookies=cookies) cookies=cookies)
raise_for_http_status(r) raise_for_http_status(r)
@@ -393,7 +400,9 @@ async def update_mcp_market_config(
api.login(update_data.token) api.login(update_data.token)
body = {'filter': {}, 'page_number': 1, 'page_size': 1, 'search': None} body = {'filter': {}, 'page_number': 1, 'page_size': 1, 'search': None}
cookies = api.get_cookies(update_data.token) cookies = api.get_cookies(update_data.token)
r = api.session.put(url=api.mcp_base_url, headers=api.builder_headers(api.headers), json=body, cookies=cookies) headers = api.builder_headers(api.headers)
headers['Authorization'] = f'Bearer {update_data.token}'
r = api.session.put(url=api.mcp_base_url, headers=headers, json=body, cookies=cookies)
raise_for_http_status(r) raise_for_http_status(r)
except Exception as e: except Exception as e:
api_logger.warning(f"Token validation failed for ModelScope MCP market: {str(e)}") api_logger.warning(f"Token validation failed for ModelScope MCP market: {str(e)}")