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

Fix URLComponent build method and update test_url_component

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-08 23:41:35 -03:00
commit 8139b743c2
2 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/langflow/components/data/URL.py
View file

@ -15,7 +15,7 @@ class URLComponent(CustomComponent):
"urls": {"display_name": "URL"}, "urls": {"display_name": "URL"},
} }
async def build( def build(
self, self,
urls: list[str], urls: list[str],
) -> list[Record]: ) -> list[Record]:

3
tests/test_data_components.py
View file

@ -172,4 +172,5 @@ def test_directory_without_mocks():
def test_url_component(): def test_url_component():
url_component = data.URLComponent() url_component = data.URLComponent()
# the url component can be used to load the contents of a website # the url component can be used to load the contents of a website
record = url_component.build("https://example.com") records = url_component.build(["https://langflow.org"])
assert all(record.data for record in records)