@autobe/agent 0.30.0-dev.20260315 → 0.30.0

package/src/orchestrate/analyze/histories/transformAnalyzeWriteSectionHistory.ts

@@ -1,191 +1,191 @@
-import {
-  AutoBeAnalyze,
-  AutoBeAnalyzeModuleSection,
-  AutoBeAnalyzeScenarioEvent,
-  AutoBeAnalyzeUnitSection,
-  AutoBeAnalyzeWriteModuleEvent,
-  AutoBeAnalyzeWriteUnitEvent,
-} from "@autobe/interface";
-import { StringUtil } from "@autobe/utils";
-import { v7 } from "uuid";
-
-import { AutoBeSystemPromptConstant } from "../../../constants/AutoBeSystemPromptConstant";
-import { AutoBeContext } from "../../../context/AutoBeContext";
-import { IAutoBeOrchestrateHistory } from "../../../structures/IAutoBeOrchestrateHistory";
-import { AutoBePreliminaryController } from "../../common/AutoBePreliminaryController";
-import {
-  FixedAnalyzeTemplateFeature,
-  buildFixedAnalyzeCanonicalSourceContent,
-  buildFixedAnalyzeExpandedTemplate,
-} from "../structures/FixedAnalyzeTemplate";
-
-export const transformAnalyzeWriteSectionHistory = (
-  ctx: AutoBeContext,
-  props: {
-    scenario: AutoBeAnalyzeScenarioEvent;
-    file: AutoBeAnalyze.IFileScenario;
-    moduleEvent: AutoBeAnalyzeWriteModuleEvent;
-    unitEvent: AutoBeAnalyzeWriteUnitEvent;
-    allUnitEvents: AutoBeAnalyzeWriteUnitEvent[];
-    moduleIndex: number;
-    unitIndex: number;
-    feedback?: string;
-    preliminary: null | AutoBePreliminaryController<"previousAnalysisSections">;
-  },
-): IAutoBeOrchestrateHistory => {
-  const moduleSection: AutoBeAnalyzeModuleSection | undefined =
-    props.moduleEvent.moduleSections[props.moduleIndex];
-  const unitSection: AutoBeAnalyzeUnitSection | undefined =
-    props.unitEvent.unitSections[props.unitIndex];
-
-  // Find the file template for scope context (using expanded template for conditional modules)
-  const expandedTemplate = buildFixedAnalyzeExpandedTemplate(
-    (props.scenario.features ?? []) as FixedAnalyzeTemplateFeature[],
-  );
-  const fileTemplate = expandedTemplate.find(
-    (t) => t.filename === props.file.filename,
-  );
-
-  // Build scope summary for all 6 files
-  const fileScopeSummary = expandedTemplate
-    .map(
-      (t) =>
-        `- **${t.filename}**: ${t.description}${t.filename === props.file.filename ? " ← **YOU ARE HERE**" : ""}`,
-    )
-    .join("\n");
-
-  // Build canonical source declaration
-  const canonicalSourceDeclaration = buildFixedAnalyzeCanonicalSourceContent();
-
-  return {
-    histories: [
-      {
-        id: v7(),
-        created_at: new Date().toISOString(),
-        type: "systemMessage",
-        text: AutoBeSystemPromptConstant.ANALYZE_WRITE_SECTION,
-      },
-      ...(props.preliminary?.getHistories() ?? []),
-      {
-        id: v7(),
-        created_at: new Date().toISOString(),
-        type: "assistantMessage",
-        text: StringUtil.trim`
-        ## Language
-
-        The language of the document is ${JSON.stringify(props.scenario.language ?? "en-US")}.
-
-        ## Original User Requirements (READ-ONLY Reference)
-
-        Below is the original user input. ALL your sections MUST be traceable to these requirements.
-        Do NOT invent features, constraints, or behaviors not stated or directly implied below.
-
-        ${ctx
-          .histories()
-          .filter((h) => h.type === "userMessage")
-          .flatMap((h) =>
-            h.type === "userMessage"
-              ? h.contents.filter((c) => c.type === "text").map((c) => c.text)
-              : [],
-          )
-          .join("\n\n---\n\n")}
-
-        ## AUTHORITATIVE Scope Reference
-
-        **Service Prefix**: ${props.scenario.prefix}
-        **Actors**: ${JSON.stringify(props.scenario.actors.map((a) => ({ name: a.name, kind: a.kind })))}
-        **Domain Entities**:
-        ${props.scenario.entities.map((e) => `- **${e.name}**: ${e.attributes.slice(0, 5).join(", ")}${e.relationships?.length ? ` | ${e.relationships.join(", ")}` : ""}`).join("\n")}
-
-        **CRITICAL**: You MUST NOT reference entities, actors, or features not listed above.
-        If actors are [guest, member], do NOT introduce "admin" or "moderator".
-        If entities are [Todo, User], do NOT introduce "Project" or "Label".
-
-        ## File Scope Context
-
-        **Current File**: ${props.file.filename}
-        **File Scope**: ${fileTemplate?.description ?? props.file.reason}
-
-        ### All SRS Files (Fixed 6-File Structure)
-
-        ${fileScopeSummary}
-
-        ### Canonical Source Declaration
-
-        ${canonicalSourceDeclaration}
-
-        ## Document Context
-
-        **Document Title**: ${props.moduleEvent.title}
-        **Document Summary**: ${props.moduleEvent.summary}
-
-        ## Parent Module Section
-
-        **Module Index**: ${props.moduleIndex}
-        **Title**: ${moduleSection?.title ?? "Unknown"}
-        **Purpose**: ${moduleSection?.purpose ?? "Unknown"}
-        **Content**: ${moduleSection?.content ?? "No content"}
-
-        ## Parent Unit Section
-
-        **Unit Index**: ${props.unitIndex}
-        **Title**: ${unitSection?.title ?? "Unknown"}
-        **Purpose**: ${unitSection?.purpose ?? "Unknown"}
-        **Content**: ${unitSection?.content ?? "Unknown"}
-
-        ## CRITICAL: Value Consistency
-
-        **You MUST use the EXACT same values defined in parent sections above.**
-        If the parent section says "10MB file limit", you MUST use 10MB, not 25MB or 5MB.
-        If the parent section says "5 attachments maximum", you MUST use 5, not 10.
-        Any deviation will cause the review to REJECT your output.
-
-        ## CRITICAL: No Duplicate Content
-
-        Each section MUST contain unique information:
-        - Do NOT restate requirements already implied by sibling sections' keywords
-        - Define each Entity.attribute ONLY in the first section that introduces it
-        - Subsequent sections referencing the same attribute should use: "(defined in [Section Name])"
-        - Do NOT duplicate state transitions, operations, or permission rules across sections
-
-        ## Keywords to Address
-
-        You MUST create sections that address these keywords:
-
-        ${unitSection?.keywords?.map((kw, i) => `${i + 1}. ${kw}`)?.join("\n") ?? "No keywords"}
-
-        ## Sibling Unit Summaries (do NOT duplicate this content)
-
-        ${props.allUnitEvents
-          .flatMap((unitEvt, mi) =>
-            (unitEvt?.unitSections ?? [])
-              .map((unit, ui) => {
-                if (mi === props.moduleIndex && ui === props.unitIndex)
-                  return null;
-                return `- **Module ${mi + 1} > ${unit.title}**: ${(unit.keywords ?? []).slice(0, 5).join(", ")}`;
-              })
-              .filter(Boolean),
-          )
-          .join("\n")}
-
-        ## Your Task
-
-        Create detailed sections (#### level) that address ALL the keywords above.
-        Use natural language for all requirements.
-        ${
-          props.feedback
-            ? `
-        ## Previous Attempt Feedback
-
-        Your previous attempt was rejected. Please address these issues:
-
-        ${props.feedback}
-        `
-            : ""
-        }
-      `,
-      },
-    ],
-    userMessage: `Create detailed sections for "${unitSection?.title ?? "Unknown"}".`,
-  };
-};
+import {
+  AutoBeAnalyze,
+  AutoBeAnalyzeModuleSection,
+  AutoBeAnalyzeScenarioEvent,
+  AutoBeAnalyzeUnitSection,
+  AutoBeAnalyzeWriteModuleEvent,
+  AutoBeAnalyzeWriteUnitEvent,
+} from "@autobe/interface";
+import { StringUtil } from "@autobe/utils";
+import { v7 } from "uuid";
+
+import { AutoBeSystemPromptConstant } from "../../../constants/AutoBeSystemPromptConstant";
+import { AutoBeContext } from "../../../context/AutoBeContext";
+import { IAutoBeOrchestrateHistory } from "../../../structures/IAutoBeOrchestrateHistory";
+import { AutoBePreliminaryController } from "../../common/AutoBePreliminaryController";
+import {
+  FixedAnalyzeTemplateFeature,
+  buildFixedAnalyzeCanonicalSourceContent,
+  buildFixedAnalyzeExpandedTemplate,
+} from "../structures/FixedAnalyzeTemplate";
+
+export const transformAnalyzeWriteSectionHistory = (
+  ctx: AutoBeContext,
+  props: {
+    scenario: AutoBeAnalyzeScenarioEvent;
+    file: AutoBeAnalyze.IFileScenario;
+    moduleEvent: AutoBeAnalyzeWriteModuleEvent;
+    unitEvent: AutoBeAnalyzeWriteUnitEvent;
+    allUnitEvents: AutoBeAnalyzeWriteUnitEvent[];
+    moduleIndex: number;
+    unitIndex: number;
+    feedback?: string;
+    preliminary: null | AutoBePreliminaryController<"previousAnalysisSections">;
+  },
+): IAutoBeOrchestrateHistory => {
+  const moduleSection: AutoBeAnalyzeModuleSection | undefined =
+    props.moduleEvent.moduleSections[props.moduleIndex];
+  const unitSection: AutoBeAnalyzeUnitSection | undefined =
+    props.unitEvent.unitSections[props.unitIndex];
+
+  // Find the file template for scope context (using expanded template for conditional modules)
+  const expandedTemplate = buildFixedAnalyzeExpandedTemplate(
+    (props.scenario.features ?? []) as FixedAnalyzeTemplateFeature[],
+  );
+  const fileTemplate = expandedTemplate.find(
+    (t) => t.filename === props.file.filename,
+  );
+
+  // Build scope summary for all 6 files
+  const fileScopeSummary = expandedTemplate
+    .map(
+      (t) =>
+        `- **${t.filename}**: ${t.description}${t.filename === props.file.filename ? " ← **YOU ARE HERE**" : ""}`,
+    )
+    .join("\n");
+
+  // Build canonical source declaration
+  const canonicalSourceDeclaration = buildFixedAnalyzeCanonicalSourceContent();
+
+  return {
+    histories: [
+      {
+        id: v7(),
+        created_at: new Date().toISOString(),
+        type: "systemMessage",
+        text: AutoBeSystemPromptConstant.ANALYZE_WRITE_SECTION,
+      },
+      ...(props.preliminary?.getHistories() ?? []),
+      {
+        id: v7(),
+        created_at: new Date().toISOString(),
+        type: "assistantMessage",
+        text: StringUtil.trim`
+        ## Language
+
+        The language of the document is ${JSON.stringify(props.scenario.language ?? "en-US")}.
+
+        ## Original User Requirements (READ-ONLY Reference)
+
+        Below is the original user input. ALL your sections MUST be traceable to these requirements.
+        Do NOT invent features, constraints, or behaviors not stated or directly implied below.
+
+        ${ctx
+          .histories()
+          .filter((h) => h.type === "userMessage")
+          .flatMap((h) =>
+            h.type === "userMessage"
+              ? h.contents.filter((c) => c.type === "text").map((c) => c.text)
+              : [],
+          )
+          .join("\n\n---\n\n")}
+
+        ## AUTHORITATIVE Scope Reference
+
+        **Service Prefix**: ${props.scenario.prefix}
+        **Actors**: ${JSON.stringify(props.scenario.actors.map((a) => ({ name: a.name, kind: a.kind })))}
+        **Domain Entities**:
+        ${props.scenario.entities.map((e) => `- **${e.name}**: ${e.attributes.slice(0, 5).join(", ")}${e.relationships?.length ? ` | ${e.relationships.join(", ")}` : ""}`).join("\n")}
+
+        **CRITICAL**: You MUST NOT reference entities, actors, or features not listed above.
+        If actors are [guest, member], do NOT introduce "admin" or "moderator".
+        If entities are [Todo, User], do NOT introduce "Project" or "Label".
+
+        ## File Scope Context
+
+        **Current File**: ${props.file.filename}
+        **File Scope**: ${fileTemplate?.description ?? props.file.reason}
+
+        ### All SRS Files (Fixed 6-File Structure)
+
+        ${fileScopeSummary}
+
+        ### Canonical Source Declaration
+
+        ${canonicalSourceDeclaration}
+
+        ## Document Context
+
+        **Document Title**: ${props.moduleEvent.title}
+        **Document Summary**: ${props.moduleEvent.summary}
+
+        ## Parent Module Section
+
+        **Module Index**: ${props.moduleIndex}
+        **Title**: ${moduleSection?.title ?? "Unknown"}
+        **Purpose**: ${moduleSection?.purpose ?? "Unknown"}
+        **Content**: ${moduleSection?.content ?? "No content"}
+
+        ## Parent Unit Section
+
+        **Unit Index**: ${props.unitIndex}
+        **Title**: ${unitSection?.title ?? "Unknown"}
+        **Purpose**: ${unitSection?.purpose ?? "Unknown"}
+        **Content**: ${unitSection?.content ?? "Unknown"}
+
+        ## CRITICAL: Value Consistency
+
+        **You MUST use the EXACT same values defined in parent sections above.**
+        If the parent section says "10MB file limit", you MUST use 10MB, not 25MB or 5MB.
+        If the parent section says "5 attachments maximum", you MUST use 5, not 10.
+        Any deviation will cause the review to REJECT your output.
+
+        ## CRITICAL: No Duplicate Content
+
+        Each section MUST contain unique information:
+        - Do NOT restate requirements already implied by sibling sections' keywords
+        - Define each Entity.attribute ONLY in the first section that introduces it
+        - Subsequent sections referencing the same attribute should use: "(defined in [Section Name])"
+        - Do NOT duplicate state transitions, operations, or permission rules across sections
+
+        ## Keywords to Address
+
+        You MUST create sections that address these keywords:
+
+        ${unitSection?.keywords?.map((kw, i) => `${i + 1}. ${kw}`)?.join("\n") ?? "No keywords"}
+
+        ## Sibling Unit Summaries (do NOT duplicate this content)
+
+        ${props.allUnitEvents
+          .flatMap((unitEvt, mi) =>
+            (unitEvt?.unitSections ?? [])
+              .map((unit, ui) => {
+                if (mi === props.moduleIndex && ui === props.unitIndex)
+                  return null;
+                return `- **Module ${mi + 1} > ${unit.title}**: ${(unit.keywords ?? []).slice(0, 5).join(", ")}`;
+              })
+              .filter(Boolean),
+          )
+          .join("\n")}
+
+        ## Your Task
+
+        Create detailed sections (#### level) that address ALL the keywords above.
+        Use natural language for all requirements.
+        ${
+          props.feedback
+            ? `
+        ## Previous Attempt Feedback
+
+        Your previous attempt was rejected. Please address these issues:
+
+        ${props.feedback}
+        `
+            : ""
+        }
+      `,
+      },
+    ],
+    userMessage: `Create detailed sections for "${unitSection?.title ?? "Unknown"}".`,
+  };
+};

