@wowok/agent-mcp 2.5.1 → 2.5.3

package/dist/schema/project/index.js

@@ -78,10 +78,32 @@ export const ProjectOperationInputSchema = z
         "aggregate_risks",
         "generate_deployment_doc",
         "trace_substeps",
+        "get_project_status",
+        "load_project",
+        "fork_project",
+        "export_project",
+        "get_project_files",
+        "validate_stage_unit",
+        "get_stage_assessment",
+        "get_stage_risk_policy",
+        "get_doc_relations",
+        "update_deployment_doc",
+        "validate_validation_chain",
+        "apply_improvement",
+        "get_improvement_queue",
+        "get_flywheel_config",
+        "rollback_stage",
+        "get_reversibility",
+        "assemble_context",
     ])
-        .describe("Action to perform. Two categories:\n" +
+        .describe("Action to perform. Six categories:\n" +
         "1. Graph Query (10): list_projects, list_objects, shareable, dependencies, referenced_by, pre_publish_check, cross_project_refs, graph_stats, save_graph, load_graph.\n" +
-        "2. Project-Based Deployment 5-Stage (4): Call IN ORDER — analyze_intent (Stage 2) → aggregate_risks (Stage 3) → generate_deployment_doc (Stage 4) → trace_substeps (Stage 5). Each action's output includes `next_action` telling you which action to call next. Stage 1 (Project Naming) uses the `project` + `version` params."),
+        "2. Project-Based Deployment 5-Stage (4): Call IN ORDER — analyze_intent (Stage 2) → aggregate_risks (Stage 3) → generate_deployment_doc (Stage 4) → trace_substeps (Stage 5). Each action's output includes `next_action` telling you which action to call next. Stage 1 (Project Naming) uses the `project` + `version` params.\n" +
+        "3. Project Memory Layer (5): load_project, fork_project, export_project, get_project_files, get_project_status.\n" +
+        "4. Nested-Autonomy Stage Unit (6): validate_stage_unit, get_stage_assessment, get_stage_risk_policy, get_doc_relations, update_deployment_doc, validate_validation_chain.\n" +
+        "5. Distillation Reflow (3): apply_improvement (review + apply a proposal — config_override writes overrides/, source_patch writes patches/), get_improvement_queue (list pending/approved proposals), get_flywheel_config (query reflow loop status).\n" +
+        "6. Rollback & Reversibility (2): rollback_stage (evaluate + execute stage rollback with file cleanup — blocked if has_published_object), get_reversibility (query object lifecycle reversibility matrix for risk reminders).\n" +
+        "7. Context Assembly (1): assemble_context (query on-chain object states + extract embedded history + optional events, and assemble a structured semantic context for LLM consumption — bridges raw facts to business-understandable context). Required input: context_objects (array of object IDs/names). Optional: include_events (query recent state-transition events)."),
     project: z
         .string()
         .optional()
@@ -95,6 +117,15 @@ export const ProjectOperationInputSchema = z
         .string()
         .optional()
         .describe("Object name or address (for dependencies / referenced_by / pre_publish_check)"),
+    fork_to_version: z
+        .string()
+        .regex(/^v\d+$/)
+        .optional()
+        .describe("Target version for fork_project (e.g. 'v2'). Must differ from source 'version'."),
+    fork_to_prefix: z
+        .string()
+        .optional()
+        .describe("Target prefix for fork_project. Defaults to source 'project' prefix when omitted."),
     user_intent: z
         .string()
         .min(1)
@@ -204,6 +235,28 @@ export const ProjectOperationInputSchema = z
         .boolean()
         .optional()
         .describe("Whether to run deployment-scanner validation after generating the doc. Defaults to true."),
