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

refactor: Update instantiate_class function to support CustomComponent and Component inheritance

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

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

@ -37,7 +37,7 @@ async def instantiate_class(
params_copy = params.copy() params_copy = params.copy()
# Remove code from params # Remove code from params
class_object: Type["CustomComponent"] = eval_custom_component_code(params_copy.pop("code")) class_object: Type["CustomComponent"] = eval_custom_component_code(params_copy.pop("code"))
custom_component: "CustomComponent" = class_object( custom_component: "CustomComponent" | "Component" = class_object(
user_id=user_id, user_id=user_id,
parameters=params_copy, parameters=params_copy,
vertex=vertex, vertex=vertex,
@ -47,10 +47,10 @@ async def instantiate_class(
) )
with warnings.catch_warnings(): with warnings.catch_warnings():
warnings.filterwarnings("ignore", category=PydanticDeprecatedSince20) warnings.filterwarnings("ignore", category=PydanticDeprecatedSince20)
if base_type == "custom_components": if issubclass(class_object, Component):
return await build_custom_component(params=params_copy, custom_component=custom_component)
elif base_type == "component":
return await build_component(params=params_copy, custom_component=custom_component, vertex=vertex) return await build_component(params=params_copy, custom_component=custom_component, vertex=vertex)
elif issubclass(class_object, CustomComponent):
return await build_custom_component(params=params_copy, custom_component=custom_component)
else: else:
raise ValueError(f"Base type {base_type} not found.") raise ValueError(f"Base type {base_type} not found.")