diff --git a/label_studio/core/storage.py b/label_studio/core/storage.py index 05a6926d1608..c6f099a52ec3 100644 --- a/label_studio/core/storage.py +++ b/label_studio/core/storage.py @@ -66,7 +66,10 @@ def url(self, name, storage_url=False, *args, **kwargs): class CustomS3Boto3Storage(StorageProxyMixin, S3Boto3Storage): - pass + def url(self, name, storage_url=False, *args, **kwargs): + if storage_url is True: + return super().url(name, *args, **kwargs) + return f"s3://{settings.AWS_STORAGE_BUCKET_NAME}/{name}" class CustomAzureStorage(StorageProxyMixin, AzureStorage): diff --git a/label_studio/data_import/models.py b/label_studio/data_import/models.py index 8a92bd3ed7ee..8b7dbc626778 100644 --- a/label_studio/data_import/models.py +++ b/label_studio/data_import/models.py @@ -113,7 +113,7 @@ def read_task_from_hypertext_body(self): def read_task_from_uploaded_file(self): logger.debug('Read 1 task from uploaded file {}'.format(self.filepath)) if settings.CLOUD_FILE_STORAGE_ENABLED: - tasks = [{'data': {settings.DATA_UNDEFINED_NAME: self.filepath}}] + tasks = [{'data': {settings.DATA_UNDEFINED_NAME: self.file.storage.url(self.file.name)}}] else: tasks = [{'data': {settings.DATA_UNDEFINED_NAME: self.url}}] return tasks