Langchain-Chatchat/server/agent/tools_select.py

90 lines
3.2 KiB
Python
Raw Normal View History

from langchain.tools import Tool
from server.agent.tools import *
# tools = [
# Tool.from_function(
# func=calculate,
# name="计算器工具",
# description="进行简单的数学运算, 只是简单的, 使用Wolfram数学工具进行更复杂的运算",
# ),
# Tool.from_function(
# func=translate,
# name="翻译工具",
# description="如果你无法访问互联网,并且需要翻译各种语言,应该使用这个工具"
# ),
# Tool.from_function(
# func=weathercheck,
# name="天气查询工具",
# description="无需访问互联网使用这个工具查询中国各地未来24小时的天气",
# ),
# Tool.from_function(
# func=shell,
# name="shell工具",
# description="使用命令行工具输出",
# ),
# Tool.from_function(
# func=knowledge_search_more,
# name="知识库查询工具",
# description="优先访问知识库来获取答案",
# ),
# Tool.from_function(
# func=search_internet,
# name="互联网查询工具",
# description="如果你无法访问互联网这个工具可以帮助你访问Bing互联网来解答问题",
# ),
# Tool.from_function(
# func=wolfram,
# name="Wolfram数学工具",
# description="高级的数学运算工具,能够完成非常复杂的数学问题"
# ),
# Tool.from_function(
# func=youtube_search,
# name="Youtube搜索工具",
# description="使用这个工具在Youtube上搜索视频"
# ]
## 请注意如果你是为了使用AgentLM在这里你应该使用英文版本下面的内容是英文版本。
tools = [
Tool.from_function(
func=calculate,
name="Simple Calculator Tool",
description="Perform simple mathematical operations, Just simple, Use Wolfram Math Tool for more complex operations"
),
Tool.from_function(
func=translate,
name="Translation Tool",
description="Use this tool if you can't access the internet and need to translate various languages"
),
Tool.from_function(
func=weathercheck,
name="Weather Checking Tool",
description="Check the weather for various places in China for the next 24 hours without needing internet access"
),
Tool.from_function(
func=shell,
name="Shell Tool",
description="Use command line tool output"
),
Tool.from_function(
func=knowledge_search_more,
name="Knowledge Base Tool",
description="Prioritize accessing the knowledge base to get answers"
),
Tool.from_function(
func=search_internet,
name="Internet Tool",
description="If you can't access the internet, this tool can help you access Bing to answer questions"
),
Tool.from_function(
func=wolfram,
name="Wolfram Math Tool",
description="Use this tool to perform more complex mathematical operations"
),
Tool.from_function(
func=youtube_search,
name="Youtube Search Tool",
description="Use this tool to search for videos on Youtube"
)
]
tool_names = [tool.name for tool in tools]