ChatQnA can support authorized LLM service.

Signed-off-by: Yugar-1 <[email protected]>
opea-project · Feb 8, 2025 · 317f7a0 · 317f7a0
1 parent ecb7f7b
commit 317f7a0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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,10 @@ 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,