diff --git a/src/backend/langflow/field_typing/__init__.py b/src/backend/langflow/field_typing/__init__.py index 08578ef44..a206054f4 100644 --- a/src/backend/langflow/field_typing/__init__.py +++ b/src/backend/langflow/field_typing/__init__.py @@ -1,16 +1,35 @@ from typing import Any -from .constants import (AgentExecutor, BaseChatMemory, BaseLanguageModel, - BaseLLM, BaseLoader, BaseMemory, BaseOutputParser, - BasePromptTemplate, BaseRetriever, Callable, Chain, - ChatPromptTemplate, Data, Document, Embeddings, - NestedDict, Object, Prompt, PromptTemplate, - TextSplitter, Tool, VectorStore) +from .constants import ( + AgentExecutor, + BaseChatMemory, + BaseLanguageModel, + BaseLLM, + BaseLoader, + BaseMemory, + BaseOutputParser, + BasePromptTemplate, + BaseRetriever, + Callable, + Chain, + ChatPromptTemplate, + Data, + Document, + Embeddings, + NestedDict, + Object, + Prompt, + PromptTemplate, + TextSplitter, + Tool, + VectorStore, +) from .range_spec import RangeSpec def import_template_field(): from langflow.template.field.base import TemplateField + return TemplateField @@ -21,12 +40,11 @@ def __getattr__(name: str) -> Any: elif name == "RangeSpec": return RangeSpec # The other names should work as if they were imported from constants + import langflow.field_typing.constants as constants + return getattr(constants, name) - - - __all__ = [ "NestedDict", "Data", diff --git a/src/backend/langflow/template/field/base.py b/src/backend/langflow/template/field/base.py index 7571f5209..23a5c767c 100644 --- a/src/backend/langflow/template/field/base.py +++ b/src/backend/langflow/template/field/base.py @@ -1,4 +1,3 @@ -from abc import ABC from typing import Any, Callable, Optional, Union from pydantic import BaseModel, ConfigDict, Field, field_serializer