CalvinBackup/invariant-gateway
1
0
Fork 0
mirror of https://github.com/invariantlabs-ai/invariant-gateway.git synced 2026-08-16 05:50:21 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #3 from invariantlabs-ai/anthropic-implement

Browse Source 
correct anthropic header key
This commit is contained in:
Hemang Sarkar
2025-02-19 14:42:41 +01:00
committed by GitHub
parent d6e551cb3b 931504975f
commit 9ab00a4e73
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
proxy/routes/anthropic.py
+5 -3
View File
@@ -27,6 +27,8 @@ CONTENT_BLOCK_START = "content_block_start"
CONTENT_BLOCK_DELTA = "content_block_delta" CONTENT_BLOCK_DELTA = "content_block_delta"
CONTENT_BLOCK_STOP = "content_block_stop" CONTENT_BLOCK_STOP = "content_block_stop"
HEADER_AUTHORIZATION = "x-api-key"
def validate_headers( def validate_headers(
x_api_key: str = Header(None) x_api_key: str = Header(None)
): ):
@@ -54,17 +56,17 @@ async def anthropic_proxy(
if request.headers.get( if request.headers.get(
"invariant-authorization" "invariant-authorization"
) is None and "|invariant-auth:" not in request.headers.get("authorization"): ) is None and "|invariant-auth:" not in request.headers.get(HEADER_AUTHORIZATION):
raise HTTPException(status_code=400, detail=MISSING_INVARIANT_AUTH_API_KEY) raise HTTPException(status_code=400, detail=MISSING_INVARIANT_AUTH_API_KEY)
if request.headers.get("invariant-authorization"): if request.headers.get("invariant-authorization"):
invariant_authorization = request.headers.get("invariant-authorization") invariant_authorization = request.headers.get("invariant-authorization")
else: else:
authorization = request.headers.get("x-api-key") authorization = request.headers.get(HEADER_AUTHORIZATION)
api_keys = authorization.split("|invariant-auth: ") api_keys = authorization.split("|invariant-auth: ")
invariant_authorization = f"Bearer {api_keys[1].strip()}" invariant_authorization = f"Bearer {api_keys[1].strip()}"
# Update the authorization header to pass the OpenAI API Key to the OpenAI API # Update the authorization header to pass the OpenAI API Key to the OpenAI API
headers["x-api-key"] = f"{api_keys[0].strip()}" headers[HEADER_AUTHORIZATION] = f"{api_keys[0].strip()}"
request_body = await request.body() request_body = await request.body()