feat(file_manager): add support for file extension attribute

- Introduced `EXTENSION` to `FileAttribute` enum. - Updated `get_attr` function to handle the new `EXTENSION` attribute.
langgenius · Sep 27, 2024 · 58f700c · 58f700c
1 parent 3b01c0e
commit 58f700c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/api/core/file/enums.py b/api/core/file/enums.py
@@ -48,6 +48,7 @@ class FileAttribute(str, Enum):
     MIME_TYPE = "mime_type"
     TRANSFER_METHOD = "transfer_method"
     URL = "url"
+    EXTENSION = "extension"
 
 
 class ArrayFileAttribute(str, Enum):

diff --git a/api/core/file/file_manager.py b/api/core/file/file_manager.py
@@ -26,6 +26,8 @@ def get_attr(*, file: "File", attr: "FileAttribute"):
             return file.transfer_method.value
         case FileAttribute.URL:
             return file.remote_url
+        case FileAttribute.EXTENSION:
+            return file.extension
         case _:
             raise ValueError(f"Invalid file attribute: {attr}")