jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Update langflow imports in validate.py

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-01 15:10:34 -03:00
commit 13135d3948
1 changed files with 3 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

26
src/backend/langflow/utils/validate.py
View file

@ -4,6 +4,8 @@ import importlib
from types import FunctionType from types import FunctionType
from typing import Dict from typing import Dict
from langflow.field_typing.constants import CUSTOM_COMPONENT_SUPPORTED_TYPES
def add_type_ignores(): def add_type_ignores():
if not hasattr(ast, "TypeIgnore"): if not hasattr(ast, "TypeIgnore"):
@ -266,29 +268,7 @@ def get_default_imports(code_string):
# Add more imports from the typing module as needed # Add more imports from the typing module as needed
} }
langflow_imports = [ langflow_imports = list(CUSTOM_COMPONENT_SUPPORTED_TYPES.keys())
"AgentExecutor",
"BaseChatMemory",
"BaseLanguageModel",
"BaseLLM",
"BaseLoader",
"BaseMemory",
"BaseOutputParser",
"BasePromptTemplate",
"BaseRetriever",
"Callable",
"Chain",
"ChatPromptTemplate",
"Data",
"Document",
"Embeddings",
"NestedDict",
"Object",
"PromptTemplate",
"TextSplitter",
"Tool",
"VectorStore",
]
necessary_imports = find_names_in_code(code_string, langflow_imports) necessary_imports = find_names_in_code(code_string, langflow_imports)
langflow_module = importlib.import_module("langflow.field_typing") langflow_module = importlib.import_module("langflow.field_typing")
default_imports.update({name: getattr(langflow_module, name) for name in necessary_imports}) default_imports.update({name: getattr(langflow_module, name) for name in necessary_imports})