LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2026-03-04 04:44:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: implement internal API authentication mechanism

Browse Source
This commit is contained in:
Alex
2025-12-04 15:52:45 +00:00
parent 9b9f95710a
commit e68da34c13
5 changed files with 23 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
application/worker.py
View File

@@ -109,6 +109,10 @@ def download_file(url, params, dest_path):
def upload_index(full_path, file_data):
files = None
try:
headers = {}
if settings.INTERNAL_KEY:
headers["X-Internal-Key"] = settings.INTERNAL_KEY
if settings.VECTOR_STORE == "faiss":
faiss_path = full_path + "/index.faiss"
pkl_path = full_path + "/index.pkl"
@@ -129,10 +133,13 @@ def upload_index(full_path, file_data):
urljoin(settings.API_URL, "/api/upload_index"),
files=files,
data=file_data,
headers=headers,
)
else:
response = requests.post(
urljoin(settings.API_URL, "/api/upload_index"), data=file_data
urljoin(settings.API_URL, "/api/upload_index"),
data=file_data,
headers=headers,
)
response.raise_for_status()
except (requests.RequestException, FileNotFoundError) as e: