mirror of
https://github.com/googleapis/genai-toolbox.git
synced 2026-02-06 05:04:58 -05:00
4aabb4aaca
To facilitate the transition of moving invocation implementation to Source, we will have to move parameter to `internal/util`. This approach is crucial because certain parameters may not be fully resolvable pre-implementation. Since both `internal/sources` and `internal/tools` will need access to `parameters`, it will be more relevant to move parameters implementation to utils.
95 lines
3.0 KiB
Go
95 lines
3.0 KiB
Go
// Copyright 2025 Google LLC
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package tools_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
"github.com/googleapis/genai-toolbox/internal/tools"
|
|
"github.com/googleapis/genai-toolbox/internal/util/parameters"
|
|
)
|
|
|
|
func TestGetMcpManifestMetadata(t *testing.T) {
|
|
authServices := []parameters.ParamAuthService{
|
|
{
|
|
Name: "my-google-auth-service",
|
|
Field: "auth_field",
|
|
},
|
|
{
|
|
Name: "other-auth-service",
|
|
Field: "other_auth_field",
|
|
}}
|
|
tcs := []struct {
|
|
desc string
|
|
name string
|
|
description string
|
|
authInvoke []string
|
|
params parameters.Parameters
|
|
wantMetadata map[string]any
|
|
}{
|
|
{
|
|
desc: "basic manifest without metadata",
|
|
name: "basic",
|
|
description: "foo bar",
|
|
authInvoke: []string{},
|
|
params: parameters.Parameters{parameters.NewStringParameter("string-param", "string parameter")},
|
|
wantMetadata: nil,
|
|
},
|
|
{
|
|
desc: "with auth invoke metadata",
|
|
name: "basic",
|
|
description: "foo bar",
|
|
authInvoke: []string{"auth1", "auth2"},
|
|
params: parameters.Parameters{parameters.NewStringParameter("string-param", "string parameter")},
|
|
wantMetadata: map[string]any{"toolbox/authInvoke": []string{"auth1", "auth2"}},
|
|
},
|
|
{
|
|
desc: "with auth param metadata",
|
|
name: "basic",
|
|
description: "foo bar",
|
|
authInvoke: []string{},
|
|
params: parameters.Parameters{parameters.NewStringParameterWithAuth("string-param", "string parameter", authServices)},
|
|
wantMetadata: map[string]any{
|
|
"toolbox/authParam": map[string][]string{
|
|
"string-param": []string{"my-google-auth-service", "other-auth-service"},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
desc: "with auth invoke and auth param metadata",
|
|
name: "basic",
|
|
description: "foo bar",
|
|
authInvoke: []string{"auth1", "auth2"},
|
|
params: parameters.Parameters{parameters.NewStringParameterWithAuth("string-param", "string parameter", authServices)},
|
|
wantMetadata: map[string]any{
|
|
"toolbox/authInvoke": []string{"auth1", "auth2"},
|
|
"toolbox/authParam": map[string][]string{
|
|
"string-param": []string{"my-google-auth-service", "other-auth-service"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
for _, tc := range tcs {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
got := tools.GetMcpManifest(tc.name, tc.description, tc.authInvoke, tc.params)
|
|
gotM := got.Metadata
|
|
if diff := cmp.Diff(tc.wantMetadata, gotM); diff != "" {
|
|
t.Fatalf("unexpected metadata (-want +got):\n%s", diff)
|
|
}
|
|
})
|
|
}
|
|
}
|