diff --git a/web/api/models.py b/web/api/models.py index 844ce8b..c0206ae 100644 --- 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: