diff --git a/src/backend/base/langflow/graph/vertex/base.py b/src/backend/base/langflow/graph/vertex/base.py index c2968df2a..f3891bb75 100644 --- a/src/backend/base/langflow/graph/vertex/base.py +++ b/src/backend/base/langflow/graph/vertex/base.py @@ -4,20 +4,14 @@ import inspect import types from enum import Enum from typing import TYPE_CHECKING, Any, AsyncIterator, Callable, Dict, Iterator, List, Optional - from loguru import logger -from langflow.graph.schema import ( - INPUT_COMPONENTS, - INPUT_FIELD_NAME, - OUTPUT_COMPONENTS, - InterfaceComponentTypes, - ResultData, -) +from langflow.graph.schema import INPUT_COMPONENTS, OUTPUT_COMPONENTS, InterfaceComponentTypes, ResultData from langflow.graph.utils import UnbuiltObject, UnbuiltResult from langflow.graph.vertex.utils import generate_result from langflow.interface.initialize import loading from langflow.interface.listing import lazy_load_dict +from langflow.schema.schema import INPUT_FIELD_NAME from langflow.services.deps import get_storage_service from langflow.utils.constants import DIRECT_TYPES from langflow.utils.schemas import ChatOutputResponse diff --git a/src/backend/base/langflow/services/deps.py b/src/backend/base/langflow/services/deps.py index e1cc0c2c2..ec3a5ec84 100644 --- a/src/backend/base/langflow/services/deps.py +++ b/src/backend/base/langflow/services/deps.py @@ -8,7 +8,6 @@ if TYPE_CHECKING: from langflow.services.cache.service import CacheService from langflow.services.chat.service import ChatService - from langflow.services.credentials.service import VariableService from langflow.services.database.service import DatabaseService from langflow.services.monitor.service import MonitorService from langflow.services.plugins.service import PluginService @@ -18,6 +17,7 @@ if TYPE_CHECKING: from langflow.services.storage.service import StorageService from langflow.services.store.service import StoreService from langflow.services.task.service import TaskService + from langflow.services.variable.service import VariableService def get_socket_service() -> "SocketIOService": diff --git a/src/backend/base/langflow/services/credentials/__init__.py b/src/backend/base/langflow/services/variable/__init__.py similarity index 100% rename from src/backend/base/langflow/services/credentials/__init__.py rename to src/backend/base/langflow/services/variable/__init__.py diff --git a/src/backend/base/langflow/services/credentials/factory.py b/src/backend/base/langflow/services/variable/factory.py similarity index 85% rename from src/backend/base/langflow/services/credentials/factory.py rename to src/backend/base/langflow/services/variable/factory.py index 24a652c4e..aac384807 100644 --- a/src/backend/base/langflow/services/credentials/factory.py +++ b/src/backend/base/langflow/services/variable/factory.py @@ -1,7 +1,7 @@ from typing import TYPE_CHECKING -from langflow.services.credentials.service import VariableService from langflow.services.factory import ServiceFactory +from langflow.services.variable.service import VariableService if TYPE_CHECKING: from langflow.services.settings.service import SettingsService diff --git a/src/backend/base/langflow/services/credentials/service.py b/src/backend/base/langflow/services/variable/service.py similarity index 100% rename from src/backend/base/langflow/services/credentials/service.py rename to src/backend/base/langflow/services/variable/service.py