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

feat(langflow): add function support to build_template_from_class function

Browse source 
feat(langflow): add function base class to InitializeAgentNode

The `build_template_from_class` function in `ChainCreator` now supports adding a function to the chain. This is done by passing `add_function=True` to the function.

In `InitializeAgentNode`, the `base_classes` attribute has been updated to include the `function` base class. This allows the node to be used as a function in the chain.
This commit is contained in:
Gabriel Almeida 2023-04-28 23:30:12 -03:00
commit 659491f9a2
2 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/interface/chains/base.py
View file

@ -37,7 +37,9 @@ class ChainCreator(LangChainTypeCreator):
try: try:
if name in get_custom_nodes(self.type_name).keys(): if name in get_custom_nodes(self.type_name).keys():
return get_custom_nodes(self.type_name)[name] return get_custom_nodes(self.type_name)[name]
return build_template_from_class(name, self.type_to_loader_dict) return build_template_from_class(
name, self.type_to_loader_dict, add_function=True
)
except ValueError as exc: except ValueError as exc:
raise ValueError("Chain not found") from exc raise ValueError("Chain not found") from exc
except AttributeError as exc: except AttributeError as exc:

2
src/backend/langflow/template/nodes.py
View file

@ -216,7 +216,7 @@ class InitializeAgentNode(FrontendNode):
], ],
) )
description: str = """Construct a json agent from an LLM and tools.""" description: str = """Construct a json agent from an LLM and tools."""
base_classes: list[str] = ["AgentExecutor"] base_classes: list[str] = ["AgentExecutor", "function"]
def to_dict(self): def to_dict(self):
return super().to_dict() return super().to_dict()