From fd7da8720f1afbafffb85412656b51f120c75fc2 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 8 Mar 2024 11:35:55 -0300 Subject: [PATCH] Set up __init__ for all components --- src/backend/langflow/components/__init__.py | 22 ++++++++-- .../langflow/components/agents/__init__.py | 17 +++++++ .../langflow/components/chains/__init__.py | 17 +++++++ .../langflow/components/data/__init__.py | 7 +++ .../components/embeddings/__init__.py | 19 ++++++++ .../components/experimental/__init__.py | 25 +++++++++++ .../langflow/components/helpers/__init__.py | 13 ++++++ .../langflow/components/inputs/__init__.py | 4 ++ .../components/model_specs/__init__.py | 39 ++++++++++++++++ .../langflow/components/models/__init__.py | 28 ++++++++++++ .../components/models/base/__init__.py | 3 ++ .../langflow/components/outputs/__init__.py | 4 ++ .../components/retrievers/__init__.py | 11 +++++ .../components/textsplitters/__init__.py | 9 ++++ .../langflow/components/toolkits/__init__.py | 15 +++++++ .../components/vectorstores/__init__.py | 44 +++++++++++++++++++ .../components/vectorstores/base/__init__.py | 3 ++ 17 files changed, 276 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/components/__init__.py b/src/backend/langflow/components/__init__.py index 765042210..8ecc4c1a9 100644 --- a/src/backend/langflow/components/__init__.py +++ b/src/backend/langflow/components/__init__.py @@ -1,4 +1,18 @@ -from langflow.interface.custom.custom_component import CustomComponent - - -__all__ = ["CustomComponent"] +__all__ = [ + "agents", + "chains", + "data", + "documentloaders", + "embeddings", + "experimental", + "helpers", + "inputs", + "memories", + "model_specs", + "models", + "outputs", + "retrievers", + "textsplitters", + "toolkits", + "vectorstores", +] diff --git a/src/backend/langflow/components/agents/__init__.py b/src/backend/langflow/components/agents/__init__.py index e69de29bb..c003c80c2 100644 --- a/src/backend/langflow/components/agents/__init__.py +++ b/src/backend/langflow/components/agents/__init__.py @@ -0,0 +1,17 @@ +from .AgentInitializer import AgentInitializerComponent +from .CSVAgent import CSVAgentComponent +from .JsonAgent import JsonAgentComponent +from .OpenAIConversationalAgent import ConversationalAgent +from .SQLAgent import SQLAgentComponent +from .VectorStoreAgent import VectorStoreAgentComponent +from .VectorStoreRouterAgent import VectorStoreRouterAgentComponent + +__all__ = [ + "AgentInitializerComponent", + "CSVAgentComponent", + "JsonAgentComponent", + "ConversationalAgent", + "SQLAgentComponent", + "VectorStoreAgentComponent", + "VectorStoreRouterAgentComponent", +] diff --git a/src/backend/langflow/components/chains/__init__.py b/src/backend/langflow/components/chains/__init__.py index e69de29bb..365a80eb6 100644 --- a/src/backend/langflow/components/chains/__init__.py +++ b/src/backend/langflow/components/chains/__init__.py @@ -0,0 +1,17 @@ +from .ConversationChain import ConversationChainComponent +from .LLMChain import LLMChainComponent +from .LLMCheckerChain import LLMCheckerChainComponent +from .LLMMathChain import LLMMathChainComponent +from .RetrievalQA import RetrievalQAComponent +from .RetrievalQAWithSourcesChain import RetrievalQAWithSourcesChainComponent +from .SQLGenerator import SQLGeneratorComponent + +__all__ = [ + "ConversationChainComponent", + "LLMChainComponent", + "LLMCheckerChainComponent", + "LLMMathChainComponent", + "RetrievalQAComponent", + "RetrievalQAWithSourcesChainComponent", + "SQLGeneratorComponent", +] diff --git a/src/backend/langflow/components/data/__init__.py b/src/backend/langflow/components/data/__init__.py index e69de29bb..a977ab54f 100644 --- a/src/backend/langflow/components/data/__init__.py +++ b/src/backend/langflow/components/data/__init__.py @@ -0,0 +1,7 @@ +from .APIRequest import APIRequest +from .Directory import DirectoryComponent +from .File import FileComponent +from .FileLoader import FileLoaderComponent +from .URL import URLComponent + +__all__ = ["APIRequest", "DirectoryComponent", "FileComponent", "FileLoaderComponent", "URLComponent"] diff --git a/src/backend/langflow/components/embeddings/__init__.py b/src/backend/langflow/components/embeddings/__init__.py index e69de29bb..f2d1bc48e 100644 --- a/src/backend/langflow/components/embeddings/__init__.py +++ b/src/backend/langflow/components/embeddings/__init__.py @@ -0,0 +1,19 @@ +from .AmazonBedrockEmbeddings import AmazonBedrockEmeddingsComponent +from .AzureOpenAIEmbeddings import AzureOpenAIEmbeddingsComponent +from .CohereEmbeddings import CohereEmbeddingsComponent +from .HuggingFaceEmbeddings import HuggingFaceEmbeddingsComponent +from .HuggingFaceInferenceAPIEmbeddings import HuggingFaceInferenceAPIEmbeddingsComponent +from .OllamaEmbeddings import OllamaEmbeddingsComponent +from .OpenAIEmbeddings import OpenAIEmbeddingsComponent +from .VertexAIEmbeddings import VertexAIEmbeddingsComponent + +__all__ = [ + "AmazonBedrockEmeddingsComponent", + "AzureOpenAIEmbeddingsComponent", + "CohereEmbeddingsComponent", + "HuggingFaceEmbeddingsComponent", + "HuggingFaceInferenceAPIEmbeddingsComponent", + "OllamaEmbeddingsComponent", + "OpenAIEmbeddingsComponent", + "VertexAIEmbeddingsComponent", +] diff --git a/src/backend/langflow/components/experimental/__init__.py b/src/backend/langflow/components/experimental/__init__.py index e69de29bb..5771f4353 100644 --- a/src/backend/langflow/components/experimental/__init__.py +++ b/src/backend/langflow/components/experimental/__init__.py @@ -0,0 +1,25 @@ +from .ClearMessageHistory import ClearMessageHistoryComponent +from .ExtractDataFromRecord import ExtractKeyFromRecordComponent +from .GetNotified import GetNotifiedComponent +from .ListFlows import ListFlowsComponent +from .MergeRecords import MergeRecordsComponent +from .MessageHistory import MessageHistoryComponent +from .Notify import NotifyComponent +from .RunFlow import RunFlowComponent +from .RunnableExecutor import RunnableExecComponent +from .SQLExecutor import SQLExecutorComponent +from .TextToRecord import TextToRecordComponent + +__all__ = [ + "ClearMessageHistoryComponent", + "ExtractKeyFromRecordComponent", + "GetNotifiedComponent", + "ListFlowsComponent", + "MergeRecordsComponent", + "MessageHistoryComponent", + "NotifyComponent", + "RunFlowComponent", + "RunnableExecComponent", + "SQLExecutorComponent", + "TextToRecordComponent", +] diff --git a/src/backend/langflow/components/helpers/__init__.py b/src/backend/langflow/components/helpers/__init__.py index e69de29bb..f659a0438 100644 --- a/src/backend/langflow/components/helpers/__init__.py +++ b/src/backend/langflow/components/helpers/__init__.py @@ -0,0 +1,13 @@ +from .CustomComponent import Component +from .DocumentToRecord import DocumentToRecordComponent +from .IDGenerator import UUIDGeneratorComponent +from .PythonFunction import PythonFunctionComponent +from .RecordsAsText import RecordsAsTextComponent + +__all__ = [ + "Component", + "DocumentToRecordComponent", + "UUIDGeneratorComponent", + "PythonFunctionComponent", + "RecordsAsTextComponent", +] diff --git a/src/backend/langflow/components/inputs/__init__.py b/src/backend/langflow/components/inputs/__init__.py index e69de29bb..c8b4f1645 100644 --- a/src/backend/langflow/components/inputs/__init__.py +++ b/src/backend/langflow/components/inputs/__init__.py @@ -0,0 +1,4 @@ +from .ChatInput import ChatInput +from .TextInput import TextInput + +__all__ = ["ChatInput", "TextInput"] diff --git a/src/backend/langflow/components/model_specs/__init__.py b/src/backend/langflow/components/model_specs/__init__.py index e69de29bb..15474a28e 100644 --- a/src/backend/langflow/components/model_specs/__init__.py +++ b/src/backend/langflow/components/model_specs/__init__.py @@ -0,0 +1,39 @@ +from .AmazonBedrockSpecs import AmazonBedrockComponent +from .AnthropicLLMSpecs import AnthropicLLM +from .AnthropicSpecs import AnthropicComponent +from .AzureChatOpenAISpecs import AzureChatOpenAISpecsComponent +from .BaiduQianfanChatEndpointsSpecs import QianfanChatEndpointComponent +from .BaiduQianfanLLMEndpointsSpecs import QianfanLLMEndpointComponent +from .CTransformersSpecs import CTransformersComponent +from .ChatAnthropicSpecs import ChatAnthropicComponent +from .ChatLiteLLMSpecs import ChatLiteLLMComponent +from .ChatOllamaEndpointSpecs import ChatOllamaComponent +from .ChatOpenAISpecs import ChatOpenAIComponent +from .ChatVertexAISpecs import ChatVertexAIComponent +from .CohereSpecs import CohereComponent +from .GoogleGenerativeAISpecs import GoogleGenerativeAIComponent +from .HuggingFaceEndpointsSpecs import HuggingFaceEndpointsComponent +from .LlamaCppSpecs import LlamaCppComponent +from .OllamaLLMSpecs import OllamaLLM +from .VertexAISpecs import VertexAIComponent + +__all__ = [ + "AmazonBedrockComponent", + "AnthropicLLM", + "AnthropicComponent", + "AzureChatOpenAISpecsComponent", + "QianfanChatEndpointComponent", + "QianfanLLMEndpointComponent", + "CTransformersComponent", + "ChatAnthropicComponent", + "ChatLiteLLMComponent", + "ChatOllamaComponent", + "ChatOpenAIComponent", + "ChatVertexAIComponent", + "CohereComponent", + "GoogleGenerativeAIComponent", + "HuggingFaceEndpointsComponent", + "LlamaCppComponent", + "OllamaLLM", + "VertexAIComponent", +] diff --git a/src/backend/langflow/components/models/__init__.py b/src/backend/langflow/components/models/__init__.py index e69de29bb..651132f53 100644 --- a/src/backend/langflow/components/models/__init__.py +++ b/src/backend/langflow/components/models/__init__.py @@ -0,0 +1,28 @@ +from .AmazonBedrockModel import AmazonBedrockComponent +from .AnthropicModel import AnthropicLLM +from .AzureOpenAIModel import AzureChatOpenAIComponent +from .BaiduQianfanChatModel import QianfanChatEndpointComponent +from .CTransformersModel import CTransformersComponent +from .CohereModel import CohereComponent +from .GoogleGenerativeAIModel import GoogleGenerativeAIComponent +from .HuggingFaceModel import HuggingFaceEndpointsComponent +from .LlamaCppModel import LlamaCppComponent +from .OllamaModel import ChatOllamaComponent +from .OpenAIModel import OpenAIModelComponent +from .VertexAiModel import ChatVertexAIComponent + +__all__ = [ + "AmazonBedrockComponent", + "AnthropicLLM", + "AzureChatOpenAIComponent", + "QianfanChatEndpointComponent", + "CTransformersComponent", + "CohereComponent", + "GoogleGenerativeAIComponent", + "HuggingFaceEndpointsComponent", + "LlamaCppComponent", + "ChatOllamaComponent", + "OpenAIModelComponent", + "ChatVertexAIComponent", + "base", +] diff --git a/src/backend/langflow/components/models/base/__init__.py b/src/backend/langflow/components/models/base/__init__.py index e69de29bb..921f10336 100644 --- a/src/backend/langflow/components/models/base/__init__.py +++ b/src/backend/langflow/components/models/base/__init__.py @@ -0,0 +1,3 @@ +from .model import LCModelComponent + +__all__ = ["LCModelComponent"] diff --git a/src/backend/langflow/components/outputs/__init__.py b/src/backend/langflow/components/outputs/__init__.py index e69de29bb..cf395bb86 100644 --- a/src/backend/langflow/components/outputs/__init__.py +++ b/src/backend/langflow/components/outputs/__init__.py @@ -0,0 +1,4 @@ +from .ChatOutput import ChatOutput +from .TextOutput import TextOutput + +__all__ = ["ChatOutput", "TextOutput"] diff --git a/src/backend/langflow/components/retrievers/__init__.py b/src/backend/langflow/components/retrievers/__init__.py index e69de29bb..c01701e13 100644 --- a/src/backend/langflow/components/retrievers/__init__.py +++ b/src/backend/langflow/components/retrievers/__init__.py @@ -0,0 +1,11 @@ +from .AmazonKendra import AmazonKendraRetrieverComponent +from .MetalRetriever import MetalRetrieverComponent +from .MultiQueryRetriever import MultiQueryRetrieverComponent +from .VectaraSelfQueryRetriver import VectaraSelfQueryRetriverComponent + +__all__ = [ + "AmazonKendraRetrieverComponent", + "MetalRetrieverComponent", + "MultiQueryRetrieverComponent", + "VectaraSelfQueryRetriverComponent", +] diff --git a/src/backend/langflow/components/textsplitters/__init__.py b/src/backend/langflow/components/textsplitters/__init__.py index e69de29bb..eb2d6af13 100644 --- a/src/backend/langflow/components/textsplitters/__init__.py +++ b/src/backend/langflow/components/textsplitters/__init__.py @@ -0,0 +1,9 @@ +from .CharacterTextSplitter import CharacterTextSplitterComponent +from .LanguageRecursiveTextSplitter import LanguageRecursiveTextSplitterComponent +from .RecursiveCharacterTextSplitter import RecursiveCharacterTextSplitterComponent + +__all__ = [ + "CharacterTextSplitterComponent", + "LanguageRecursiveTextSplitterComponent", + "RecursiveCharacterTextSplitterComponent", +] diff --git a/src/backend/langflow/components/toolkits/__init__.py b/src/backend/langflow/components/toolkits/__init__.py index e69de29bb..189b43159 100644 --- a/src/backend/langflow/components/toolkits/__init__.py +++ b/src/backend/langflow/components/toolkits/__init__.py @@ -0,0 +1,15 @@ +from .JsonToolkit import JsonToolkitComponent +from .Metaphor import MetaphorToolkit +from .OpenAPIToolkit import OpenAPIToolkitComponent +from .VectorStoreInfo import VectorStoreInfoComponent +from .VectorStoreRouterToolkit import VectorStoreRouterToolkitComponent +from .VectorStoreToolkit import VectorStoreToolkitComponent + +__all__ = [ + "JsonToolkitComponent", + "MetaphorToolkit", + "OpenAPIToolkitComponent", + "VectorStoreInfoComponent", + "VectorStoreRouterToolkitComponent", + "VectorStoreToolkitComponent", +] diff --git a/src/backend/langflow/components/vectorstores/__init__.py b/src/backend/langflow/components/vectorstores/__init__.py index e69de29bb..a2ddfa53d 100644 --- a/src/backend/langflow/components/vectorstores/__init__.py +++ b/src/backend/langflow/components/vectorstores/__init__.py @@ -0,0 +1,44 @@ +from .Chroma import ChromaComponent +from .ChromaSearch import ChromaSearchComponent +from .FAISS import FAISSComponent +from .FAISSSearch import FAISSSearchComponent +from .MongoDBAtlasVector import MongoDBAtlasComponent +from .MongoDBAtlasVectorSearch import MongoDBAtlasSearchComponent +from .Pinecone import PineconeComponent +from .PineconeSearch import PineconeSearchComponent +from .Qdrant import QdrantComponent +from .QdrantSearch import QdrantSearchComponent +from .Redis import RedisComponent +from .RedisSearch import RedisSearchComponent +from .SupabaseVectorStore import SupabaseComponent +from .SupabaseVectorStoreSearch import SupabaseSearchComponent +from .Vectara import VectaraComponent +from .VectaraSearch import VectaraSearchComponent +from .Weaviate import WeaviateVectorStoreComponent +from .WeaviateSearch import WeaviateSearchVectorStore +from .pgvector import PGVectorComponent +from .pgvectorSearch import PGVectorSearchComponent + +__all__ = [ + "ChromaComponent", + "ChromaSearchComponent", + "FAISSComponent", + "FAISSSearchComponent", + "MongoDBAtlasComponent", + "MongoDBAtlasSearchComponent", + "PineconeComponent", + "PineconeSearchComponent", + "QdrantComponent", + "QdrantSearchComponent", + "RedisComponent", + "RedisSearchComponent", + "SupabaseComponent", + "SupabaseSearchComponent", + "VectaraComponent", + "VectaraSearchComponent", + "WeaviateVectorStoreComponent", + "WeaviateSearchVectorStore", + "base", + "PGVectorComponent", + "PGVectorSearchComponent", +] diff --git a/src/backend/langflow/components/vectorstores/base/__init__.py b/src/backend/langflow/components/vectorstores/base/__init__.py index e69de29bb..93e42c4aa 100644 --- a/src/backend/langflow/components/vectorstores/base/__init__.py +++ b/src/backend/langflow/components/vectorstores/base/__init__.py @@ -0,0 +1,3 @@ +from .model import LCVectorStoreComponent + +__all__ = ["LCVectorStoreComponent"]