fix key check in agents,copilot

This commit is contained in:
ramnique 2025-01-29 10:45:37 +05:30
parent 3d13471723
commit b0f38b4e4e
2 changed files with 4 additions and 2 deletions

View file

@ -27,7 +27,8 @@ def require_api_key(f):
return jsonify({'error': 'Missing or invalid authorization header'}), 401 return jsonify({'error': 'Missing or invalid authorization header'}), 401
token = auth_header.split('Bearer ')[1] token = auth_header.split('Bearer ')[1]
if token != os.environ.get('API_KEY', 'test'): actual = os.environ.get('API_KEY', '').strip()
if actual and token != actual:
return jsonify({'error': 'Invalid API key'}), 403 return jsonify({'error': 'Invalid API key'}), 403
return f(*args, **kwargs) return f(*args, **kwargs)

View file

@ -34,7 +34,8 @@ def require_api_key(f):
return jsonify({'error': 'Missing or invalid authorization header'}), 401 return jsonify({'error': 'Missing or invalid authorization header'}), 401
token = auth_header.split('Bearer ')[1] token = auth_header.split('Bearer ')[1]
if token != os.environ.get('API_KEY', 'test'): actual = os.environ.get('API_KEY', '').strip()
if actual and token != actual:
return jsonify({'error': 'Invalid API key'}), 403 return jsonify({'error': 'Invalid API key'}), 403
return f(*args, **kwargs) return f(*args, **kwargs)