formatting

src/backend/langflow/__main__.py

@@ -2,12 +2,12 @@ import logging
 import multiprocessing
 import platform
 from pathlib import Path
-from langflow.settings import settings
 
 import typer
 from fastapi.staticfiles import StaticFiles
 
 from langflow.main import create_app
+from langflow.settings import settings
 
 logger = logging.getLogger(__name__)
 

src/backend/langflow/api/endpoints.py

@@ -1,13 +1,12 @@
 from typing import Any, Dict
 
 from fastapi import APIRouter, HTTPException
-from langflow.api.base import Code, ValidationResponse
 
+from langflow.api.base import Code, ValidationResponse
 from langflow.interface.run import process_graph
 from langflow.interface.types import build_langchain_types_dict
 from langflow.utils.validate import validate_code
 
-
 # build router
 router = APIRouter()
 

src/backend/langflow/custom/customs.py

@@ -1,6 +1,5 @@
 from langflow.template import nodes
 
-
 CUSTOM_NODES = {
     "prompts": {**nodes.ZeroShotPromptNode().to_dict()},
     "tools": {**nodes.PythonFunctionNode().to_dict(), **nodes.ToolNode().to_dict()},

src/backend/langflow/graph/__init__.py

@@ -1,4 +1,4 @@
+from langflow.graph.base import Edge, Node
 from langflow.graph.graph import Graph
-from langflow.graph.base import Node, Edge
 
 __all__ = ["Graph", "Node", "Edge"]

src/backend/langflow/interface/base.py

@@ -1,8 +1,9 @@
-from typing import Any, Dict, List, Optional
-from pydantic import BaseModel
 from abc import ABC, abstractmethod
-from langflow.template.template import Template, Field, FrontendNode
+from typing import Any, Dict, List, Optional
 
+from pydantic import BaseModel
+
+from langflow.template.base import Field, FrontendNode, Template
 
 # Assuming necessary imports for Field, Template, and FrontendNode classes
 

src/backend/langflow/interface/chains/base.py

@@ -1,9 +1,11 @@
 from typing import Dict, List
-from langflow.interface.base import LangChainTypeCreator
-from langflow.utils.util import build_template_from_function
-from langflow.settings import settings
+
 from langchain.chains import loading as chains_loading
 
+from langflow.interface.base import LangChainTypeCreator
+from langflow.settings import settings
+from langflow.utils.util import build_template_from_function
+
 # Assuming necessary imports for Field, Template, and FrontendNode classes
 
 

src/backend/langflow/interface/custom/types.py

@@ -1,8 +1,8 @@
-from langflow.utils import validate
+from typing import Callable, Optional
+
 from pydantic import BaseModel, validator
 
-
-from typing import Callable, Optional
+from langflow.utils import validate
 
 
 class Function(BaseModel):

src/backend/langflow/interface/custom_lists.py

@@ -2,10 +2,10 @@
 from typing import Any
 
 from langchain import llms, requests
-from langchain.llms.openai import OpenAIChat
 from langchain.agents import agent_toolkits
-from langflow.interface.importing.utils import import_class
+from langchain.llms.openai import OpenAIChat
 
+from langflow.interface.importing.utils import import_class
 
 llm_type_to_cls_dict = llms.type_to_cls_dict
 llm_type_to_cls_dict["openai-chat"] = OpenAIChat

src/backend/langflow/interface/llms/base.py

@@ -1,8 +1,9 @@
+from typing import Dict, List
+
+from langflow.interface.base import LangChainTypeCreator
 from langflow.interface.custom_lists import llm_type_to_cls_dict
 from langflow.settings import settings
-from langflow.interface.base import LangChainTypeCreator
 from langflow.utils.util import build_template_from_class
-from typing import Dict, List
 
 
 class LLMCreator(LangChainTypeCreator):

src/backend/langflow/interface/memories/base.py

@@ -1,8 +1,9 @@
+from typing import Dict, List
+
+from langflow.interface.base import LangChainTypeCreator
 from langflow.interface.custom_lists import memory_type_to_cls_dict
 from langflow.settings import settings
-from langflow.interface.base import LangChainTypeCreator
 from langflow.utils.util import build_template_from_class
-from typing import Dict, List
 
 
 class MemoryCreator(LangChainTypeCreator):

src/backend/langflow/interface/prompts/base.py

@@ -1,10 +1,12 @@
-from langchain.prompts import loading
-from langflow.interface.base import LangChainTypeCreator
-from langflow.utils.util import build_template_from_function
-from langflow.settings import settings
-from langflow.custom.customs import get_custom_nodes
 from typing import Dict, List
 
+from langchain.prompts import loading
+
+from langflow.custom.customs import get_custom_nodes
+from langflow.interface.base import LangChainTypeCreator
+from langflow.settings import settings
+from langflow.utils.util import build_template_from_function
+
 
 class PromptCreator(LangChainTypeCreator):
     type_name: str = "prompts"

src/backend/langflow/interface/run.py

@@ -3,9 +3,9 @@ import io
 import re
 from typing import Any, Dict
 
+from langflow.graph.graph import Graph
 from langflow.interface import loading
 from langflow.utils import payload
-from langflow.graph.graph import Graph
 
 
 def process_graph(data_graph: Dict[str, Any]):
@@ -51,6 +51,7 @@ def get_result_and_thought_using_graph(loaded_langchain, message: str):
     except Exception as e:
         result = f"Error: {str(e)}"
         thought = ""
+        raise e
     return result, thought
 
 

src/backend/langflow/interface/tools/base.py

@@ -1,27 +1,27 @@
-from langflow.custom import customs
-from langflow.interface.tools.constants import (
-    ALL_TOOLS_NAMES,
-    CUSTOM_TOOLS,
-    FILE_TOOLS,
-)
-from langflow.template.base import Field
-from langflow.template.base import Template
-from langflow.utils import util
-from langflow.settings import settings
-from langflow.interface.base import LangChainTypeCreator
 from typing import Dict, List
+
 from langchain.agents.load_tools import (
     _BASE_TOOLS,
     _EXTRA_LLM_TOOLS,
     _EXTRA_OPTIONAL_TOOLS,
     _LLM_TOOLS,
 )
+
+from langflow.custom import customs
+from langflow.interface.base import LangChainTypeCreator
+from langflow.interface.tools.constants import (
+    ALL_TOOLS_NAMES,
+    CUSTOM_TOOLS,
+    FILE_TOOLS,
+)
 from langflow.interface.tools.util import (
     get_tool_by_name,
-    get_tools_dict,
     get_tool_params,
+    get_tools_dict,
 )
-
+from langflow.settings import settings
+from langflow.template.base import Field, Template
+from langflow.utils import util
 
 TOOL_INPUTS = {
     "str": Field(

src/backend/langflow/interface/tools/constants.py

@@ -1,8 +1,8 @@
-from langchain.agents.load_tools import get_all_tool_names
 from langchain.agents import Tool
-from langflow.interface.custom.types import PythonFunction
+from langchain.agents.load_tools import get_all_tool_names
 from langchain.tools.json.tool import JsonSpec
 
+from langflow.interface.custom.types import PythonFunction
 
 FILE_TOOLS = {"JsonSpec": JsonSpec}
 CUSTOM_TOOLS = {"Tool": Tool, "PythonFunction": PythonFunction}

src/backend/langflow/interface/tools/util.py

@@ -1,6 +1,7 @@
 import ast
 import inspect
 from typing import Dict, Union
+
 from langchain.agents.load_tools import (
     _BASE_TOOLS,
     _EXTRA_LLM_TOOLS,
@@ -8,6 +9,7 @@ from langchain.agents.load_tools import (
     _LLM_TOOLS,
 )
 from langchain.agents.tools import Tool
+
 from langflow.interface.tools.constants import CUSTOM_TOOLS, FILE_TOOLS
 
 

src/backend/langflow/interface/wrappers/base.py

@@ -1,10 +1,12 @@
-from langchain import requests
-from langflow.interface.base import LangChainTypeCreator
-from langflow.utils.util import build_template_from_class
-from langflow.settings import settings
-from langflow.custom.customs import get_custom_nodes
 from typing import Dict, List
 
+from langchain import requests
+
+from langflow.custom.customs import get_custom_nodes
+from langflow.interface.base import LangChainTypeCreator
+from langflow.settings import settings
+from langflow.utils.util import build_template_from_class
+
 
 class WrapperCreator(LangChainTypeCreator):
     type_name: str = "wrappers"

src/backend/langflow/template/base.py

@@ -1,8 +1,7 @@
-from pydantic import BaseModel
-
-
-from typing import Any, Union
 from abc import ABC
+from typing import Any, Union
+
+from pydantic import BaseModel
 
 
 class FieldCreator(BaseModel, ABC):

src/backend/langflow/template/nodes.py

@@ -1,6 +1,6 @@
-from langflow.template.base import Field, Template
-from langflow.template.base import FrontendNode
 from langchain.agents.mrkl import prompt
+
+from langflow.template.base import Field, FrontendNode, Template
 from langflow.utils.constants import DEFAULT_PYTHON_FUNCTION
 
 

src/backend/langflow/utils/util.py

@@ -3,7 +3,6 @@ import inspect
 import re
 from typing import Dict, Optional
 
-
 from langflow.utils import constants
 
 

tests/test_custom_types.py

@@ -1,7 +1,7 @@
 # Test this:
+import pytest
 from langflow.interface.custom.types import PythonFunction
 from langflow.utils import constants
-import pytest
 
 
 def test_python_function():

tests/test_graph.py

@@ -1,8 +1,9 @@
 import json
-from langflow.graph import Edge, Node, Graph
+
 import pytest
-from langflow.utils.payload import build_json, get_root_node
 from langchain.agents import AgentExecutor
+from langflow.graph import Edge, Graph, Node
+from langflow.utils.payload import build_json, get_root_node
 
 # Test cases for the graph module
 

tests/test_loading.py

@@ -1,10 +1,10 @@
 import json
-from langflow.graph import Graph
-import pytest
 
-from langflow import load_flow_from_json
-from langflow.utils.payload import get_root_node
+import pytest
 from langchain.agents import AgentExecutor
+from langflow import load_flow_from_json
+from langflow.graph import Graph
+from langflow.utils.payload import get_root_node
 
 
 def test_load_flow_from_json():

tests/test_validate_code.py

@@ -1,12 +1,13 @@
+from unittest import mock
+
+import pytest
 from langflow.utils.validate import (
     create_function,
+    execute_function,
     extract_function_name,
     validate_code,
-    execute_function,
 )
-import pytest
 from requests.exceptions import MissingSchema
-from unittest import mock
 
 
 def test_validate_code():