🔧 fix(custom_component.py): fix import statement for CUSTOM_COMPONENT_SUPPORTED_TYPES

🔧 fix(custom_component.py): fix argument type in return statement of CustomComponent.get_args()
🔧 fix(custom_components.py): make description attribute optional in CustomComponentFrontendNode class
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-11 18:33:13 -03:00
commit 8f20105b4b
2 changed files with 4 additions and 2 deletions

View file

@ -1,7 +1,7 @@
from typing import Any, Callable, List, Optional, Union from typing import Any, Callable, List, Optional, Union
from uuid import UUID from uuid import UUID
from fastapi import HTTPException from fastapi import HTTPException
from langflow.interface.custom.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
from langflow.services.getters import get_db_service from langflow.services.getters import get_db_service
@ -108,6 +108,7 @@ class CustomComponent(Component, extra=Extra.allow):
), ),
}, },
) )
# TODO arg type should be Data if it is None
return args return args
@property @property

View file

@ -2,6 +2,7 @@ from langflow.template.field.base import TemplateField
from langflow.template.frontend_node.base import FrontendNode from langflow.template.frontend_node.base import FrontendNode
from langflow.template.template.base import Template from langflow.template.template.base import Template
from langflow.interface.custom.constants import DEFAULT_CUSTOM_COMPONENT_CODE from langflow.interface.custom.constants import DEFAULT_CUSTOM_COMPONENT_CODE
from typing import Optional
class CustomComponentFrontendNode(FrontendNode): class CustomComponentFrontendNode(FrontendNode):
@ -24,7 +25,7 @@ class CustomComponentFrontendNode(FrontendNode):
) )
], ],
) )
description: str = "Create any custom component you want!" description: Optional[str] = None
base_classes: list[str] = [] base_classes: list[str] = []
def to_dict(self): def to_dict(self):