[AISOS-2294] Add decompose draft review step before Jira task creation

.gitignore

@@ -7,6 +7,7 @@ specs/
 docs/github-app-setup.md
 
 # Python
+.mypy_cache/
 __pycache__/
 *.py[cod]
 *$py.class

CLAUDE.md

@@ -120,11 +120,16 @@ podman rm $(podman ps -a --filter name=forge- -q)
 
 ## Jira Comment Syntax
 
-| Prefix | Effect |
-|--------|--------|
-| `!` | Revision request — triggers regeneration with feedback |
+| Prefix / Command | Effect |
+|------------------|--------|
+| `!` | Revision request — triggers regeneration/revision with feedback |
 | `?` or `@forge ask` | Question — triggers Q&A answer |
 | `>option N` | RCA option selection (RCA Option Gate only) |
+| `/forge approve` | Approve draft (Epic plan or Tasks) to provision tickets |
+| `/forge remove ID` | Remove proposed draft item by ID and re-sequence |
+| `/forge exclude ID` | Toggle exclusion of proposed draft item by ID |
+| `/forge update ID key=val` | Update fields (`summary`, `description`, `repo`) of proposed draft item |
+| `/forge add key=val` | Add a new proposed item to draft |
 | _(no prefix)_ | Informational — workflow ignores it |
 
 ## GitHub PR Comment Commands

docs/developer-guide.md

@@ -858,6 +858,18 @@ curl -X POST http://localhost:8000/api/v1/webhooks/github \
 | `/forge skip-gate <name>` | Skip named CI check | CI stages |
 | `/forge unskip-gate <name>` | Remove a skip | CI stages |
 
+### Jira comment commands
+
+These commands are used on the parent Jira ticket during the draft review stages (Epic Plan and Tasks).
+
+| Command | Effect | Active at |
+|---------|--------|-----------|
+| `/forge approve` | Approve draft, provision sub-tickets, and delete draft attachment | `plan_approval_gate`, `task_approval_gate` |
+| `/forge remove <ID>` | Remove a draft item by local sequential ID | `plan_approval_gate`, `task_approval_gate` |
+| `/forge exclude <ID>` | Toggle the exclusion flag of a draft item | `plan_approval_gate`, `task_approval_gate` |
+| `/forge update <ID> key=val` | Update fields (`summary`, `description`, `repo`) of a draft item | `plan_approval_gate`, `task_approval_gate` |
+| `/forge add key=val` | Add a new proposed item to the draft | `plan_approval_gate`, `task_approval_gate` |
+
 ### Jira labels
 
 | Label | Meaning |

docs/guide/feature-workflow.md

@@ -60,22 +60,29 @@ Forge generates a behavioral specification from the approved PRD, typically usin
 
 Forge breaks the feature into logical epics — high-level areas of work that map to implementation phases.
 
-**Human action:** Review the epic plan. You have four options at this stage:
+By default, Forge uses an interactive **Draft Review Flow** at this stage (unless YOLO mode is active):
+1. Instead of creating Jira tickets immediately, Forge serializes the proposed epics into `forge-stories-draft.json` and uploads it as an attachment on the Feature ticket.
+2. Forge posts a markdown table comment on the Feature ticket outlining the proposed Epics.
+3. The workflow pauses at `plan_approval_gate`.
 
