refac

backend/open_webui/utils/files.py

@@ -6,6 +6,7 @@ import re
 from pathlib import Path
 from typing import Optional
 
+import aiofiles
 from fastapi import (
     APIRouter,
     Depends,
@@ -75,7 +76,7 @@ async def get_image_base64_from_url(url: str, user=None) -> Optional[str]:
             # file-ID resolver which enforces ownership/access checks.
             return await get_image_base64_from_file_id(url, user=user)
 
-    except Exception as e:
+    except Exception:
         return None
 
 
@@ -200,15 +201,15 @@ async def get_image_base64_from_file_id(id: str, user=None) -> Optional[str]:
 
         # Check if the file already exists in the cache
         if file_path.is_file():
-            with open(file_path, 'rb') as image_file:
-                encoded_string = base64.b64encode(image_file.read()).decode('utf-8')
-                content_type = mimetypes.guess_type(file_path.name)[0] or (file.meta or {}).get('content_type')
-                if not content_type and ENABLE_IMAGE_CONTENT_TYPE_EXTENSION_FALLBACK:
-                    content_type = _IMAGE_MIME_FALLBACK.get(file_path.suffix.lower())
-                if not content_type:
-                    return None
-                return f'data:{content_type};base64,{encoded_string}'
+            async with aiofiles.open(file_path, 'rb') as image_file:
+                encoded_string = base64.b64encode(await image_file.read()).decode('utf-8')
+            content_type = mimetypes.guess_type(file_path.name)[0] or (file.meta or {}).get('content_type')
+            if not content_type and ENABLE_IMAGE_CONTENT_TYPE_EXTENSION_FALLBACK:
+                content_type = _IMAGE_MIME_FALLBACK.get(file_path.suffix.lower())
+            if not content_type:
+                return None
+            return f'data:{content_type};base64,{encoded_string}'
         else:
             return None
-    except Exception as e:
+    except Exception:
         return None