d0aa3261f1
🐛 fix(schemas.py): change json.dumps to orjson_dumps for improved performance and compatibility with orjson library 🐛 fix(utils.py): change json.loads to orjson.loads for improved performance and compatibility with orjson library 🐛 fix(loading.py): change json.loads to orjson.loads for improved performance and compatibility with orjson library 🐛 fix(utils.py): change json.loads to orjson.loads for improved performance and compatibility with orjson library 🐛 fix(vector_store.py): change json.loads to orjson.loads for improved performance and compatibility with orjson library 🐛 fix(types.py): change json.loads to orjson.loads for improved performance and compatibility with orjson library 🐛 fix(process.py): change json.loads to orjson.loads for improved performance and compatibility with orjson library ✨ feat(server.ts): change port variable case from lowercase port to uppercase PORT to improve semantics ✨ feat(server.ts): add support for process.env.PORT environment variable to be able to run app on a configurable port 🔧 fix(base.py): import orjson instead of json to improve performance and compatibility 🔧 fix(frontend_node/llms.py): use orjson_dumps instead of json.dumps to improve performance and compatibility 🔧 fix(frontend_node/utilities.py): use orjson_dumps instead of json.dumps to improve performance and compatibility 🔧 fix(test_cache.py): import orjson and use orjson_dumps instead of json.dumps to improve performance and compatibility 🔧 fix(test_database.py): import correct json encoder and decoder functions to fix import errors 🔧 fix(test_database.py): replace json.dumps and json.loads with orjson_dumps and orjson.loads for better performance and compatibility 🔧 fix(test_loading.py): remove unused import statement
73 lines
2.1 KiB
Python
73 lines
2.1 KiB
Python
import json
|
|
from langflow.database.models.base import orjson_dumps
|
|
import orjson
|
|
from langflow.graph import Graph
|
|
|
|
import pytest
|
|
from langflow.interface.run import (
|
|
build_langchain_object_with_caching,
|
|
)
|
|
|
|
|
|
def get_graph(_type="basic"):
|
|
"""Get a graph from a json file"""
|
|
if _type == "basic":
|
|
path = pytest.BASIC_EXAMPLE_PATH
|
|
elif _type == "complex":
|
|
path = pytest.COMPLEX_EXAMPLE_PATH
|
|
elif _type == "openapi":
|
|
path = pytest.OPENAPI_EXAMPLE_PATH
|
|
|
|
with open(path, "r") as f:
|
|
flow_graph = json.load(f)
|
|
return flow_graph["data"]
|
|
|
|
|
|
@pytest.fixture
|
|
def basic_data_graph():
|
|
return get_graph()
|
|
|
|
|
|
@pytest.fixture
|
|
def complex_data_graph():
|
|
return get_graph("complex")
|
|
|
|
|
|
@pytest.fixture
|
|
def openapi_data_graph():
|
|
return get_graph("openapi")
|
|
|
|
|
|
def langchain_objects_are_equal(obj1, obj2):
|
|
return str(obj1) == str(obj2)
|
|
|
|
|
|
# Test build_langchain_object_with_caching
|
|
def test_build_langchain_object_with_caching(basic_data_graph):
|
|
build_langchain_object_with_caching.clear_cache()
|
|
graph = build_langchain_object_with_caching(basic_data_graph)
|
|
assert graph is not None
|
|
|
|
|
|
# Test build_graph
|
|
def test_build_graph(basic_data_graph):
|
|
graph = Graph.from_payload(basic_data_graph)
|
|
assert graph is not None
|
|
assert len(graph.nodes) == len(basic_data_graph["nodes"])
|
|
assert len(graph.edges) == len(basic_data_graph["edges"])
|
|
|
|
|
|
# Test cache size limit
|
|
def test_cache_size_limit(basic_data_graph):
|
|
build_langchain_object_with_caching.clear_cache()
|
|
for i in range(11):
|
|
modified_data_graph = basic_data_graph.copy()
|
|
nodes = modified_data_graph["nodes"]
|
|
node_id = nodes[0]["id"]
|
|
# Now we replace all instances ode node_id with a new id in the json
|
|
json_string = orjson_dumps(modified_data_graph)
|
|
modified_json_string = json_string.replace(node_id, f"{node_id}_{i}")
|
|
modified_data_graph_new_id = orjson.loads(modified_json_string)
|
|
build_langchain_object_with_caching(modified_data_graph_new_id)
|
|
|
|
assert len(build_langchain_object_with_caching.cache) == 10
|