-| Action | How |
-|--------|-----|
-| Approve | Change label to `forge:plan-approved` |
-| Ask a question | Comment with `?` prefix — Forge answers without re-decomposing |
-| Revise one epic | `!` comment on the **specific epic sub-ticket** — Forge updates only that epic |
-| Redo the full decomposition | `!` comment on the **feature ticket** — Forge regenerates all epics with your feedback |
+**Human action:** Review the epic plan draft. You have several options at this stage:
+
+| Action | How | Description |
+|--------|-----|-------------|
+| **Approve** | Comment `/forge approve` OR set label to `forge:plan-approved` | Forge provisions the Epic sub-tickets on Jira from the draft, deletes the draft attachment, and advances to Task Generation. |
+| **Direct Edit** | Use `/forge` commands (e.g. `/forge update`, `/forge remove`, etc.) | Directly modify the draft attachment and regenerate the proposal comment. See [Jira Labels & Comments](labels.md) for a list of commands. |
+| **Ask a question** | Comment with `?` prefix or `@forge ask` | Forge answers your question without regenerating the draft. |
+| **Request revisions** | Comment with `!` prefix followed by your feedback | Forge uses LLM assistance to revise the entire draft JSON and update the proposal comment with your feedback. |
+
+If `forge:yolo` mode is active, the draft review is bypassed. Epics are created in Jira immediately, and the workflow automatically proceeds to Task Generation.
 
 ```mermaid
 flowchart TD
     Gate([plan_approval_gate])
-    Gate -->|forge:plan-approved| Next[Generate Tasks]
+    Gate -->|forge:plan-approved or /forge approve| Next[Generate Tasks]
     Gate -->|"? on feature ticket"| QA[Answer Question]
     Gate -->|"! on feature ticket"| Regen[Regenerate All Epics]
-    Gate -->|"! on epic sub-ticket"| Update[Update Single Epic]
+    Gate -->|"/forge update/remove/exclude/add"| Update[Modify Draft]
     QA --> Gate
     Regen --> Gate
     Update --> Gate
@@ -87,22 +94,29 @@ flowchart TD
 
 Forge generates granular implementation tasks scoped to individual repositories. Each task is sized to fit in a single container execution pass.
 
-**Human action:** Review the tasks. You have four options at this stage:
+By default, Forge uses an interactive **Draft Review Flow** at this stage (unless YOLO mode is active):
+1. Instead of creating Jira tickets immediately, Forge serializes the proposed tasks into `forge-tasks-draft.json` and uploads it as an attachment on the Feature ticket.
+2. Forge posts a markdown table comment on the Feature ticket outlining the proposed Tasks.
+3. The workflow pauses at `task_approval_gate`.
+
+**Human action:** Review the task draft. You have several options at this stage:
+
+| Action | How | Description |
+|--------|-----|-------------|
+| **Approve** | Comment `/forge approve` OR set label to `forge:task-approved` | Forge provisions the Task sub-tickets on Jira from the draft, deletes the draft attachment, and advances to Implementation. |
+| **Direct Edit** | Use `/forge` commands (e.g. `/forge update`, `/forge remove`, etc.) | Directly modify the draft attachment and regenerate the proposal comment. See [Jira Labels & Comments](labels.md) for a list of commands. |
+| **Ask a question** | Comment with `?` prefix or `@forge ask` | Forge answers your question without regenerating the draft. |
+| **Request revisions** | Comment with `!` prefix followed by your feedback | Forge uses LLM assistance to revise the entire draft JSON and update the proposal comment with your feedback. |
 
-| Action | How |
-|--------|-----|
-| Approve | Change label to `forge:task-approved` |
-| Ask a question | Comment with `?` prefix — Forge answers without regenerating |
-| Revise one task | `!` comment on the **specific task sub-ticket** — Forge updates only that task |
-| Regenerate all tasks | `!` comment on the **feature or epic ticket** — Forge regenerates the full task list with your feedback |
+If `forge:yolo` mode is active, the draft review is bypassed. Tasks are created in Jira immediately, and the workflow automatically proceeds to Implementation.
 
 ```mermaid
 flowchart TD
     Gate([task_approval_gate])
-    Gate -->|forge:task-approved| Next[Implement Tasks]
+    Gate -->|forge:task-approved or /forge approve| Next[Implement Tasks]
     Gate -->|"? on ticket"| QA[Answer Question]
     Gate -->|"! on feature/epic"| Regen[Regenerate All Tasks]
