From f178e29ef45a36a67712f1ca12560aa275b43279 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 5 Aug 2023 23:17:27 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20fix(test=5Fllms=5Ftemplate.py):?= =?UTF-8?q?=20update=20import=20statement=20to=20use=20get=5Fsettings=5Fma?= =?UTF-8?q?nager=20function=20from=20langflow.services.utils=20module=20?= =?UTF-8?q?=F0=9F=94=A7=20fix(test=5Fprompts=5Ftemplate.py):=20update=20im?= =?UTF-8?q?port=20statement=20to=20use=20get=5Fsettings=5Fmanager=20functi?= =?UTF-8?q?on=20from=20langflow.services.utils=20module=20=F0=9F=94=A7=20f?= =?UTF-8?q?ix(test=5Fvectorstore=5Ftemplate.py):=20update=20import=20state?= =?UTF-8?q?ment=20to=20use=20get=5Fsettings=5Fmanager=20function=20from=20?= =?UTF-8?q?langflow.services.utils=20module=20=F0=9F=94=A7=20fix(test=5Fll?= =?UTF-8?q?ms=5Ftemplate.py):=20update=20assertion=20to=20use=20settings?= =?UTF-8?q?=20from=20settings=5Fmanager=20instead=20of=20settings=20module?= =?UTF-8?q?=20=F0=9F=94=A7=20fix(test=5Fprompts=5Ftemplate.py):=20update?= =?UTF-8?q?=20assertion=20to=20use=20settings=20from=20settings=5Fmanager?= =?UTF-8?q?=20instead=20of=20settings=20module=20=F0=9F=94=A7=20fix(test?= =?UTF-8?q?=5Fvectorstore=5Ftemplate.py):=20update=20assertion=20to=20use?= =?UTF-8?q?=20settings=20from=20settings=5Fmanager=20instead=20of=20settin?= =?UTF-8?q?gs=20module?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_llms_template.py | 5 +++-- tests/test_prompts_template.py | 5 +++-- tests/test_vectorstore_template.py | 5 +++-- 3 files changed, 9 insertions(+), 6 deletions(-) diff --git a/tests/test_llms_template.py b/tests/test_llms_template.py index d8f9e96f3..f1b76e18e 100644 --- a/tests/test_llms_template.py +++ b/tests/test_llms_template.py @@ -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): diff --git a/tests/test_prompts_template.py b/tests/test_prompts_template.py index fa7a683bd..dde313c20 100644 --- a/tests/test_prompts_template.py +++ b/tests/test_prompts_template.py @@ -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): diff --git a/tests/test_vectorstore_template.py b/tests/test_vectorstore_template.py index bac950ee1..6ae4843ac 100644 --- a/tests/test_vectorstore_template.py +++ b/tests/test_vectorstore_template.py @@ -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)