package/src/orchestrate/analyze/histories/transformAnalyzeWriteSectionPatchHistory.ts

@@ -1,122 +1,122 @@
-import {
-  AutoBeAnalyze,
-  AutoBeAnalyzeModuleSection,
-  AutoBeAnalyzeScenarioEvent,
-  AutoBeAnalyzeUnitSection,
-  AutoBeAnalyzeWriteModuleEvent,
-  AutoBeAnalyzeWriteSectionEvent,
-  AutoBeAnalyzeWriteUnitEvent,
-} from "@autobe/interface";
-import { StringUtil } from "@autobe/utils";
-import { v7 } from "uuid";
-
-import { AutoBeSystemPromptConstant } from "../../../constants/AutoBeSystemPromptConstant";
-import { AutoBeContext } from "../../../context/AutoBeContext";
-import { IAutoBeOrchestrateHistory } from "../../../structures/IAutoBeOrchestrateHistory";
-import { AutoBePreliminaryController } from "../../common/AutoBePreliminaryController";
-
-export const transformAnalyzeWriteSectionPatchHistory = (
-  _ctx: AutoBeContext,
-  props: {
-    scenario: AutoBeAnalyzeScenarioEvent;
-    file: AutoBeAnalyze.IFileScenario;
-    moduleEvent: AutoBeAnalyzeWriteModuleEvent;
-    unitEvent: AutoBeAnalyzeWriteUnitEvent;
-    moduleIndex: number;
-    unitIndex: number;
-    previousSectionEvent: AutoBeAnalyzeWriteSectionEvent;
-    feedback: string;
-    preliminary: null | AutoBePreliminaryController<"previousAnalysisSections">;
-    sectionIndices?: number[] | null;
-  },
-): IAutoBeOrchestrateHistory => {
-  const moduleSection: AutoBeAnalyzeModuleSection | undefined =
-    props.moduleEvent.moduleSections[props.moduleIndex];
-  const unitSection: AutoBeAnalyzeUnitSection | undefined =
-    props.unitEvent.unitSections[props.unitIndex];
-
-  const hasSectionTargets =
-    props.sectionIndices != null && props.sectionIndices.length > 0;
-  const targetSet = hasSectionTargets ? new Set(props.sectionIndices) : null;
-
-  const previousOutputBlock = hasSectionTargets
-    ? `The following sections were generated. Sections marked [NEEDS FIX] must be corrected.
-Sections marked [APPROVED] must be returned EXACTLY as-is, character-for-character.
-
-${props.previousSectionEvent.sectionSections
-  .map(
-    (s, i) => `
-### Section ${i + 1}: ${s.title} ${targetSet!.has(i) ? "[NEEDS FIX]" : "[APPROVED - DO NOT MODIFY]"}
-
-${s.content}
-`,
-  )
-  .join("\n---\n")}`
-    : `The following sections were generated but REJECTED by review:
-
-${props.previousSectionEvent.sectionSections
-  .map(
-    (s, i) => `
-### Section ${i + 1}: ${s.title}
-
-${s.content}
-`,
-  )
-  .join("\n---\n")}`;
-
-  const taskBlock = hasSectionTargets
-    ? `Fix ONLY sections marked [NEEDS FIX] (section indices: ${props.sectionIndices!.map((i) => i + 1).join(", ")}).
-Return ALL sections (both fixed and unchanged) in the same sectionSections format.
-Sections marked [APPROVED] MUST be returned EXACTLY as they appear above -- do NOT modify them.`
-    : `Fix ONLY the issues identified above. Return ALL sections
-(both fixed and unchanged) in the same sectionSections format.
-Do NOT rewrite sections that were not flagged in the feedback.`;
-
-  return {
-    histories: [
-      {
-        id: v7(),
-        created_at: new Date().toISOString(),
-        type: "systemMessage",
-        text: AutoBeSystemPromptConstant.ANALYZE_WRITE_SECTION_PATCH,
-      },
-      ...(props.preliminary?.getHistories() ?? []),
-      {
-        id: v7(),
-        created_at: new Date().toISOString(),
-        type: "assistantMessage",
-        text: StringUtil.trim`
-        ## Parent Context
-
-        **Module**: ${moduleSection?.title ?? "Unknown"}
-        **Unit**: ${unitSection?.title ?? "Unknown"}
-        **Purpose**: ${unitSection?.purpose ?? "Unknown"}
-
-        ## AUTHORITATIVE Scope Reference
-
-        **Service Prefix**: ${props.scenario.prefix}
-        **Actors**: ${JSON.stringify(props.scenario.actors.map((a) => ({ name: a.name, kind: a.kind })))}
-        **Domain Entities**:
-        ${props.scenario.entities.map((e) => `- **${e.name}**: ${e.attributes.slice(0, 5).join(", ")}${e.relationships?.length ? ` | ${e.relationships.join(", ")}` : ""}`).join("\n")}
-
-        **CRITICAL**: You MUST NOT reference entities, actors, or features not listed above.
-
-        ## Previous Output
-
-        ${previousOutputBlock}
-
-        ## Review Feedback
-
-        The review REJECTED the above output for these reasons:
-
-        ${props.feedback}
-
-        ## Your Task
-
-        ${taskBlock}
-        `,
-      },
-    ],
-    userMessage: `Fix the rejected sections for "${unitSection?.title ?? "Unknown"}" based on the review feedback above.`,
-  };
-};
+import {
+  AutoBeAnalyze,
+  AutoBeAnalyzeModuleSection,
+  AutoBeAnalyzeScenarioEvent,
+  AutoBeAnalyzeUnitSection,
+  AutoBeAnalyzeWriteModuleEvent,
+  AutoBeAnalyzeWriteSectionEvent,
+  AutoBeAnalyzeWriteUnitEvent,
+} from "@autobe/interface";
+import { StringUtil } from "@autobe/utils";
+import { v7 } from "uuid";
+
+import { AutoBeSystemPromptConstant } from "../../../constants/AutoBeSystemPromptConstant";
+import { AutoBeContext } from "../../../context/AutoBeContext";
+import { IAutoBeOrchestrateHistory } from "../../../structures/IAutoBeOrchestrateHistory";
+import { AutoBePreliminaryController } from "../../common/AutoBePreliminaryController";
+
+export const transformAnalyzeWriteSectionPatchHistory = (
+  _ctx: AutoBeContext,
+  props: {
+    scenario: AutoBeAnalyzeScenarioEvent;
+    file: AutoBeAnalyze.IFileScenario;
+    moduleEvent: AutoBeAnalyzeWriteModuleEvent;
+    unitEvent: AutoBeAnalyzeWriteUnitEvent;
+    moduleIndex: number;
+    unitIndex: number;
+    previousSectionEvent: AutoBeAnalyzeWriteSectionEvent;
+    feedback: string;
+    preliminary: null | AutoBePreliminaryController<"previousAnalysisSections">;
+    sectionIndices?: number[] | null;
+  },
+): IAutoBeOrchestrateHistory => {
+  const moduleSection: AutoBeAnalyzeModuleSection | undefined =
+    props.moduleEvent.moduleSections[props.moduleIndex];
+  const unitSection: AutoBeAnalyzeUnitSection | undefined =
+    props.unitEvent.unitSections[props.unitIndex];
+
+  const hasSectionTargets =
+    props.sectionIndices != null && props.sectionIndices.length > 0;
+  const targetSet = hasSectionTargets ? new Set(props.sectionIndices) : null;
+
+  const previousOutputBlock = hasSectionTargets
+    ? `The following sections were generated. Sections marked [NEEDS FIX] must be corrected.
+Sections marked [APPROVED] must be returned EXACTLY as-is, character-for-character.
+
+${props.previousSectionEvent.sectionSections
+  .map(
+    (s, i) => `
+### Section ${i + 1}: ${s.title} ${targetSet!.has(i) ? "[NEEDS FIX]" : "[APPROVED - DO NOT MODIFY]"}
+
+${s.content}
+`,
+  )
+  .join("\n---\n")}`
+    : `The following sections were generated but REJECTED by review:
+
+${props.previousSectionEvent.sectionSections
+  .map(
+    (s, i) => `
+### Section ${i + 1}: ${s.title}
+
+${s.content}
+`,
+  )
+  .join("\n---\n")}`;
+
+  const taskBlock = hasSectionTargets
+    ? `Fix ONLY sections marked [NEEDS FIX] (section indices: ${props.sectionIndices!.map((i) => i + 1).join(", ")}).
+Return ALL sections (both fixed and unchanged) in the same sectionSections format.
+Sections marked [APPROVED] MUST be returned EXACTLY as they appear above -- do NOT modify them.`
+    : `Fix ONLY the issues identified above. Return ALL sections
+(both fixed and unchanged) in the same sectionSections format.
+Do NOT rewrite sections that were not flagged in the feedback.`;
+
+  return {
+    histories: [
+      {
+        id: v7(),
+        created_at: new Date().toISOString(),
+        type: "systemMessage",
+        text: AutoBeSystemPromptConstant.ANALYZE_WRITE_SECTION_PATCH,
+      },
+      ...(props.preliminary?.getHistories() ?? []),
+      {
+        id: v7(),
+        created_at: new Date().toISOString(),
+        type: "assistantMessage",
+        text: StringUtil.trim`
+        ## Parent Context
+
+        **Module**: ${moduleSection?.title ?? "Unknown"}
+        **Unit**: ${unitSection?.title ?? "Unknown"}
+        **Purpose**: ${unitSection?.purpose ?? "Unknown"}
+
+        ## AUTHORITATIVE Scope Reference
+
+        **Service Prefix**: ${props.scenario.prefix}
+        **Actors**: ${JSON.stringify(props.scenario.actors.map((a) => ({ name: a.name, kind: a.kind })))}
+        **Domain Entities**:
+        ${props.scenario.entities.map((e) => `- **${e.name}**: ${e.attributes.slice(0, 5).join(", ")}${e.relationships?.length ? ` | ${e.relationships.join(", ")}` : ""}`).join("\n")}
+
+        **CRITICAL**: You MUST NOT reference entities, actors, or features not listed above.
+
+        ## Previous Output
+
+        ${previousOutputBlock}
+
+        ## Review Feedback
+
+        The review REJECTED the above output for these reasons:
+
+        ${props.feedback}
+
+        ## Your Task
+
+        ${taskBlock}
+        `,
+      },
+    ],
+    userMessage: `Fix the rejected sections for "${unitSection?.title ?? "Unknown"}" based on the review feedback above.`,
+  };
+};