-    Gate -->|"! on task sub-ticket"| Update[Update Single Task]
+    Gate -->|"/forge update/remove/exclude/add"| Update[Modify Draft]
     QA --> Gate
     Regen --> Gate
     Update --> Gate

docs/guide/labels.md

@@ -51,13 +51,33 @@ Standalone Tasks and Epics can be processed with the standard `forge:managed` la
 
 **Starting a workflow:** Create a Jira issue and add `forge:managed`. Forge detects the issue type and begins the appropriate pipeline: Feature/Story, Bug, or standalone Task/Epic takeover.
 
-**Approving a stage:** When Forge posts a PRD, spec, or other artifact, it sets the `forge:*-pending` label. Change it to `forge:*-approved` to advance the workflow. Do not add the approved label manually before Forge posts — it won't be recognized until the pending state is set.
+**Approving a stage:** When Forge posts an artifact (such as a PRD or Spec), it sets the `forge:*-pending` label. You can approve it by changing the label to `forge:*-approved` to advance the workflow. For draft-based stages (Epic Plan and Tasks), you can also approve by commenting `/forge approve` on the ticket.
 
-**Requesting revisions:** Start a comment with `!` followed by your feedback. Forge regenerates the artifact and resets the pending label.
+**Interactive Draft Review:** For Epic Decomposition and Task Generation stages, Forge uses a draft-based review flow by default (unless `forge:yolo` mode is active).
+1. Instead of creating sub-tickets immediately, Forge serializes the proposed items into a JSON draft file (`forge-stories-draft.json` or `forge-tasks-draft.json`) and uploads it as a Jira attachment.
+2. Forge posts a formatted markdown table comment on the ticket detailing the proposed plan.
+3. While the stage is pending, you can modify the draft directly using **Jira comment commands** (see below) or request a natural language revision.
+4. Once you approve (via `/forge approve` or setting the approved label), Forge downloads the draft, provisions the actual Jira tickets from it, and deletes the draft attachment.
 
-**Asking questions:** Start a comment with `?` or `@forge ask`. Forge answers without advancing or regenerating.
+### Jira Comment Commands
 
-**Informational comments:** Comments without a recognized prefix (`!`, `?`, `@forge ask`, `>option`) are ignored by the workflow — use them for team discussion without triggering Forge.
+For stages using the draft-based review flow (Epic Plan and Tasks), you can post comments on the parent ticket with the following commands:
+
+| Command | Description | Example |
+|---------|-------------|---------|
+| `/forge approve` | Approve the draft, provision all non-excluded items as Jira tickets, and delete the draft attachment. | `/forge approve` |
+| `/forge remove <ID>` | Remove a draft item by its local sequential ID. Remaining items are automatically re-sequenced. | `/forge remove 3` |
+| `/forge exclude <ID>` | Toggle the exclusion flag of a draft item. Excluded items are skipped during ticket provisioning. | `/forge exclude 2` |
+| `/forge update <ID> key=val` | Update fields of a draft item (supported keys: `summary`, `description`, `repo`). | `/forge update 1 repo="my-org/custom-repo"` |
+| `/forge add key=val` | Add a new proposed item to the draft. | `/forge add summary="New Story" repo="my-org/repo"` |
+
+*Note: Successful command/revision comments are automatically edited by Forge to prepend `✅`. If a command or revision fails, Forge posts a comment detailing the error with a leading `❌`.*
+
+**Requesting revisions:** Start a comment with `!` followed by your feedback (e.g., `! update the repositories to use the new service`). For standard artifacts, Forge regenerates them. For drafts, Forge uses LLM assistance to revise the draft JSON attachment and update the proposed plan table.
+
+**Asking questions:** Start a comment with `?` or `@forge ask`. Forge answers without advancing or regenerating/modifying the drafts.
+
+**Informational comments:** Comments without a recognized prefix (such as `!`, `?`, `@forge ask`, `>option`, or `/forge`) are ignored by the workflow — use them for team discussion without triggering Forge.
 
 **Handling failures:** When `forge:blocked` appears, read the Forge comment for the error. Fix the underlying issue if needed, then add `forge:retry`.
 

