add checkpoint token comparison

8004611·

peyton-alt·3w ago·2 files·+569 added/-13 removed

Sessions

cea04e74bd49View transcript

Changes

2

• cmd/entire/cli

  • Mcheckpoint_tokens.go+182/-13
  • Msessions_test.go+387

4 unmodified lines

encoding/json"
"fmt"
"io"
"strconv"
"strings"

"github.com/entireio/cli/cmd/entire/cli/agent"

Code

type checkpointTokensComparison struct {
  BaselineCheckpointID string                       `json:"baseline_checkpoint_id"`
  TargetCheckpointID   string                       `json:"target_checkpoint_id"`
  Status               string                       `json:"status"`
  Total                *checkpointTokensMetricDelta `json:"total,omitempty"`
  CacheRead            *checkpointTokensMetricDelta `json:"cache_read,omitempty"`
  APICalls             *checkpointTokensMetricDelta `json:"api_calls,omitempty"`
  Qualification        string                       `json:"qualification"`
  Limitations          []string                     `json:"limitations,omitempty"`
}

func newCheckpointTokensCmd() *cobra.Command {
  var jsonFlag bool
  var compareFlag string

cmd := &cobra.Command{
    Use:   "tokens <checkpoint-id>",
    Args: cobra.ExactArgs(1),
    RunE: func(cmd *cobra.Command, args []string) error {
      return runCheckpointTokens(cmd.Context(), cmd, args[0], jsonFlag)
    },
  }

cmd.Flags().BoolVar(&jsonFlag, "json", false, "Output as JSON")
  cmd.Flags().StringVar(&compareFlag, "compare", "", "Compare against a baseline checkpoint ID")
  return cmd
}