diff --git a/comps/cores/mega/orchestrator.py b/comps/cores/mega/orchestrator.py index 97ee2a76b..aab8a4ef5 100644 --- a/comps/cores/mega/orchestrator.py +++ b/comps/cores/mega/orchestrator.py @@ -25,6 +25,7 @@ logger = CustomLogger("comps-core-orchestrator") LOGFLAG = os.getenv("LOGFLAG", False) +API_KEY_TOKENS = os.getenv("API_KEY_TOKEN", "EMPTY") class OrchestratorMetrics: @@ -259,7 +260,7 @@ async def execute( response = requests.post( url=endpoint, data=json.dumps(inputs), - headers={"Content-type": "application/json"}, + headers={"Content-type": "application/json", "Authorization": f"Bearer {API_KEY_TOKENS}"}, proxies={"http": None}, stream=True, timeout=1000,