From 17e70ee2b77efdcad7e02f5ec2d5964484847b9a Mon Sep 17 00:00:00 2001 From: harini-venkataraman Date: Fri, 10 Jan 2025 11:52:20 +0530 Subject: [PATCH] Import enhancements --- .../prompt_studio_core_v2/prompt_studio_helper.py | 3 +-- .../utils/file_storage/helpers/prompt_studio_file_helper.py | 3 +-- prompt-service/src/unstract/prompt_service/helper.py | 3 +-- 3 files changed, 3 insertions(+), 6 deletions(-) diff --git a/backend/prompt_studio/prompt_studio_core_v2/prompt_studio_helper.py b/backend/prompt_studio/prompt_studio_core_v2/prompt_studio_helper.py index f5ccf5419..c5b17a0a6 100644 --- a/backend/prompt_studio/prompt_studio_core_v2/prompt_studio_helper.py +++ b/backend/prompt_studio/prompt_studio_core_v2/prompt_studio_helper.py @@ -55,9 +55,8 @@ from prompt_studio.prompt_studio_v2.models import ToolStudioPrompt from unstract.sdk.constants import LogLevel from unstract.sdk.exceptions import IndexingError, SdkError -from unstract.sdk.file_storage import FileStorage, FileStorageProvider +from unstract.sdk.file_storage import EnvHelper, FileStorage, FileStorageProvider from unstract.sdk.file_storage.constants import StorageType -from unstract.sdk.file_storage.env_helper import EnvHelper from unstract.sdk.index import Index from unstract.sdk.prompt import PromptTool from unstract.sdk.utils.tool_utils import ToolUtils diff --git a/backend/utils/file_storage/helpers/prompt_studio_file_helper.py b/backend/utils/file_storage/helpers/prompt_studio_file_helper.py index 8b6811069..3cf83ff9e 100644 --- a/backend/utils/file_storage/helpers/prompt_studio_file_helper.py +++ b/backend/utils/file_storage/helpers/prompt_studio_file_helper.py @@ -5,9 +5,8 @@ from typing import Any, Union from file_management.file_management_helper import FileManagerHelper -from unstract.sdk.file_storage import FileStorage +from unstract.sdk.file_storage import EnvHelper, FileStorage from unstract.sdk.file_storage.constants import StorageType -from unstract.sdk.file_storage.env_helper import EnvHelper from utils.file_storage.constants import FileStorageConstants, FileStorageKeys from unstract.core.utilities import UnstractUtils diff --git a/prompt-service/src/unstract/prompt_service/helper.py b/prompt-service/src/unstract/prompt_service/helper.py index e451a2030..8f609e46d 100644 --- a/prompt-service/src/unstract/prompt_service/helper.py +++ b/prompt-service/src/unstract/prompt_service/helper.py @@ -24,9 +24,8 @@ from unstract.flags.feature_flag import check_feature_flag_status if check_feature_flag_status(FeatureFlag.REMOTE_FILE_STORAGE): - from unstract.sdk.file_storage import FileStorage, FileStorageProvider + from unstract.sdk.file_storage import EnvHelper, FileStorage, FileStorageProvider from unstract.sdk.file_storage.constants import StorageType - from unstract.sdk.file_storage.env_helper import EnvHelper load_dotenv()