🔀 chore(field_typing): reorganize imports and update __all__ list for better readability and maintainability

🔥 chore(field_typing): remove unused base.py file as it is no longer needed
🔀 chore(constants): update import statements and add type aliases for better readability and maintainability
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-12 19:23:13 -03:00
commit 4366b02c4f
3 changed files with 60 additions and 13 deletions

View file

@ -1,8 +1,53 @@
from .base import NestedDict, Data
from .constants import *
__all__ = (
["NestedDict", "Data"]
+ list(LANGCHAIN_BASE_TYPES.keys())
+ list(CUSTOM_COMPONENT_SUPPORTED_TYPES.keys())
# LANGCHAIN_BASE_TYPES = {
# "Chain": Chain,
# "AgentExecutor": AgentExecutor,
# "Tool": Tool,
# "BaseLLM": BaseLLM,
# "PromptTemplate": PromptTemplate,
# "BaseLoader": BaseLoader,
# "Document": Document,
# "TextSplitter": TextSplitter,
# "VectorStore": VectorStore,
# "Embeddings": Embeddings,
# "BaseRetriever": BaseRetriever,
# "BaseOutputParser": BaseOutputParser,
# "BaseMemory": BaseMemory,
# "BaseChatMemory": BaseChatMemory,
# }
from .constants import (
Tool,
PromptTemplate,
Chain,
BaseChatMemory,
BaseLLM,
BaseLoader,
BaseMemory,
BaseOutputParser,
BaseRetriever,
VectorStore,
Embeddings,
TextSplitter,
Document,
AgentExecutor,
NestedDict,
Data,
)
__all__ = [
"NestedDict",
"Data",
"Tool",
"PromptTemplate",
"Chain",
"BaseChatMemory",
"BaseLLM",
"BaseLoader",
"BaseMemory",
"BaseOutputParser",
"BaseRetriever",
"VectorStore",
"Embeddings",
"TextSplitter",
"Document",
"AgentExecutor",
]

View file

@ -1,5 +0,0 @@
from typing import Union, Dict, Any
# Type alias for more complex dicts
NestedDict = Dict[str, Union[str, Dict]]
Data = Any

View file

@ -10,7 +10,14 @@ from langchain.schema.memory import BaseMemory
from langchain.text_splitter import TextSplitter
from langchain.tools import Tool
from langchain.vectorstores.base import VectorStore
from langflow.field_typing import Data, NestedDict
from typing import Union, Dict
# Type alias for more complex dicts
NestedDict = Dict[str, Union[str, Dict]]
class Data:
pass
LANGCHAIN_BASE_TYPES = {