diff --git a/go.mod b/go.mod index cbb03aae8..99107c3b8 100644 --- a/go.mod +++ b/go.mod @@ -15,8 +15,8 @@ require ( github.com/jaronnie/genius v0.5.2 github.com/jhump/protoreflect v1.17.0 github.com/jzero-io/desc v1.1.0 + github.com/mark3labs/mcp-go v0.29.0 github.com/jzero-io/jzero-contrib v0.24.0 - github.com/mark3labs/mcp-go v0.28.0 github.com/mitchellh/go-homedir v1.1.0 github.com/moby/patternmatcher v0.6.0 github.com/pkg/errors v0.9.1 diff --git a/go.sum b/go.sum index 30225aaea..71009abe1 100644 --- a/go.sum +++ b/go.sum @@ -158,8 +158,8 @@ github.com/lib/pq v1.10.9 h1:YXG7RB+JIjhP29X+OtkiDnYaXQwpS4JEWq7dtCCRUEw= github.com/lib/pq v1.10.9/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o= github.com/logrusorgru/aurora v2.0.3+incompatible h1:tOpm7WcpBTn4fjmVfgpQq0EfczGlG91VSDkswnjF5A8= github.com/logrusorgru/aurora v2.0.3+incompatible/go.mod h1:7rIyQOR62GCctdiQpZ/zOJlFyk6y+94wXzv6RNZgaR4= -github.com/mark3labs/mcp-go v0.28.0 h1:7yl4y5D1KYU2f/9Uxp7xfLIggfunHoESCRbrjcytcLM= -github.com/mark3labs/mcp-go v0.28.0/go.mod h1:rXqOudj/djTORU/ThxYx8fqEVj/5pvTuuebQ2RC7uk4= +github.com/mark3labs/mcp-go v0.29.0 h1:sH1NBcumKskhxqYzhXfGc201D7P76TVXiT0fGVhabeI= +github.com/mark3labs/mcp-go v0.29.0/go.mod h1:rXqOudj/djTORU/ThxYx8fqEVj/5pvTuuebQ2RC7uk4= github.com/mattn/go-colorable v0.1.14 h1:9A9LHSqF/7dyVVX6g0U9cwm9pG3kP9gSzcuIPHPsaIE= github.com/mattn/go-colorable v0.1.14/go.mod h1:6LmQG8QLFO4G5z1gPvYEzlUgJ2wF+stgPZH1UqBm1s8= github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY= diff --git a/pkg/cobra-mcp/mcp.go b/pkg/cobra-mcp/mcp.go index b9ed2dff4..6fb0ff8dc 100644 --- a/pkg/cobra-mcp/mcp.go +++ b/pkg/cobra-mcp/mcp.go @@ -90,7 +90,7 @@ func (s *MCPServer) handleToolCall(cmd *cobra.Command) func(ctx context.Context, var fullArgs []string fullArgs = append(fullArgs, getFullCommandPath(cmd)...) - for key, val := range request.Params.Arguments { + for key, val := range request.GetArguments() { if key == "args" { continue } @@ -98,7 +98,7 @@ func (s *MCPServer) handleToolCall(cmd *cobra.Command) func(ctx context.Context, fullArgs = append(fullArgs, fmt.Sprintf("%v", val)) } - if args, ok := request.Params.Arguments["args"].([]any); ok { + if args, ok := request.GetArguments()["args"].([]any); ok { for _, arg := range args { fullArgs = append(fullArgs, fmt.Sprintf("%v", arg)) }