Display formatted table when run with gh models eval (#76)

Yuzuki-S · web-flow · commit 7fc3515d5047 · 2025-07-22T15:45:35.000+10:00
diff --git a/cmd/eval/eval.go b/cmd/eval/eval.go
@@ -10,13 +10,21 @@ import (
 	"time"
 
 	"github.com/MakeNowJust/heredoc"
+	"github.com/cli/go-gh/v2/pkg/tableprinter"
 	"github.com/github/gh-models/internal/azuremodels"
 	"github.com/github/gh-models/pkg/command"
 	"github.com/github/gh-models/pkg/prompt"
 	"github.com/github/gh-models/pkg/util"
+	"github.com/mgutz/ansi"
 	"github.com/spf13/cobra"
 )
 
+var (
+	lightGrayUnderline = ansi.ColorFunc("white+du")
+	red                = ansi.ColorFunc("red")
+	green              = ansi.ColorFunc("green")
+)
+
 // EvaluationSummary represents the overall evaluation summary
 type EvaluationSummary struct {
 	Name        string       `json:"name"`
@@ -167,6 +175,7 @@ func (h *evalCommandHandler) runEvaluation(ctx context.Context) error {
 
 	for i, testCase := range h.evalFile.TestData {
 		if !h.jsonOutput {
+			h.cfg.WriteToOut("-------------------------\n")
 			h.cfg.WriteToOut(fmt.Sprintf("Running test case %d/%d...\n", i+1, totalTests))
 		}
 
@@ -235,30 +244,58 @@ func (h *evalCommandHandler) runEvaluation(ctx context.Context) error {
 }
 
 func (h *evalCommandHandler) printTestResult(result TestResult, testPassed bool) {
+	printer := h.cfg.NewTablePrinter()
 	if testPassed {
-		h.cfg.WriteToOut("  &#10003; PASSED\n")
+		printer.AddField("Result", tableprinter.WithColor(lightGrayUnderline))
+		printer.AddField("&#10003; PASSED", tableprinter.WithColor(green))
+		printer.EndRow()
 	} else {
-		h.cfg.WriteToOut("  &#10007; FAILED\n")
+		printer.AddField("Result", tableprinter.WithColor(lightGrayUnderline))
+		printer.AddField("&#10007; FAILED", tableprinter.WithColor(red))
+		printer.EndRow()
 		// Show the first 100 characters of the model response when test fails
 		preview := result.ModelResponse
 		if len(preview) > 100 {
 			preview = preview[:100] + "..."
 		}
-		h.cfg.WriteToOut(fmt.Sprintf("    Model Response: %s\n", preview))
+
+		printer.AddField("Model Response", tableprinter.WithColor(lightGrayUnderline))
+		printer.AddField(preview)
+		printer.EndRow()
 	}
 
+	err := printer.Render()
+	if err != nil {
+		return
+	}
+
+	h.cfg.WriteToOut("\n")
+
+	table := h.cfg.NewTablePrinter()
+	table.AddHeader([]string{"EVALUATION", "RESULT", "SCORE", "CRITERIA"}, tableprinter.WithColor(lightGrayUnderline))
 	// Show evaluation details
 	for _, evalResult := range result.EvaluationResults {
-		status := "&#10003;"
+		status, color := "&#10003;", green
 		if !evalResult.Passed {
-			status = "&#10007;"
+			status, color = "&#10007;", red
 		}
-		h.cfg.WriteToOut(fmt.Sprintf("    %s %s (score: %.2f)\n",
-			status, evalResult.EvaluatorName, evalResult.Score))
+		table.AddField(evalResult.EvaluatorName)
+		table.AddField(status, tableprinter.WithColor(color))
+		table.AddField(fmt.Sprintf("%.2f", evalResult.Score), tableprinter.WithColor(color))
+
 		if evalResult.Details != "" {
-			h.cfg.WriteToOut(fmt.Sprintf("      %s\n", evalResult.Details))
+			table.AddField(evalResult.Details)
+		} else {
+			table.AddField("")
 		}
+		table.EndRow()
+	}
+
+	err = table.Render()
+	if err != nil {
+		return
 	}
+
 	h.cfg.WriteToOut("\n")
 }
 
diff --git a/cmd/eval/eval_test.go b/cmd/eval/eval_test.go
@@ -312,7 +312,8 @@ evaluators:
 		require.Contains(t, output, "Failing Test")
 		require.Contains(t, output, "Running test case")
 		require.Contains(t, output, "FAILED")
-		require.Contains(t, output, "Model Response: actual model response")
+		require.Contains(t, output, "Model Response")
+		require.Contains(t, output, "actual model response")
 	})
 
 	t.Run("json output format", func(t *testing.T) {
