@shivaedev/agent-pi 0.2.0 → 0.3.0

package/dist/makeOperations.js

@@ -0,0 +1,55 @@
+import { Effect, Exit, Scope, Stream } from "effect";
+import { UnknownAcceptance } from "@shivaedev/agent/failures.ts";
+import { protocolFailure } from "#adapters/failure.ts";
+import { openSession } from "#adapters/openSession.ts";
+import { textOf } from "#adapters/textOf.ts";
+export const makeOperations = Effect.fnUntraced(function* (options) {
+    const runs = yield* Effect.sync(() => new Map());
+    const evidence = new Map();
+    function lookup(receipt) {
+        return Effect.suspend(() => {
+            const run = runs.get(receipt.operationId);
+            return receipt.backend === "pi" && receipt.reference.version === 1 && run?.receipt.reference.value === receipt.reference.value
+                ? Effect.succeed(run)
+                : Effect.fail(protocolFailure("run is not open in this backend scope"));
+        });
+    }
+    const submit = Effect.fnUntraced(function* (request, resume) {
+        const run = yield* openSession(options, request, resume);
+        yield* run.prompt(textOf(request.input), "steer").pipe(Effect.mapError((error) => {
+            if (error._tag === "Rejected") {
+                evidence.set(request.operationId, { evidence: error.message, operationId: request.operationId, type: "not-accepted" });
+                return error;
+            }
+            return new UnknownAcceptance({ backend: "pi", message: error.message, operationId: request.operationId });
+        }));
+        runs.set(request.operationId, run);
+        evidence.set(request.operationId, { receipt: run.receipt, type: "accepted" });
+        yield* run.acknowledge;
+        yield* Effect.addFinalizer(() => Effect.sync(() => {
+            runs.delete(request.operationId);
+        }));
+        return run.receipt;
+    });
+    const open = Effect.fnUntraced(function* (request, resume) {
+        if (evidence.has(request.operationId)) {
+            return yield* Effect.fail(protocolFailure("operation already attempted; reconcile instead"));
+        }
+        evidence.set(request.operationId, { operationId: request.operationId, reason: "submission has no acceptance evidence yet", type: "uncertain" });
+        const runScope = yield* Scope.fork(yield* Effect.scope);
+        return yield* submit(request, resume).pipe(Effect.provideService(Scope.Scope, runScope), Effect.onError((cause) => Scope.close(runScope, Exit.failCause(cause))));
+    });
+    return {
+        events: (receipt) => Stream.unwrap(lookup(receipt).pipe(Effect.map((run) => run.events))),
+        inspect: (receipt) => lookup(receipt).pipe(Effect.flatMap((run) => run.inspect), Effect.catchTag("ProtocolFailure", () => Effect.succeed({ reason: "no live execution evidence in this backend scope", type: "missing" }))),
+        lookup,
+        open,
+        reconcile: (operationId) => Effect.sync(() => evidence.get(operationId) ?? {
+            operationId,
+            reason: "Pi has no operation-id lookup; missing local evidence does not prove non-acceptance",
+            type: "uncertain",
+        }),
+        resume: (reference, request) => reference.version === 1 ? open(request, reference.value) : Effect.fail(protocolFailure("unsupported session reference version")),
+    };
+});
+//# sourceMappingURL=makeOperations.js.map