src/forge/config.py

@@ -339,6 +339,11 @@ def ignored_ci_checks(self) -> list[str]:
         default=0.5, description="Webhook acknowledgment timeout in seconds"
     )
 
+    yolo_mode: bool = Field(
+        default=False,
+        description="Autonomous mode - skip all artifact approval gates",
+    )
+
     # Container Configuration
     container_image: str = Field(
         default="localhost/forge-dev:latest",

src/forge/integrations/agents/agent.py

@@ -1272,6 +1272,89 @@ async def answer_question(
         logger.info(f"Generated answer ({len(result)} chars)")
         return result.strip() if result else ""
 
+    async def revise_draft_with_feedback(
+        self,
+        draft_content: str,
+        feedback: str,
+        context: dict[str, Any] | None = None,
+    ) -> str:
+        """Revise draft content based on user feedback.
+
+        Uses the 'revision-draft' prompt template to guide the LLM to output
+        the revised draft JSON.
+
+        Args:
+            draft_content: The current draft JSON content.
+            feedback: Natural language feedback.
+            context: Optional context from the workflow state.
+
+        Returns:
+            The updated draft JSON string.
+        """
+        from langchain_core.output_parsers import StrOutputParser
+
+        # Format context into a readable string/JSON
+        context_str = json.dumps(context, indent=2) if context else "None provided"
+
+        # Load the prompt template using project's load_prompt
+        prompt_text = load_prompt(
+            "revision-draft",
+            draft_content=draft_content,
+            feedback=feedback,
+            context=context_str,
+        )
+
+        model = self._create_model()
+        chain = model | StrOutputParser()
+
+        logger.info("Revising draft using direct LangChain model chain")
+        response = await chain.ainvoke(prompt_text)
+
+        # Strip preamble/narration and validate as JSON
+        cleaned_text = response.strip()
+
+        # Check markdown code blocks first
+        pattern = r"```(?:json)?\s*([\s\S]*?)\s*```"
+        match = re.search(pattern, cleaned_text)
+        if match:
+            cleaned_text = match.group(1).strip()
+        else:
+            # If no code block, look for the JSON object/list boundary
+            # Find the first occurrence of '{' or '[' and the last of '}' or ']'
+            start_brace = cleaned_text.find("{")
+            start_bracket = cleaned_text.find("[")
+
+            # Determine which starts first
+            start_idx = -1
+            if start_brace != -1 and start_bracket != -1:
+                start_idx = min(start_brace, start_bracket)
+            elif start_brace != -1:
+                start_idx = start_brace
+            elif start_bracket != -1:
+                start_idx = start_bracket
+
+            if start_idx != -1:
+                # Find the last brace or bracket
+                end_brace = cleaned_text.rfind("}")
+                end_bracket = cleaned_text.rfind("]")
+                end_idx = max(end_brace, end_bracket)
+
+                if end_idx > start_idx:
+                    cleaned_text = cleaned_text[start_idx : end_idx + 1].strip()
+
+        try:
+            parsed_json = json.loads(cleaned_text)
+            validated_json_str = json.dumps(parsed_json, indent=2)
+            logger.info(
+                f"Successfully revised draft and validated JSON ({len(validated_json_str)} chars)"
+            )
+            return validated_json_str
+        except json.JSONDecodeError as e:
+            logger.error(f"Failed to parse LLM response as valid JSON: {e}\nResponse: {response}")
+            raise ValueError(
+                f"Failed to parse LLM response as valid JSON: {e}\nResponse: {response}"
+            )
+
     async def close(self) -> None:
         """Close the agent and cleanup resources."""
         pass