From cce5a82a82fb79d38a849c74130ef1aecc95dff5 Mon Sep 17 00:00:00 2001
From: pyr0ball <pyroballpcs@gmail.com>
Date: Wed, 25 Feb 2026 08:29:56 -0800
Subject: [PATCH] =?UTF-8?q?feat:=20wizard=5Fgenerate=20=E2=80=94=20feedbac?=
 =?UTF-8?q?k=20+=20previous=5Fresult=20support=20for=20iterative=20refinem?=
 =?UTF-8?q?ent?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 scripts/task_runner.py    |  7 +++++
 tests/test_task_runner.py | 61 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 68 insertions(+)

diff --git a/scripts/task_runner.py b/scripts/task_runner.py
index 6c817d1..99c3000 100644
--- a/scripts/task_runner.py
+++ b/scripts/task_runner.py
@@ -110,6 +110,13 @@ def _run_wizard_generate(section: str, input_data: dict) -> str:
         return _re.sub(r"\{(\w+)\}", replacer, tmpl)
 
     prompt = _safe_format(template, {k: str(v) for k, v in input_data.items()})
+    # Append iterative refinement context if provided
+    previous_result = input_data.get("previous_result", "")
+    feedback = input_data.get("feedback", "")
+    if previous_result:
+        prompt += f"\n\n---\nPrevious output:\n{previous_result}"
+    if feedback:
+        prompt += f"\n\nUser feedback / requested changes:\n{feedback}\n\nPlease revise accordingly."
     from scripts.llm_router import LLMRouter
     return LLMRouter().complete(prompt)
 
diff --git a/tests/test_task_runner.py b/tests/test_task_runner.py
index e3de98c..8d28226 100644
--- a/tests/test_task_runner.py
+++ b/tests/test_task_runner.py
@@ -293,3 +293,64 @@ def test_wizard_generate_stores_result_as_json(tmp_path):
     payload = _json.loads(row[1])
     assert payload["section"] == "career_summary"
     assert payload["result"] == "Experienced Python developer."
+
+
+def test_wizard_generate_feedback_appended_to_prompt(tmp_path):
+    """feedback and previous_result fields in input_data are appended to the prompt."""
+    from unittest.mock import patch, MagicMock
+    db = tmp_path / "t.db"
+    from scripts.db import init_db, insert_task
+    init_db(db)
+
+    captured_prompts = []
+
+    def mock_complete(prompt):
+        captured_prompts.append(prompt)
+        return "Revised career summary."
+
+    import json as _json
+    params = _json.dumps({
+        "section": "career_summary",
+        "input": {
+            "resume_text": "10 years Python dev",
+            "previous_result": "Original summary text.",
+            "feedback": "Make it shorter and focus on leadership.",
+        }
+    })
+    task_id, _ = insert_task(db, "wizard_generate", 0, params=params)
+
+    with patch("scripts.llm_router.LLMRouter") as MockRouter:
+        MockRouter.return_value.complete.side_effect = mock_complete
+        from scripts.task_runner import _run_task
+        _run_task(db, task_id, "wizard_generate", 0, params=params)
+
+    assert len(captured_prompts) == 1
+    prompt_used = captured_prompts[0]
+    assert "Original summary text." in prompt_used
+    assert "Make it shorter and focus on leadership." in prompt_used
+    assert "Please revise accordingly." in prompt_used
+
+
+def test_wizard_generate_no_feedback_no_revision_block(tmp_path):
+    """When no feedback/previous_result provided, prompt has no revision block."""
+    from unittest.mock import patch
+    db = tmp_path / "t.db"
+    from scripts.db import init_db, insert_task
+    init_db(db)
+
+    captured_prompts = []
+
+    import json as _json
+    params = _json.dumps({
+        "section": "career_summary",
+        "input": {"resume_text": "5 years QA engineer"}
+    })
+    task_id, _ = insert_task(db, "wizard_generate", 0, params=params)
+
+    with patch("scripts.llm_router.LLMRouter") as MockRouter:
+        MockRouter.return_value.complete.side_effect = lambda p: (captured_prompts.append(p) or "Summary.")
+        from scripts.task_runner import _run_task
+        _run_task(db, task_id, "wizard_generate", 0, params=params)
+
+    assert "Please revise accordingly." not in captured_prompts[0]
+    assert "Previous output:" not in captured_prompts[0]