refactor: orchestrator pre-selects features for all agents

Replace agent-initiated feature selection with orchestrator pre-selection
for both coding and testing agents. This ensures Mission Control displays
correct feature numbers for testing agents (previously showed "Feature #0").

Key changes:

MCP Server (mcp_server/feature_mcp.py):
- Add feature_get_by_id tool for agents to fetch assigned feature details
- Remove obsolete tools: feature_get_next, feature_claim_next,
  feature_claim_for_testing, feature_get_for_regression
- Remove helper functions and unused imports (text, OperationalError, func)

Orchestrator (parallel_orchestrator.py):
- Change running_testing_agents from list to dict[int, Popen]
- Add claim_feature_for_testing() with random selection
- Add release_testing_claim() method
- Pass --testing-feature-id to spawned testing agents
- Use unified [Feature #X] output format for both agent types

Agent Entry Points:
- autonomous_agent_demo.py: Add --testing-feature-id CLI argument
- agent.py: Pass testing_feature_id to get_testing_prompt()

Prompt Templates:
- coding_prompt.template.md: Update to use feature_get_by_id
- testing_prompt.template.md: Update workflow for pre-assigned features
- prompts.py: Update pre-claimed headers for both agent types

WebSocket (server/websocket.py):
- Simplify tracking with unified [Feature #X] pattern
- Remove testing-specific parsing code

Assistant (server/services/assistant_chat_session.py):
- Update help text with current available tools

Co-Authored-By: Claude Opus 4.5 <[email protected]>

Author: leonvanzyl

.claude/templates/coding_prompt.template.md

@@ -29,9 +29,6 @@ Then use MCP tools to check feature status:
 ```
 # 6. Get progress statistics (passing/total counts)
 Use the feature_get_stats tool
-
-# 7. Get the next feature to work on
-Use the feature_get_next tool
 ```
 
 Understanding the `app_spec.txt` is critical - it contains the full requirements
@@ -48,7 +45,7 @@ chmod +x init.sh
 
 Otherwise, start servers manually and document the process.
 
-### STEP 3: CHOOSE ONE FEATURE TO IMPLEMENT
+### STEP 3: GET YOUR ASSIGNED FEATURE
 
 #### TEST-DRIVEN DEVELOPMENT MINDSET (CRITICAL)
 
@@ -63,20 +60,17 @@ Features are **test cases** that drive development. This is test-driven developm
 - WRONG: "Flashcard page doesn't exist yet" → skip feature
 - RIGHT: "Flashcard page doesn't exist yet" → build flashcard page → implement filter → test feature
 
-Get the next feature to implement:
+**Note:** Your feature has been pre-assigned by the orchestrator. Use `feature_get_by_id` with your assigned feature ID to get the details.
 
-```
-# Get the highest-priority pending feature
-Use the feature_get_next tool
-```
-
-Once you've retrieved the feature, **immediately mark it as in-progress**:
+Once you've retrieved the feature, **mark it as in-progress** (if not already):
 
 ```
-# Mark feature as in-progress to prevent other sessions from working on it
-Use the feature_mark_in_progress tool with feature_id=42
+# Mark feature as in-progress
+Use the feature_mark_in_progress tool with feature_id={your_assigned_id}
 ```
 
+If you get "already in-progress" error, that's OK - continue with implementation.
+
 Focus on completing one feature perfectly and completing its testing steps in this session before moving on to other features.
 It's ok if you only complete one feature in this session, as there will be more sessions later that continue to make progress.
 
@@ -337,10 +331,10 @@ The feature tools exist to reduce token usage. **DO NOT make exploratory queries
 # 1. Get progress stats (passing/in_progress/total counts)
 feature_get_stats
 
-# 2. Get the NEXT feature to work on (one feature only)
-feature_get_next
+# 2. Get your assigned feature details
+feature_get_by_id with feature_id={your_assigned_id}
 
-# 3. Mark a feature as in-progress (call immediately after feature_get_next)
+# 3. Mark a feature as in-progress
 feature_mark_in_progress with feature_id={id}
 
 # 4. Mark a feature as passing (after verification)
@@ -349,7 +343,7 @@ feature_mark_passing with feature_id={id}
 # 5. Mark a feature as failing (if you discover it's broken)
 feature_mark_failing with feature_id={id}
 
-# 6. Skip a feature (moves to end of queue) - ONLY when blocked by dependency
+# 6. Skip a feature (moves to end of queue) - ONLY when blocked by external dependency
 feature_skip with feature_id={id}
 
 # 7. Clear in-progress status (when abandoning a feature)
@@ -361,8 +355,9 @@ feature_clear_in_progress with feature_id={id}
 - Do NOT try to fetch lists of all features
 - Do NOT query features by category
 - Do NOT list all pending features
+- Your feature is pre-assigned by the orchestrator - use `feature_get_by_id` to get details
 
-**You do NOT need to see all features.** The feature_get_next tool tells you exactly what to work on. Trust it.
+**You do NOT need to see all features.** Work on your assigned feature only.
 
 ---
 

.claude/templates/testing_prompt.template.md

@@ -40,17 +40,15 @@ chmod +x init.sh
 
 Otherwise, start servers manually.
 
-### STEP 3: CLAIM A FEATURE TO TEST
+### STEP 3: GET YOUR ASSIGNED FEATURE
 
-Atomically claim ONE passing feature for regression testing:
+Your feature has been pre-assigned by the orchestrator. Use `feature_get_by_id` to get the details:
 
 ```
-Use the feature_claim_for_testing tool
+Use the feature_get_by_id tool with feature_id={your_assigned_id}
 ```
 
-This atomically claims a random passing feature that:
-- Is not being worked on by coding agents
-- Is not already being tested by another testing agent
+The orchestrator has already claimed this feature for testing (set `testing_in_progress=true`).
 
 **CRITICAL:** You MUST call `feature_release_testing` when done, regardless of pass/fail.
 
@@ -157,9 +155,8 @@ echo "[Testing] Session complete - verified/fixed feature #{id}" >> claude-progr
 
 ### Feature Management
 - `feature_get_stats` - Get progress overview (passing/in_progress/total counts)
-- `feature_claim_for_testing` - **USE THIS** - Atomically claim a feature for testing
+- `feature_get_by_id` - Get your assigned feature details
 - `feature_release_testing` - **REQUIRED** - Release claim after testing (pass tested_ok=true/false)
-- `feature_get_for_regression` - (Legacy) Get random passing features without claiming
 - `feature_mark_failing` - Mark a feature as failing (when you find a regression)
 - `feature_mark_passing` - Mark a feature as passing (after fixing a regression)
 

agent.py

@@ -117,6 +117,7 @@ async def run_autonomous_agent(
     yolo_mode: bool = False,
     feature_id: Optional[int] = None,
     agent_type: Optional[str] = None,
+    testing_feature_id: Optional[int] = None,
 ) -> None:
     """
     Run the autonomous agent loop.
@@ -128,6 +129,7 @@ async def run_autonomous_agent(
         yolo_mode: If True, skip browser testing in coding agent prompts
         feature_id: If set, work only on this specific feature (used by orchestrator for coding agents)
         agent_type: Type of agent: "initializer", "coding", "testing", or None (auto-detect)
+        testing_feature_id: For testing agents, the pre-claimed feature ID to test
     """
     print("\n" + "=" * 70)
     print("  AUTONOMOUS CODING AGENT")
@@ -220,7 +222,7 @@ async def run_autonomous_agent(
         if agent_type == "initializer":
             prompt = get_initializer_prompt(project_dir)
         elif agent_type == "testing":
-            prompt = get_testing_prompt(project_dir)
+            prompt = get_testing_prompt(project_dir, testing_feature_id)
         elif feature_id:
             # Single-feature mode (used by orchestrator for coding agents)
             prompt = get_single_feature_prompt(feature_id, project_dir, yolo_mode)

autonomous_agent_demo.py

@@ -141,6 +141,13 @@ def parse_args() -> argparse.Namespace:
         help="Agent type (used by orchestrator to spawn specialized subprocesses)",
     )
 
+    parser.add_argument(
+        "--testing-feature-id",
+        type=int,
+        default=None,
+        help="Feature ID to regression test (used by orchestrator for testing agents)",
+    )
+
     # Testing agent configuration
     parser.add_argument(
         "--testing-ratio",
@@ -197,6 +204,7 @@ def main() -> None:
                     yolo_mode=args.yolo,
                     feature_id=args.feature_id,
                     agent_type=args.agent_type,
+                    testing_feature_id=args.testing_feature_id,
                 )
             )
         else:

client.py

@@ -54,21 +54,20 @@ def get_playwright_headless() -> bool:
 FEATURE_MCP_TOOLS = [
     # Core feature operations
     "mcp__features__feature_get_stats",
-    "mcp__features__feature_get_next",
-    "mcp__features__feature_claim_next",  # Atomic get+claim for parallel execution
-    "mcp__features__feature_get_for_regression",
+    "mcp__features__feature_get_by_id",  # Get assigned feature details
     "mcp__features__feature_mark_in_progress",
     "mcp__features__feature_mark_passing",
     "mcp__features__feature_mark_failing",  # Mark regression detected
     "mcp__features__feature_skip",
     "mcp__features__feature_create_bulk",
     "mcp__features__feature_create",
     "mcp__features__feature_clear_in_progress",
+    "mcp__features__feature_release_testing",  # Release testing claim
     # Dependency management
     "mcp__features__feature_add_dependency",
     "mcp__features__feature_remove_dependency",
     "mcp__features__feature_set_dependencies",
-    # Parallel execution support
+    # Query tools
     "mcp__features__feature_get_ready",
     "mcp__features__feature_get_blocked",
     "mcp__features__feature_get_graph",