From 3df00c9e0aa0788a4b59b45b0da4bd74257aa687 Mon Sep 17 00:00:00 2001
From: "claude[bot]" <41898282+claude[bot]@users.noreply.github.com>
Date: Fri, 27 Feb 2026 00:13:26 +0000
Subject: [PATCH] fix: correct context bomb zeros for lines and language counts
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- Remove Stats.TotalLines (not available in API response)
- Remove '· N lines' from the context bomb template
- Change Stats.Languages from map[string]int to []string since the
  API only returns language names, not per-language counts
- Change irSummary from a typed struct to map[string]any so that
  summary fields are read dynamically, avoiding silent zero-fills
  if API field names change
- Update languageList template func to accept []string and render
  plain names (e.g. 'go, json') instead of 'go (0), json (0)'

Fixes #7

Co-authored-by: Grey Newell <greynewell@users.noreply.github.com>
---
 internal/api/client.go      | 45 +++++++++++++++++++------------------
 internal/template/render.go | 10 +++------
 2 files changed, 26 insertions(+), 29 deletions(-)

diff --git a/internal/api/client.go b/internal/api/client.go
index c8ca839..df6e079 100644
--- a/internal/api/client.go
+++ b/internal/api/client.go
@@ -31,16 +31,10 @@ type Client struct {
 
 // SupermodelIR is the raw response from the Supermodel API /v1/graphs/supermodel endpoint.
 type SupermodelIR struct {
-	Repo     string           `json:"repo"`
-	Summary  irSummary        `json:"summary"`
-	Metadata irMetadata       `json:"metadata"`
-	Domains  []irDomain       `json:"domains"`
-}
-
-type irSummary struct {
-	FilesProcessed  int     `json:"filesProcessed"`
-	Functions       int     `json:"functions"`
-	PrimaryLanguage *string `json:"primaryLanguage"`
+	Repo     string         `json:"repo"`
+	Summary  map[string]any `json:"summary"`
+	Metadata irMetadata     `json:"metadata"`
+	Domains  []irDomain     `json:"domains"`
 }
 
 type irMetadata struct {
@@ -67,13 +61,21 @@ func (ir *SupermodelIR) toProjectGraph(projectName string) *ProjectGraph {
 	if len(ir.Metadata.Languages) > 0 {
 		lang = ir.Metadata.Languages[0]
 	}
-	if ir.Summary.PrimaryLanguage != nil && *ir.Summary.PrimaryLanguage != "" {
-		lang = *ir.Summary.PrimaryLanguage
+	if v, ok := ir.Summary["primaryLanguage"]; ok && v != nil {
+		if s, ok := v.(string); ok && s != "" {
+			lang = s
+		}
 	}
 
-	langMap := make(map[string]int, len(ir.Metadata.Languages))
-	for _, l := range ir.Metadata.Languages {
-		langMap[l] = 0 // count not available from API
+	// Extract integer fields from the free-form summary map.
+	// JSON numbers unmarshal as float64 in map[string]any.
+	summaryInt := func(key string) int {
+		if v, ok := ir.Summary[key]; ok {
+			if n, ok := v.(float64); ok {
+				return int(n)
+			}
+		}
+		return 0
 	}
 
 	domains := make([]Domain, 0, len(ir.Domains))
@@ -96,9 +98,9 @@ func (ir *SupermodelIR) toProjectGraph(projectName string) *ProjectGraph {
 		Language: lang,
 		Domains:  domains,
 		Stats: Stats{
-			TotalFiles:     ir.Summary.FilesProcessed,
-			TotalFunctions: ir.Summary.Functions,
-			Languages:      langMap,
+			TotalFiles:     summaryInt("filesProcessed"),
+			TotalFunctions: summaryInt("functions"),
+			Languages:      ir.Metadata.Languages,
 		},
 		UpdatedAt: time.Now(),
 	}
@@ -127,10 +129,9 @@ type Domain struct {
 
 // Stats holds codebase statistics.
 type Stats struct {
-	TotalFiles     int            `json:"total_files"`
-	TotalFunctions int            `json:"total_functions"`
-	TotalLines     int            `json:"total_lines"`
-	Languages      map[string]int `json:"languages,omitempty"`
+	TotalFiles     int      `json:"total_files"`
+	TotalFunctions int      `json:"total_functions"`
+	Languages      []string `json:"languages,omitempty"`
 }
 
 // JobStatus is the async envelope returned by the Supermodel API.
diff --git a/internal/template/render.go b/internal/template/render.go
index e44d15d..b5d0d26 100644
--- a/internal/template/render.go
+++ b/internal/template/render.go
@@ -21,7 +21,7 @@ const contextBombTmpl = `# Uncompact Context — {{.ProjectName}}
 **Language:** {{.Graph.Language}}{{if .Graph.Framework}}
 **Framework:** {{.Graph.Framework}}{{end}}{{if .Graph.Description}}
 **Description:** {{.Graph.Description}}{{end}}
-**Codebase:** {{.Graph.Stats.TotalFiles}} files · {{.Graph.Stats.TotalFunctions}} functions · {{.Graph.Stats.TotalLines}} lines
+**Codebase:** {{.Graph.Stats.TotalFiles}} files · {{.Graph.Stats.TotalFunctions}} functions
 {{- if .Graph.Stats.Languages}}
 
 **Languages:** {{languageList .Graph.Stats.Languages}}{{end}}
@@ -73,12 +73,8 @@ func Render(graph *api.ProjectGraph, projectName string, opts RenderOptions) (st
 
 	funcMap := gotmpl.FuncMap{
 		"join": strings.Join,
-		"languageList": func(langs map[string]int) string {
-			parts := make([]string, 0, len(langs))
-			for lang, count := range langs {
-				parts = append(parts, fmt.Sprintf("%s (%d)", lang, count))
-			}
-			return strings.Join(parts, ", ")
+		"languageList": func(langs []string) string {
+			return strings.Join(langs, ", ")
 		},
 	}