types

type APIError struct {
	Msg string
}

type Command interface {
	Run(cmd *cobra.Command, args []string) error
	SynthesizeRequest(input interface{}) (interface{}, error)
	InputHandler(cmd *cobra.Command) interface{}
	Execute(req interface{}) error
	DefineFlags()
}

type CommandLineInput struct {
	GroupByDimension    []string
	GroupByTag          []string
	FilterByValues      map[string]string
	IsFilterByTag       bool
	TagFilterValue      string
	IsFilterByDimension bool
	Start               string
	End                 string
	ExcludeDiscounts    bool
	Interval            string
	PrintFormat         string
	Metrics             []string
	SortByDate          bool
	OpenAIAPIKey        string
	PineconeIndex       string
	PineconeAPIKey      string
}

type CostAndUsage struct {
	Rows  []table.Row
	Total string
}

type CostAndUsageOutputType struct {
	Services       map[int]Service
	Granularity    string
	Start          string
	End            string
	Dimensions     []string
	Tags           []string
	SortBy         string
	OpenAIAPIKey   string
	PineconeAPIKey string
	PineconeIndex  string
}

type CostAndUsageRequestType struct {
	Granularity                string
	GroupBy                    []string
	GroupByTag                 []string
	Time                       Time
	IsFilterByTagEnabled       bool
	IsFilterByDimensionEnabled bool
	TagFilterValue             string
	DimensionFilter            map[string]string
	ExcludeDiscounts           bool
	Alias                      string
	Rates                      []string
	PrintFormat                string
	Metrics                    []string
	SortByDate                 bool
	OpenAIAPIKey               string
	PineconeIndex              string
	PineconeAPIKey             string
}

type CostAndUsageRequestWithResourcesType struct {
	Granularity      string
	GroupBy          []string
	Tag              string
	Time             Time
	IsFilterEnabled  bool
	FilterType       string
	TagFilterValue   string
	Rates            []string
	ExcludeDiscounts bool
}

type CostAndUsageStdoutType struct {
	Granularity string
	Services    []Service
}

type CostDataReader interface {
	ExtractGroupBySelections() ([]string, []string)
	ExtractFilterBySelection() (FilterBySelections, error)
	ExtractStartAndEndDates() (string, string, error)
	ExtractPrintPreferences() PrintOptions
}

type DateInterval struct {
	End   string
	Start string
}

type Dimension struct {
	Key   string
	Value []string
}

type Error struct {
	Msg string
}

type Filter struct {
	Dimensions []Dimension
	Tags       []Tag
}

type FilterBySelections struct {
	Tags                string
	Dimensions          map[string]string
	IsFilterByTag       bool
	IsFilterByDimension bool
}

type ForecastCommandLineInput struct {
	FilterByValues          Filter
	Granularity             string
	PredictionIntervalLevel int32
	Start                   string
	End                     string
}

type ForecastPrintData struct {
	Forecast *costexplorer.GetCostForecastOutput
	Filters  []string
}

type ForecastResults struct {
	MeanValue                    string
	PredictionIntervalLowerBound string
	PredictionIntervalUpperBound string
	TimePeriod                   DateInterval
}

type ForecastStdoutType struct {
	Forecast   []ForecastResults
	FilteredBy string
	Total      Total
}

type GetCostAndUsageAPI interface {
	GetCostAndUsage(ctx context.Context,
		optFns ...func(*costexplorer.Options)) (*costexplorer.GetCostAndUsageOutput, error)
}

type GetCostForecastAPI interface {
	GetCostForecast(ctx context.Context, params *costexplorer.GetCostForecastInput, optFns ...func(*costexplorer.Options)) (*costexplorer.GetCostForecastOutput, error)
}

type GetCostForecastReport struct{}

type GetCostForecastRequest struct {
	Time                    Time
	Granularity             string
	Metric                  string
	Filter                  Filter
	PredictionIntervalLevel int32
}

type GetDimensionValuesRequest struct {
	Dimension string
	Time      Time
}

type InputType struct {
	Services     []Service
	Granularity  string
	Start        string
	End          string
	Dimensions   []string
	Tags         []string
	SortBy       string
	OpenAIAPIKey string
}

type Metrics struct {
	Name          string
	Amount        string
	NumericAmount float64
	Unit          string
	UsageQuantity float64
}

type PresetError struct {
	Msg string
}

type PresetParams struct {
	Alias             string
	Dimension         []string
	Tag               string
	Filter            map[string]string
	FilterType        string
	FilterByDimension bool
	FilterByTag       bool
	ExcludeDiscounts  bool
	CommandSyntax     string
	Description       []string
	Granularity       string
	PrintFormat       string
	Metric            []string
}

type PrintOptions struct {
	IsSortByDate     bool
	ExcludeDiscounts bool
	Format           string
	OpenAIKey        string
	Granularity      string
	Metric           string
	PineconeIndex    string
	PineconeAPIKey   string
}

type PrintWriterType int

type Service struct {
	Keys    []string
	Name    string
	Metrics []Metrics
	Start   string
	End     string
}

type SortBy int

type Table interface {
	Writer(interface{})
	Header()
	Footer(row table.Row)
	AddRows(rows []table.Row)
	Style()
}

type Tag struct {
	Key   string
	Value []string
}

type Time struct {
	Start string
	End   string
}

type Total struct {
	Amount string
	Unit   string
}

type UpsertResponse struct {
	UpsertedCount int `json:"upsertedCount"`
}

type VectorStoreInput struct {
	Items []VectorStoreItem
}

type VectorStoreItem struct {
	ID              string
	EmbeddingText   string
	EmbeddingVector []float32
	Metadata        VectorStoreItemMetadata
}

type VectorStoreItemMetadata struct {
	StartDate   string
	EndDate     string
	Granularity string
	Dimensions  string
	Tags        string
	Cost        string
}