+    gas_budget: z
+        .object({
+        network: z.enum(["testnet", "mainnet"]),
+        already_spent_gas: z
+            .number()
+            .int()
+            .nonnegative()
+            .optional()
+            .describe("Gas already spent on-chain in MIST (from telemetry). Defaults to 0."),
+        estimated_rerun_gas: z
+            .number()
+            .int()
+            .nonnegative()
+            .optional()
+            .describe("Estimated gas to re-run remaining ops in MIST. Auto-computed from estimateDeploymentGas when omitted."),
+        warning: z
+            .string()
+            .optional()
+            .describe("Custom warning text. Defaults to a computed cost summary."),
+    })
+        .optional()
+        .describe("Gas budget input for stage 4/5 gate (§3.5). Non-blocking; only surfaces a WARN on mainnet."),
     network: z
         .enum(["testnet", "mainnet"])
         .optional()
@@ -249,8 +302,81 @@ export const ProjectOperationInputSchema = z
         .strict())
         .optional()
         .describe("Substep records for trace_substeps."),
+    target_stage: z
+        .number()
+        .int()
+        .min(1)
+        .max(5)
+        .optional()
+        .describe("Stage number (1-5) for validate_stage_unit, get_stage_assessment, get_stage_risk_policy, and rollback_stage (the stage to rollback to). Required by these 4 actions."),
+    proposal_id: z
+        .string()
+        .optional()
+        .describe("Improvement proposal ID (for apply_improvement). Required by apply_improvement."),
+    review_status: z
+        .enum(["approved", "rejected", "applied"])
+        .optional()
+        .describe("New review status to set (for apply_improvement). 'approved' triggers application (config_override or source_patch); 'rejected' marks as rejected; 'applied' marks as already applied."),
+    queue_filter_status: z
+        .enum(["pending", "approved", "rejected", "applied"])
+        .optional()
+        .describe("Filter the improvement queue by review_status (for get_improvement_queue)."),
+    queue_filter_type: z
+        .string()
+        .optional()
+        .describe("Filter the improvement queue by proposal type (for get_improvement_queue)."),
+    queue_filter_priority: z
+        .string()
+        .optional()
+        .describe("Filter the improvement queue by priority P0/P1/P2/P3 (for get_improvement_queue)."),
+    dry_run: z
+        .boolean()
+        .optional()
+        .describe("When true, rollback_stage only evaluates the policy and returns the cleanup plan WITHOUT executing the rollback. Defaults to false (execute)."),
+    query_object_type: z
+        .string()
+        .optional()
+        .describe("Object type for get_reversibility query (e.g. 'service', 'machine', 'guard'). When omitted, returns the full matrix."),
+    query_lifecycle_state: z
+        .enum(["created", "referenced", "published", "bound"])
+        .optional()
+        .describe("Object lifecycle state for get_reversibility query. When omitted, returns all lifecycle states for the given object_type."),
+    context_objects: z
+        .array(z.string())
+        .optional()
+        .describe("Object IDs (0x...) or LocalMark names to assemble context for. Required when action='assemble_context'. The current on-chain state of each object will be queried and enriched with semantic annotations."),
+    include_events: z
+        .boolean()
+        .optional()
+        .describe("When true, assemble_context also queries recent state-transition events (NewOrderEvent, ProgressEvent, ArbEvent, etc.). Events are the real-time transition log — convergent and bounded. Defaults to false. Note: object-embedded history (ProgressSession.forwards, Order.disputes) is ALWAYS extracted automatically — no flag needed."),
+    context_network: z
+        .enum(["testnet", "mainnet"])
+        .optional()
+        .describe("Network for assemble_context queries. Defaults to 'testnet' if omitted."),
 })
     .strict();
