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

🐛 fix(custom_component.py): handle case when return_type is None to prevent errors

Browse source 
🐛 fix(test_custom_component.py): update assertion to expect return_type as a list instead of a string
🐛 fix(test_vectorstore_template.py): update assertion to check if all vectorstores in settings are present in the response
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-09 17:50:03 -03:00
commit 8c41415a32
3 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/test_vectorstore_template.py
View file

@ -9,4 +9,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.VECTORSTORES)
settings_vecs = set(settings.VECTORSTORES)
assert all(vs in vectorstores for vs in settings_vecs)