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

Add menu path to component template

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-23 11:49:54 -03:00
commit 2b882bd276
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/langflow/interface/types.py
View file

@ -374,7 +374,7 @@ def build_valid_menu(valid_components):
for menu_item in valid_components["menu"]: for menu_item in valid_components["menu"]:
menu_name = menu_item["name"] menu_name = menu_item["name"]
valid_menu[menu_name] = {} valid_menu[menu_name] = {}
menu_path = menu_item["path"]
for component in menu_item["components"]: for component in menu_item["components"]:
logger.debug(f"Building component: {component.get('name'), component.get('output_types')}") logger.debug(f"Building component: {component.get('name'), component.get('output_types')}")
try: try:
@ -383,10 +383,12 @@ def build_valid_menu(valid_components):
component_output_types = component["output_types"] component_output_types = component["output_types"]
component_extractor = CustomComponent(code=component_code) component_extractor = CustomComponent(code=component_code)
component_extractor.is_check_valid() component_extractor.validate()
component_template = build_langchain_template_custom_component(component_extractor) component_template = build_langchain_template_custom_component(component_extractor)
component_template["output_types"] = component_output_types component_template["output_types"] = component_output_types
full_path = f"{menu_path}/{component.get('file')}"
component_template["full_path"] = full_path
if len(component_output_types) == 1: if len(component_output_types) == 1:
component_name = component_output_types[0] component_name = component_output_types[0]
else: else: