mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 13:52:43 +08:00
25 lines
737 B
Python
25 lines
737 B
Python
import os
|
|
|
|
from fastapi import Depends, HTTPException, status
|
|
from fastapi.security import APIKeyHeader
|
|
|
|
_SESSION_API_KEY = os.getenv('SESSION_API_KEY')
|
|
_SESSION_API_KEY_HEADER = APIKeyHeader(name='X-Session-API-Key', auto_error=False)
|
|
|
|
|
|
def check_session_api_key(
|
|
session_api_key: str | None = Depends(_SESSION_API_KEY_HEADER),
|
|
):
|
|
"""Check the session API key and throw an exception if incorrect. Having this as a dependency
|
|
means it appears in OpenAPI Docs
|
|
"""
|
|
if session_api_key != _SESSION_API_KEY:
|
|
raise HTTPException(status.HTTP_401_UNAUTHORIZED)
|
|
|
|
|
|
def get_dependencies() -> list[Depends]:
|
|
result = []
|
|
if _SESSION_API_KEY:
|
|
result.append(Depends(check_session_api_key))
|
|
return result
|