recipe: harden auth/metrics, add rate limiting and tests
This commit is contained in:
+1
-2
@@ -25,7 +25,6 @@ async def recognize(
|
||||
image: Annotated[UploadFile, File(...)],
|
||||
prompt: Annotated[str, Form(...)],
|
||||
generation_config: Annotated[str, Form()] = "{}",
|
||||
api_key_override: Annotated[str | None, Form()] = None,
|
||||
) -> dict:
|
||||
if image.content_type not in ALLOWED_TYPES:
|
||||
raise HTTPException(status_code=415, detail="Unsupported image type")
|
||||
@@ -34,7 +33,7 @@ async def recognize(
|
||||
if len(image_data) > MAX_IMAGE_BYTES:
|
||||
raise HTTPException(status_code=413, detail="Image is too large")
|
||||
|
||||
api_key = api_key_override or os.getenv("GEMINI_API_KEY")
|
||||
api_key = os.getenv("GEMINI_API_KEY")
|
||||
if not api_key:
|
||||
raise HTTPException(status_code=503, detail="Gemini is not configured")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user