Skip to content

Commit 63dfc45

Browse files
tjhoptjhop
committed
chore(tools): s/arguments/args/g
Signed-off-by: TJ Hoplock <[email protected]>
1 parent 8115539 commit 63dfc45

File tree

1 file changed

+28
-28
lines changed

1 file changed

+28
-28
lines changed

‎pkg/mcp/tools.go‎

Lines changed: 28 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -162,14 +162,14 @@ var (
162162
)
163163

164164
func queryToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
165-
arguments := request.Params.Arguments
166-
query, ok := arguments["query"].(string)
165+
args := request.Params.Arguments
166+
query, ok := args["query"].(string)
167167
if !ok {
168168
return nil, errors.New("query must be a string")
169169
}
170170

171171
ts := time.Now()
172-
if argTs, ok := arguments["timestamp"].(string); ok {
172+
if argTs, ok := args["timestamp"].(string); ok {
173173
parsedTs, err := mcpProm.ParseTimestamp(argTs)
174174
if err != nil {
175175
return nil, fmt.Errorf("failed to get ts from args: %#v", argTs)
@@ -183,8 +183,8 @@ func queryToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.Ca
183183
}
184184

185185
func rangeQueryToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
186-
arguments := request.Params.Arguments
187-
query, ok := arguments["query"].(string)
186+
args := request.Params.Arguments
187+
query, ok := args["query"].(string)
188188
if !ok {
189189
return nil, errors.New("query must be a string")
190190
}
@@ -197,7 +197,7 @@ func rangeQueryToolHandler(ctx context.Context, request mcp.CallToolRequest) (*m
197197
// Convert seconds to nanoseconds such that time.Duration parses correctly.
198198
step := time.Duration(resolution) * time.Second
199199

200-
if argEndTime, ok := arguments["end_time"].(string); ok {
200+
if argEndTime, ok := args["end_time"].(string); ok {
201201
parsedEndTime, err := mcpProm.ParseTimestamp(argEndTime)
202202
if err != nil {
203203
return nil, fmt.Errorf("failed to parse end_time %s from args: %w", argEndTime, err)
@@ -206,7 +206,7 @@ func rangeQueryToolHandler(ctx context.Context, request mcp.CallToolRequest) (*m
206206
endTs = parsedEndTime
207207
}
208208

209-
if argStartTime, ok := arguments["start_time"].(string); ok {
209+
if argStartTime, ok := args["start_time"].(string); ok {
210210
parsedStartTime, err := mcpProm.ParseTimestamp(argStartTime)
211211
if err != nil {
212212
return nil, fmt.Errorf("failed to parse start_time %s from args: %w", argStartTime, err)
@@ -215,7 +215,7 @@ func rangeQueryToolHandler(ctx context.Context, request mcp.CallToolRequest) (*m
215215
startTs = parsedStartTime
216216
}
217217

218-
if argStep, ok := arguments["step"].(string); ok {
218+
if argStep, ok := args["step"].(string); ok {
219219
parsedStep, err := time.ParseDuration(argStep)
220220
if err != nil {
221221
return nil, fmt.Errorf("failed to parse duration %s for step: %w", argStep, err)
@@ -228,8 +228,8 @@ func rangeQueryToolHandler(ctx context.Context, request mcp.CallToolRequest) (*m
228228
}
229229

230230
func seriesToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
231-
arguments := request.Params.Arguments
232-
argMatchers, ok := arguments["matchers"].([]any)
231+
args := request.Params.Arguments
232+
argMatchers, ok := args["matchers"].([]any)
233233
if !ok {
234234
return nil, errors.New("matchers must be an array")
235235
}
@@ -242,7 +242,7 @@ func seriesToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.C
242242
endTs := time.Now()
243243
startTs := endTs.Add(DefaultLookbackDelta)
244244

245-
if argEndTime, ok := arguments["end_time"].(string); ok {
245+
if argEndTime, ok := args["end_time"].(string); ok {
246246
parsedEndTime, err := mcpProm.ParseTimestamp(argEndTime)
247247
if err != nil {
248248
return nil, fmt.Errorf("failed to parse end_time %s from args: %w", argEndTime, err)
@@ -251,7 +251,7 @@ func seriesToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.C
251251
endTs = parsedEndTime
252252
}
253253

254-
if argStartTime, ok := arguments["start_time"].(string); ok {
254+
if argStartTime, ok := args["start_time"].(string); ok {
255255
parsedStartTime, err := mcpProm.ParseTimestamp(argStartTime)
256256
if err != nil {
257257
return nil, fmt.Errorf("failed to parse start_time %s from args: %w", argStartTime, err)
@@ -265,8 +265,8 @@ func seriesToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.C
265265
}
266266

267267
func labelNamesToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
268-
arguments := request.Params.Arguments
269-
argMatchers, ok := arguments["matchers"].([]any)
268+
args := request.Params.Arguments
269+
argMatchers, ok := args["matchers"].([]any)
270270
if !ok {
271271
return nil, errors.New("matchers must be an array")
272272
}
@@ -279,7 +279,7 @@ func labelNamesToolHandler(ctx context.Context, request mcp.CallToolRequest) (*m
279279
endTs := time.Now()
280280
startTs := endTs.Add(DefaultLookbackDelta)
281281

282-
if argEndTime, ok := arguments["end_time"].(string); ok {
282+
if argEndTime, ok := args["end_time"].(string); ok {
283283
parsedEndTime, err := mcpProm.ParseTimestamp(argEndTime)
284284
if err != nil {
285285
return nil, fmt.Errorf("failed to parse end_time %s from args: %w", argEndTime, err)
@@ -288,7 +288,7 @@ func labelNamesToolHandler(ctx context.Context, request mcp.CallToolRequest) (*m
288288
endTs = parsedEndTime
289289
}
290290

291-
if argStartTime, ok := arguments["start_time"].(string); ok {
291+
if argStartTime, ok := args["start_time"].(string); ok {
292292
parsedStartTime, err := mcpProm.ParseTimestamp(argStartTime)
293293
if err != nil {
294294
return nil, fmt.Errorf("failed to parse start_time %s from args: %w", argStartTime, err)
@@ -302,13 +302,13 @@ func labelNamesToolHandler(ctx context.Context, request mcp.CallToolRequest) (*m
302302
}
303303

304304
func labelValuesToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
305-
arguments := request.Params.Arguments
306-
label, ok := arguments["label"].(string)
305+
args := request.Params.Arguments
306+
label, ok := args["label"].(string)
307307
if !ok {
308308
return nil, errors.New("label must be a string")
309309
}
310310

311-
argMatchers, ok := arguments["matchers"].([]any)
311+
argMatchers, ok := args["matchers"].([]any)
312312
if !ok {
313313
return nil, errors.New("matchers must be an array")
314314
}
@@ -321,7 +321,7 @@ func labelValuesToolHandler(ctx context.Context, request mcp.CallToolRequest) (*
321321
endTs := time.Now()
322322
startTs := endTs.Add(DefaultLookbackDelta)
323323

324-
if argEndTime, ok := arguments["end_time"].(string); ok {
324+
if argEndTime, ok := args["end_time"].(string); ok {
325325
parsedEndTime, err := mcpProm.ParseTimestamp(argEndTime)
326326
if err != nil {
327327
return nil, fmt.Errorf("failed to parse end_time %s from args: %w", argEndTime, err)
@@ -330,7 +330,7 @@ func labelValuesToolHandler(ctx context.Context, request mcp.CallToolRequest) (*
330330
endTs = parsedEndTime
331331
}
332332

333-
if argStartTime, ok := arguments["start_time"].(string); ok {
333+
if argStartTime, ok := args["start_time"].(string); ok {
334334
parsedStartTime, err := mcpProm.ParseTimestamp(argStartTime)
335335
if err != nil {
336336
return nil, fmt.Errorf("failed to parse start_time %s from args: %w", argStartTime, err)
@@ -389,20 +389,20 @@ func targetsToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.
389389
}
390390

391391
func targetsMetadataToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
392-
arguments := request.Params.Arguments
392+
args := request.Params.Arguments
393393

394394
matchTarget := ""
395-
if argMatchTarget, ok := arguments["match_target"].(string); ok {
395+
if argMatchTarget, ok := args["match_target"].(string); ok {
396396
matchTarget = argMatchTarget
397397
}
398398

399399
metric := ""
400-
if argMetric, ok := arguments["metric"].(string); ok {
400+
if argMetric, ok := args["metric"].(string); ok {
401401
metric = argMetric
402402
}
403403

404404
limit := ""
405-
if argLimit, ok := arguments["limit"].(string); ok {
405+
if argLimit, ok := args["limit"].(string); ok {
406406
limit = argLimit
407407
}
408408

@@ -413,15 +413,15 @@ func targetsMetadataToolHandler(ctx context.Context, request mcp.CallToolRequest
413413
// // Metadata returns metadata about metrics currently scraped by the metric name.
414414
// Metadata(ctx context.Context, metric, limit string) (map[string][]Metadata, error)
415415
func metricMetadataToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
416-
arguments := request.Params.Arguments
416+
args := request.Params.Arguments
417417

418418
metric := ""
419-
if argMetric, ok := arguments["metric"].(string); ok {
419+
if argMetric, ok := args["metric"].(string); ok {
420420
metric = argMetric
421421
}
422422

423423
limit := ""
424-
if argLimit, ok := arguments["limit"].(string); ok {
424+
if argLimit, ok := args["limit"].(string); ok {
425425
limit = argLimit
426426
}
427427

0 commit comments

Comments
 (0)