refactor settings

tests/test_graph.py

@@ -12,7 +12,6 @@ from langflow.graph.vertex.types import (
     FileToolVertex,
     LLMVertex,
     ToolkitVertex,
-    WrapperVertex,
 )
 from langflow.processing.process import get_result_and_thought
 from langflow.utils.payload import get_root_node
@@ -292,11 +291,11 @@ def test_file_tool_node_build(openapi_graph):
     assert not Path(file_path).exists()
 
 
-def test_wrapper_node_build(openapi_graph):
-    wrapper_node = get_node_by_type(openapi_graph, WrapperVertex)
-    assert wrapper_node is not None
-    built_object = wrapper_node.build()
-    assert built_object is not None
+# def test_wrapper_node_build(openapi_graph):
+#     wrapper_node = get_node_by_type(openapi_graph, WrapperVertex)
+#     assert wrapper_node is not None
+#     built_object = wrapper_node.build()
+#     assert built_object is not None
 
 
 def test_get_result_and_thought(basic_graph):

tests/test_llms_template.py

@@ -7,7 +7,7 @@ def test_llms_settings(client: TestClient):
     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.LLMS)
 
 
 # def test_hugging_face_hub(client: TestClient):

tests/test_prompts_template.py

@@ -7,7 +7,7 @@ def test_prompts_settings(client: TestClient):
     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.PROMPTS)
 
 
 def test_prompt_template(client: TestClient):

tests/test_vectorstore_template.py

@@ -9,4 +9,4 @@ 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)
+    assert set(vectorstores.keys()) == set(settings.VECTORSTORES)