package/dist/makeOperations.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"makeOperations.js","sourceRoot":"","sources":["../src/makeOperations.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AACrD,OAAO,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AAEjE,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AACvD,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAG7C,MAAM,CAAC,MAAM,cAAc,GAAG,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,OAAkB;IAC5E,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,GAAG,EAA0D,CAAC,CAAC;IACzG,MAAM,QAAQ,GAAG,IAAI,GAAG,EAA0B,CAAC;IACnD,SAAS,MAAM,CAAC,OAAgB;QAC/B,OAAO,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE;YAC1B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAC1C,OAAO,OAAO,CAAC,OAAO,KAAK,IAAI,IAAI,OAAO,CAAC,SAAS,CAAC,OAAO,KAAK,CAAC,IAAI,GAAG,EAAE,OAAO,CAAC,SAAS,CAAC,KAAK,KAAK,OAAO,CAAC,SAAS,CAAC,KAAK;gBAC7H,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC;gBACrB,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,uCAAuC,CAAC,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;IACJ,CAAC;IACD,MAAM,MAAM,GAAG,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,OAAqB,EAAE,MAA0B;QAC5F,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,WAAW,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QACzD,KAAK,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,OAAO,CAAC,CAAC,IAAI,CACrD,MAAM,CAAC,QAAQ,CAAC,CAAC,KAAK,EAAE,EAAE;YACzB,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC/B,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,QAAQ,EAAE,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE,OAAO,CAAC,WAAW,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;gBACvH,OAAO,KAAK,CAAC;YACd,CAAC;YACD,OAAO,IAAI,iBAAiB,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;QAC3G,CAAC,CAAC,CACF,CAAC;QACF,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QACnC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;QAC9E,KAAK,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC;QACvB,KAAK,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,GAAG,EAAE,CAC/B,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE;YAChB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QAClC,CAAC,CAAC,CACF,CAAC;QACF,OAAO,GAAG,CAAC,OAAO,CAAC;IACpB,CAAC,CAAC,CAAC;IACH,MAAM,IAAI,GAAG,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,OAAqB,EAAE,MAA0B;QAC1F,IAAI,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;YACvC,OAAO,KAAK,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,gDAAgD,CAAC,CAAC,CAAC;QAC9F,CAAC;QACD,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,WAAW,EAAE,OAAO,CAAC,WAAW,EAAE,MAAM,EAAE,2CAA2C,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;QAChJ,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACxD,OAAO,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,IAAI,CACzC,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,KAAK,EAAE,QAAQ,CAAC,EAC5C,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CACvE,CAAC;IACH,CAAC,CAAC,CAAC;IACH,OAAO;QACN,MAAM,EAAE,CAAC,OAAgB,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QAClG,OAAO,EAAE,CAAC,OAAgB,EAAE,EAAE,CAC7B,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CACnB,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,EACpC,MAAM,CAAC,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE,CACvC,MAAM,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,kDAAkD,EAAE,IAAI,EAAE,SAAkB,EAAE,CAAC,CACxG,CACD;QACF,MAAM;QACN,IAAI;QACJ,SAAS,EAAE,CAAC,WAAmB,EAAE,EAAE,CAClC,MAAM,CAAC,IAAI,CACV,GAAmB,EAAE,CACpB,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI;YAC5B,WAAW;YACX,MAAM,EAAE,qFAAqF;YAC7F,IAAI,EAAE,WAAW;SACjB,CACF;QACF,MAAM,EAAE,CAAC,SAA+D,EAAE,OAAqB,EAAE,EAAE,CAClG,SAAS,CAAC,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,uCAAuC,CAAC,CAAC;KACjI,CAAC;AACH,CAAC,CAAC,CAAC"}

package/dist/pi.d.ts

@@ -1,2 +1,5 @@
-export {};
+import { Layer } from "effect";
+import { Backend } from "@shivaedev/agent/backend.ts";
+import type { PiOptions } from "#PiOptions.ts";
+export declare function pi(options?: PiOptions): Layer.Layer<Backend>;
 //# sourceMappingURL=pi.d.ts.map

package/dist/pi.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"pi.d.ts","sourceRoot":"","sources":["../src/pi.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC"}
+{"version":3,"file":"pi.d.ts","sourceRoot":"","sources":["../src/pi.ts"],"names":[],"mappings":"AAAA,OAAO,EAAU,KAAK,EAAE,MAAM,QAAQ,CAAC;AACvC,OAAO,EAAE,OAAO,EAA0B,MAAM,6BAA6B,CAAC;AAI9E,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAE/C,wBAAgB,EAAE,CAAC,OAAO,GAAE,SAAc,GAAG,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAoChE"}

package/dist/pi.js

