Skip to content

Commit 28ca76c

Browse files
Use NewToolResultErrorFromErr
GitOrigin-RevId: f59beb6
1 parent 63765da commit 28ca76c

1 file changed

Lines changed: 7 additions & 7 deletions

File tree

pkg/postgres/tools.go

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -232,18 +232,18 @@ func queryPostgres(postgresRepo *Repo) server.ServerTool {
232232

233233
config, err := pgx.ParseConfig(connectionInfo.ExternalConnectionString)
234234
if err != nil {
235-
return mcp.NewToolResultErrorf("Error parsing connection string: %s", err.Error()), nil
235+
return mcp.NewToolResultErrorFromErr("Error parsing connection string", err), nil
236236
}
237237
conn, err := pgx.ConnectConfig(ctx, config)
238238
if err != nil {
239-
return mcp.NewToolResultErrorf("Error connecting to database: %s", err.Error()), nil
239+
return mcp.NewToolResultErrorFromErr("Error connecting to database", err), nil
240240
}
241241
defer conn.Close(ctx)
242242

243243
// Wrap all queries in a READ ONLY transaction
244244
tx, err := conn.BeginTx(ctx, pgx.TxOptions{AccessMode: pgx.ReadOnly})
245245
if err != nil {
246-
return mcp.NewToolResultErrorf("Error beginning transaction: %s", err.Error()), nil
246+
return mcp.NewToolResultErrorFromErr("Error beginning transaction", err), nil
247247
}
248248

249249
// Make sure we roll back the transaction if it's not committed successfully
@@ -253,7 +253,7 @@ func queryPostgres(postgresRepo *Repo) server.ServerTool {
253253

254254
rows, err := tx.Query(ctx, sqlQuery)
255255
if err != nil {
256-
return mcp.NewToolResultErrorf("Error executing query: %s", err.Error()), nil
256+
return mcp.NewToolResultErrorFromErr("Error executing query", err), nil
257257
}
258258
defer rows.Close()
259259

@@ -269,7 +269,7 @@ func queryPostgres(postgresRepo *Repo) server.ServerTool {
269269
for rows.Next() {
270270
values, err := rows.Values()
271271
if err != nil {
272-
return mcp.NewToolResultErrorf("Error reading row values: %s", err.Error()), nil
272+
return mcp.NewToolResultErrorFromErr("Error reading row values", err), nil
273273
}
274274

275275
rowMap := make(map[string]interface{})
@@ -290,12 +290,12 @@ func queryPostgres(postgresRepo *Repo) server.ServerTool {
290290

291291
// Check for any errors encountered during iteration
292292
if err := rows.Err(); err != nil {
293-
return mcp.NewToolResultErrorf("Error iterating rows: %s", err.Error()), nil
293+
return mcp.NewToolResultErrorFromErr("Error iterating rows", err), nil
294294
}
295295

296296
respJSON, err := json.Marshal(results)
297297
if err != nil {
298-
return mcp.NewToolResultErrorf("Error marshaling results: %s", err.Error()), nil
298+
return mcp.NewToolResultErrorFromErr("Error marshaling results", err), nil
299299
}
300300

301301
return mcp.NewToolResultText(string(respJSON)), nil

0 commit comments

Comments
 (0)