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

Refactor eval_custom_component_code function

Browse source 
signature
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-23 08:40:59 -03:00
commit 686b97e695
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/backend/langflow/interface/importing/utils.py
View file

@ -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)