mirror of
https://github.com/googleapis/genai-toolbox.git
synced 2026-02-19 03:14:29 -05:00
f59a06bd10
## Description Tool `invoke()` and `RequiresClientAuthorization()` takes a new input argument -- Resource Manager. Resource manager will be used to retrieve Source in the next step. In order to achieve the goal, this PR implements the follows: * move resource manager from the server package to a new package to prevent import cycles (between server and mcp) * added a new interface in `tools.go` to prevent import cycle (between resources and tools package) * add new input argument in all tools ## PR Checklist > Thank you for opening a Pull Request! Before submitting your PR, there are a > few things you can do to make sure it goes smoothly: - [x] Make sure you reviewed [CONTRIBUTING.md](https://github.com/googleapis/genai-toolbox/blob/main/CONTRIBUTING.md) - [x] Make sure to open an issue as a [bug/issue](https://github.com/googleapis/genai-toolbox/issues/new/choose) before writing your code! That way we can discuss the change, evaluate designs, and agree on the general idea - [x] Ensure the tests and linter pass - [x] Code coverage does not decrease (if any source code was changed) - [x] Appropriate docs were updated (if necessary) - [x] Make sure to add `!` if this involve a breaking change
104 lines
3.5 KiB
Go
104 lines
3.5 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 resources_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
"github.com/googleapis/genai-toolbox/internal/auth"
|
|
"github.com/googleapis/genai-toolbox/internal/prompts"
|
|
"github.com/googleapis/genai-toolbox/internal/server/resources"
|
|
"github.com/googleapis/genai-toolbox/internal/sources"
|
|
"github.com/googleapis/genai-toolbox/internal/sources/alloydbpg"
|
|
"github.com/googleapis/genai-toolbox/internal/tools"
|
|
)
|
|
|
|
func TestUpdateServer(t *testing.T) {
|
|
newSources := map[string]sources.Source{
|
|
"example-source": &alloydbpg.Source{
|
|
Config: alloydbpg.Config{
|
|
Name: "example-alloydb-source",
|
|
Kind: "alloydb-postgres",
|
|
},
|
|
},
|
|
}
|
|
newAuth := map[string]auth.AuthService{"example-auth": nil}
|
|
newTools := map[string]tools.Tool{"example-tool": nil}
|
|
newToolsets := map[string]tools.Toolset{
|
|
"example-toolset": {
|
|
ToolsetConfig: tools.ToolsetConfig{
|
|
Name: "example-toolset",
|
|
},
|
|
Tools: []*tools.Tool{},
|
|
},
|
|
}
|
|
newPrompts := map[string]prompts.Prompt{"example-prompt": nil}
|
|
newPromptsets := map[string]prompts.Promptset{
|
|
"example-promptset": {
|
|
PromptsetConfig: prompts.PromptsetConfig{
|
|
Name: "example-promptset",
|
|
},
|
|
Prompts: []*prompts.Prompt{},
|
|
},
|
|
}
|
|
resMgr := resources.NewResourceManager(newSources, newAuth, newTools, newToolsets, newPrompts, newPromptsets)
|
|
|
|
gotSource, _ := resMgr.GetSource("example-source")
|
|
if diff := cmp.Diff(gotSource, newSources["example-source"]); diff != "" {
|
|
t.Errorf("error updating server, sources (-want +got):\n%s", diff)
|
|
}
|
|
|
|
gotAuthService, _ := resMgr.GetAuthService("example-auth")
|
|
if diff := cmp.Diff(gotAuthService, newAuth["example-auth"]); diff != "" {
|
|
t.Errorf("error updating server, authServices (-want +got):\n%s", diff)
|
|
}
|
|
|
|
gotTool, _ := resMgr.GetTool("example-tool")
|
|
if diff := cmp.Diff(gotTool, newTools["example-tool"]); diff != "" {
|
|
t.Errorf("error updating server, tools (-want +got):\n%s", diff)
|
|
}
|
|
|
|
gotToolset, _ := resMgr.GetToolset("example-toolset")
|
|
if diff := cmp.Diff(gotToolset, newToolsets["example-toolset"]); diff != "" {
|
|
t.Errorf("error updating server, toolset (-want +got):\n%s", diff)
|
|
}
|
|
|
|
gotPrompt, _ := resMgr.GetPrompt("example-prompt")
|
|
if diff := cmp.Diff(gotPrompt, newPrompts["example-prompt"]); diff != "" {
|
|
t.Errorf("error updating server, prompts (-want +got):\n%s", diff)
|
|
}
|
|
|
|
gotPromptset, _ := resMgr.GetPromptset("example-promptset")
|
|
if diff := cmp.Diff(gotPromptset, newPromptsets["example-promptset"]); diff != "" {
|
|
t.Errorf("error updating server, promptset (-want +got):\n%s", diff)
|
|
}
|
|
|
|
updateSource := map[string]sources.Source{
|
|
"example-source2": &alloydbpg.Source{
|
|
Config: alloydbpg.Config{
|
|
Name: "example-alloydb-source2",
|
|
Kind: "alloydb-postgres",
|
|
},
|
|
},
|
|
}
|
|
|
|
resMgr.SetResources(updateSource, newAuth, newTools, newToolsets, newPrompts, newPromptsets)
|
|
gotSource, _ = resMgr.GetSource("example-source2")
|
|
if diff := cmp.Diff(gotSource, updateSource["example-source2"]); diff != "" {
|
|
t.Errorf("error updating server, sources (-want +got):\n%s", diff)
|
|
}
|
|
}
|