klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-08 05:39:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Limit max audio file size allowed for transcription on API endpoint

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2023-11-26 14:19:46 -08:00
parent a79604b601
commit b249bbb5b5
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/khoj/routers/api.py
View File

@@ -595,6 +595,11 @@ async def transcribe(request: Request, common: CommonQueryParams, file: UploadFi
audio_filename = f"{user.uuid}-{str(uuid.uuid4())}.webm" audio_filename = f"{user.uuid}-{str(uuid.uuid4())}.webm"
user_message: str = None user_message: str = None
# If the file is too large, return an unprocessable entity error
if file.size > 10 * 1024 * 1024:
logger.warning(f"Audio file too large to transcribe. Audio file size: {file.size}. Exceeds 10Mb limit.")
return Response(content="Audio size larger than 10Mb limit", status_code=422)
# Transcribe the audio from the request # Transcribe the audio from the request
try: try:
# Store the audio from the request in a temporary file # Store the audio from the request in a temporary file
@@ -627,7 +632,7 @@ async def transcribe(request: Request, common: CommonQueryParams, file: UploadFi
update_telemetry_state( update_telemetry_state(
request=request, request=request,
telemetry_type="api", telemetry_type="api",
api="speech_to_text", api="transcribe",
**common.__dict__, **common.__dict__,
) )