astrocode-workflow 0.1.11 → 0.1.16

package/dist/agents/registry.d.ts

@@ -1,6 +1,6 @@
 import type { AgentConfig } from "@opencode-ai/sdk";
 import type { AstrocodeConfig } from "../config/schema";
 export declare function createAstroAgents(opts: {
-    systemDefaultModel: string;
+    systemDefaultModel?: string;
     pluginConfig: AstrocodeConfig;
 }): Record<string, AgentConfig>;

package/dist/index.js

@@ -6,11 +6,14 @@ import { createAstroTools } from "./tools";
 import { createContinuationEnforcer } from "./hooks/continuation-enforcer";
 import { createToolOutputTruncatorHook } from "./hooks/tool-output-truncator";
 import { createToastManager } from "./ui/toasts";
+import { createAstroAgents } from "./agents/registry";
 console.log("Astrocode plugin loading...");
 const Astrocode = async (ctx) => {
     const repoRoot = ctx.directory;
     // Always load config first - this provides defaults even in limited mode
     let pluginConfig = loadAstrocodeConfig(repoRoot);
+    // Create agents for registration
+    const agents = createAstroAgents({ pluginConfig });
     // Always ensure .astro directories exist, even in limited mode
     const paths = getAstroPaths(repoRoot, pluginConfig.db.path);
     ensureAstroDirs(paths);
@@ -48,6 +51,8 @@ const Astrocode = async (ctx) => {
     }
     return {
         name: "Astrocode",
+        // Register agents
+        agents,
         // Merge agents + slash commands into system config
         config: configHandler,
         // Register tools

package/dist/state/db.js

@@ -2,6 +2,7 @@ import fs from "node:fs";
 import path from "node:path";
 import { SCHEMA_SQL, SCHEMA_VERSION } from "./schema";
 import { nowISO } from "../shared/time";
+import { info } from "../shared/log";
 import { createDatabaseAdapter } from "./adapters";
 /** Ensure directory exists for a file path. */
 function ensureParentDir(filePath) {
@@ -55,6 +56,19 @@ export function ensureSchema(db, opts) {
     }
     try {
         db.exec(SCHEMA_SQL);
+        // Migrations for existing databases
+        // Add created_at to stage_runs if missing (introduced in schema version 2)
+        try {
+            const columns = db.prepare("PRAGMA table_info(stage_runs)").all();
+            const hasCreatedAt = columns.some(col => col.name === 'created_at');
+            if (!hasCreatedAt) {
+                db.exec("ALTER TABLE stage_runs ADD COLUMN created_at TEXT NOT NULL DEFAULT ''");
+                info("[Astrocode] Added created_at column to stage_runs table");
+            }
+        }
+        catch (e) {
+            // Column might already exist or table doesn't exist, ignore
+        }
         const row = db.prepare("SELECT schema_version FROM repo_state WHERE id = 1").get();
         if (!row) {
             const now = nowISO();

package/dist/tools/stage.js

@@ -219,8 +219,9 @@ export function createAstroStageCompleteTool(opts) {
                     const specPath = path.join(repoRoot, ".astro", "spec.md");
                     if (fs.existsSync(specPath) && fs.statSync(specPath).size > 100) {
                         // Skip spec
-                        db.prepare("INSERT INTO stage_runs (stage_run_id, run_id, stage_key, status, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?)")
-                            .run(newId("stage"), rid, "spec", "skipped", now, now);
+                        const specIndex = pipeline.indexOf("spec");
+                        db.prepare("INSERT INTO stage_runs (stage_run_id, run_id, stage_key, stage_index, status, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?)")
+                            .run(newId("stage"), rid, "spec", specIndex, "skipped", now, now);
                         next = "implement";
                     }
                 }

package/dist/workflow/directives.js

@@ -85,11 +85,15 @@ export function buildStageDirective(opts) {
         `Output contract (strict):`,
         `1) Baton markdown (short, structured)`,
         `2) ASTRO JSON between markers:`,
-        `   ${"```"}`,
         `   ${"<!-- ASTRO_JSON_BEGIN -->"}`,
-        `   {...}`,
+        `   {`,
+        `     "schema_version": 1,`,
+        `     "stage_key": "${stage_key}",`,
+        `     "status": "ok",`,
+        `     "summary": "Brief summary of work done",`,
+        `     ...`,
+        `   }`,
         `   ${"<!-- ASTRO_JSON_END -->"}`,
-        `   ${"```"}`,
         ``,
         `ASTRO JSON requirements:`,
         `- stage_key must be "${stage_key}"`,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "astrocode-workflow",
-  "version": "0.1.11",
+  "version": "0.1.16",
   "type": "module",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/agents/registry.ts

@@ -116,7 +116,7 @@ function stageVerifyPermissions(): PermissionMap {
 }
 
 export function createAstroAgents(opts: {
-  systemDefaultModel: string;
+  systemDefaultModel?: string;
   pluginConfig: AstrocodeConfig;
 }): Record<string, AgentConfig> {
   const { systemDefaultModel, pluginConfig } = opts;

package/src/index.ts

@@ -7,6 +7,7 @@ import { createAstroTools } from "./tools";
 import { createContinuationEnforcer } from "./hooks/continuation-enforcer";
 import { createToolOutputTruncatorHook } from "./hooks/tool-output-truncator";
 import { createToastManager } from "./ui/toasts";
+import { createAstroAgents } from "./agents/registry";
 import { info, warn } from "./shared/log";
 
 console.log("Astrocode plugin loading...");
@@ -17,6 +18,9 @@ const Astrocode: Plugin = async (ctx) => {
   // Always load config first - this provides defaults even in limited mode
   let pluginConfig = loadAstrocodeConfig(repoRoot);
 
+  // Create agents for registration
+  const agents = createAstroAgents({ pluginConfig });
+
   // Always ensure .astro directories exist, even in limited mode
   const paths = getAstroPaths(repoRoot, pluginConfig.db.path);
   ensureAstroDirs(paths);
@@ -62,6 +66,9 @@ const Astrocode: Plugin = async (ctx) => {
   return {
     name: "Astrocode",
 
+    // Register agents
+    agents,
+
     // Merge agents + slash commands into system config
     config: configHandler,
 

package/src/state/db.ts

@@ -68,6 +68,19 @@ export function ensureSchema(db: SqliteDb, opts?: { allowAutoMigrate?: boolean;
   try {
     db.exec(SCHEMA_SQL);
 
+    // Migrations for existing databases
+    // Add created_at to stage_runs if missing (introduced in schema version 2)
+    try {
+      const columns = db.prepare("PRAGMA table_info(stage_runs)").all() as { name: string }[];
+      const hasCreatedAt = columns.some(col => col.name === 'created_at');
+      if (!hasCreatedAt) {
+        db.exec("ALTER TABLE stage_runs ADD COLUMN created_at TEXT NOT NULL DEFAULT ''");
+        info("[Astrocode] Added created_at column to stage_runs table");
+      }
+    } catch (e) {
+      // Column might already exist or table doesn't exist, ignore
+    }
+
     const row = db.prepare("SELECT schema_version FROM repo_state WHERE id = 1").get() as { schema_version?: number } | undefined;
 
     if (!row) {

package/src/tools/stage.ts

@@ -300,8 +300,9 @@ export function createAstroStageCompleteTool(opts: { ctx: any; config: Astrocode
           const specPath = path.join(repoRoot, ".astro", "spec.md");
           if (fs.existsSync(specPath) && fs.statSync(specPath).size > 100) {
             // Skip spec
-            db.prepare("INSERT INTO stage_runs (stage_run_id, run_id, stage_key, status, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?)")
-              .run(newId("stage"), rid, "spec", "skipped", now, now);
+            const specIndex = pipeline.indexOf("spec");
+            db.prepare("INSERT INTO stage_runs (stage_run_id, run_id, stage_key, stage_index, status, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?)")
+              .run(newId("stage"), rid, "spec", specIndex, "skipped", now, now);
             next = "implement";
           }
         }

package/src/workflow/directives.ts

@@ -140,11 +140,15 @@ export function buildStageDirective(opts: {
         `Output contract (strict):`,
         `1) Baton markdown (short, structured)`,
         `2) ASTRO JSON between markers:`,
-        `   ${"```"}`,
         `   ${"<!-- ASTRO_JSON_BEGIN -->"}`,
-        `   {...}`,
+        `   {`,
+        `     "schema_version": 1,`,
+        `     "stage_key": "${stage_key}",`,
+        `     "status": "ok",`,
+        `     "summary": "Brief summary of work done",`,
+        `     ...`,
+        `   }`,
         `   ${"<!-- ASTRO_JSON_END -->"}`,
-        `   ${"```"}`,
         ``,
         `ASTRO JSON requirements:`,
         `- stage_key must be "${stage_key}"`,