From 686b97e69595f826503d10a1799d8f17a2e9add9 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 23 Nov 2023 08:40:59 -0300 Subject: [PATCH] Refactor eval_custom_component_code function signature --- src/backend/langflow/interface/importing/utils.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/backend/langflow/interface/importing/utils.py b/src/backend/langflow/interface/importing/utils.py index fb28bbdc8..f3276d952 100644 --- a/src/backend/langflow/interface/importing/utils.py +++ b/src/backend/langflow/interface/importing/utils.py @@ -9,7 +9,6 @@ from langchain.chains.base import Chain from langchain.chat_models.base import BaseChatModel from langchain.prompts import PromptTemplate from langchain.tools import BaseTool - from langflow.interface.custom.custom_component import CustomComponent from langflow.interface.wrappers.base import wrapper_creator from langflow.utils import validate @@ -181,7 +180,7 @@ def get_function(code): return validate.create_function(code, function_name) -def eval_custom_component_code(code: str) -> CustomComponent: +def eval_custom_component_code(code: str) -> Type[CustomComponent]: """Evaluate custom component code""" class_name = validate.extract_class_name(code) return validate.create_class(code, class_name)