@@ -1,2 +1,40 @@
-export {};
+import { Effect, Layer } from "effect";
+import { Backend } from "@shivaedev/agent/backend.ts";
+import { listPiModels } from "#adapters/listPiModels.ts";
+import { textOf } from "#adapters/textOf.ts";
+import { makeOperations } from "#makeOperations.ts";
+export function pi(options = {}) {
+    return Layer.effect(Backend, Effect.gen(function* () {
+        const operations = yield* makeOperations(options);
+        return {
+            backend: "pi",
+            events: operations.events,
+            features: {
+                imageInput: { reason: "only text input is implemented", support: "unsupported" },
+                interruption: {
+                    implementation: { interrupt: (receipt) => operations.lookup(receipt).pipe(Effect.flatMap((run) => run.interrupt)) },
+                    support: "supported",
+                },
+                models: { implementation: { list: listPiModels }, support: "supported" },
+                queuedInput: {
+                    implementation: {
+                        queue: (receipt, input) => operations.lookup(receipt).pipe(Effect.flatMap((run) => run.prompt(textOf(input), "followUp"))),
+                    },
+                    support: "supported",
+                },
+                sessions: { implementation: { resume: operations.resume }, support: "supported" },
+                steering: {
+                    implementation: {
+                        steer: (receipt, input) => operations.lookup(receipt).pipe(Effect.flatMap((run) => run.prompt(textOf(input), "steer"))),
+                    },
+                    support: "supported",
+                },
+                transcript: { reason: "no provider history reader is implemented", support: "unsupported" },
+            },
+            inspect: operations.inspect,
+            reconcile: operations.reconcile,
+            start: (request) => operations.open(request, undefined),
+        };
+    }));
+}
 //# sourceMappingURL=pi.js.map

package/dist/pi.js.map

