From 7cfd5171d7a1c95606be376b07d023e542be9df8 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sun, 10 Dec 2023 20:30:22 -0300 Subject: [PATCH] Import langflow.field_typing.constants in __init__.py --- src/backend/langflow/field_typing/__init__.py | 36 ++++++++++++++----- src/backend/langflow/template/field/base.py | 1 - 2 files changed, 27 insertions(+), 10 deletions(-) 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