Procházet zdrojové kódy

Merge pull request #11636 from OliveiraHermogenes/fix-api-audio-mimetype-check

fix: audio api endpoint filetype check
Timothy Jaeryang Baek před 1 měsícem
rodič
revize
4025d61c33
1 změnil soubory, kde provedl 3 přidání a 1 odebrání
  1. 3 1
      backend/open_webui/routers/audio.py

+ 3 - 1
backend/open_webui/routers/audio.py

@@ -625,7 +625,9 @@ def transcription(
 ):
     log.info(f"file.content_type: {file.content_type}")
 
-    if file.content_type not in ["audio/mpeg", "audio/wav", "audio/ogg", "audio/x-m4a"]:
+    supported_filetypes = ("audio/mpeg", "audio/wav", "audio/ogg", "audio/x-m4a")
+
+    if not file.content_type.startswith(supported_filetypes):
         raise HTTPException(
             status_code=status.HTTP_400_BAD_REQUEST,
             detail=ERROR_MESSAGES.FILE_NOT_SUPPORTED,