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 fastapi.testclient import TestClient
from langflow.settings import settings from langflow.services.utils import get_settings_manager
def test_llms_settings(client: TestClient): def test_llms_settings(client: TestClient):
settings_manager = get_settings_manager()
response = client.get("api/v1/all") response = client.get("api/v1/all")
assert response.status_code == 200 assert response.status_code == 200
json_response = response.json() json_response = response.json()
llms = json_response["llms"] 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): # 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 fastapi.testclient import TestClient
from langflow.settings import settings from langflow.services.utils import get_settings_manager
def test_prompts_settings(client: TestClient): def test_prompts_settings(client: TestClient):
settings_manager = get_settings_manager()
response = client.get("api/v1/all") response = client.get("api/v1/all")
assert response.status_code == 200 assert response.status_code == 200
json_response = response.json() json_response = response.json()
prompts = json_response["prompts"] 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): def test_prompt_template(client: TestClient):

5
tests/test_vectorstore_template.py
View file

@ -1,12 +1,13 @@
from fastapi.testclient import TestClient 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 # check that all agents are in settings.agents
# are in json_response["agents"] # are in json_response["agents"]
def test_vectorstores_settings(client: TestClient): def test_vectorstores_settings(client: TestClient):
settings_manager = get_settings_manager()
response = client.get("api/v1/all") response = client.get("api/v1/all")
assert response.status_code == 200 assert response.status_code == 200
json_response = response.json() json_response = response.json()
vectorstores = json_response["vectorstores"] vectorstores = json_response["vectorstores"]
assert set(vectorstores.keys()) == set(settings.VECTORSTORES) assert set(vectorstores.keys()) == set(settings_manager.settings.VECTORSTORES)