cmd

type DisplayMetricsResult struct {
	Metric     string
	Timestamps string
	Values     string
}

type GenericToolInput struct {
	Flags map[string]any `json:"flags"`
	Args  []string       `json:"args"`
}

type GenericToolOutput struct {
	Output map[string]any `json:"output"`
	Error  string         `json:"error,omitempty"`
}

type JSONSchema struct {
	Type        string                 `json:"type"`
	Properties  map[string]*JSONSchema `json:"properties,omitempty"`
	Items       *JSONSchema            `json:"items,omitempty"`
	Description string                 `json:"description,omitempty"`
	Default     any                    `json:"default,omitempty"`
}

type LogGroupOutput struct {
	Sample         string `json:"sample"`
	Namespace      string `json:"namespace"`
	Pod            string `json:"pod"`
	Count          string `json:"count"`
	LastOccurrence string `json:"last_occurrence"`
}

type MessageConfig struct {
	FollowupOptions []string `json:"followupOptions"`

	FollowupType string `json:"followupType"`

	Question string `json:"question"`
}

type MetricsQueryResponse struct {
	MetricsQuery struct {
		Results []MetricsResponse `json:"results"`
	} `json:"metrics_list"`
}

type MetricsResponse struct {
	QueryKey string          `json:"query_key"`
	Payload  []MetricsResult `json:"payload"`
}

type MetricsResult struct {
	Metric     map[string]string `json:"metric"`
	Timestamps []float64         `json:"timestamps"`
	Values     []float64         `json:"values"`
}

type NubiToolInput struct {
	Query string `json:"query" jsonschema:"The user's question or request in human-readable form."`
}

type NubiToolOutput struct {
	Response map[string]any `json:"response" jsonschema:"The response from the Nubi agent."`
}

type SecurityImageRecommendation struct {
	Severity       string          `json:"severity"`
	Recommendation json.RawMessage `json:"recommendation"`
	Image          string          `json:"image"`
	ID             string          `json:"id"`
	Namespace      string          `json:"namespace"`
	WorkloadName   string          `json:"workload_name"`
	PackageID      string          `json:"package_id"`
}

type SecurityImageResponse struct {
	Recommendation struct {
		Rows []SecurityImageRecommendation `json:"rows"`
	} `json:"recommendation"`
	RecommendationAggregate struct {
		Rows []struct {
			Count int `json:"count"`
		} `json:"rows"`
	} `json:"recommendation_aggregate"`
}