From 5b277913cf0f1affc62c7ef06b2b0d6e46fea92a Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Fri, 31 Mar 2023 14:00:47 -0300 Subject: [PATCH] fix: improvement to tools dict name --- src/backend/langflow/interface/tools/constants.py | 4 ++-- src/backend/langflow/interface/tools/util.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/interface/tools/constants.py b/src/backend/langflow/interface/tools/constants.py index b5db816a5..744f9cfd8 100644 --- a/src/backend/langflow/interface/tools/constants.py +++ b/src/backend/langflow/interface/tools/constants.py @@ -4,8 +4,8 @@ from langflow.interface.custom.types import PythonFunction from langchain.tools.json.tool import JsonSpec -OTHER_TOOLS = {"JsonSpec": JsonSpec} +FILE_TOOLS = {"JsonSpec": JsonSpec} CUSTOM_TOOLS = {"Tool": Tool, "PythonFunction": PythonFunction} ALL_TOOLS_NAMES = set( - get_all_tool_names() + list(CUSTOM_TOOLS.keys()) + list(OTHER_TOOLS.keys()) + get_all_tool_names() + list(CUSTOM_TOOLS.keys()) + list(FILE_TOOLS.keys()) ) diff --git a/src/backend/langflow/interface/tools/util.py b/src/backend/langflow/interface/tools/util.py index 2ec273e67..92ea40b88 100644 --- a/src/backend/langflow/interface/tools/util.py +++ b/src/backend/langflow/interface/tools/util.py @@ -8,7 +8,7 @@ from langchain.agents.load_tools import ( _LLM_TOOLS, ) from langchain.agents.tools import Tool -from langflow.interface.tools.constants import CUSTOM_TOOLS, OTHER_TOOLS +from langflow.interface.tools.constants import CUSTOM_TOOLS, FILE_TOOLS def get_tools_dict(): @@ -20,7 +20,7 @@ def get_tools_dict(): **{k: v[0] for k, v in _EXTRA_LLM_TOOLS.items()}, **{k: v[0] for k, v in _EXTRA_OPTIONAL_TOOLS.items()}, **CUSTOM_TOOLS, - **OTHER_TOOLS, + **FILE_TOOLS, }