@nt-ai-lab/deterministic-agent-workflow-pi 0.5.2 → 0.5.4

package/dist/features/pi-extension/entrypoint/pi-workflow-extension.d.ts

@@ -1,6 +1,6 @@
-import type { BaseWorkflowState, RehydratableWorkflow } from '@nt-ai-lab/deterministic-agent-workflow-engine';
+import type { BaseWorkflowState, RehydratableWorkflow, TransitionContext } from '@nt-ai-lab/deterministic-agent-workflow-engine';
 import type { PiWorkflowExtension, PiWorkflowExtensionConfig } from '../../../platform/domain/pi-workflow-extension-types';
-export declare const PI_IDLE_RECOVERY_MESSAGE = "You have stopped. You should never stop until the workflow is complete unless your current state permits stopping.";
+export declare const PI_IDLE_RECOVERY_MESSAGE: string;
 export declare const PI_SESSION_BRANCH_BLOCK_MESSAGE = "Pi session tree navigation and forks are disabled while a workflow is active.";
 /** @riviere-role cli-entrypoint */
-export declare function createPiWorkflowExtension<TWorkflow extends RehydratableWorkflow<TState>, TState extends BaseWorkflowState<TStateName>, TDeps, TStateName extends string = string, TOperation extends string = string>(config: PiWorkflowExtensionConfig<TWorkflow, TState, TDeps, TStateName, TOperation>): PiWorkflowExtension;
+export declare function createPiWorkflowExtension<TWorkflow extends RehydratableWorkflow<TState>, TState extends BaseWorkflowState<TStateName>, TDeps, TStateName extends string = string, TOperation extends string = string, TTransitionContext extends TransitionContext<TState, TStateName> = TransitionContext<TState, TStateName>>(config: PiWorkflowExtensionConfig<TWorkflow, TState, TDeps, TStateName, TOperation, TTransitionContext>): PiWorkflowExtension;

package/dist/features/pi-extension/entrypoint/pi-workflow-extension.js

@@ -1,6 +1,6 @@
 import { join } from 'node:path';
 import { WorkflowEngine } from '@nt-ai-lab/deterministic-agent-workflow-engine';
-import { createPreToolUseHandler, createWorkflowRunner, getRepositoryName, } from '@nt-ai-lab/deterministic-agent-workflow-cli';
+import { createPreToolUseHandler, createWorkflowRunner, formatStopPreventionMessage, getRepositoryName, } from '@nt-ai-lab/deterministic-agent-workflow-cli';
 import { createStore } from '@nt-ai-lab/deterministic-agent-workflow-event-store';
 import { Type } from 'typebox';
 import { parsePiCommandArguments } from '../../../platform/domain/pi-command-arguments.js';
@@ -12,7 +12,7 @@ const DEFAULT_COMMAND_NAME = 'workflow';
 const DEFAULT_TOOL_NAME = 'workflow';
 const INITIALIZATION_PENDING_REASON = 'Pi workflow initialization has not completed safely. Tool execution is blocked.';
 const INACTIVE_WORKFLOW_REASON = 'Pi workflow is inactive. Run the workflow init command before using workflow operations.';
