From 7243fdc6744bf27e9c06daac738130c1186e572d Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Fri, 8 Mar 2024 17:07:54 -0300
Subject: [PATCH] Remove unused PythonFunction node from customs.py

---
 src/backend/langflow/custom/customs.py            |  1 -
 src/backend/langflow/interface/tools/constants.py | 15 +++++++++++----
 2 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/src/backend/langflow/custom/customs.py b/src/backend/langflow/custom/customs.py
index fb22c56b3..82e87f1e5 100644
--- a/src/backend/langflow/custom/customs.py
+++ b/src/backend/langflow/custom/customs.py
@@ -7,7 +7,6 @@ CUSTOM_NODES = {
     # },
     "tools": {
         "PythonFunctionTool": frontend_node.tools.PythonFunctionToolNode(),
-        "PythonFunction": frontend_node.tools.PythonFunctionNode(),
         "Tool": frontend_node.tools.ToolNode(),
     },
     "agents": {
diff --git a/src/backend/langflow/interface/tools/constants.py b/src/backend/langflow/interface/tools/constants.py
index 0ac37a0a4..2561d9996 100644
--- a/src/backend/langflow/interface/tools/constants.py
+++ b/src/backend/langflow/interface/tools/constants.py
@@ -1,18 +1,25 @@
 from langchain import tools
 from langchain.agents import Tool
-from langchain.agents.load_tools import _BASE_TOOLS, _EXTRA_LLM_TOOLS, _EXTRA_OPTIONAL_TOOLS, _LLM_TOOLS
+from langchain.agents.load_tools import (
+    _BASE_TOOLS,
+    _EXTRA_LLM_TOOLS,
+    _EXTRA_OPTIONAL_TOOLS,
+    _LLM_TOOLS,
+)
 from langchain.tools.json.tool import JsonSpec
+
 from langflow.interface.importing.utils import import_class
-from langflow.interface.tools.custom import PythonFunction, PythonFunctionTool
+from langflow.interface.tools.custom import PythonFunctionTool
 
 FILE_TOOLS = {"JsonSpec": JsonSpec}
 CUSTOM_TOOLS = {
     "Tool": Tool,
     "PythonFunctionTool": PythonFunctionTool,
-    "PythonFunction": PythonFunction,
 }
 
-OTHER_TOOLS = {tool: import_class(f"langchain_community.tools.{tool}") for tool in tools.__all__}
+OTHER_TOOLS = {
+    tool: import_class(f"langchain_community.tools.{tool}") for tool in tools.__all__
+}
 
 ALL_TOOLS_NAMES = {
     **_BASE_TOOLS,