mirror of
https://github.com/Tencent/WeKnora.git
synced 2026-06-04 13:30:32 +08:00
4715b10642
- Introduced a new built-in agent, "Wiki Researcher," designed for navigating and answering questions based on Wiki knowledge bases, complete with multilingual support. - Added a corresponding system prompt that outlines the agent's role, mission, and workflow for effective knowledge graph traversal. - Updated the agent configuration to include specific tools and parameters tailored for the Wiki Researcher, enhancing its functionality and user interaction. - Removed deprecated wiki tools from the agent service to streamline the toolset and improve performance. These changes significantly enhance the capabilities of the agent system, providing users with a specialized tool for in-depth research and information retrieval from Wiki sources.
75 lines
3.3 KiB
Go
75 lines
3.3 KiB
Go
package tools
|
|
|
|
// maxFunctionNameLength is the maximum length for a tool/function name
|
|
// imposed by the OpenAI API.
|
|
const maxFunctionNameLength = 64
|
|
|
|
// Tool names constants
|
|
const (
|
|
ToolThinking = "thinking"
|
|
ToolTodoWrite = "todo_write"
|
|
ToolGrepChunks = "grep_chunks"
|
|
ToolKnowledgeSearch = "knowledge_search"
|
|
ToolListKnowledgeChunks = "list_knowledge_chunks"
|
|
ToolQueryKnowledgeGraph = "query_knowledge_graph"
|
|
ToolGetDocumentInfo = "get_document_info"
|
|
ToolDatabaseQuery = "database_query"
|
|
ToolDataAnalysis = "data_analysis"
|
|
ToolDataSchema = "data_schema"
|
|
ToolWebSearch = "web_search"
|
|
ToolWebFetch = "web_fetch"
|
|
ToolFinalAnswer = "final_answer"
|
|
// Skills-related tools (only available when skills are enabled)
|
|
ToolExecuteSkillScript = "execute_skill_script"
|
|
ToolReadSkill = "read_skill"
|
|
// Wiki-related tools (only available when wiki KBs are in scope)
|
|
ToolWikiReadPage = "wiki_read_page"
|
|
ToolWikiSearch = "wiki_search"
|
|
)
|
|
|
|
// AvailableTool defines a simple tool metadata used by settings APIs.
|
|
type AvailableTool struct {
|
|
Name string `json:"name"`
|
|
Label string `json:"label"`
|
|
Description string `json:"description"`
|
|
}
|
|
|
|
// AvailableToolDefinitions returns the list of tools exposed to the UI.
|
|
// Keep this in sync with registered tools in this package.
|
|
func AvailableToolDefinitions() []AvailableTool {
|
|
return []AvailableTool{
|
|
{Name: ToolThinking, Label: "思考", Description: "动态和反思性的问题解决思考工具"},
|
|
{Name: ToolTodoWrite, Label: "制定计划", Description: "创建结构化的研究计划"},
|
|
{Name: ToolGrepChunks, Label: "关键词搜索", Description: "快速定位包含特定关键词的文档和分块"},
|
|
{Name: ToolKnowledgeSearch, Label: "语义搜索", Description: "理解问题并查找语义相关内容"},
|
|
{Name: ToolListKnowledgeChunks, Label: "查看文档分块", Description: "获取文档完整分块内容"},
|
|
{Name: ToolQueryKnowledgeGraph, Label: "查询知识图谱", Description: "从知识图谱中查询关系"},
|
|
{Name: ToolGetDocumentInfo, Label: "获取文档信息", Description: "查看文档元数据"},
|
|
{Name: ToolDatabaseQuery, Label: "查询数据库", Description: "查询数据库中的信息"},
|
|
{Name: ToolDataAnalysis, Label: "数据分析", Description: "理解数据文件并进行数据分析"},
|
|
{Name: ToolDataSchema, Label: "查看数据元信息", Description: "获取表格文件的元信息"},
|
|
{Name: ToolReadSkill, Label: "读取技能", Description: "按需读取技能内容以学习专业能力"},
|
|
{Name: ToolExecuteSkillScript, Label: "执行技能脚本", Description: "在沙箱环境中执行技能脚本"},
|
|
{Name: ToolFinalAnswer, Label: "提交最终回答", Description: "提交最终回答给用户"},
|
|
{Name: ToolWikiReadPage, Label: "读取Wiki页面", Description: "读取指定的Wiki页面内容"},
|
|
{Name: ToolWikiSearch, Label: "搜索Wiki", Description: "在Wiki中搜索页面"},
|
|
}
|
|
}
|
|
|
|
// DefaultAllowedTools returns the default allowed tools list.
|
|
func DefaultAllowedTools() []string {
|
|
return []string{
|
|
ToolThinking,
|
|
ToolTodoWrite,
|
|
ToolKnowledgeSearch,
|
|
ToolGrepChunks,
|
|
ToolListKnowledgeChunks,
|
|
ToolQueryKnowledgeGraph,
|
|
ToolGetDocumentInfo,
|
|
ToolDatabaseQuery,
|
|
ToolDataAnalysis,
|
|
ToolDataSchema,
|
|
ToolFinalAnswer,
|
|
}
|
|
}
|