agent-afk 5.310.5 → 5.310.6

package/dist/agent/file-checkpoint/file-checkpoint.d.ts

@@ -0,0 +1,15 @@
+export interface FileCheckpointRegistry {
+    snapshotFile(absPath: string): Promise<void>;
+    readonly turnId: string;
+}
+export interface FileSnapshot {
+    absPath: string;
+    snapshotPath: string;
+    isNewFile: boolean;
+    mode?: number;
+}
+export declare function decodePathKey(key: string): string;
+export declare const NEW_FILE_SENTINEL = "__afk_new_file__\n";
+export declare function createFileCheckpointRegistry(sessionId: string, turnId: string, _stateDir?: string): FileCheckpointRegistry;
+export declare function hasTurnCheckpoint(sessionId: string, turnId: string, _stateDir?: string): boolean;
+export declare function listTurnSnapshots(sessionId: string, turnId: string, _stateDir?: string): FileSnapshot[];

package/dist/agent/file-checkpoint/rewind-files.d.ts

@@ -0,0 +1,11 @@
+import type { ProviderRewindResult } from '../provider.js';
+import type { ToolHandlerContext } from '../tools/types.js';
+export interface RewindFilesDeps {
+    sessionId: string;
+    enableFileCheckpointing: boolean;
+    writePolicy?: ToolHandlerContext;
+    _stateDir?: string;
+}
+export declare function rewindFiles(deps: RewindFilesDeps, turnId: string, options?: {
+    dryRun?: boolean;
+}): Promise<ProviderRewindResult>;

package/dist/agent/file-checkpoint/rewind-staging.d.ts

@@ -0,0 +1,2 @@
+export declare function pinRewindParent(target: string): () => void;
+export declare function stageRewindFile(target: string, content: Buffer, validate: () => void, mode?: number): void;

package/dist/agent/file-checkpoint/write-policy.d.ts

@@ -0,0 +1,3 @@
+import type { ToolHandlerContext } from '../tools/types.js';
+export declare function validatedWriteTarget(path: string, context: ToolHandlerContext | undefined, cwd?: string): Promise<string>;
+export declare function currentWritePolicy(dispatcher: object | undefined, fallback: ToolHandlerContext): ToolHandlerContext;

package/dist/agent/provider.d.ts