+const GateValidationSchema = z
+    .object({
+    stage: z.number().int().min(1).max(5),
+    passed: z.boolean().describe("True when no FAIL checks exist."),
+    checks: z.array(z.object({
+        dimension: z.enum([
+            "semantic",
+            "tools",
+            "reversibility",
+            "risk",
+            "isolation",
+            "gas_budget",
+        ]),
+        status: z.enum(["PASS", "FAIL", "WARN"]),
+        message: z.string(),
+    })),
+    blocking_issues: z.array(z.string()),
+    warnings: z.array(z.string()),
+    can_proceed: z.boolean(),
+})
+    .optional()
+    .describe("Stage gate validation result (6 dimensions: semantic/tools/reversibility/risk/isolation/gas_budget).");
 const ItemsResult = (action) => z.object({
     action: z.literal(action),
     items: z
@@ -279,10 +405,14 @@ const AnalyzeIntentResultSchema = z
     puzzles: z.record(z.string(), PuzzleEntrySchema),
     overall_completeness: z.enum(["complete", "partial", "empty"]),
     next_step: z.string(),
+    recommended_creation_order: z
+        .array(z.string())
+        .describe("Recommended object creation order based on dependency topology (e.g. ['permission','progress','guard','allocation','machine','service'])."),
     next_action: z
         .enum(["aggregate_risks", "none"])
         .describe("Next project_operation action to call. 'aggregate_risks' when puzzles are complete enough; 'none' if user must fill missing fields first."),
     warnings: z.array(z.string()),
+    gate_validation: GateValidationSchema,
 })
     .strict();
 const AggregateRisksResultSchema = z
@@ -330,6 +460,16 @@ const AggregateRisksResultSchema = z
         entry_id: z.string(),
     }))
         .optional(),
+    fix_suggestions_grouped: z
+        .record(z.string(), z.array(z.object({
+        risk_rule_id: z.string(),
+        severity: z.enum(["CRITICAL", "HIGH", "MEDIUM", "LOW", "INFO"]),
+        message: z.string(),
+        fix_suggestion: z.string(),
+        entry_id: z.string(),
+    })))
+        .optional()
+        .describe("Fix suggestions grouped by object_type for better readability."),
     calibrated_at: z.number().int(),
     warnings: z.array(z.string()),
     can_proceed: z
@@ -341,6 +481,7 @@ const AggregateRisksResultSchema = z
     user_action_required: z
         .string()
         .describe("Human-readable next-step guidance."),
+    gate_validation: GateValidationSchema,
 })
     .strict();
 const GenerateDeploymentDocResultSchema = z
@@ -367,6 +508,30 @@ const GenerateDeploymentDocResultSchema = z
     })
         .optional()
         .describe("Scanner validation result. Present when run_scanner=true."),
+    estimated_gas: z
+        .object({
+        total_gas_mist: z.number().int().describe("Estimated total gas in MIST (1 WOW = 10^9 MIST)."),
+        per_object: z.array(z.object({
+            entry_id: z.string(),
+            object_type: z.string(),
+            create_gas_mist: z.number().int(),
+            publish_gas_mist: z.number().int(),
+        })),
+        publish_gas_mist: z.number().int().describe("Gas for publish operations only."),
+    })
+        .optional()
+        .describe("Gas cost estimate for deploying all objects."),
+    persisted: z
+        .object({
+        all_files: z
+            .array(z.string())
+            .describe("Project-relative paths of all generated files (27 total)."),
+        all_stages_passed: z
+            .boolean()
+            .describe("Whether all 5 stage gates passed validation."),
+    })
+        .optional()
+        .describe("Nested-autonomy doc set persistence result (§11). Present when the doc set was successfully written to disk."),
     can_proceed: z
         .boolean()
         .describe("Whether the user can proceed to stage 5. False when D-errors exist."),
@@ -374,6 +539,7 @@ const GenerateDeploymentDocResultSchema = z
         .enum(["trace_substeps", "fix_doc_errors", "none"])
         .describe("Next project_operation action. 'trace_substeps' when can_proceed=true; 'fix_doc_errors' when D-errors exist; 'none' for other cases."),
     user_action_required: z.string(),
