mirror of
https://gitee.com/dromara/mayfly-go
synced 2026-05-20 18:05:33 +08:00
21 lines
487 B
Go
21 lines
487 B
Go
package tools
|
|
|
|
import (
|
|
"mayfly-go/internal/ai/agent"
|
|
"mayfly-go/pkg/logx"
|
|
)
|
|
|
|
func Init() {
|
|
if queryTableTool, err := GetQueryTableInfo(); err != nil {
|
|
logx.Errorf("agent tool - 获取QueryTableInfo工具失败: %v", err)
|
|
} else {
|
|
agent.RegisterTool(agent.ToolTypeDb, queryTableTool)
|
|
}
|
|
|
|
if queryDataTool, err := GetQueryData(); err != nil {
|
|
logx.Errorf("agent tool - 获取QueryData工具失败: %v", err)
|
|
} else {
|
|
agent.RegisterTool(agent.ToolTypeDb, queryDataTool)
|
|
}
|
|
}
|