diff --git a/rag-worker/api_client.py b/rag-worker/api_client.py index d3758e5..56f5ed0 100644 --- a/rag-worker/api_client.py +++ b/rag-worker/api_client.py @@ -1,10 +1,10 @@ from __future__ import annotations import httpx -from config import API_BASE_URL, RAG_WORKER_SECRET, WORKER_ID +from config import API_BASE_URL, INTERNAL_API_KEY, WORKER_ID _auth_headers = { - "Authorization": f"Bearer {RAG_WORKER_SECRET}", + "x-internal-api-key": INTERNAL_API_KEY, "X-Worker-Id": WORKER_ID, } diff --git a/rag-worker/config.py b/rag-worker/config.py index 9c1c632..6ef0083 100644 --- a/rag-worker/config.py +++ b/rag-worker/config.py @@ -12,7 +12,7 @@ if _dotenv_path.exists(): # NestJS 内部 API API_BASE_URL = os.getenv("API_BASE_URL", "http://127.0.0.1:3000") -RAG_WORKER_SECRET = os.getenv("RAG_WORKER_SECRET", "") +INTERNAL_API_KEY = os.getenv("INTERNAL_API_KEY", os.getenv("RAG_WORKER_SECRET", "")) # SiliconFlow SILICONFLOW_API_KEY = os.getenv("SILICONFLOW_API_KEY", "")