@@ -118,6 +118,7 @@ export type ProviderEvent = {
     sessionId?: string;
 } | {
     type: 'turn.completed';
+    userMessageId?: string;
     usage: ProviderUsage;
     sessionId?: string;
 } | {
@@ -156,6 +157,7 @@ export type ProviderEvent = {
 };
 export interface ProviderUserTurn {
     content: string | ContentBlockParam[];
+    userMessageId?: string;
     sessionId?: string;
 }
 export interface ProviderQueryArgs {

package/dist/agent/providers/anthropic-direct/query/session-state.d.ts

@@ -14,6 +14,8 @@ export interface SessionState {
     autoCompactThreshold: number | undefined;
     readonly journalSync: JournalSync<MessageParam>;
     readonly messageJournal: MessageJournal | undefined;
+    enableFileCheckpointing: boolean;
+    currentTurnId: string | undefined;
 }
 export declare function createSessionState(opts: {
     model: string;
@@ -25,4 +27,5 @@ export declare function createSessionState(opts: {
     autoCompactThreshold?: number;
     initialUsageInputTokens?: number;
     messageJournal?: MessageJournal;
+    enableFileCheckpointing?: boolean;
 }): SessionState;

package/dist/agent/providers/anthropic-direct/query-options.d.ts

@@ -46,5 +46,6 @@ export interface AnthropicDirectQueryOptions {
     throttleQueue?: import('./throttle-queue.js').ThrottleQueue;
     fastModeController?: import('../../fast-mode.js').FastModeController;
     beforeNextRound?: () => string | undefined;
+    enableFileCheckpointing?: boolean;
 }
 export declare function starterModels(): ProviderModelInfo[];

package/dist/agent/providers/anthropic-direct/query-runtime.d.ts

@@ -57,7 +57,7 @@ export declare class AnthropicDirectQuery implements ProviderQuery {
         oldAccountId: string;
         swapped: boolean;
     } | null>;
-    rewindFiles(_userMessageId: string, _options?: {
+    rewindFiles(userMessageId: string, options?: {
         dryRun?: boolean;
     }): Promise<ProviderRewindResult>;
     compact(): Promise<ProviderCompactResult>;

package/dist/agent/providers/openai-compatible/query.d.ts

@@ -56,6 +56,7 @@ export declare class OpenAICompatibleQuery implements ProviderQuery, TurnDriverC
     get closed(): boolean;
     lastUsage: ProviderUsage | null;
     private readonly autoCompactThreshold;
+    private readonly checkpoint;
     constructor(opts: OpenAICompatibleQueryOptions);
     activeOpenAITools(): OpenAIFunctionTool[] | undefined;
     [Symbol.asyncIterator](): AsyncIterator<ProviderEvent>;
@@ -81,7 +82,7 @@ export declare class OpenAICompatibleQuery implements ProviderQuery, TurnDriverC
     getContextUsage(): Promise<ProviderContextUsage>;
     mcpServerStatus(): Promise<ProviderMcpServerStatus[]>;
     accountInfo(): Promise<ProviderAccountInfo>;
-    rewindFiles(_userMessageId: string, _options?: {
+    rewindFiles(userMessageId: string, options?: {
         dryRun?: boolean;
     }): Promise<ProviderRewindResult>;
     journalSnapshot(): ReturnType<OpenAIJournalWiring['snapshot']>;

package/dist/agent/providers/openai-compatible/query.file-checkpoint.d.ts

@@ -0,0 +1,14 @@
+import type { ToolHandlerContext } from '../../tools/types.js';
+import type { ProviderEvent, ProviderRewindResult } from '../../provider.js';
+export declare class FileCheckpointState {
+    private readonly enabled;
+    private readonly sessionId;
+    private currentTurnId;
+    constructor(enabled: boolean, sessionId: string);
+    begin(dispatcher: object | undefined, userMessageId?: string): string | undefined;
+    end(dispatcher: object | undefined): void;
+    events(events: AsyncIterable<ProviderEvent>, dispatcher: object | undefined, checkpointId: string | undefined): AsyncGenerator<ProviderEvent>;
+    rewind(userMessageId: string, options?: {
+        dryRun?: boolean;
+    }, writePolicy?: ToolHandlerContext): Promise<ProviderRewindResult>;
+}

package/dist/agent/providers/shared/chain-provider-close.d.ts

@@ -0,0 +1,2 @@
+import type { ProviderQuery } from '../../provider.js';
+export declare function chainProviderClose<Q extends ProviderQuery>(query: Q, closeProvider: () => void): Q;

package/dist/agent/session/stream-consumer.d.ts

@@ -20,5 +20,5 @@ export declare function parsePersistedOutput(content: string): {
     sizeBytes: number;
     absolutePath: string;
 } | null;
-export declare function usageToMetadata(usage: ProviderUsage, sessionId: string | undefined): ResponseMetadata;
+export declare function usageToMetadata(usage: ProviderUsage, sessionId: string | undefined, userMessageId?: string): ResponseMetadata;
 export declare function transformProviderEvent(event: ProviderEvent, deps: TransformDeps): OutputEvent | null;

package/dist/agent/tools/dispatcher.d.ts

@@ -16,6 +16,7 @@ import type { TraceSink } from '../trace/index.js';
 import { type UserAttention } from './user-yield.js';
 import { type DetachableToolRegistry } from './detach-bash.js';
 import type { ProcessJobRegistry } from '../shell-jobs/process-jobs.js';
+import { FileCheckpointWiring } from './dispatcher.file-checkpoint.js';
 export { defaultConcurrencyClassifier } from './dispatch-batching.js';
 export declare const REPEAT_CIRCUIT_BREAKER_THRESHOLD = 8;
 export { DEFAULT_MAX_CONCURRENT_SAFE_TOOL_CALLS, resolveMaxConcurrentSafeToolCalls };
@@ -79,6 +80,7 @@ export declare class SessionToolDispatcher implements ToolDispatcher {
     private readonly userAttention;
     private readonly detachRegistry;
     private readonly processJobs;
+    readonly fileCheckpointWiring: FileCheckpointWiring;
     private readonly bashOutputTailReporter;
     private readonly gateState;
     private readonly repeatFailureGuard;

package/dist/agent/tools/dispatcher.file-checkpoint.d.ts

@@ -0,0 +1,9 @@
+import type { FileCheckpointRegistry } from '../file-checkpoint/file-checkpoint.js';
+export type { FileCheckpointRegistry };
+export declare class FileCheckpointWiring {
+    private _registry;
+    set(registry: FileCheckpointRegistry | undefined): void;
+    contextSpread(): {
+        fileCheckpoint: FileCheckpointRegistry;
+    } | Record<never, never>;
+}

package/dist/agent/tools/types.d.ts

@@ -8,6 +8,7 @@ import type { SpawnedPidRegistry } from './handlers/pid-registry.js';
 import type { UserAttention } from './user-yield.js';
 import type { DetachableToolRegistry } from './detach-registry.js';
 import type { ProcessJobRegistry } from '../shell-jobs/process-jobs.js';
+import type { FileCheckpointRegistry } from '../file-checkpoint/file-checkpoint.js';
 export interface ToolHandlerContext {
     resolveBase?: string;
     cwd?: string;
@@ -23,6 +24,7 @@ export interface ToolHandlerContext {
     userAttention?: UserAttention;
     detachRegistry?: DetachableToolRegistry;
     processJobs?: ProcessJobRegistry;
+    fileCheckpoint?: FileCheckpointRegistry;
 }
 export type ToolHandler = (input: unknown, signal: AbortSignal, context?: ToolHandlerContext) => Promise<ToolResult>;
 export type ConcurrencyClassifier = (toolName: string, input?: unknown) => boolean;

package/dist/agent/types/message-types.d.ts

@@ -2,6 +2,7 @@ import type { DiffPayload } from '../../utils/diff.js';
 import type { ToolFailureClass } from '../trace/types.js';
 export type MessageRole = 'user' | 'assistant';
 export interface ResponseMetadata extends Record<string, unknown> {
+    userMessageId?: string;
     sessionId?: string;
     stopReason?: string | null;
     resultSubtype?: string;