diff --git a/src/backend/langflow/cache/utils.py b/src/backend/langflow/cache/utils.py index 7dcc6b935..763a2af0c 100644 --- a/src/backend/langflow/cache/utils.py +++ b/src/backend/langflow/cache/utils.py @@ -1,13 +1,13 @@ import contextlib +import functools +import hashlib import json import os import tempfile +from collections import OrderedDict from pathlib import Path import dill # type: ignore -import functools -from collections import OrderedDict -import hashlib def memoize_dict(maxsize=128): diff --git a/tests/test_graph.py b/tests/test_graph.py index 65a9d0ea6..c007bdd78 100644 --- a/tests/test_graph.py +++ b/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