langflow/tests/test_vectorstore_template.py
Gabriel Luiz Freitas Almeida 3d8d44538f Merge branch 'release' into dev
2023-08-10 10:10:28 -03:00

14 lines
566 B
Python

from fastapi.testclient import TestClient
from langflow.services.utils import get_settings_manager
# check that all agents are in settings.agents
# are in json_response["agents"]
def test_vectorstores_settings(client: TestClient):
settings_manager = get_settings_manager()
response = client.get("api/v1/all")
assert response.status_code == 200
json_response = response.json()
vectorstores = json_response["vectorstores"]
settings_vecs = set(settings_manager.settings.VECTORSTORES)
assert all(vs in vectorstores for vs in settings_vecs)