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

Update CustomComponent class to include a status

Browse source 
attribute
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-21 22:46:00 -03:00
commit b2c8702db1
1 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -3,6 +3,7 @@ from uuid import UUID
import yaml import yaml
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
@ -21,14 +22,20 @@ class CustomComponent(Component):
code_class_base_inheritance: ClassVar[str] = "CustomComponent" code_class_base_inheritance: ClassVar[str] = "CustomComponent"
function_entrypoint_name: ClassVar[str] = "build" function_entrypoint_name: ClassVar[str] = "build"
function: Optional[Callable] = None function: Optional[Callable] = None
return_type_valid_list: List[str] = list(CUSTOM_COMPONENT_SUPPORTED_TYPES.keys())
repr_value: Optional[Any] = "" repr_value: Optional[Any] = ""
user_id: Optional[Union[UUID, str]] = None user_id: Optional[Union[UUID, str]] = None
status: Optional[str] = None
def __init__(self, **data): def __init__(self, **data):
super().__init__(**data) super().__init__(**data)
@property
def return_type_valid_list(self):
return list(CUSTOM_COMPONENT_SUPPORTED_TYPES.keys())
def custom_repr(self): def custom_repr(self):
if self.status:
self.repr_value = self.status
if isinstance(self.repr_value, dict): if isinstance(self.repr_value, dict):
return yaml.dump(self.repr_value) return yaml.dump(self.repr_value)
if isinstance(self.repr_value, str): if isinstance(self.repr_value, str):