Replaces python-magic with built-in mimetypes

libraries/python/assistant-drive/assistant_drive/drive.py

@@ -1,21 +1,20 @@
 import io
 import json
 import logging
+import mimetypes
 import os
 import pathlib
 from contextlib import contextmanager
 from datetime import datetime
 from shutil import rmtree
 from typing import Any, BinaryIO, ContextManager, Iterator, TypeVar
 
-import magic
 from context import Context
 from pydantic import BaseModel, Field
 
 # from pydantic_settings import BaseSettings
 
 logger = logging.getLogger(__name__)
-mime = magic.Magic(mime=True)
 
 
 class DriveConfig(BaseModel):
@@ -59,8 +58,8 @@ def __str__(self) -> str:
 
     @staticmethod
     def from_bytes(content: BinaryIO, filename: str, dir: str | None = None) -> "FileMetadata":
-        content_type = mime.from_buffer(content.read())
-        content.seek(0)
+        mime_type, _ = mimetypes.guess_type(filename)
+        content_type = mime_type or "application/octet-stream"
         size = len(content.read())
         return FileMetadata(filename=filename, dir=dir, content_type=content_type, size=size)
 

libraries/python/assistant-drive/pyproject.toml

@@ -9,7 +9,7 @@ dependencies = [
     "context>=0.1.0",
     "pydantic>=2.6.1",
     "pydantic-settings>=2.5.2",
-    "python-magic>=0.4.27",
+    # "python-magic>=0.4.27",
     # "python-magic-bin>=0.4.14",
 ]