From f3224278a8bff3758b23a227eb5c203fde9b40fb Mon Sep 17 00:00:00 2001
From: beads/crew/fang <steve.yegge@gmail.com>
Date: Wed, 31 Dec 2025 13:04:01 -0800
Subject: [PATCH] fix: use lightweight molecule discovery in bd mol progress
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Replace findInProgressMolecules (which loads full subgraphs) with
findInProgressMoleculeIDs (which only returns IDs). This ensures
auto-discovery is efficient for mega-molecules.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
---
 cmd/bd/mol_progress.go | 38 ++++++++++++++++++++++++++++++++++----
 1 file changed, 34 insertions(+), 4 deletions(-)

diff --git a/cmd/bd/mol_progress.go b/cmd/bd/mol_progress.go
index 2d87b6ad..da9deb48 100644
--- a/cmd/bd/mol_progress.go
+++ b/cmd/bd/mol_progress.go
@@ -1,10 +1,12 @@
 package main
 
 import (
+	"context"
 	"fmt"
 	"os"
 
 	"github.com/spf13/cobra"
+	"github.com/steveyegge/beads/internal/storage"
 	"github.com/steveyegge/beads/internal/types"
 	"github.com/steveyegge/beads/internal/ui"
 	"github.com/steveyegge/beads/internal/utils"
@@ -53,9 +55,9 @@ Example:
 			}
 			moleculeID = resolved
 		} else {
-			// Infer from in_progress work
-			molecules := findInProgressMolecules(ctx, store, actor)
-			if len(molecules) == 0 {
+			// Infer from in_progress work - use lightweight discovery
+			moleculeIDs := findInProgressMoleculeIDs(ctx, store, actor)
+			if len(moleculeIDs) == 0 {
 				if jsonOutput {
 					outputJSON([]interface{}{})
 					return
@@ -65,7 +67,7 @@ Example:
 				return
 			}
 			// Show progress for first molecule
-			moleculeID = molecules[0].MoleculeID
+			moleculeID = moleculeIDs[0]
 		}
 
 		stats, err := store.GetMoleculeProgress(ctx, moleculeID)
@@ -108,6 +110,34 @@ Example:
 	},
 }
 
+// findInProgressMoleculeIDs finds molecule IDs with in_progress steps for an agent.
+// This is a lightweight version that only returns IDs without loading subgraphs.
+func findInProgressMoleculeIDs(ctx context.Context, s storage.Storage, agent string) []string {
+	// Query for in_progress issues
+	status := types.StatusInProgress
+	filter := types.IssueFilter{Status: &status}
+	if agent != "" {
+		filter.Assignee = &agent
+	}
+	inProgressIssues, err := s.SearchIssues(ctx, "", filter)
+	if err != nil || len(inProgressIssues) == 0 {
+		return nil
+	}
+
+	// For each in_progress issue, find its parent molecule
+	seen := make(map[string]bool)
+	var moleculeIDs []string
+	for _, issue := range inProgressIssues {
+		moleculeID := findParentMolecule(ctx, s, issue.ID)
+		if moleculeID != "" && !seen[moleculeID] {
+			seen[moleculeID] = true
+			moleculeIDs = append(moleculeIDs, moleculeID)
+		}
+	}
+
+	return moleculeIDs
+}
+
 // printMoleculeProgressStats prints molecule progress in human-readable format
 func printMoleculeProgressStats(stats *types.MoleculeProgressStats) {
 	fmt.Printf("Molecule: %s (%s)\n", ui.RenderAccent(stats.MoleculeID), stats.MoleculeTitle)