-export const PI_IDLE_RECOVERY_MESSAGE = 'You have stopped. You should never stop until the workflow is complete unless your current state permits stopping.';
+export const PI_IDLE_RECOVERY_MESSAGE = formatStopPreventionMessage();
 export const PI_SESSION_BRANCH_BLOCK_MESSAGE = 'Pi session tree navigation and forks are disabled while a workflow is active.';
 const workflowToolParameters = Type.Object({
     operation: Type.String({ description: 'Workflow operation, for example init or transition' }),
@@ -40,23 +40,28 @@ export function createPiWorkflowExtension(config) {
         questionToolName: PI_QUESTION_TOOL,
         customGates: config.customGates,
     });
+    function buildEngineDeps(ctx, store) {
+        const sessionId = ctx.sessionManager.getSessionId();
+        const now = () => new Date().toISOString();
+        const note = translationNote(toolName);
+        return {
+            store,
+            getPluginRoot: () => config.pluginRoot,
+            getEnvFilePath: () => join(config.pluginRoot, '.pi', 'unused.env'),
+            getRepositoryName: () => getRepositoryName(ctx.cwd),
+            readFile: (path) => readWorkflowInstruction(path, note),
+            appendToFile: () => undefined,
+            now,
+            transcriptReader: new PiTranscriptReader(() => ctx.sessionManager.getBranch()),
+            sessionContext: { getMainSessionId: () => sessionId },
+        };
+    }
     function useEngine(ctx, operation) {
         const store = createStore(databasePath);
         try {
+            const engineDeps = buildEngineDeps(ctx, store);
+            const now = engineDeps.now;
             const sessionId = ctx.sessionManager.getSessionId();
-            const now = () => new Date().toISOString();
-            const note = translationNote(toolName);
-            const engineDeps = {
-                store,
-                getPluginRoot: () => config.pluginRoot,
-                getEnvFilePath: () => join(config.pluginRoot, '.pi', 'unused.env'),
-                getRepositoryName: () => getRepositoryName(ctx.cwd),
-                readFile: (path) => readWorkflowInstruction(path, note),
-                appendToFile: () => undefined,
-                now,
-                transcriptReader: new PiTranscriptReader(() => ctx.sessionManager.getBranch()),
-                sessionContext: { getMainSessionId: () => sessionId },
-            };
             const platform = {
                 getPluginRoot: () => config.pluginRoot,
                 now,
@@ -71,8 +76,22 @@ export function createPiWorkflowExtension(config) {
             store.db.close();
         }
     }
+    function hasPersistedWorkflowState(sessionId) {
+        const store = createStore(databasePath);
+        try {
+            return store.hasSessionStarted(sessionId);
+        }
+        finally {
+            store.db.close();
+        }
+    }
     function markInitializationFailed(ctx, sessionId, detail) {
-        const reason = `Pi workflow initialization failed: ${detail}`;
+        return failSession(ctx, sessionId, `Pi workflow initialization failed: ${detail}`);
+    }
+    function markSafetyUnavailable(ctx, sessionId, detail) {
+        return failSession(ctx, sessionId, `Pi workflow safety is unavailable: ${detail}`);
+    }
+    function failSession(ctx, sessionId, reason) {
         initializationBySession.set(sessionId, {
             type: 'failed',
             reason,
@@ -209,7 +228,7 @@ export function createPiWorkflowExtension(config) {
         }
         catch (error) {
             return {
-                output: markInitializationFailed(ctx, session.sessionId, `Workflow operation could not establish safe state: ${String(error)}`),
+                output: markSafetyUnavailable(ctx, session.sessionId, `Workflow operation could not establish safe state: ${String(error)}`),
                 exitCode: 1,
             };
         }
@@ -224,7 +243,7 @@ export function createPiWorkflowExtension(config) {
             }
             sessionStartsById.set(session.sessionId, event);
             try {
-                const hasPersistedWorkflow = useEngine(ctx, (engine) => engine.hasSessionStarted(session.sessionId));
+                const hasPersistedWorkflow = hasPersistedWorkflowState(session.sessionId);
                 const hasTranscriptWorkflow = hasPiWorkflowMarker(ctx.sessionManager.getEntries());
                 if (!hasPersistedWorkflow && !hasTranscriptWorkflow) {
                     initializationBySession.set(session.sessionId, { type: 'inactive' });
@@ -270,7 +289,7 @@ export function createPiWorkflowExtension(config) {
             catch (error) {
                 return {
                     block: true,
-                    reason: markInitializationFailed(ctx, session.sessionId, `Tool safety could not be established: ${String(error)}`),
+                    reason: markSafetyUnavailable(ctx, session.sessionId, `Tool safety could not be established: ${String(error)}`),
                 };
             }
         });
@@ -298,11 +317,11 @@ export function createPiWorkflowExtension(config) {
                 const result = useEngine(ctx, (engine) => engine.checkStopping(session.sessionId, 'stop'));
                 if (result.type === 'blocked' && ctx.isIdle() && !ctx.hasPendingMessages()) {
                     recoveredAssistantBySession.set(session.sessionId, settlement.id);
-                    pi.sendUserMessage(PI_IDLE_RECOVERY_MESSAGE);
+                    pi.sendUserMessage(formatStopPreventionMessage(result.output, config.stopPreventionMessage));
                 }
             }
             catch (error) {
-                markInitializationFailed(ctx, session.sessionId, `Stopping safety could not be established: ${String(error)}`);
+                markSafetyUnavailable(ctx, session.sessionId, `Stopping safety could not be established: ${String(error)}`);
             }
         });
         const blockSessionBranching = (ctx) => {

package/dist/platform/domain/pi-workflow-extension-types.d.ts

@@ -1,5 +1,5 @@
 import type { ExtensionFactory } from '@earendil-works/pi-coding-agent';
-import type { BaseWorkflowState, RehydratableWorkflow, WorkflowDefinition } from '@nt-ai-lab/deterministic-agent-workflow-engine';
+import type { BaseWorkflowState, RehydratableWorkflow, TransitionContext, WorkflowDefinition } from '@nt-ai-lab/deterministic-agent-workflow-engine';
 import type { PlatformContext, PreToolUseHandlerConfig, RouteMap } from '@nt-ai-lab/deterministic-agent-workflow-cli';
 /** @riviere-role value-object */
 export type PiWorkflowExtension = ExtensionFactory;
@@ -23,12 +23,13 @@ export type PiSessionIdResult = {
     readonly reason: string;
 };
 /** @riviere-role value-object */
-export type PiWorkflowExtensionConfig<TWorkflow extends RehydratableWorkflow<TState>, TState extends BaseWorkflowState<TStateName>, TDeps, TStateName extends string = string, TOperation extends string = string> = Omit<PreToolUseHandlerConfig<TWorkflow, TState, TStateName>, 'questionToolName'> & {
-    readonly workflowDefinition: WorkflowDefinition<TWorkflow, TState, TDeps, TStateName, TOperation>;
+export type PiWorkflowExtensionConfig<TWorkflow extends RehydratableWorkflow<TState>, TState extends BaseWorkflowState<TStateName>, TDeps, TStateName extends string = string, TOperation extends string = string, TTransitionContext extends TransitionContext<TState, TStateName> = TransitionContext<TState, TStateName>> = Omit<PreToolUseHandlerConfig<TWorkflow, TState, TStateName>, 'questionToolName'> & {
+    readonly workflowDefinition: WorkflowDefinition<TWorkflow, TState, TDeps, TStateName, TOperation, TTransitionContext>;
     readonly routes: RouteMap<TWorkflow, TState>;
     readonly buildWorkflowDeps: (platform: PlatformContext) => TDeps;
     readonly pluginRoot: string;
     readonly databasePath?: string;
     readonly commandName?: string;
     readonly toolName?: string;
+    readonly stopPreventionMessage?: string;
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nt-ai-lab/deterministic-agent-workflow-pi",
-  "version": "0.5.2",
+  "version": "0.5.4",
   "private": false,
   "type": "module",
   "engines": {
@@ -23,9 +23,9 @@
   },
   "dependencies": {
     "typebox": "^1.3.7",
-    "@nt-ai-lab/deterministic-agent-workflow-cli": "0.4.5",
-    "@nt-ai-lab/deterministic-agent-workflow-event-store": "0.4.5",
-    "@nt-ai-lab/deterministic-agent-workflow-engine": "0.4.5"
+    "@nt-ai-lab/deterministic-agent-workflow-cli": "0.4.7",
+    "@nt-ai-lab/deterministic-agent-workflow-engine": "0.4.7",
+    "@nt-ai-lab/deterministic-agent-workflow-event-store": "0.4.7"
   },
   "devDependencies": {
     "@earendil-works/pi-coding-agent": "^0.84.4",