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

Fix code validation and return type in

Browse source 
CustomComponent
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-23 11:49:13 -03:00
commit 4d285000cf
1 changed files with 10 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/backend/langflow/interface/custom/custom_component.py
View file

@ -5,6 +5,7 @@ from uuid import UUID
import yaml import yaml
from cachetools import TTLCache, cachedmethod from cachetools import TTLCache, cachedmethod
from fastapi import HTTPException from fastapi import HTTPException
from langflow.field_typing.constants import CUSTOM_COMPONENT_SUPPORTED_TYPES from langflow.field_typing.constants import CUSTOM_COMPONENT_SUPPORTED_TYPES
from langflow.interface.custom.component import Component from langflow.interface.custom.component import Component
from langflow.interface.custom.directory_reader import DirectoryReader from langflow.interface.custom.directory_reader import DirectoryReader
@ -73,8 +74,9 @@ class CustomComponent(Component):
"traceback": f"Type hint '{type_hint}' is used but not imported in the code.", "traceback": f"Type hint '{type_hint}' is used but not imported in the code.",
} }
raise HTTPException(status_code=400, detail=error_detail) raise HTTPException(status_code=400, detail=error_detail)
return True
def is_check_valid(self) -> bool: def validate(self) -> bool:
return self._class_template_validation(self.code) if self.code else False return self._class_template_validation(self.code) if self.code else False
def get_code_tree(self, code: str): def get_code_tree(self, code: str):
@ -125,7 +127,7 @@ class CustomComponent(Component):
return build_methods[0] return build_methods[0]
@property @property
def get_function_entrypoint_return_type(self) -> List[str]: def get_function_entrypoint_return_type(self) -> List[Any]:
build_method = self.get_build_method() build_method = self.get_build_method()
if not build_method: if not build_method:
return build_method return build_method
@ -149,13 +151,15 @@ class CustomComponent(Component):
@property @property
def get_main_class_name(self): def get_main_class_name(self):
if not self.code:
return ""
tree = self.get_code_tree(self.code) tree = self.get_code_tree(self.code)
base_name = self.code_class_base_inheritance base_name = self.code_class_base_inheritance
method_name = self.function_entrypoint_name method_name = self.function_entrypoint_name
classes = [] classes = []
for item in tree.get("classes"): for item in tree.get("classes", []):
if base_name in item["bases"]: if base_name in item["bases"]:
method_names = [method["name"] for method in item["methods"]] method_names = [method["name"] for method in item["methods"]]
if method_name in method_names: if method_name in method_names:
@ -166,11 +170,13 @@ class CustomComponent(Component):
@property @property
def build_template_config(self): def build_template_config(self):
if not self.code:
return {}
tree = self.get_code_tree(self.code) tree = self.get_code_tree(self.code)
attributes = [ attributes = [
main_class["attributes"] main_class["attributes"]
for main_class in tree.get("classes") for main_class in tree.get("classes", [])
if main_class["name"] == self.get_main_class_name if main_class["name"] == self.get_main_class_name
] ]
# Get just the first item # Get just the first item