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

refactor: Update build_component_and_get_results function to handle custom components and base components separately

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-21 20:21:24 -03:00
commit 2affde4dbf
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/interface/initialize/loading.py
View file

@ -66,9 +66,9 @@ async def build_component_and_get_results(
)
with warnings.catch_warnings():
warnings.filterwarnings("ignore", category=PydanticDeprecatedSince20)
if base_type == "custom_components":
if base_type == "custom_components" and isinstance(custom_component, CustomComponent):
return await build_custom_component(params=params_copy, custom_component=custom_component)
elif base_type == "component":
elif base_type == "component" and isinstance(custom_component, Component):
return await build_component(params=params_copy, custom_component=custom_component)
else:
raise ValueError(f"Base type {base_type} not found.")