CalvinBackup/invariant-gateway
1
0
Fork 0
mirror of https://github.com/invariantlabs-ai/invariant-gateway.git synced 2026-02-12 14:32:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update gemini route to include streamGenerateContent in allowed endpoints response.

Browse Source
This commit is contained in:
Hemang
2025-06-04 15:04:56 +02:00
parent f184c488e8
commit 24d47c4585
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
gateway/integrations/guardrails.py
View File

@@ -364,11 +364,11 @@ async def check_guardrails(
json={
"messages": messages,
"policies": [g.content for g in guardrails],
"parameters": context.guardrails_parameters or {}
"parameters": context.guardrails_parameters or {},
},
headers={
"Authorization": context.get_guardrailing_authorization(),
"Accept": "application/json"
"Accept": "application/json",
},
timeout=5,
)

6
gateway/routes/gemini.py
View File

@@ -58,8 +58,10 @@ async def gemini_generate_content_gateway(
if endpoint not in ["generateContent", "streamGenerateContent"]:
return Response(
content="Invalid endpoint - the only endpoints supported are: \
/api/v1/gateway/gemini/<version>/models/<model-name>:generateContent or \
/api/v1/gateway/<dataset-name>/gemini/<version>models/<model-name>:generateContent",
/api/v1/gateway/gemini/<version>/models/<model-name>:generateContent \
/api/v1/gateway/<dataset-name>/gemini/<version>models/<model-name>:generateContent \
/api/v1/gateway/gemini/<version>/models/<model-name>:streamGenerateContent or \
/api/v1/gateway/<dataset-name>/gemini/<version>models/<model-name>:streamGenerateContent",
status_code=400,
)
headers = {