internal/mcp: add a test for complete mapping of CallToolParams fields
Add a test to verify that the toWireParams helper maps all fields of CallToolParams. Change-Id: I81c28791734d60c793c76809c6b93d27e78c3c36 Reviewed-on: https://go-review.googlesource.com/c/tools/+/677975 LUCI-TryBot-Result: Go LUCI <golang-scoped@luci-project-accounts.iam.gserviceaccount.com> Reviewed-by: Jonathan Amsterdam <jba@google.com>
This commit is contained in:
parent
0ff70be797
commit
facec97821
1 changed files with 39 additions and 4 deletions
|
|
@ -265,17 +265,26 @@ func (cs *ClientSession) CallTool(ctx context.Context, params *CallToolParams[js
|
|||
// CallTool is a helper to call a tool with any argument type. It returns an
|
||||
// error if params.Arguments fails to marshal to JSON.
|
||||
func CallTool[TArgs any](ctx context.Context, cs *ClientSession, params *CallToolParams[TArgs]) (*CallToolResult, error) {
|
||||
wireParams, err := toWireParams(params)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return cs.CallTool(ctx, wireParams)
|
||||
}
|
||||
|
||||
func toWireParams[TArgs any](params *CallToolParams[TArgs]) (*CallToolParams[json.RawMessage], error) {
|
||||
data, err := json.Marshal(params.Arguments)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to marshal arguments: %v", err)
|
||||
}
|
||||
// TODO(rfindley): write a test that guarantees this copying is total.
|
||||
wireParams := &CallToolParams[json.RawMessage]{
|
||||
// The field mapping here must be kept up to date with the CallToolParams.
|
||||
// This is partially enforced by TestToWireParams, which verifies that all
|
||||
// comparable fields are mapped.
|
||||
return &CallToolParams[json.RawMessage]{
|
||||
Meta: params.Meta,
|
||||
Name: params.Name,
|
||||
Arguments: data,
|
||||
}
|
||||
return cs.CallTool(ctx, wireParams)
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (cs *ClientSession) SetLevel(ctx context.Context, params *SetLevelParams) error {
|
||||
|
|
|
|||
Loading…
Reference in a new issue