github/genai-toolbox
1
1
Fork 0
mirror of https://github.com/googleapis/genai-toolbox.git synced 2026-05-02 03:00:36 -04:00
Code Issues Packages Projects Releases 14 Wiki Activity

chore: refactor parameters logic to allow for more complex functionality (#24)

Browse Source 
Refactors parameter logic to allow for more complex functionality (e.g.
arrays) in the future.

---------

Co-authored-by: Wenxin Du <117315983+duwenxin99@users.noreply.github.com>
This commit is contained in:
Kurtis Van Gent
2024-10-23 13:04:10 -06:00
committed by GitHub
parent 546ca41237
commit de14c6f669
8 changed files with 392 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
internal/tools/cloud_sql_pg.go
View File

@@ -28,12 +28,12 @@ const CloudSQLPgSQLGenericKind string = "cloud-sql-postgres-generic"
var _ Config = CloudSQLPgGenericConfig{}
type CloudSQLPgGenericConfig struct {
Name string `yaml:"name"`
Kind string `yaml:"kind"`
Source string `yaml:"source"`
Description string `yaml:"description"`
Statement string `yaml:"statement"`
Parameters []Parameter `yaml:"parameters"`
Name string `yaml:"name"`
Kind string `yaml:"kind"`
Source string `yaml:"source"`
Description string `yaml:"description"`
Statement string `yaml:"statement"`
Parameters Parameters `yaml:"parameters"`
}
func (cfg CloudSQLPgGenericConfig) toolKind() string {
@@ -60,7 +60,7 @@ func (cfg CloudSQLPgGenericConfig) Initialize(srcs map[string]sources.Source) (T
Source: s,
Statement: cfg.Statement,
Parameters: cfg.Parameters,
manifest: Manifest{cfg.Description, cfg.Parameters},
manifest: ToolManifest{cfg.Description, generateManifests(cfg.Parameters)},
}
return t, nil
}
@@ -73,8 +73,8 @@ type CloudSQLPgGenericTool struct {
Kind string `yaml:"kind"`
Source sources.CloudSQLPgSource
Statement string
Parameters []Parameter `yaml:"parameters"`
manifest Manifest
Parameters Parameters `yaml:"parameters"`
manifest ToolManifest
}
func (t CloudSQLPgGenericTool) Invoke(params []any) (string, error) {
@@ -100,6 +100,6 @@ func (t CloudSQLPgGenericTool) ParseParams(data map[string]any) ([]any, error) {
return ParseParams(t.Parameters, data)
}
func (t CloudSQLPgGenericTool) Manifest() Manifest {
func (t CloudSQLPgGenericTool) Manifest() ToolManifest {
return t.manifest
}