🐛 fix(custom_component.py): fix error message typo in CustomComponent class
🐛 fix(directory_reader.py): fix error message typo in DirectoryReader class
This commit is contained in:
parent
15643263a8
commit
082a61eadd
2 changed files with 12 additions and 5 deletions
|
|
@ -2,6 +2,7 @@ from typing import Any, Callable, List, Optional
|
||||||
from fastapi import HTTPException
|
from fastapi import HTTPException
|
||||||
from langflow.interface.custom.constants import CUSTOM_COMPONENT_SUPPORTED_TYPES
|
from langflow.interface.custom.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.utils import validate
|
from langflow.utils import validate
|
||||||
|
|
||||||
|
|
@ -38,12 +39,18 @@ class CustomComponent(Component, extra=Extra.allow):
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
|
||||||
# TODO: Create the logic to validate what the Custom Component
|
reader = DirectoryReader("", False)
|
||||||
# should have as a prerequisite to be able to execute
|
if reader.is_type_hint_used_but_not_imported("Optional", code):
|
||||||
return True
|
raise HTTPException(
|
||||||
|
status_code=400,
|
||||||
|
detail={
|
||||||
|
"error": "Type hint Error",
|
||||||
|
"traceback": "Name type hint 'Optional' is not defined",
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
def is_check_valid(self) -> bool:
|
def is_check_valid(self) -> bool:
|
||||||
return self._class_template_validation(self.code) if self.code else False
|
return self._class_template_validation(self.code)
|
||||||
|
|
||||||
def get_code_tree(self, code: str):
|
def get_code_tree(self, code: str):
|
||||||
return super().get_code_tree(code)
|
return super().get_code_tree(code)
|
||||||
|
|
|
||||||
|
|
@ -196,7 +196,7 @@ class DirectoryReader:
|
||||||
elif not self.validate_build(file_content):
|
elif not self.validate_build(file_content):
|
||||||
return False, "Missing build function"
|
return False, "Missing build function"
|
||||||
elif self.is_type_hint_used_but_not_imported("Optional", file_content):
|
elif self.is_type_hint_used_but_not_imported("Optional", file_content):
|
||||||
return False, "Name hint type 'Optional' is not defined"
|
return False, "Name type hint 'Optional' is not defined"
|
||||||
else:
|
else:
|
||||||
if self.compress_code_field:
|
if self.compress_code_field:
|
||||||
file_content = str(StringCompressor(file_content).compress_string())
|
file_content = str(StringCompressor(file_content).compress_string())
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue