fix: Ensure importability of modules not included in __init__.py files (#5965)

* fix: ensure that modules not included in __init__.py files are importable

* test: add test for module import in custom component
This commit is contained in:
Gabriel Luiz Freitas Almeida 2025-01-28 09:18:40 -03:00 • committed by GitHub
commit 0ef54c5ad9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 27 additions and 1 deletions

View file

@ -127,6 +127,26 @@ class MyComponent(CustomComponent):
assert result.value == "test"
def test_create_class_module_import():
code = """
from langflow.custom import CustomComponent
from PIL import ImageDraw
class ExternalClass:
def __init__(self, value):
self.value = value
class MyComponent(CustomComponent):
def build(self):
return ExternalClass("test")
"""
class_name = "MyComponent"
created_class = create_class(code, class_name)
instance = created_class()
result = instance.build()
assert result.value == "test"
def test_create_class_with_multiple_external_classes():
code = """
from langflow.custom import CustomComponent