48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
import sys
|
|
import os
|
|
|
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
|
|
|
|
from server.agent.math import calculate
|
|
from server.agent.translator import translate
|
|
from server.agent.weather import weathercheck
|
|
from server.agent.shell import shell
|
|
from langchain.agents import Tool
|
|
from server.agent.search_knowledge import search_knowledge
|
|
from server.agent.search_internet import search_internet
|
|
|
|
tools = [
|
|
Tool.from_function(
|
|
func=calculate,
|
|
name="计算器工具",
|
|
description="进行简单的数学运算"
|
|
),
|
|
Tool.from_function(
|
|
func=translate,
|
|
name="翻译工具",
|
|
description="翻译各种语言"
|
|
),
|
|
Tool.from_function(
|
|
func=weathercheck,
|
|
name="天气查询工具",
|
|
description="查询天气",
|
|
),
|
|
Tool.from_function(
|
|
func=shell,
|
|
name="shell工具",
|
|
description="使用命令行工具输出",
|
|
),
|
|
Tool.from_function(
|
|
func=search_knowledge,
|
|
name="知识库查询工具",
|
|
description="使用西交利物浦大学大数据专业的本专业数据库来解答问题",
|
|
),
|
|
Tool.from_function(
|
|
func=search_internet,
|
|
name="互联网查询工具",
|
|
description="访问Bing互联网来解答问题",
|
|
),
|
|
|
|
]
|
|
tool_names = [tool.name for tool in tools]
|