Skip to content

Commit d6ecc06

Browse files
author
jeromechoo
committed
Add bearer token option for passing token
1 parent 5c5d764 commit d6ecc06

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

diffbot_mcp_server.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,9 +14,13 @@ def __init__(self):
1414
self.token = os.getenv('DIFFBOT_TOKEN')
1515
try:
1616
request = Request(get_http_request())
17-
token = request.query_params.get('token')
18-
if token:
19-
self.token = token
17+
auth_header = request.headers.get('Authorization', '')
18+
if auth_header.lower().startswith('bearer '):
19+
self.token = auth_header[7:]
20+
else:
21+
token = request.query_params.get('token')
22+
if token:
23+
self.token = token
2024
except Exception:
2125
# Not an http request, use token in env
2226
pass

0 commit comments

Comments
 (0)