Merge pull request #108 from tjmlabs/csv-bug-fix

Handle csv mime type
tjmlabs · Dec 4, 2024 · 0a7c52f · 0a7c52f
2 parents 69c183d + 388ce36
commit 0a7c52f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/web/api/models.py b/web/api/models.py
@@ -20,7 +20,8 @@
 from django_stubs_ext.db.models import TypedModelMeta
 from pdf2image import convert_from_bytes
 from pgvector.django import HalfVectorField
-from tenacity import retry, retry_if_exception_type, stop_after_attempt, wait_fixed
+from tenacity import (retry, retry_if_exception_type, stop_after_attempt,
+                      wait_fixed)
 
 logger = logging.getLogger(__name__)
 
@@ -52,6 +53,7 @@ def get_extension_from_mime(mime_type):
         "image/png": ".png",
         "image/gif": ".gif",
         "application/pdf": ".pdf",
+        "application/csv": ".csv",
     }
 
     if mime_type in hardcode_mimetypes: