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

Refactor instantiate_custom_component function signature

Browse source 
This commit refactors the function signature of the `instantiate_custom_component` function in `loading.py`. The `class_object` parameter is now annotated with the `Type["CustomComponent"]` type hint to improve type safety.
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-01-30 08:54:41 -03:00
commit 0c16cbed04
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -119,8 +119,8 @@ async def instantiate_based_on_type(class_object, base_type, node_type, params,
async def instantiate_custom_component(node_type, class_object, params, user_id):
params_copy = params.copy()
class_object: "CustomComponent" = eval_custom_component_code(params_copy.pop("code"))
custom_component = class_object(user_id=user_id)
class_object: Type["CustomComponent"] = eval_custom_component_code(params_copy.pop("code"))
custom_component: "CustomComponent" = class_object(user_id=user_id)
if "retriever" in params_copy and hasattr(params_copy["retriever"], "as_retriever"):
params_copy["retriever"] = params_copy["retriever"].as_retriever()