🔀 merge(base.py): add SQLDatabaseChain to the ChainCreator's type_dict to enable support for SQL database chains

📦 chore(pyproject.toml): add langchain-experimental package as a dependency
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-02 22:05:52 -03:00
commit d397f4457f
3 changed files with 19 additions and 1 deletions

View file

@ -8,6 +8,7 @@ from langflow.template.frontend_node.chains import ChainFrontendNode
from langflow.utils.logger import logger
from langflow.utils.util import build_template_from_class, build_template_from_method
from langchain import chains
from langchain_experimental.sql import SQLDatabaseChain
# Assuming necessary imports for Field, Template, and FrontendNode classes
@ -35,6 +36,8 @@ class ChainCreator(LangChainTypeCreator):
}
from langflow.interface.chains.custom import CUSTOM_CHAINS
self.type_dict["SQLDatabaseChain"] = SQLDatabaseChain
self.type_dict.update(CUSTOM_CHAINS)
# Filter according to settings.chains
self.type_dict = {