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

Merge branch 'release' into dev

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-10 10:10:28 -03:00
commit 3d8d44538f
18 changed files with 468 additions and 283 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/test_custom_component.py
View file

@ -241,7 +241,7 @@ def test_custom_component_get_function_entrypoint_return_type():
code=code_default, function_entrypoint_name="build"
)
return_type = custom_component.get_function_entrypoint_return_type
assert return_type == "Document"
assert return_type == ["Document"]
def test_custom_component_get_main_class_name():
@ -436,7 +436,7 @@ class MyClass(CustomComponent):
custom_component = CustomComponent(code=my_code, function_entrypoint_name="build")
return_type = custom_component.get_function_entrypoint_return_type
assert return_type is None
assert return_type == []
def test_custom_component_get_main_class_name_no_main_class():
@ -469,7 +469,7 @@ def test_build_config_no_code():
component = CustomComponent(code=None)
assert component.get_function_entrypoint_args == ""
assert component.get_function_entrypoint_return_type == ""
assert component.get_function_entrypoint_return_type == []
@pytest.fixture

3
tests/test_vectorstore_template.py
View file

@ -10,4 +10,5 @@ def test_vectorstores_settings(client: TestClient):
assert response.status_code == 200
json_response = response.json()
vectorstores = json_response["vectorstores"]
assert set(vectorstores.keys()) == set(settings_manager.settings.VECTORSTORES)
settings_vecs = set(settings_manager.settings.VECTORSTORES)
assert all(vs in vectorstores for vs in settings_vecs)