mirror of
https://github.com/hanzoai/o11y-mcp.git
synced 2026-10-30 18:26:37 +00:00
PRADEEP KUMARandGitHub
776b0bfde2
aggregate log under the hood uses query builder only but now llm passes multiple input to the tool to filter. search log is generic log search not tight with error logs or logs for services. ideally we can even remove other logs related tools like search log by service, get error logs but for now keeping.
75 lines
1.9 KiB
Go
75 lines
1.9 KiB
Go
package tools
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
// parseAggregateLogsArgs validates and parses arguments for the aggregate_logs tool.
|
|
func parseAggregateLogsArgs(args map[string]any) (*AggregateRequest, error) {
|
|
service, _ := args["service"].(string)
|
|
severity, _ := args["severity"].(string)
|
|
filter, _ := args["filter"].(string)
|
|
filterExpr := buildLogFilterExpr(filter, service, severity, "")
|
|
|
|
return parseAggregateArgs(args, "logs", filterExpr)
|
|
}
|
|
|
|
// SearchLogsRequest holds the parsed parameters for a log search query.
|
|
type SearchLogsRequest struct {
|
|
FilterExpression string
|
|
Limit int
|
|
Offset int
|
|
StartTime int64
|
|
EndTime int64
|
|
}
|
|
|
|
func parseSearchLogsArgs(args map[string]any) (*SearchLogsRequest, error) {
|
|
query, _ := args["query"].(string)
|
|
service, _ := args["service"].(string)
|
|
severity, _ := args["severity"].(string)
|
|
searchText, _ := args["searchText"].(string)
|
|
filterExpr := buildLogFilterExpr(query, service, severity, searchText)
|
|
|
|
limit, err := intArg(args, "limit", 100)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
offset, err := intArg(args, "offset", 0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
startTime, endTime, err := resolveTimestamps(args, "1h")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &SearchLogsRequest{
|
|
FilterExpression: filterExpr,
|
|
Limit: limit,
|
|
Offset: offset,
|
|
StartTime: startTime,
|
|
EndTime: endTime,
|
|
}, nil
|
|
}
|
|
|
|
// buildLogFilterExpr combines with log specific filters.
|
|
func buildLogFilterExpr(query, service, severity, searchText string) string {
|
|
var parts []string
|
|
if query != "" {
|
|
parts = append(parts, query)
|
|
}
|
|
if service != "" {
|
|
parts = append(parts, fmt.Sprintf("service.name = '%s'", service))
|
|
}
|
|
if severity != "" {
|
|
parts = append(parts, fmt.Sprintf("severity_text = '%s'", severity))
|
|
}
|
|
if searchText != "" {
|
|
parts = append(parts, fmt.Sprintf("body CONTAINS '%s'", searchText))
|
|
}
|
|
return strings.Join(parts, " AND ")
|
|
}
|