瀏覽代碼

updates to formatting

Chris Pietschmann 2 月之前
父節點
當前提交
a29f83c4e7

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

@@ -228,7 +228,6 @@ class GCSStorageProvider(StorageProvider):
         LocalStorageProvider.delete_all_files()
 
 
-
 class AzureStorageProvider(StorageProvider):
     def __init__(self):
         self.endpoint = AZURE_STORAGE_ENDPOINT
@@ -246,7 +245,9 @@ class AzureStorageProvider(StorageProvider):
             self.blob_service_client = BlobServiceClient(
                 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]:
         """Handles uploading of the file to Azure Blob Storage."""
@@ -293,7 +294,7 @@ class AzureStorageProvider(StorageProvider):
 
         # Always delete from local storage
         LocalStorageProvider.delete_all_files()
-    
+
 
 def get_storage_provider(storage_provider: str):
     if storage_provider == "local":

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

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