CalvinBackup/invariant-gateway
1
0
Fork 0
mirror of https://github.com/invariantlabs-ai/invariant-gateway.git synced 2026-03-07 02:00:52 +00: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

8
proxy/routes/anthropic.py
View File

@@ -27,6 +27,8 @@ CONTENT_BLOCK_START = "content_block_start"
CONTENT_BLOCK_DELTA = "content_block_delta"
CONTENT_BLOCK_STOP = "content_block_stop"
HEADER_AUTHORIZATION = "x-api-key"
def validate_headers(
x_api_key: str = Header(None)
):
@@ -54,17 +56,17 @@ async def anthropic_proxy(
if request.headers.get(
"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)
if request.headers.get("invariant-authorization"):
invariant_authorization = request.headers.get("invariant-authorization")
else:
authorization = request.headers.get("x-api-key")
authorization = request.headers.get(HEADER_AUTHORIZATION)
api_keys = authorization.split("|invariant-auth: ")
invariant_authorization = f"Bearer {api_keys[1].strip()}"
# 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()