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

Revert "chore: Access params by maps for execute-sql tools" (#1095)

Browse Source 
This reverts commit 6455ba964b.
This commit is contained in:
Wenxin Du
2025-08-06 16:13:36 -04:00
committed by GitHub
parent 6455ba964b
commit 5fda4d4ed1
6 changed files with 18 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
internal/tools/mssql/mssqlexecutesql/mssqlexecutesql.go
View File

@@ -118,10 +118,10 @@ type Tool struct {
}
func (t Tool) Invoke(ctx context.Context, params tools.ParamValues) (any, error) {
paramsMap := params.AsMap()
sql, ok := paramsMap["sql"].(string)
sliceParams := params.AsSlice()
sql, ok := sliceParams[0].(string)
if !ok {
return nil, fmt.Errorf("unable to get cast %s", paramsMap["sql"])
return nil, fmt.Errorf("unable to get cast %s", sliceParams[0])
}
// Log the query executed for debugging.

6
internal/tools/mysql/mysqlexecutesql/mysqlexecutesql.go
View File

@@ -118,10 +118,10 @@ type Tool struct {
}
func (t Tool) Invoke(ctx context.Context, params tools.ParamValues) (any, error) {
paramsMap := params.AsMap()
sql, ok := paramsMap["sql"].(string)
sliceParams := params.AsSlice()
sql, ok := sliceParams[0].(string)
if !ok {
return nil, fmt.Errorf("unable to get cast %s", paramsMap["sql"])
return nil, fmt.Errorf("unable to get cast %s", sliceParams[0])
}
// Log the query executed for debugging.

6
internal/tools/neo4j/neo4jexecutecypher/neo4jexecutecypher.go
View File

@@ -124,10 +124,10 @@ type Tool struct {
}
func (t Tool) Invoke(ctx context.Context, params tools.ParamValues) (any, error) {
paramsMap := params.AsMap()
cypherStr, ok := paramsMap["sql"].(string)
sliceParams := params.AsSlice()
cypherStr, ok := sliceParams[0].(string)
if !ok {
return nil, fmt.Errorf("unable to get cast %s", paramsMap["sql"])
return nil, fmt.Errorf("unable to get cast %s", sliceParams[0])
}
if cypherStr == "" {

6
internal/tools/postgres/postgresexecutesql/postgresexecutesql.go
View File

@@ -120,10 +120,10 @@ type Tool struct {
}
func (t Tool) Invoke(ctx context.Context, params tools.ParamValues) (any, error) {
paramsMap := params.AsMap()
sql, ok := paramsMap["sql"].(string)
sliceParams := params.AsSlice()
sql, ok := sliceParams[0].(string)
if !ok {
return nil, fmt.Errorf("unable to get cast %s", paramsMap["sql"])
return nil, fmt.Errorf("unable to get cast %s", sliceParams[0])
}
// Log the query executed for debugging.
logger, err := util.LoggerFromContext(ctx)

6
internal/tools/spanner/spannerexecutesql/spannerexecutesql.go
View File

@@ -146,10 +146,10 @@ func processRows(iter *spanner.RowIterator) ([]any, error) {
}
func (t Tool) Invoke(ctx context.Context, params tools.ParamValues) (any, error) {
paramsMap := params.AsMap()
sql, ok := paramsMap["sql"].(string)
sliceParams := params.AsSlice()
sql, ok := sliceParams[0].(string)
if !ok {
return nil, fmt.Errorf("unable to get cast %s", paramsMap["sql"])
return nil, fmt.Errorf("unable to get cast %s", sliceParams[0])
}
// Log the query executed for debugging.

6
internal/tools/tidb/tidbexecutesql/tidbexecutesql.go
View File

@@ -116,10 +116,10 @@ type Tool struct {
}
func (t Tool) Invoke(ctx context.Context, params tools.ParamValues) (any, error) {
paramsMap := params.AsMap()
sql, ok := paramsMap["sql"].(string)
sliceParams := params.AsSlice()
sql, ok := sliceParams[0].(string)
if !ok {
return nil, fmt.Errorf("unable to get cast %s", paramsMap["sql"])
return nil, fmt.Errorf("unable to get cast %s", sliceParams[0])
}
// Log the query executed for debugging.