AI/open-webui
1
0
Fork 1
mirror of https://git.mirrors.martin98.com/https://github.com/open-webui/open-webui synced 2025-08-17 11:45:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

updates to formatting

Browse Source
This commit is contained in:
Chris Pietschmann 2025-02-18 15:17:49 -05:00
parent 4c352ff974
commit a29f83c4e7
2 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
backend/open_webui/storage/provider.py
View File

@ -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":

8
backend/open_webui/test/apps/webui/storage/test_provider.py
View File

@ -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, azure.storage.blob, "BlobClient", lambda *args, **kwargs: mock_blob_client
"BlobClient",
lambda *args, **kwargs: mock_blob_client
) )