+    gate_validation: GateValidationSchema,
 })
     .strict();
 const TraceSubstepsResultSchema = z
@@ -427,6 +593,488 @@ const TraceSubstepsResultSchema = z
         .enum(["deploy", "fix_substeps", "none"])
         .describe("Next action. 'deploy' when can_proceed=true (execute substeps via onchain_operations); 'fix_substeps' when issues exist; 'none' when done."),
     user_action_required: z.string(),
+    gate_validation: GateValidationSchema,
+})
+    .strict();
+const GetProjectStatusResultSchema = z
+    .object({
+    action: z.literal("get_project_status"),
+    project: z.string(),
+    version: z.string(),
+    stage: z.number().int().min(1).max(5),
+    stage_name: z.string(),
+    stage_description: z.string(),
+    next_action: z.string().describe("The next project_operation action to call."),
+    can_proceed: z.boolean(),
+    last_updated: z.number().int().optional().describe("Timestamp of last stage update (ms)."),
+    risk_status: z.string().optional(),
+    risk_can_proceed: z.boolean().optional(),
+    doc_generated: z.boolean().optional(),
+    trace_coherent: z.boolean().optional(),
+})
+    .strict();
+const LoadProjectListItemSchema = z.object({
+    prefix: z.string(),
+    version: z.string(),
+    dir_name: z.string(),
+    initialized: z.boolean(),
+    archived: z.boolean(),
+    needs_migration: z.boolean().describe("Whether schema_version differs from current."),
+    forked_from: z.string().optional(),
+    chain_exists: z.boolean().describe("Whether the project prefix exists on-chain."),
+    sync_status: z
+        .enum(["synced", "local_only", "archived", "uninitialized"])
+        .describe("Cross-validation of local filesystem vs chain truth."),
+});
+const LoadProjectResultSchema = z
+    .object({
+    action: z.literal("load_project"),
+    items: z.array(LoadProjectListItemSchema).optional(),
+    total: z.number().int().optional(),
+    project: z.string().optional(),
+    version: z.string().optional(),
+    local_exists: z.boolean().optional(),
+    manifest: z
+        .object({
+        schema_version: z.string(),
+        mcp_version: z.string(),
+        prefix: z.string(),
+        version: z.string(),
+        created_at: z.string(),
+        updated_at: z.string(),
+        forked_from: z.string().optional(),
+        needs_recalibration: z.boolean().optional(),
+    })
+        .optional(),
+    stage_state: z
+        .object({
+        stage: z.number().int().min(1).max(5),
+        updated_at: z.number().int(),
+        has_published_object: z.boolean(),
+        archived: z.boolean(),
+    })
+        .partial()
+        .optional(),
+    sync_status: z
+        .object({
+        local_exists: z.boolean(),
+        initialized: z.boolean(),
+        needs_migration: z.boolean(),
+        archived: z.boolean(),
+    })
+        .optional(),
+    migration: z
+        .object({
+        migrated: z.boolean().describe("Whether a migration was performed."),
+        from_version: z.string(),
+        to_version: z.string(),
+        steps_executed: z.array(z.string()),
+        backup_path: z.string().optional(),
+        needs_recalibration: z.boolean(),
+        message: z.string(),
+    })
+        .optional(),
+    message: z.string().optional(),
+})
+    .strict();
+const ForkProjectResultSchema = z
+    .object({
+    action: z.literal("fork_project"),
+    new_prefix: z.string(),
+    new_version: z.string(),
+    new_path: z.string().describe("Absolute filesystem path of the new project directory."),
+    forked_from: z
+        .string()
+        .describe("Absolute path of the source project directory."),
+    copied_files: z
+        .array(z.string())
+        .describe("Project-relative paths of copied files (excludes stage.json + audit.jsonl)."),
+    copied_file_count: z.number().int(),
+    warning: z
+        .string()
+        .describe("Project isolation notice — forked project must re-create its own objects."),
+    next_action: z
+        .literal("analyze_intent")
+        .describe("The forked project starts at stage 2 (analyze_intent)."),
+})
+    .strict();
+const ExportProjectResultSchema = z
+    .object({
+    action: z.literal("export_project"),
+    project: z.string(),
+    version: z.string(),
+    files: z.array(z.object({
+        path: z.string().describe("Project-relative file path."),
+        size: z.number().int().describe("File size in bytes."),
+        kind: z.enum(["json", "markdown", "jsonl", "other"]),
+        content: z.string().describe("File content (text)."),
+    })),
+    total_files: z.number().int(),
+})
+    .strict();
+const GetProjectFilesResultSchema = z
+    .object({
+    action: z.literal("get_project_files"),
+    project: z.string(),
+    version: z.string(),
+    files: z.array(z.object({
+        path: z.string().describe("Project-relative file path."),
+        size: z.number().int().describe("File size in bytes."),
+        kind: z.enum(["json", "markdown", "jsonl", "other"]),
+        mtime: z.number().int().describe("Last modified time (ms)."),
+    })),
+    total_files: z.number().int(),
+})
+    .strict();
+const ValidateStageUnitResultSchema = z
+    .object({
+    action: z.literal("validate_stage_unit"),
+    project: z.string(),
+    version: z.string(),
+    stage: z.number().int().min(1).max(5),
+    valid: z
+        .boolean()
+        .describe("True when all 4 stage unit files exist and pass consistency checks."),
+    missing_files: z
+        .array(z.string())
+        .describe("Project-relative paths of missing stage unit files (empty if all present)."),
+    checks: z.array(z.object({
+        check_id: z.string(),
+        status: z.enum(["pass", "fail", "warn"]),
+        message: z.string(),
+    })),
+    blocking_issues: z.array(z.string()),
+    stage_passed: z
+        .boolean()
+        .optional()
+        .describe("From verification.json stage_passed field (undefined if file missing)."),
+})
+    .strict();
+const GetStageAssessmentResultSchema = z
+    .object({
+    action: z.literal("get_stage_assessment"),
+    project: z.string(),
+    version: z.string(),
+    stage: z.number().int().min(1).max(5),
+    found: z.boolean(),
+    assessment: z
+        .object({
+        stage: z.number().int().min(1).max(5),
+        assessed_at: z.number().int(),
+        goal_completion: z.record(z.number()),
+        findings: z.array(z.object({
+            finding_id: z.string(),
+            severity: z.enum(["info", "warn", "critical"]),
+            criterion_id: z.string(),
+            description: z.string(),
+            evidence: z.unknown(),
+        })),
+        recommendations: z.array(z.string()),
+    })
+        .optional()
+        .describe("The stage assessment object (from assessment.json)."),
+})
+    .strict();
+const GetStageRiskPolicyResultSchema = z
+    .object({
+    action: z.literal("get_stage_risk_policy"),
+    project: z.string(),
+    version: z.string(),
+    stage: z.number().int().min(1).max(5),
+    found: z.boolean(),
+    risk_policy: z
+        .object({
+        stage: z.number().int().min(1).max(5),
+        risks: z.array(z.object({
+            risk_id: z.string(),
+            category: z.enum([
+                "naming",
+                "dependency",
+                "operational",
+                "financial",
+                "security",
+            ]),
+            description: z.string(),
+            likelihood: z.enum(["low", "medium", "high"]),
+            impact: z.enum(["low", "medium", "high", "critical"]),
+            mitigation: z.string(),
+            owner: z.enum(["ai", "user", "system"]),
+        })),
+        active_controls: z.array(z.object({
+            control_id: z.string(),
+            control_type: z.enum(["preventive", "detective", "corrective"]),
+            target_risk: z.string(),
+            implementation: z.string(),
+            verified: z.boolean(),
+        })),
+        residual_risks: z.array(z.string()),
+    })
+        .optional()
+        .describe("The stage risk policy object (from risk-policy.json)."),
+})
+    .strict();
+const GetDocRelationsResultSchema = z
+    .object({
+    action: z.literal("get_doc_relations"),
+    project: z.string(),
+    version: z.string(),
+    found: z.boolean(),
+    relations: z
+        .object({
+        version: z.string(),
+        generated_at: z.number().int(),
+        project_invariants: z.array(z.object({
+            invariant_id: z.string(),
+            description: z.string(),
+            verified_by_stages: z.array(z.number().int()),
+        })),
+        stage_units: z.array(z.object({
+            stage: z.number().int().min(1).max(5),
+            goal_id: z.string(),
+            assessment_id: z.string(),
+            risk_policy_id: z.string(),
+            verification_id: z.string(),
+            artifacts: z.array(z.string()),
+        })),
+        derivations: z.array(z.object({
+            source_type: z.enum(["project_invariant", "project_goal"]),
+            source_id: z.string(),
+            target_type: z.enum(["stage_goal", "stage_risk", "stage_verification"]),
+            target_stage: z.number().int().min(1).max(5),
+            target_id: z.string(),
+            derivation_logic: z.string(),
+        })),
+        propagation_rules: z.array(z.object({
+            rule_id: z.string(),
+            trigger: z.object({
+                stage: z.number().int(),
+                event: z.enum(["verification_passed", "verification_failed"]),
+            }),
+            effect: z.object({
+                target: z.enum(["project_invariant", "next_stage"]),
+                action: z.enum(["verify", "block"]),
+            }),
+        })),
+        impact_map: z.record(z.array(z.string())),
+    })
+        .optional()
+        .describe("The validation chain graph (from relations.json)."),
+})
+    .strict();
+const UpdateDeploymentDocResultSchema = z
+    .object({
+    action: z.literal("update_deployment_doc"),
+    project: z.string(),
+    version: z.string(),
+    updated: z.boolean().describe("Whether the deployment doc set was regenerated."),
+    updated_files: z
+        .array(z.string())
+        .describe("Project-relative paths of regenerated files."),
+    updated_file_count: z.number().int(),
+    all_stages_passed: z
+        .boolean()
+        .describe("Whether all stages passed their D-checks after the update."),
+    impact_summary: z
+        .object({
+        stages_affected: z.array(z.number().int()).describe("Stage numbers whose content changed."),
+        relations_rebuilt: z.boolean().describe("Whether relations.json was regenerated."),
+    })
+        .describe("Summary of what changed (based on relations.json impact analysis)."),
+    next_action: z
+        .enum(["validate_validation_chain", "trace_substeps", "none"])
+        .describe("Next action: validate the chain after update, or proceed to trace_substeps."),
+})
+    .strict();
+const ValidateValidationChainResultSchema = z
+    .object({
+    action: z.literal("validate_validation_chain"),
+    project: z.string(),
+    version: z.string(),
+    valid: z
+        .boolean()
+        .describe("True when derivation, propagation, cross-stage, and impact-map checks all pass."),
+    derivation_checks: z.array(z.object({
+        check_id: z.string(),
+        status: z.enum(["pass", "fail", "warn"]),
+        message: z.string(),
+    })),
+    propagation_checks: z.array(z.object({
+        check_id: z.string(),
+        status: z.enum(["pass", "fail", "warn"]),
+        message: z.string(),
+    })),
+    cross_stage_checks: z.array(z.object({
+        check_id: z.string(),
+        status: z.enum(["pass", "fail", "warn"]),
+        message: z.string(),
+    })),
+    impact_map_checks: z.array(z.object({
+        check_id: z.string(),
+        status: z.enum(["pass", "fail", "warn"]),
+        message: z.string(),
+    })),
+    blocking_issues: z.array(z.string()),
+    warnings: z.array(z.string()),
+})
+    .strict();
+const ApplyImprovementResultSchema = z
+    .object({
+    action: z.literal("apply_improvement"),
+    proposal_id: z.string(),
+    previous_status: z.string(),
+    new_status: z.string(),
+    applied: z
+        .boolean()
+        .describe("Whether the proposal was applied (true for approved+config_override, true for approved+source_patch producing patch file)."),
+    apply_mode: z.enum(["config_override", "source_patch"]),
+    override_file: z
+        .string()
+        .optional()
+        .describe("Path of the override file written (config_override only)."),
+    patch_file: z
+        .string()
+        .optional()
+        .describe("Path of the patch file written (source_patch only)."),
+    message: z.string(),
+    next_action: z
+        .enum(["get_improvement_queue", "get_flywheel_config", "none"])
+        .describe("Next action: check more proposals or view flywheel status."),
+})
+    .strict();
+const GetImprovementQueueResultSchema = z
+    .object({
+    action: z.literal("get_improvement_queue"),
+    total: z.number().int(),
+    filtered_count: z.number().int(),
+    proposals: z.array(z.object({
+        proposal_id: z.string(),
+        source_diagnosis_id: z.string(),
+        type: z.string(),
+        priority: z.string(),
+        automation: z.enum(["full", "semi", "manual"]),
+        apply_mode: z.enum(["config_override", "source_patch"]),
+        override_category: z.string().optional(),
+        title: z.string(),
+        description: z.string(),
+        target: z.string(),
+        review_status: z.enum(["pending", "approved", "rejected", "applied"]),
+        ab_test_recommended: z.boolean(),
+        persisted_at: z.string().optional(),
+        reviewed_at: z.string().optional(),
+    })),
+    summary: z.string(),
+})
+    .strict();
+const GetFlywheelConfigResultSchema = z
+    .object({
+    action: z.literal("get_flywheel_config"),
+    queue_path: z.string(),
+    overrides_dir: z.string(),
+    patches_dir: z.string(),
+    total_proposals: z.number().int(),
+    pending_count: z.number().int(),
+    approved_count: z.number().int(),
+    applied_count: z.number().int(),
+    rejected_count: z.number().int(),
+    override_files: z.array(z.string()),
+    recent_proposals: z.array(z.object({
+        proposal_id: z.string(),
+        type: z.string(),
+        priority: z.string(),
+        review_status: z.enum(["pending", "approved", "rejected", "applied"]),
+        title: z.string(),
+    })),
+    reflow_status: z
+        .enum(["active", "idle", "blocked"])
+        .describe("active=pending proposals exist; idle=no proposals; blocked=approved but not applied."),
+})
+    .strict();
+const RollbackStageResultSchema = z
+    .object({
+    action: z.literal("rollback_stage"),
+    success: z.boolean().describe("Whether the rollback was executed (or would be, if dry_run=false)."),
+    dry_run: z.boolean().describe("Whether this was a policy evaluation only (no execution)."),
+    target_stage: z.number().int().min(1).max(5),
+    previous_stage: z.number().int().min(1).max(5),
+    allowed: z.boolean().describe("Whether the rollback is allowed by policy."),
+    block_reason: z.string().optional().describe("Present when allowed=false — explains why rollback is blocked."),
+    files_to_clean: z
+        .array(z.string())
+        .describe("Files/directories that will be (or were) deleted during rollback."),
+    files_cleaned: z
+        .array(z.string())
+        .describe("Files/directories actually deleted (empty in dry_run mode)."),
+    warnings: z.array(z.string()).describe("Non-blocking warnings (e.g., data loss reminder)."),
+    iteration_strategy: z.object({
+        strategy: z.enum(["in_place", "fork"]),
+        reason: z.string(),
+        fork_to_version: z.string().optional(),
+        published_objects: z.array(z.string()).optional(),
+    }),
+    message: z.string().describe("Human-readable summary."),
+})
+    .strict();
+const GetReversibilityResultSchema = z
+    .object({
+    action: z.literal("get_reversibility"),
+    query: z
+        .object({
+        object_type: z.string().optional(),
+        lifecycle_state: z.string().optional(),
+    })
+        .describe("The query parameters used (empty = full matrix returned)."),
+    entries: z.array(z.object({
+        object_type: z.string(),
+        lifecycle_state: z.enum(["created", "referenced", "published", "bound"]),
+        struct_reversible: z.enum(["reversible", "limited", "immutable"]),
+        behavior_reversible: z.boolean(),
+        gas_impact: z.enum(["low", "medium", "high", "none"]),
+        explanation: z.string(),
+    })),
+    total: z.number().int().describe("Number of matrix entries returned."),
+})
+    .strict();
+const AssembleContextResultSchema = z
+    .object({
+    action: z.literal("assemble_context"),
+    objects: z
+        .array(z.unknown())
+        .describe("Current on-chain states of the requested objects (from query_objects)."),
+    events: z
+        .array(z.unknown())
+        .optional()
+        .describe("Recent state-transition events (only when include_events=true). " +
+        "Includes NewOrderEvent, ProgressEvent, ArbEvent, etc."),
+    embedded_history: z
+        .record(z.string(), z.unknown())
+        .optional()
+        .describe("Operation history extracted from object states — NOT from tx queries. " +
+        "Progress objects: session.forwards (who did what, when, accomplished). " +
+        "Order objects: disputes, claimed_by, progress_object. " +
+        "This is the CORRECT history source: convergent, no infinite recursion."),
+    semantic_summary: z
+        .unknown()
+        .optional()
+        .describe("Business semantic annotations generated by buildDataSemantic — " +
+        "maps raw object fields to business-meaningful concepts for LLM consumption."),
+    context_digest: z
+        .string()
+        .describe("Unique digest for this assembled context (SHA of object IDs + versions + timestamp). " +
+        "Use for caching or to detect state changes between two assembly calls."),
+    next_action: z
+        .string()
+        .describe("Suggested next action: use onchain_transaction to verify a specific past tx, " +
+        "or proceed with onchain_operations to modify state, or re-call assemble_context " +
+        "after state changes."),
+    flywheel_collected: z
+        .object({
+        signature_recorded: z.boolean().describe("Whether the experience signature was recorded in the real-time experience layer."),
+        proposal_created: z.boolean().describe("Whether an improvement proposal was created for the offline distillation flywheel."),
+        proposal_id: z.string().optional().describe("The proposal ID, if created."),
+    })
+        .optional()
+        .describe("Distillation flywheel collection result. Non-blocking — may be undefined if collection fails. " +
+        "When present, the assembled context has been fed back into the experience layer and/or improvement queue."),
 })
     .strict();
 export const ProjectOperationOutputSchema = z
@@ -465,6 +1113,23 @@ export const ProjectOperationOutputSchema = z
         AggregateRisksResultSchema,
         GenerateDeploymentDocResultSchema,
         TraceSubstepsResultSchema,
+        GetProjectStatusResultSchema,
+        LoadProjectResultSchema,
+        ForkProjectResultSchema,
+        ExportProjectResultSchema,
+        GetProjectFilesResultSchema,
+        ValidateStageUnitResultSchema,
+        GetStageAssessmentResultSchema,
+        GetStageRiskPolicyResultSchema,
+        GetDocRelationsResultSchema,
+        UpdateDeploymentDocResultSchema,
+        ValidateValidationChainResultSchema,
+        ApplyImprovementResultSchema,
+        GetImprovementQueueResultSchema,
+        GetFlywheelConfigResultSchema,
+        RollbackStageResultSchema,
+        GetReversibilityResultSchema,
+        AssembleContextResultSchema,
     ]).describe("Query result — shape depends on action"),
 })
     .strict();