Merge pull request #423 from hnshah/ren/preserve-requested-quick-sources

tests/test_planner_v3.py

@@ -92,6 +92,41 @@ class PlannerV3Tests(unittest.TestCase):
         self.assertEqual(1, len(plan.subqueries))
         self.assertEqual(["reddit", "x"], plan.subqueries[0].sources)
 
+    def test_quick_mode_preserves_explicit_requested_sources(self):
+        raw = {
+            "intent": "product",
+            "freshness_mode": "balanced_recent",
+            "cluster_mode": "debate",
+            "subqueries": [
+                {
+                    "label": "primary",
+                    "search_query": "AI coding agents",
+                    "ranking_query": "What are people saying about AI coding agents?",
+                    "sources": ["reddit", "youtube", "grounding", "digg"],
+                    "weight": 1.0,
+                }
+            ],
+        }
+        plan = planner._sanitize_plan(
+            raw,
+            "AI coding agents",
+            ["reddit", "youtube", "grounding", "digg"],
+            ["reddit", "youtube", "grounding", "digg"],
+            "quick",
+        )
+        self.assertIn("digg", plan.subqueries[0].sources)
+
+    def test_quick_mode_preserves_explicit_requested_sources_in_fallback_plan(self):
+        plan = planner.plan_query(
+            topic="AI coding agents",
+            available_sources=["reddit", "youtube", "github"],
+            requested_sources=["reddit", "github"],
+            depth="quick",
+            provider=None,
+            model=None,
+        )
+        self.assertIn("github", plan.subqueries[0].sources)
+
     def test_default_comparison_uses_all_capable_sources(self):
         plan = planner.plan_query(
             topic="codex vs claude code",