Merge branch 'dev' into 146-improve-conversationchain-node

tests/test_custom_types.py

@@ -1,6 +1,6 @@
 # Test this:
 import pytest
-from langflow.interface.custom.types import PythonFunction
+from langflow.interface.tools.custom import PythonFunction
 from langflow.utils import constants
 
 

tests/test_graph.py

@@ -3,6 +3,7 @@ from typing import Type, Union
 
 import pytest
 from langchain.agents import AgentExecutor
+from langchain.llms.fake import FakeListLLM
 from langflow.graph import Edge, Graph, Node
 from langflow.graph.nodes import (
     AgentNode,
@@ -14,9 +15,8 @@ from langflow.graph.nodes import (
     ToolNode,
     WrapperNode,
 )
-from langflow.utils.payload import build_json, get_root_node
 from langflow.interface.run import get_result_and_thought_using_graph
-from langchain.llms.fake import FakeListLLM
+from langflow.utils.payload import build_json, get_root_node
 
 # Test cases for the graph module
 

tests/test_vectorstore_template.py

@@ -0,0 +1,12 @@
+from fastapi.testclient import TestClient
+from langflow.settings import settings
+
+
+# check that all agents are in settings.agents
+# are in json_response["agents"]
+def test_vectorstores_settings(client: TestClient):
+    response = client.get("/all")
+    assert response.status_code == 200
+    json_response = response.json()
+    vectorstores = json_response["vectorstores"]
+    assert set(vectorstores.keys()) == set(settings.vectorstores)