caidaoli commited on
Commit
f7dfec8
·
verified ·
1 Parent(s): a6b4ed2

Update openai_ondemand_adapter.py

Browse files
Files changed (1) hide show
  1. openai_ondemand_adapter.py +2 -0
openai_ondemand_adapter.py CHANGED
@@ -21,6 +21,8 @@ def check_private_key():
21
  for header in SAFE_HEADERS:
22
  key = request.headers.get(header)
23
  if key:
 
 
24
  break
25
  if not key or key != PRIVATE_KEY:
26
  return jsonify({"error": "Unauthorized, must provide correct Authorization or X-API-KEY"}), 401
 
21
  for header in SAFE_HEADERS:
22
  key = request.headers.get(header)
23
  if key:
24
+ if header == "Authorization" and key.startswith("Bearer "):
25
+ key = key[len("Bearer "):].strip()
26
  break
27
  if not key or key != PRIVATE_KEY:
28
  return jsonify({"error": "Unauthorized, must provide correct Authorization or X-API-KEY"}), 401