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

🔧 fix(test_llms_template.py): update import statement to use get_settings_manager function from langflow.services.utils module

Browse source 
🔧 fix(test_prompts_template.py): update import statement to use get_settings_manager function from langflow.services.utils module
🔧 fix(test_vectorstore_template.py): update import statement to use get_settings_manager function from langflow.services.utils module
🔧 fix(test_llms_template.py): update assertion to use settings from settings_manager instead of settings module
🔧 fix(test_prompts_template.py): update assertion to use settings from settings_manager instead of settings module
🔧 fix(test_vectorstore_template.py): update assertion to use settings from settings_manager instead of settings module
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-05 23:17:27 -03:00
commit f178e29ef4
3 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/test_llms_template.py
View file

@ -1,13 +1,14 @@
from fastapi.testclient import TestClient
from langflow.settings import settings
from langflow.services.utils import get_settings_manager
def test_llms_settings(client: TestClient):
settings_manager = get_settings_manager()
response = client.get("api/v1/all")
assert response.status_code == 200
json_response = response.json()
llms = json_response["llms"]
assert set(llms.keys()) == set(settings.LLMS)
assert set(llms.keys()) == set(settings_manager.settings.LLMS)
# def test_hugging_face_hub(client: TestClient):

5
tests/test_prompts_template.py
View file

@ -1,13 +1,14 @@
from fastapi.testclient import TestClient
from langflow.settings import settings
from langflow.services.utils import get_settings_manager
def test_prompts_settings(client: TestClient):
settings_manager = get_settings_manager()
response = client.get("api/v1/all")
assert response.status_code == 200
json_response = response.json()
prompts = json_response["prompts"]
assert set(prompts.keys()) == set(settings.PROMPTS)
assert set(prompts.keys()) == set(settings_manager.settings.PROMPTS)
def test_prompt_template(client: TestClient):

5
tests/test_vectorstore_template.py
View file

@ -1,12 +1,13 @@
from fastapi.testclient import TestClient
from langflow.settings import settings
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"]
assert set(vectorstores.keys()) == set(settings.VECTORSTORES)
assert set(vectorstores.keys()) == set(settings_manager.settings.VECTORSTORES)