jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: improvement to tools dict name

Browse source
This commit is contained in:
Gabriel Almeida 2023-03-31 14:00:47 -03:00
commit 5b277913cf
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/interface/tools/constants.py
View file

@ -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())
)

4
src/backend/langflow/interface/tools/util.py
View file

@ -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,
}