57 lines
1.5 KiB
Smarty
57 lines
1.5 KiB
Smarty
|
package mcpTool
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"github.com/mark3labs/mcp-go/mcp"
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
RegisterTool(&{{.Name | title}}{})
|
||
|
|
}
|
||
|
|
|
||
|
|
type {{.Name | title}} struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
// {{.Description}}
|
||
|
|
func (t *{{.Name | title}}) Handle(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||
|
|
// TODO: 实现工具逻辑
|
||
|
|
// 参数示例:
|
||
|
|
// {{- range .Params}}
|
||
|
|
// {{.Name}} := request.GetArguments()["{{.Name}}"]
|
||
|
|
// {{- end}}
|
||
|
|
return &mcp.CallToolResult{
|
||
|
|
Content: []mcp.Content{
|
||
|
|
{{- range .Response}}
|
||
|
|
mcp.{{.Type | title}}Content{
|
||
|
|
Type: "{{.Type}}",
|
||
|
|
// TODO: 填充{{.Type}}内容
|
||
|
|
},
|
||
|
|
{{- end}}
|
||
|
|
},
|
||
|
|
}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t *{{.Name | title}}) New() mcp.Tool {
|
||
|
|
return mcp.NewTool("{{.Name}}",
|
||
|
|
mcp.WithDescription("{{.Description}}"),
|
||
|
|
{{- range .Params}}
|
||
|
|
mcp.With{{.Type | title}}("{{.Name}}",
|
||
|
|
{{- if .Required}}mcp.Required(),{{end}}
|
||
|
|
mcp.Description("{{.Description}}"),
|
||
|
|
{{- if .Default}}
|
||
|
|
{{- if eq .Type "string"}}
|
||
|
|
mcp.DefaultString("{{.Default}}"),
|
||
|
|
{{- else if eq .Type "number"}}
|
||
|
|
mcp.DefaultNumber({{.Default}}),
|
||
|
|
{{- else if eq .Type "boolean"}}
|
||
|
|
mcp.DefaultBoolean({{if or (eq .Default "true") (eq .Default "True")}}true{{else}}false{{end}}),
|
||
|
|
{{- else if eq .Type "array"}}
|
||
|
|
// 注意：数组默认值需要在后端代码中预处理为正确的格式
|
||
|
|
// mcp.DefaultArray({{.Default}}),
|
||
|
|
{{- end}}
|
||
|
|
{{- end}}
|
||
|
|
),
|
||
|
|
{{- end}}
|
||
|
|
)
|
||
|
|
}
|