@@ -1 +1 @@
-{"version":3,"file":"pi.js","sourceRoot":"","sources":["../src/pi.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC"}
+{"version":3,"file":"pi.js","sourceRoot":"","sources":["../src/pi.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,QAAQ,CAAC;AACvC,OAAO,EAAE,OAAO,EAA0B,MAAM,6BAA6B,CAAC;AAC9E,OAAO,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AACzD,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAGpD,MAAM,UAAU,EAAE,CAAC,OAAO,GAAc,EAAE;IACzC,OAAO,KAAK,CAAC,MAAM,CAClB,OAAO,EACP,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;QACnB,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAClD,OAAO;YACN,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,UAAU,CAAC,MAAM;YACzB,QAAQ,EAAE;gBACT,UAAU,EAAE,EAAE,MAAM,EAAE,gCAAgC,EAAE,OAAO,EAAE,aAAa,EAAE;gBAChF,YAAY,EAAE;oBACb,cAAc,EAAE,EAAE,SAAS,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,EAAE;oBACnH,OAAO,EAAE,WAAW;iBACpB;gBACD,MAAM,EAAE,EAAE,cAAc,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE;gBACxE,WAAW,EAAE;oBACZ,cAAc,EAAE;wBACf,KAAK,EAAE,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC;qBAC1H;oBACD,OAAO,EAAE,WAAW;iBACpB;gBACD,QAAQ,EAAE,EAAE,cAAc,EAAE,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE;gBACjF,QAAQ,EAAE;oBACT,cAAc,EAAE;wBACf,KAAK,EAAE,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;qBACvH;oBACD,OAAO,EAAE,WAAW;iBACpB;gBACD,UAAU,EAAE,EAAE,MAAM,EAAE,2CAA2C,EAAE,OAAO,EAAE,aAAa,EAAE;aAC3F;YACD,OAAO,EAAE,UAAU,CAAC,OAAO;YAC3B,SAAS,EAAE,UAAU,CAAC,SAAS;YAC/B,KAAK,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC;SACvD,CAAC;IACH,CAAC,CAAC,CACF,CAAC;AACH,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shivaedev/agent-pi",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "description": "Pi integration for Effect applications",
   "homepage": "https://github.com/ShivaeDev/platform/tree/main/packages/agent-pi#readme",
   "bugs": {
@@ -19,7 +19,8 @@
       "@shivaedev/source": "./src/*.ts",
       "types": "./dist/*.d.ts",
       "default": "./dist/*.js"
-    }
+    },
+    "#test/*": "./src/test-support/*"
   },
   "exports": {
     "./*.ts": {
@@ -32,12 +33,33 @@
   "files": [
     "dist",
     "src",
+    "!src/**/*.test.ts",
+    "!src/**/*.spec.ts",
+    "!src/**/test-support",
     "CHANGELOG.md",
     "README.md"
   ],
+  "dependencies": {
+    "@shivaedev/agent": "0.3.0"
+  },
   "devDependencies": {
+    "@earendil-works/pi-coding-agent": "0.84.4",
+    "@effect/vitest": "4.0.6",
+    "@shivaedev/quality": "0.61.0",
+    "@shivaedev/test-story": "0.39.0",
     "@types/node": "24.19.2",
-    "typescript": "7.0.2"
+    "effect": "4.0.6",
+    "typescript": "7.0.2",
+    "vitest": "5.0.5"
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-coding-agent": "0.84.4",
+    "effect": "4.0.6"
+  },
+  "peerDependenciesMeta": {
+    "@earendil-works/pi-coding-agent": {
+      "optional": true
+    }
   },
   "engines": {
     "node": ">=24"
@@ -48,7 +70,8 @@
   },
   "scripts": {
     "build": "node --eval \"import('node:fs').then(({ rmSync }) => rmSync('dist', { force: true, recursive: true }))\" && tsc --project tsconfig.build.json",
-    "ready": "pnpm typecheck && pnpm build",
+    "ready": "pnpm typecheck && pnpm test && pnpm build",
+    "test": "vitest run --configLoader native",
     "typecheck": "tsc --noEmit"
   }
 }

package/src/PiOptions.ts

@@ -0,0 +1,4 @@
+export interface PiOptions {
+	readonly constrainedPrompt?: string;
+	readonly skills?: readonly string[];
+}

package/src/adapters/deliver.ts

@@ -0,0 +1,24 @@
+import type { AgentSession } from "@earendil-works/pi-coding-agent";
+import { Effect } from "effect";
+import type { BackendFailure } from "@shivaedev/agent/failures.ts";
+import { protocolFailure, rejected } from "#adapters/failure.ts";
+
+export function deliver(session: AgentSession, text: string, delivery: "followUp" | "steer") {
+	return Effect.callback<void, BackendFailure>((resume) => {
+		let preflight: boolean | undefined;
+		session
+			.prompt(text, {
+				preflightResult: (ok) => {
+					preflight = ok;
+					if (ok) {
+						resume(Effect.void);
+					}
+				},
+				streamingBehavior: delivery,
+			})
+			.then(
+				() => resume(Effect.void),
+				(error: unknown) => resume(Effect.fail(preflight === false ? rejected(error) : protocolFailure(error))),
+			);
+	});
+}

package/src/adapters/effort.ts

@@ -0,0 +1,15 @@
+import { Effect } from "effect";
+import { rejected } from "#adapters/failure.ts";
+
+export const piEfforts = ["off", "minimal", "low", "medium", "high", "xhigh", "max"] as const;
+export type PiThinkingLevel = (typeof piEfforts)[number];
+
+export function thinkingLevel(effort: string | undefined) {
+	if (effort === undefined) {
+		return Effect.succeed(undefined);
+	}
+	const level = piEfforts.find((candidate) => candidate === effort);
+	return level === undefined
+		? Effect.fail(rejected(`effort ${effort} is not a thinking level Pi offers; choose one of ${piEfforts.join(", ")}`))
+		: Effect.succeed(level);
+}

package/src/adapters/failure.ts

@@ -0,0 +1,14 @@
+import { BackendUnavailable, ProtocolFailure, Rejected } from "@shivaedev/agent/failures.ts";
+
+export function describe(detail: unknown): string {
+	return detail instanceof Error ? detail.message : String(detail);
+}
+export function unavailable(detail: unknown) {
+	return new BackendUnavailable({ backend: "pi", message: describe(detail) });
+}
+export function protocolFailure(detail: unknown) {
+	return new ProtocolFailure({ backend: "pi", message: describe(detail) });
+}
+export function rejected(detail: unknown) {
+	return new Rejected({ backend: "pi", message: describe(detail) });
+}

package/src/adapters/listPiModels.ts

@@ -0,0 +1,16 @@
+import { Effect } from "effect";
+import { piEfforts } from "#adapters/effort.ts";
+import { unavailable } from "#adapters/failure.ts";
+import { loadSdk } from "#adapters/runtime.ts";
+
+export const listPiModels = loadSdk.pipe(
+	Effect.flatMap((sdk) => Effect.tryPromise({ catch: unavailable, try: async () => (await sdk.ModelRuntime.create()).getAvailable() })),
+	Effect.map((models) =>
+		models.map((model, index) => ({
+			efforts: piEfforts,
+			id: `${model.provider}/${model.id}`,
+			isDefault: index === 0,
+			name: `${model.name} (${model.provider})`,
+		})),
+	),
+);

package/src/adapters/mapping.ts

@@ -0,0 +1,50 @@
+import type { AgentSessionEvent } from "@earendil-works/pi-coding-agent";
+import { Option, Schema } from "effect";
+import type { RunEvent } from "@shivaedev/agent/events.ts";
+import { messageEvents } from "./messageEvents.ts";
+
+export function rawOf(kind: string, payload: unknown) {
+	return { kind, payload: JSON.stringify(payload), source: "pi" };
+}
+const decodeOutput = Schema.decodeUnknownOption(
+	Schema.Struct({ content: Schema.Array(Schema.Struct({ text: Schema.optional(Schema.String), type: Schema.String })) }),
+);
+
+export function toAgentEvents(event: AgentSessionEvent): RunEvent[] {
+	const raw = rawOf(event.type, event);
+	switch (event.type) {
+		case "agent_start":
+			return [{ raw, state: "running", type: "execution" }];
+		case "agent_end": {
+			if (event.willRetry) {
+				return [{ raw, type: "raw" }];
+			}
+			const ending = event.messages.filter((message) => message.role === "assistant").at(-1);
+			const state = endState(ending?.stopReason);
+			return [{ raw, state, type: "execution" }];
+		}
+		case "message_end":
+			return messageEvents(event.message, raw);
+		case "tool_execution_start":
+			return [{ input: JSON.stringify(event.args), name: event.toolName, raw, toolId: event.toolCallId, type: "tool.started" }];
+		case "tool_execution_end": {
+			const output = Option.match(decodeOutput(event.result), {
+				onNone: () => "",
+				onSome: ({ content }) => content.flatMap((part) => (part.type === "text" && part.text !== undefined ? [part.text] : [])).join("\n"),
+			});
+			return [{ ok: !event.isError, output, raw, toolId: event.toolCallId, type: "tool.completed" }];
+		}
+		default:
+			return [{ raw, type: "raw" }];
+	}
+}
+
+function endState(reason: string | undefined) {
+	if (reason === "aborted") {
+		return "interrupted";
+	}
+	if (reason === "error") {
+		return "failed";
+	}
+	return "completed";
+}

package/src/adapters/messageEvents.ts

@@ -0,0 +1,50 @@
+import type { AgentSessionEvent } from "@earendil-works/pi-coding-agent";
+import type { RunEvent } from "@shivaedev/agent/events.ts";
+
+type PiMessage = Extract<AgentSessionEvent, { type: "message_end" }>["message"];
+type Raw = NonNullable<RunEvent["raw"]>;
+
+export function messageEvents(message: PiMessage, raw: Raw): RunEvent[] {
+	if (message.role === "user") {
+		const text =
+			typeof message.content === "string" ? message.content : message.content.flatMap((part) => (part.type === "text" ? [part.text] : [])).join("\n");
+		return [{ parts: [{ text, type: "text" }], raw, role: "user", type: "message" }];
+	}
+	if (message.role !== "assistant") {
+		return [{ raw, type: "raw" }];
+	}
+	const events: RunEvent[] = [];
+	const thinking = message.content.flatMap((part) => (part.type === "thinking" ? [part.thinking] : [])).join("\n");
+	const text = message.content.flatMap((part) => (part.type === "text" ? [part.text] : [])).join("\n");
+	if (thinking !== "") {
+		events.push({ raw, text: thinking, type: "thinking" });
+	}
+	if (text !== "") {
+		events.push({ parts: [{ text, type: "text" }], raw, role: "agent", type: "message" });
+	}
+	const model = `${message.provider}/${message.model}`;
+	events.push(
+		{
+			cacheReadTokens: message.usage.cacheRead,
+			cacheWriteTokens: message.usage.cacheWrite,
+			inputTokens: message.usage.input,
+			model,
+			outputTokens: message.usage.output,
+			raw,
+			scope: "delta",
+			type: "usage",
+			unit: "turn",
+		},
+		{
+			amount: message.usage.cost.total,
+			currency: "USD",
+			model,
+			raw,
+			scope: "delta",
+			source: "provider",
+			type: "cost",
+			unit: "turn",
+		},
+	);
+	return events;
+}

package/src/adapters/openSession.ts

@@ -0,0 +1,46 @@
+import { type Cause, Effect, Queue, Stream } from "effect";
+import type { RunEvent } from "@shivaedev/agent/events.ts";
+import type { Observation, Receipt, StartRequest } from "@shivaedev/agent/run.ts";
+import { deliver } from "#adapters/deliver.ts";
+import { unavailable } from "#adapters/failure.ts";
+import { rawOf, toAgentEvents } from "#adapters/mapping.ts";
+import { openRuntime } from "#adapters/runtime.ts";
+import type { PiOptions } from "#PiOptions.ts";
+
+export const openSession = Effect.fnUntraced(function* (options: PiOptions, request: StartRequest, resume: string | undefined) {
+	const { session, sessionFile } = yield* openRuntime(options, request, resume);
+	const output = yield* Queue.unbounded<RunEvent, Cause.Done>();
+	let observation: Observation = { state: "accepted", type: "known" };
+	function emit(event: RunEvent) {
+		if (event.type === "execution") {
+			observation = { state: event.state, type: "known" };
+		}
+		Queue.offerUnsafe(output, event);
+	}
+	const unsubscribe = session.subscribe((event) => {
+		for (const mapped of toAgentEvents(event)) {
+			emit(mapped);
+		}
+	});
+	yield* Effect.addFinalizer(() =>
+		Effect.sync(() => {
+			unsubscribe();
+			Queue.endUnsafe(output);
+		}),
+	);
+	const receipt: Receipt = {
+		backend: "pi",
+		operationId: request.operationId,
+		reference: { value: sessionFile, version: 1 },
+		session: { value: sessionFile, version: 1 },
+	};
+	emit({ raw: rawOf("session/opened", { sessionFile, sessionId: session.sessionId }), reference: sessionFile, type: "session.opened" });
+	return {
+		acknowledge: Effect.sync(() => emit({ receipt, type: "acknowledged" })),
+		events: Stream.fromQueue(output),
+		inspect: Effect.sync(() => observation),
+		interrupt: Effect.tryPromise({ catch: unavailable, try: () => session.abort() }),
+		prompt: (text: string, delivery: "followUp" | "steer") => deliver(session, text, delivery),
+		receipt,
+	};
+});

package/src/adapters/runtime.ts

@@ -0,0 +1,70 @@
+import { access } from "node:fs/promises";
+import type { AgentSession } from "@earendil-works/pi-coding-agent";
+import { Effect } from "effect";
+import type { StartRequest } from "@shivaedev/agent/run.ts";
+import { thinkingLevel } from "#adapters/effort.ts";
+import { rejected, unavailable } from "#adapters/failure.ts";
+import type { PiOptions } from "#PiOptions.ts";
+
+export const loadSdk = Effect.tryPromise({ catch: unavailable, try: () => import("@earendil-works/pi-coding-agent") });
+
+export const openRuntime = Effect.fnUntraced(function* (options: PiOptions, request: StartRequest, resume: string | undefined) {
+	const effort = yield* thinkingLevel(request.effort);
+	if (resume !== undefined) {
+		yield* Effect.tryPromise({ catch: () => unavailable("Pi session file is unavailable for resume"), try: () => access(resume) });
+	}
+	const sdk = yield* loadSdk;
+	const modelRuntime = yield* Effect.tryPromise({ catch: unavailable, try: () => sdk.ModelRuntime.create() });
+	const models = yield* Effect.tryPromise({ catch: unavailable, try: () => modelRuntime.getAvailable() });
+	const model = request.model === undefined ? models[0] : models.find((value) => `${value.provider}/${value.id}` === request.model);
+	if (model === undefined) {
+		return yield* Effect.fail(
+			rejected(`Pi offers no model ${request.model ?? "(default)"} with credentials; choose provider/model-id from its catalog`),
+		);
+	}
+	const session = yield* Effect.acquireRelease(
+		Effect.tryPromise({
+			catch: unavailable,
+			try: async () => {
+				const agentDir = sdk.getAgentDir();
+				const resourceLoader = new sdk.DefaultResourceLoader({
+					agentDir,
+					cwd: request.cwd,
+					...resourceOptions(options),
+				});
+				await resourceLoader.reload();
+				const { session: opened } = await sdk.createAgentSession({
+					agentDir,
+					customTools: [],
+					cwd: request.cwd,
+					model,
+					modelRuntime,
+					resourceLoader,
+					sessionManager: resume === undefined ? sdk.SessionManager.create(request.cwd) : sdk.SessionManager.open(resume, undefined, request.cwd),
+					...(options.constrainedPrompt === undefined ? {} : { tools: [] }),
+					...(effort === undefined ? {} : { thinkingLevel: effort }),
+				});
+				return opened;
+			},
+		}),
+		(openedSession: AgentSession) => Effect.promise(() => openedSession.abort()).pipe(Effect.ensuring(Effect.sync(() => openedSession.dispose()))),
+	);
+	if (session.sessionFile === undefined) {
+		return yield* Effect.fail(unavailable("Pi opened a session without a persisted file reference"));
+	}
+	return { session, sessionFile: session.sessionFile };
+});
+
+function resourceOptions(options: PiOptions) {
+	if (options.constrainedPrompt === undefined) {
+		return { additionalSkillPaths: [...(options.skills ?? [])] };
+	}
+	return {
+		noContextFiles: true,
+		noExtensions: true,
+		noPromptTemplates: true,
+		noSkills: true,
+		noThemes: true,
+		systemPrompt: options.constrainedPrompt,
+	};
+}

package/src/adapters/textOf.ts

@@ -0,0 +1,5 @@
+import type { Input } from "@shivaedev/agent/run.ts";
+
+export function textOf(input: Input) {
+	return input.parts.map((part) => part.text).join("\n");
+}

package/src/makeOperations.ts

@@ -0,0 +1,75 @@
+import { Effect, Exit, Scope, Stream } from "effect";
+import { UnknownAcceptance } from "@shivaedev/agent/failures.ts";
+import type { Receipt, Reconciliation, StartRequest } from "@shivaedev/agent/run.ts";
+import { protocolFailure } from "#adapters/failure.ts";
+import { openSession } from "#adapters/openSession.ts";
+import { textOf } from "#adapters/textOf.ts";
+import type { PiOptions } from "#PiOptions.ts";
+
+export const makeOperations = Effect.fnUntraced(function* (options: PiOptions) {
+	const runs = yield* Effect.sync(() => new Map<string, Effect.Success<ReturnType<typeof openSession>>>());
+	const evidence = new Map<string, Reconciliation>();
+	function lookup(receipt: Receipt) {
+		return Effect.suspend(() => {
+			const run = runs.get(receipt.operationId);
+			return receipt.backend === "pi" && receipt.reference.version === 1 && run?.receipt.reference.value === receipt.reference.value
+				? Effect.succeed(run)
+				: Effect.fail(protocolFailure("run is not open in this backend scope"));
+		});
+	}
+	const submit = Effect.fnUntraced(function* (request: StartRequest, resume: string | undefined) {
+		const run = yield* openSession(options, request, resume);
+		yield* run.prompt(textOf(request.input), "steer").pipe(
+			Effect.mapError((error) => {
+				if (error._tag === "Rejected") {
+					evidence.set(request.operationId, { evidence: error.message, operationId: request.operationId, type: "not-accepted" });
+					return error;
+				}
+				return new UnknownAcceptance({ backend: "pi", message: error.message, operationId: request.operationId });
+			}),
+		);
+		runs.set(request.operationId, run);
+		evidence.set(request.operationId, { receipt: run.receipt, type: "accepted" });
+		yield* run.acknowledge;
+		yield* Effect.addFinalizer(() =>
+			Effect.sync(() => {
+				runs.delete(request.operationId);
+			}),
+		);
+		return run.receipt;
+	});
+	const open = Effect.fnUntraced(function* (request: StartRequest, resume: string | undefined) {
+		if (evidence.has(request.operationId)) {
+			return yield* Effect.fail(protocolFailure("operation already attempted; reconcile instead"));
+		}
+		evidence.set(request.operationId, { operationId: request.operationId, reason: "submission has no acceptance evidence yet", type: "uncertain" });
+		const runScope = yield* Scope.fork(yield* Effect.scope);
+		return yield* submit(request, resume).pipe(
+			Effect.provideService(Scope.Scope, runScope),
+			Effect.onError((cause) => Scope.close(runScope, Exit.failCause(cause))),
+		);
+	});
+	return {
+		events: (receipt: Receipt) => Stream.unwrap(lookup(receipt).pipe(Effect.map((run) => run.events))),
+		inspect: (receipt: Receipt) =>
+			lookup(receipt).pipe(
+				Effect.flatMap((run) => run.inspect),
+				Effect.catchTag("ProtocolFailure", () =>
+					Effect.succeed({ reason: "no live execution evidence in this backend scope", type: "missing" as const }),
+				),
+			),
+		lookup,
+		open,
+		reconcile: (operationId: string) =>
+			Effect.sync(
+				(): Reconciliation =>
+					evidence.get(operationId) ?? {
+						operationId,
+						reason: "Pi has no operation-id lookup; missing local evidence does not prove non-acceptance",
+						type: "uncertain",
+					},
+			),
+		resume: (reference: { readonly value: string; readonly version: number }, request: StartRequest) =>
+			reference.version === 1 ? open(request, reference.value) : Effect.fail(protocolFailure("unsupported session reference version")),
+	};
+});

package/src/pi.ts

@@ -1 +1,44 @@
-export {};
+import { Effect, Layer } from "effect";
+import { Backend, type ConfiguredBackend } from "@shivaedev/agent/backend.ts";
+import { listPiModels } from "#adapters/listPiModels.ts";
+import { textOf } from "#adapters/textOf.ts";
+import { makeOperations } from "#makeOperations.ts";
+import type { PiOptions } from "#PiOptions.ts";
+
+export function pi(options: PiOptions = {}): Layer.Layer<Backend> {
+	return Layer.effect(
+		Backend,
+		Effect.gen(function* (): Effect.fn.Return<ConfiguredBackend> {
+			const operations = yield* makeOperations(options);
+			return {
+				backend: "pi",
+				events: operations.events,
+				features: {
+					imageInput: { reason: "only text input is implemented", support: "unsupported" },
+					interruption: {
+						implementation: { interrupt: (receipt) => operations.lookup(receipt).pipe(Effect.flatMap((run) => run.interrupt)) },
+						support: "supported",
+					},
+					models: { implementation: { list: listPiModels }, support: "supported" },
+					queuedInput: {
+						implementation: {
+							queue: (receipt, input) => operations.lookup(receipt).pipe(Effect.flatMap((run) => run.prompt(textOf(input), "followUp"))),
+						},
+						support: "supported",
+					},
+					sessions: { implementation: { resume: operations.resume }, support: "supported" },
+					steering: {
+						implementation: {
+							steer: (receipt, input) => operations.lookup(receipt).pipe(Effect.flatMap((run) => run.prompt(textOf(input), "steer"))),
+						},
+						support: "supported",
+					},
+					transcript: { reason: "no provider history reader is implemented", support: "unsupported" },
+				},
+				inspect: operations.inspect,
+				reconcile: operations.reconcile,
+				start: (request) => operations.open(request, undefined),
+			};
+		}),
+	);
+}