@themoltnet/pi-extension 0.1.0 → 0.3.0

package/README.md

@@ -119,4 +119,4 @@ import {
 } from '@themoltnet/pi-extension';
 ```
 
-See `tools/src/resolve-issue.ts` for a complete example of headless usage with `createAgentSession`.
+See `tools/src/tasks/fulfill-brief.ts` for a complete example of headless usage: it synthesizes a `fulfill_brief` Task from a GitHub issue and executes it via `createPiTaskExecutor` + `AgentRuntime`.

package/dist/index.d.ts

@@ -1,9 +1,233 @@
-import type { ExtensionAPI } from '@mariozechner/pi-coding-agent';
-export default function moltnetExtension(pi: ExtensionAPI): void;
-export { createMoltNetTools } from './moltnet-tools.js';
-export type { EnsureSnapshotOptions, SandboxConfig, SnapshotConfig, } from './snapshot.js';
-export { ensureSnapshot } from './snapshot.js';
-export { createGondolinBashOps, createGondolinEditOps, createGondolinReadOps, createGondolinWriteOps, toGuestPath, } from './tool-operations.js';
-export type { ManagedVm, VmConfig, VmCredentials } from './vm-manager.js';
-export { activateAgentEnv, findMainWorktree, loadCredentials, resumeVm, } from './vm-manager.js';
-//# sourceMappingURL=index.d.ts.map
+import { BashOperations } from '@mariozechner/pi-coding-agent';
+import { connect } from '@themoltnet/sdk';
+import { EditOperations } from '@mariozechner/pi-coding-agent';
+import { ExtensionAPI } from '@mariozechner/pi-coding-agent';
+import { ReadOperations } from '@mariozechner/pi-coding-agent';
+import { Task } from '../../../tasks/dist/index.d.ts';
+import { TaskOutput } from '../../../tasks/dist/index.d.ts';
+import { TaskReporter } from '../../../agent-runtime/dist/index.d.ts';
+import { ToolDefinition } from '@mariozechner/pi-coding-agent';
+import { VM } from '@earendil-works/gondolin';
+import { WriteOperations } from '@mariozechner/pi-coding-agent';
+
+/**
+ * Apply agent env vars to the host process, mirroring `moltnet start`.
+ * Resolves relative paths (e.g. GIT_CONFIG_GLOBAL) against the repo root.
+ */
+export declare function activateAgentEnv(agentEnv: Record<string, string | undefined>, repoRoot: string): void;
+
+export declare function buildPiJudgeRecipeManifest(inputs: PiJudgeRecipeInputs): PiJudgeRecipeManifest;
+
+export declare function computePiJudgeRecipeCid(inputs: PiJudgeRecipeInputs): PiJudgeRecipeCid;
+
+export declare function createGondolinBashOps(vm: VM, localCwd: string): BashOperations;
+
+export declare function createGondolinEditOps(vm: VM, localCwd: string): EditOperations;
+
+export declare function createGondolinReadOps(vm: VM, localCwd: string): ReadOperations;
+
+export declare function createGondolinWriteOps(vm: VM, localCwd: string): WriteOperations;
+
+/**
+ * Create all MoltNet tool definitions, ready to pass to `pi.registerTool()`.
+ */
+export declare function createMoltNetTools(config: MoltNetToolsConfig): ToolDefinition<any, any>[];
+
+/**
+ * Factory that builds a pi-specific `executeTask` function suitable for
+ * injection into `AgentRuntime`. The returned function caches the resolved
+ * checkpoint across tasks so the second task hits the snapshot cache.
+ */
+export declare function createPiTaskExecutor(opts: ExecutePiTaskOptions): (task: Task, reporter: TaskReporter) => Promise<TaskOutput>;
+
+/**
+ * Ensure a cached snapshot exists, building one if needed.
+ * Returns the absolute path to the qcow2 checkpoint file.
+ */
+export declare function ensureSnapshot(options?: EnsureSnapshotOptions): Promise<string>;
+
+export declare interface EnsureSnapshotOptions {
+    config?: SnapshotConfig;
+    onProgress?: (message: string) => void;
+    /** Max number of old snapshots to keep (default 1). */
+    maxCached?: number;
+}
+
+/**
+ * Run one attempt of `task` in a freshly-resumed Gondolin VM. Owns the full
+ * lifecycle: resume VM → wire tools → pi session → close VM. Always returns
+ * a `TaskOutput` (failures surface as `status: 'failed'`); throws only on
+ * unrecoverable setup errors.
+ */
+export declare function executePiTask(task: Task, reporter: TaskReporter, opts: ExecutePiTaskOptions): Promise<TaskOutput>;
+
+export declare interface ExecutePiTaskOptions {
+    /** MoltNet agent whose credentials the VM boots with. */
+    agentName: string;
+    /** Host cwd that the VM mounts at /workspace (defaults to `process.cwd()`). */
+    mountPath?: string;
+    /** LLM selection. */
+    provider: string;
+    model: string;
+    /** Extra hosts to allow in the sandbox egress policy. */
+    extraAllowedHosts?: string[];
+    /** Sandbox overrides (env, VFS shadows, resources). */
+    sandboxConfig?: SandboxConfig;
+    /** Forwarded to `buildPromptForTask` for per-type builders. */
+    promptExtras?: Record<string, unknown>;
+    /** Snapshot progress callback; defaults to stderr logging. */
+    onSnapshotProgress?: (message: string) => void;
+    /** Attempt number; defaults to 1. */
+    attemptN?: number;
+    /**
+     * Optional pre-resolved checkpoint path. If omitted, `ensureSnapshot` is
+     * invoked. Useful for batch execution where the caller wants to cache
+     * across tasks.
+     */
+    checkpointPath?: string;
+}
+
+/**
+ * Resolve the main worktree root (where .moltnet/ lives — it's untracked,
+ * only exists in the main worktree, not in git worktrees).
+ */
+export declare function findMainWorktree(): string;
+
+export declare function loadCredentials(agentDir: string): VmCredentials;
+
+export declare interface ManagedVm {
+    vm: VM;
+    credentials: VmCredentials;
+    mountPath: string;
+    guestWorkspace: string;
+    agentDir: string;
+}
+
+declare type MoltNetAgent = Awaited<ReturnType<typeof connect>>;
+
+declare function moltnetExtension(pi: ExtensionAPI): void;
+export default moltnetExtension;
+
+declare interface MoltNetToolsConfig {
+    getAgent(): MoltNetAgent | null;
+    getDiaryId(): string | null;
+    getSessionErrors(): readonly TrackedError[];
+    clearSessionErrors(): void;
+}
+
+declare interface PiJudgeRecipeCid {
+    cid: string;
+    manifest: PiJudgeRecipeManifest;
+}
+
+declare interface PiJudgeRecipeInputs {
+    judgePrompt: string;
+    rubric: string;
+    skillFragment?: string | null;
+    implementationSource?: string | null;
+    promptAsset?: string | null;
+    rubricAsset?: string | null;
+    skillSourcePath?: string | null;
+    overrides?: Partial<PiJudgeRecipeVersions>;
+}
+
+declare interface PiJudgeRecipeManifest {
+    kind: 'pi-judge-recipe/v1';
+    versions: PiJudgeRecipeVersions;
+    assets: {
+        promptAsset: string | null;
+        rubricAsset: string | null;
+        skillSourcePath: string | null;
+    };
+    hashes: {
+        judgePromptSha256: string;
+        rubricSha256: string;
+        skillFragmentSha256: string | null;
+        implementationSha256: string | null;
+    };
+}
+
+declare interface PiJudgeRecipeVersions {
+    pi: string | null;
+    piExtension: string | null;
+    sdk: string | null;
+}
+
+export declare function resolvePiJudgeRecipeVersions(): PiJudgeRecipeVersions;
+
+/**
+ * Resume a VM from a checkpoint, inject credentials, configure egress +
+ * TLS. Returns the managed VM handle.
+ */
+export declare function resumeVm(config: VmConfig): Promise<ManagedVm>;
+
+export declare interface SandboxConfig {
+    /** Snapshot build settings. */
+    snapshot?: {
+        /** Shell commands to run after the base setup. */
+        setupCommands?: string[];
+        /** Additional hosts to allow network access during build. */
+        allowedHosts?: string[];
+        /** Overlay disk size (default '3G'). */
+        overlaySize?: string;
+    };
+    /** VFS shadow settings — hide host paths from the guest. */
+    vfs?: {
+        /** Paths (relative to workspace root) to shadow from the host mount. */
+        shadow?: string[];
+        /** What to do with writes to shadowed paths: 'deny' or 'tmpfs' (default 'tmpfs'). */
+        shadowMode?: 'deny' | 'tmpfs';
+    };
+    /** Environment variable overrides for the guest VM (applied on top of defaults). */
+    env?: Record<string, string>;
+    /** VM resource allocation. */
+    resources?: {
+        /** Memory size in qemu syntax (default '1G'). */
+        memory?: string;
+        /** CPU count (default 2). */
+        cpus?: number;
+    };
+}
+
+/** Extract snapshot-specific config for backwards compat with ensureSnapshot. */
+export declare type SnapshotConfig = NonNullable<SandboxConfig['snapshot']>;
+
+/**
+ * Map a host-side absolute path to a guest-side /workspace path.
+ * Throws if the path escapes the workspace.
+ */
+export declare function toGuestPath(localCwd: string, localPath: string): string;
+
+declare interface TrackedError {
+    toolName: string;
+    toolCallId: string;
+    input: Record<string, unknown>;
+    error: string;
+    timestamp: number;
+}
+
+export declare interface VmConfig {
+    /** Absolute path to the qcow2 checkpoint. */
+    checkpointPath: string;
+    /** MoltNet agent name (used to resolve credentials). */
+    agentName: string;
+    /** Host directory to mount at /workspace in the VM. */
+    mountPath: string;
+    /** Additional hosts to allow in egress policy. */
+    extraAllowedHosts?: string[];
+    /** Full sandbox config (vfs shadows, env overrides). */
+    sandboxConfig?: SandboxConfig;
+}
+
+export declare interface VmCredentials {
+    moltnetJson: string;
+    agentEnvRaw: string;
+    piAuthJson: string;
+    agentEnv: Record<string, string | undefined>;
+    gitconfig: string | null;
+    sshPrivateKey: string | null;
+    sshPublicKey: string | null;
+    allowedSigners: string | null;
+}
+
+export { }