astrocode-workflow 0.4.0 → 0.4.2

package/src/tools/workflow.ts

@@ -1,5 +1,6 @@
 // src/tools/workflow.ts
 import { tool, type ToolDefinition } from "@opencode-ai/plugin/tool";
+import path from "node:path";
 import type { AstrocodeConfig } from "../config/schema";
 import type { SqliteDb } from "../state/db";
 import { withTx } from "../state/db";
@@ -19,6 +20,7 @@ import { buildStageDirective, directiveHash } from "../workflow/directives";
 import { injectChatPrompt } from "../ui/inject";
 import { nowISO } from "../shared/time";
 import { newEventId } from "../state/ids";
+
 import { debug } from "../shared/log";
 import { createToastManager } from "../ui/toasts";
 import type { AgentConfig } from "@opencode-ai/sdk";

package/src/ui/inject.ts

@@ -1,4 +1,5 @@
 // src/ui/inject.ts
+import { recordInjection, recordError } from "../shared/metrics";
 type InjectionItem = {
   ctx: any;
   sessionId: string;
@@ -33,16 +34,26 @@ async function tryInjectOnce(item: InjectionItem): Promise<void> {
   const { ctx, sessionId, text, agent = "Astro" } = item;
   const prefixedText = `[${agent}]\n\n${text}`;
 
-  const { session, prompt } = getPromptInvoker(ctx);
+  const injectionRecorder = recordInjection({ sessionId, attempts: item.attempts + 1, agent });
+  const injectionStart = injectionRecorder.start();
 
-  // IMPORTANT: force correct `this` binding
-  await prompt.call(session, {
-    path: { id: sessionId },
-    body: {
-      parts: [{ type: "text", text: prefixedText }],
-      agent,
-    },
-  });
+  try {
+    const { session, prompt } = getPromptInvoker(ctx);
+
+    // IMPORTANT: force correct `this` binding
+    await prompt.call(session, {
+      path: { id: sessionId },
+      body: {
+        parts: [{ type: "text", text: prefixedText }],
+        agent,
+      },
+    });
+
+    injectionRecorder.end(injectionStart, true);
+  } catch (error) {
+    injectionRecorder.end(injectionStart, false);
+    throw error;
+  }
 }
 
 async function runSessionQueue(sessionId: string) {
@@ -69,6 +80,7 @@ async function runSessionQueue(sessionId: string) {
           console.warn(
             `[Astrocode] Injection failed permanently after ${item.attempts} attempts: ${msg}`
           );
+          recordError("injection_failure", `Injection failed after ${item.attempts} attempts: ${msg}`);
           item.reject(err);
           continue;
         }

package/src/workflow/repair.ts

@@ -72,11 +72,11 @@ export function repairState(db: SqliteDb, config: AstrocodeConfig): RepairReport
     if (stageRuns.length < pipeline.length) {
       const existingKeys = new Set(stageRuns.map((s) => s.stage_key));
       const insert = db.prepare(
-        "INSERT INTO stage_runs (stage_run_id, run_id, stage_key, stage_index, status, updated_at) VALUES (?, ?, ?, ?, 'pending', ?)"
+        "INSERT INTO stage_runs (stage_run_id, run_id, stage_key, stage_index, status, created_at, updated_at) VALUES (?, ?, ?, ?, 'pending', ?, ?)"
       );
       pipeline.forEach((key, idx) => {
         if (!existingKeys.has(key as any)) {
-          insert.run(newStageRunId(), active.run_id, key, idx, now);
+          insert.run(newStageRunId(), active.run_id, key, idx, now, now);
           push(report, `Inserted missing stage_run ${key} for run ${active.run_id}`);
         }
       });