14 KiB
Building a pi extension on the Effect v4 setup
Practical, migration-oriented companion to
effect-v4-notes.md. That file is the Effect API cheat-sheet (v3→v4 renames, child processes, concurrency); this file is how to stand up one pi extension on the same toolchain and where to draw the line between "wrap in Effect" and "leave as plain TS".Verified against
effect@4.0.0-beta.98,@effect/platform-node@4.0.0-beta.98,@effect/tsgo@0.19.0,typescript@7.0.2(checked 2026-07-13, inextensions/subagents).npm run checkthere passes clean — use it as the reference implementation.Audience: the agents migrating
firecrawl-search,ask-user,model-info,git-info,ui-customization, andcopy-all.
0. The one rule: Effect is for the async core, not the whole file
A pi extension's public surface is plain callbacks the host calls:
export default function (pi: ExtensionAPI), pi.registerTool({ …, async execute() }),
pi.registerCommand, pi.on(...), renderers. None of that changes. Effect lives inside
those callbacks — you run an effect at the top of execute and return its value.
Reach for Effect only where you actually get something from it:
- Yes: async work that needs typed errors, cancellation via the tool
AbortSignal, timeouts, retries/polling, or a resource whose lifetime must outlive one call (child process, subscription) → child processes (git-info,copy-all), the Firecrawl SDK calls (firecrawl-search), git/gh polling (git-info). - No / barely: pure TUI popups and rendering (
ask-user,ui-customization), cross-extension channel plumbing, cost/token bookkeeping (model-info). These are synchronous or already-Promise UI code; wrapping them in Effect adds ceremony and no safety. Migrate them by keeping the logic and only touching whatever genuinely async part benefits (usually nothing).
If an extension has no async core worth typing, the "migration" may be just adopting the toolchain (§1) and leaving the body plain. Don't invent an Effect layer to have one.
1. Per-extension toolchain (copy this exactly)
Each extension is its own npm package with its own node_modules. Replicate the pinned
setup — do not float the versions.
package.json:
{
"name": "<ext>",
"private": true,
"type": "module",
"scripts": {
"check": "tsc --noEmit -p .",
"prepare": "effect-tsgo patch", // patches the Effect LS into the tsgo binary
},
"dependencies": {
"effect": "4.0.0-beta.98", // EXACT pin, no ^
"@effect/platform-node": "4.0.0-beta.98", // only if you touch fs / child processes
},
"devDependencies": {
"@effect/tsgo": "^0.19.0",
"typescript": "^7.0.2",
},
}
tsconfig.json (extends the repo root, adds the Effect language-service plugin):
{
"extends": "../../tsconfig.json",
"compilerOptions": {
"plugins": [{ "name": "@effect/language-service" }],
},
"include": ["index.ts", "src/**/*.ts", "*.test.ts"],
}
The root tsconfig.json already sets strict, module/moduleResolution: NodeNext,
verbatimModuleSyntax, allowImportingTsExtensions, target: ES2022, types: ["node"].
Keep local .ts imports written with the .ts extension (./src/manager.ts), matching
the house style.
The LSP / language-service, precisely
typescript@7is the native (tsgo) TypeScript — itstscis a Go binary, and it's whatnpm run checkruns.@effect/language-serviceis not an installed npm package here (you won't find it innode_modules). It's delivered byeffect-tsgo patch, run automatically by thepreparelifecycle script onnpm install. The patch injects the Effect Language Service into the tsgo binary; the tsconfigpluginsentry then turns on Effect-aware editor diagnostics and quickfixes (e.g. "yield missing services", floating effects).- Practical sequence for a fresh/edited extension:
npm install(runs the patch) →npm run check. If editor diagnostics look stale, re-runnpx effect-tsgo patch;npx effect-tsgo get-exe-pathprints the patched binary it resolved. - The plugin drives the editor; it does not change
tscexit codes.npm run checkis still your ground-truth green/red signal.
2. The async boundary: one ManagedRuntime + a runTool helper
Every extension that uses Effect needs exactly this shape. Build one runtime lazily,
dispose it on shutdown, and funnel every tool handler through a single helper that turns
Effect exits into what pi expects (a value, or a thrown Error; interruption → a friendly
message). This is lifted verbatim from src/runtime.ts — reuse it.
// src/runtime.ts
import { Cause, Exit, Layer, ManagedRuntime, type Effect } from "effect";
import { NodeServices } from "@effect/platform-node"; // only if you need fs / processes
// Compose your services here (see §3). NodeServices.layer =
// ChildProcessSpawner | FileSystem | Path | Crypto | Stdio | Terminal.
const AppLayer = Layer.mergeAll(NodeServices.layer /*, MyServiceLive */);
export function createRuntime() {
return ManagedRuntime.make(AppLayer);
}
export type ExtRuntime = ReturnType<typeof createRuntime>;
/** Run an effect from an async handler: value on success, thrown Error otherwise. */
export async function runTool<A, E>(
runtime: ExtRuntime,
effect: Effect.Effect<A, E>,
options: { signal?: AbortSignal; interruptMessage?: string } = {},
) {
const exit = await runtime.runPromiseExit(
effect,
options.signal ? { signal: options.signal } : undefined,
);
if (Exit.isSuccess(exit)) return exit.value;
if (Cause.hasInterruptsOnly(exit.cause)) {
throw new Error(options.interruptMessage ?? "Operation was aborted.");
}
const [first] = Cause.prettyErrors(exit.cause);
throw new Error(first?.message ?? Cause.pretty(exit.cause));
}
Wire it into the extension lifecycle (lazy build, dispose on shutdown):
// index.ts
export default function (pi: ExtensionAPI) {
let runtime: ExtRuntime | undefined;
const getRuntime = () => (runtime ??= createRuntime());
pi.registerTool({
name: "my_tool",
parameters: Type.Object({/* typebox */}),
async execute(_id, params, signal) {
return await runTool(getRuntime(), myEffect(params), {
signal,
interruptMessage: "Cancelled.",
});
},
});
pi.on("session_shutdown", async () => {
const closing = runtime;
runtime = undefined;
await closing?.dispose(); // runs all finalizers: kills scoped child processes, etc.
});
}
Key facts:
runPromiseExit(effect, { signal })— passing the tool'sAbortSignalmakes cancelling the tool interrupt the fiber; scoped resources (child processes) are torn down.runtime.runFork(effect)for fire-and-forget background work (polling loops, §5).dispose()is the single teardown hook — put resource cleanup in Effect finalizers, not in ad-hocsession_shutdowncode.- If an extension has no services (pure
Effect.tryPromisecalls, no fs/process), you can skipNodeServices.layerand even skipManagedRuntime—Effect.runPromiseExit(effect, { signal })works standalone. Add the runtime only when you have aLayerto share.
3. Services, layers, errors (house style)
Mirror the subagents code. Class-style Context.Service, Layer.effect/Layer.sync,
Data.TaggedError for domain errors.
import { Context, Data, Effect, Layer } from "effect";
// Domain error — yieldable, tag-narrowable with Effect.catchTag:
export class GitError extends Data.TaggedError("GitError")<{
readonly message: string;
}> {}
// Service: the class value is the key AND the type.
export interface GitShape {
readonly status: Effect.Effect<string, GitError>;
}
export class Git extends Context.Service<Git, GitShape>()("gitinfo/Git") {}
// Layer building it (Layer.effect can use scoped resources; Layer.sync for pure):
export const GitLive: Layer.Layer<Git, never, ChildProcessSpawner> =
Layer.effect(
Git,
Effect.gen(function* () {
const spawner = yield* ChildProcessSpawner; // dependency, provided by NodeServices.layer
return Git.of({
status: spawner
.string(ChildProcess.make("git", ["status", "--porcelain"]))
.pipe(
Effect.mapError(
(cause) => new GitError({ message: String(cause) }),
),
),
});
}),
);
Provide dependencies with Layer.provide, compose with Layer.mergeAll (see §2 AppLayer).
Full rename table (Effect.fork→forkChild, catchAll→catch, Either→Result, …) and
the child-process / concurrency APIs live in effect-v4-notes.md — don't re-derive them.
4. Recipe: wrapping a Promise SDK (firecrawl-search)
The Firecrawl client is Promise-based. Wrap each call in Effect.tryPromise with a typed
error; the callback receives an AbortSignal tied to fiber interruption — forward it to any
SDK that accepts one so tool cancellation propagates.
import { Data, Effect } from "effect";
class FirecrawlError extends Data.TaggedError("FirecrawlError")<{
readonly cause: unknown;
}> {}
const search = (client: Firecrawl, query: string) =>
Effect.tryPromise({
try: (signal) => client.search(query, {/* …, signal? if supported */}),
catch: (cause) => new FirecrawlError({ cause }),
});
There's no Layer here unless you want the client as a service. For a single tool this is
fine to call directly through runTool (or even plain Effect.runPromiseExit, §2). Keep the
existing .env/API-key reading and result truncation as plain TS — no reason to Effect-ify
string trimming.
5. Recipe: child processes + timeout + polling (git-info, copy-all)
git-info shells out to git/gh with per-command timeouts and polls on an interval;
copy-all pipes text into pbcopy. Two viable levels — pick the lightest that fits.
Simple, one-shot, small: if all you do is "run a command, capture stdout, with a
timeout," the Effect win is Effect.timeout + interruption killing the child. Use the
spawner service:
import { Effect } from "effect";
import { ChildProcess } from "effect/unstable/process";
import { ChildProcessSpawner } from "effect/unstable/process/ChildProcessSpawner";
const gitStatus = Effect.gen(function* () {
const spawner = yield* ChildProcessSpawner;
return yield* spawner
.string(ChildProcess.make("git", ["status", "--porcelain"], { cwd }))
.pipe(Effect.timeout("3 seconds")); // fails with Cause.TimeoutError (tag "TimeoutError")
});
Note the import gotcha: ChildProcessSpawner the class comes from the
.../ChildProcessSpawner submodule; the effect/unstable/process index gives you the
namespace. Provide NodeServices.layer (or ChildProcessSpawner.layer). Full command
builder / streaming / kill semantics are in effect-v4-notes.md §6.
Polling on an interval: replace setInterval with a forked, scheduled effect so it
cancels cleanly on dispose:
import { Effect, Schedule } from "effect";
const pollLoop = refresh.pipe(
Effect.catchCause(() => Effect.void), // don't let one failure kill the loop
Effect.repeat(Schedule.spaced("3 seconds")),
);
const fiber = runtime.runFork(pollLoop); // interrupted by runtime.dispose()
When to stay plain: copy-all spawning pbcopy is a trivial one-shot with no
cancellation need — the existing node:child_process + Promise wrapper is honestly fine.
Migrate it only for consistency; if you do, Effect.callback around child.once("exit", …)
(see notes §4) is the minimal wrapper. Don't add a service/layer for a clipboard write.
6. Recipe: UI popups & rendering (ask-user, ui-customization, model-info)
These are the "leave it mostly plain" cases.
ask-useris a TUI popup that resolves a Promise when the user picks. That Promise already models the one async thing. Effect adds nothing; if you want uniformity, wrap the final await inEffect.tryPromiseat the boundary and stop there. Do not build a service.ui-customizationandmodel-infoare renderers / event bookkeepers driven bypi.on(...)and cross-extension channels (shared/dashboard-state.ts). Channels are a pi-native mechanism — keep them. State counting and formatting stay synchronous TS.- If
model-infohas a periodic "live update" tick, the §5 polling pattern applies; but a plain timer here is also acceptable since there's no resource to tear down.
The migration bar for these: adopt the toolchain (§1) so they typecheck under TS7 + the Effect LS, and only touch runtime code that has a real async/resource concern.
7. Verify — scoped to the one extension
Run everything from inside the extension directory so you never trigger a root/global build or format:
cd extensions/<ext>
npm install # first time / after dep or script edits — runs `prepare` (effect-tsgo patch)
npm run check # tsc --noEmit -p . ← ground-truth green/red
npm run test # only if the extension defines tests; keep them minimal
extensions/subagents is the known-green reference: npm run check there exits 0 against
the pinned versions. If a migrated extension fails check with Effect.fork/ServiceMap/
Either errors, it's using stale v3/early-beta APIs — consult the rename table in
effect-v4-notes.md.
8. Don'ts (keep it lean)
- Don't float versions. Pin
effectand@effect/platform-nodeto the exact same4.0.0-beta.98;unstable/*can break between betas. - Don't Effect-ify pure/UI code. No service or layer for a clipboard write, a string truncation, or a popup. §0 is the test: is there a typed-error / cancellation / resource / retry concern? If not, leave it.
- Don't guess APIs. If it's not in this guide or
effect-v4-notes.mdand you haven't grepped it innode_modules/effect/dist/*.d.ts, don't write it. Watch for hallucinatedServiceMap(reverted toContext),Effect.fork(→forkChild),Effect.either(→Effect.result). - Don't hand-roll teardown. Put cleanup in Effect finalizers and let
runtime.dispose()insession_shutdownrun them. - Don't over-test. A
checkthat passes plus one focused runtime test (where behavior is non-obvious) beats a wall of defensive unit tests. - Don't run root scripts. No repo-root
tsc,prettier, ornpm run format; stay insideextensions/<ext>.