فهرست منبع

updates to formatting

Chris Pietschmann 2 ماه پیش
والد
کامیت
a29f83c4e7
2فایلهای تغییر یافته به همراه7 افزوده شده و 8 حذف شده
  1. 4 3
      backend/open_webui/storage/provider.py
  2. 3 5
      backend/open_webui/test/apps/webui/storage/test_provider.py

+ 4 - 3
backend/open_webui/storage/provider.py

@@ -228,7 +228,6 @@ class GCSStorageProvider(StorageProvider):
         LocalStorageProvider.delete_all_files()
         LocalStorageProvider.delete_all_files()
 
 
 
 
-
 class AzureStorageProvider(StorageProvider):
 class AzureStorageProvider(StorageProvider):
     def __init__(self):
     def __init__(self):
         self.endpoint = AZURE_STORAGE_ENDPOINT
         self.endpoint = AZURE_STORAGE_ENDPOINT
@@ -246,7 +245,9 @@ class AzureStorageProvider(StorageProvider):
             self.blob_service_client = BlobServiceClient(
             self.blob_service_client = BlobServiceClient(
                 account_url=self.endpoint, credential=DefaultAzureCredential()
                 account_url=self.endpoint, credential=DefaultAzureCredential()
             )
             )
-        self.container_client = self.blob_service_client.get_container_client(self.container_name)
+        self.container_client = self.blob_service_client.get_container_client(
+            self.container_name
+        )
 
 
     def upload_file(self, file: BinaryIO, filename: str) -> Tuple[bytes, str]:
     def upload_file(self, file: BinaryIO, filename: str) -> Tuple[bytes, str]:
         """Handles uploading of the file to Azure Blob Storage."""
         """Handles uploading of the file to Azure Blob Storage."""
@@ -293,7 +294,7 @@ class AzureStorageProvider(StorageProvider):
 
 
         # Always delete from local storage
         # Always delete from local storage
         LocalStorageProvider.delete_all_files()
         LocalStorageProvider.delete_all_files()
-    
+
 
 
 def get_storage_provider(storage_provider: str):
 def get_storage_provider(storage_provider: str):
     if storage_provider == "local":
     if storage_provider == "local":

+ 3 - 5
backend/open_webui/test/apps/webui/storage/test_provider.py

@@ -299,17 +299,15 @@ class TestAzureStorageProvider:
         monkeypatch.setattr(
         monkeypatch.setattr(
             azure.storage.blob,
             azure.storage.blob,
             "BlobServiceClient",
             "BlobServiceClient",
-            lambda *args, **kwargs: mock_blob_service_client
+            lambda *args, **kwargs: mock_blob_service_client,
         )
         )
         monkeypatch.setattr(
         monkeypatch.setattr(
             azure.storage.blob,
             azure.storage.blob,
             "BlobContainerClient",
             "BlobContainerClient",
-            lambda *args, **kwargs: mock_container_client
+            lambda *args, **kwargs: mock_container_client,
         )
         )
         monkeypatch.setattr(
         monkeypatch.setattr(
-            azure.storage.blob,
-            "BlobClient",
-            lambda *args, **kwargs: mock_blob_client
+            azure.storage.blob, "BlobClient", lambda *args, **kwargs: mock_blob_client
         )
         )