codemem 0.46.0-alpha.2 → 0.46.0-beta.2

package/dist/index.js

@@ -1,5 +1,5 @@
 #!/usr/bin/env node
-import { ContextOnlyReplayError, DEDUP_KEY_BACKFILL_JOB, DEFAULT_COORDINATOR_DB_PATH, DedupKeyBackfillRunner, DeviceIdentityError, EXTRACTION_BENCHMARK_QUALITY_WEIGHTS, MUTATING_TOOL_NAMES, MemoryStore, ObserverClient, ObserverOutputError, ObserverOutputTransportError, PROMPT_TRANSPORT_PROTOCOL_RANGE, REF_BACKFILL_JOB, RawEventIngestValidationError, RawEventSweeper, RefBackfillRunner, SCOPE_BACKFILL_JOB, SESSION_CONTEXT_BACKFILL_JOB, SUMMARY_DEDUP_BACKFILL_JOB, ScopeBackfillRunner, SessionContextBackfillRunner, SummaryDedupBackfillRunner, SyncRetentionRunner, TRUSTED_HOOK_MAPPER_OPTIONS, VERSION, VectorModelMigrationRunner, aiBackfillStructuredContent, applyBootstrapSnapshot, applyDistillRule, arePromptTransportProtocolRangesCompatible, atomicReplaceConfigFile, backfillMemoryDedupKeys, backfillNarrativeFromBody, backfillTagsText, backfillVectors, buildBaseUrl, buildDirectPeerAuthHeaders, buildDistillReport, buildPiFlushSignalFromEvent, buildRawEventEnvelopeFromHook, buildRawEventEnvelopeFromPiEvent, buildViewerIdentityTarget, classifyPromptTransportFailure, clonePromptPackAttempt, collectOperationalStatus, compareMemoryRoleReports, connect, connectReadOnly, coordinatorCreateGroupAction, coordinatorCreateInviteAction, coordinatorCreateScopeAction, coordinatorDisableDeviceAction, coordinatorEnrollDeviceAction, coordinatorGrantScopeMembershipAction, coordinatorImportInviteAction, coordinatorListBootstrapGrantsAction, coordinatorListDevicesAction, coordinatorListGroupsAction, coordinatorListJoinRequestsAction, coordinatorListScopeMembershipsAction, coordinatorListScopesAction, coordinatorRemoveDeviceAction, coordinatorRenameDeviceAction, coordinatorReviewJoinRequestAction, coordinatorRevokeBootstrapGrantAction, coordinatorRevokeScopeMembershipAction, coordinatorUpdateScopeAction, createBetterSqliteCoordinatorApp, deactivateLowSignalMemories, deactivateLowSignalObservations, dedupNearDuplicateMemories, detectInstallKind, draftDistillRule, ensureDeviceIdentity, ensureSchemaBootstrapped, estimateExtractionModelCost, exportMemories, extractApplyPatchPaths, extractionReplayObserverIdentity, fetchAllSnapshotPages, fingerprintPublicKey, flushRawEvents, formatHostPort, getAttributionDiagnostics, getCodememConfigPath, getExtractionBenchmarkProfile, getExtractionModelPricing, getInjectionEvalScenarioPack, getInjectionEvalScenarioPrompts, getMaintenanceJob, getMemoryArtifactReport, getMemoryRoleReport, getRawEventRelinkPlan, getRawEventRelinkReport, getRawEventStatus, getSemanticIndexDiagnostics, getSessionExtractionEval, getSessionExtractionEvalScenario, getUpdateStatus, getWorkspaceCodememConfigPath, hasPendingDedupKeyBackfill, hasPendingRefBackfill, hasPendingScopeBackfill, hasPendingSessionContextBackfill, hasPendingSummaryDedupBackfill, hasUnsyncedSharedMemoryChanges, importMemories, ingestRawEvents, initDatabase, isAutomaticRecallMeasurement, isEmbeddingDisabled, isExplicitUpdateInstallEligible, isReleaseVersionForChannel, judgeDistillReport, listMaintenanceJobs, listPerPeerScopeSyncState, listRetentionScopeIds, loadObserverConfig, loadPublicKey, loadSqliteVec, mdnsEnabled, mutateCodememConfigFile, normalizePromptTransportProtocolRange, observerBaseUrlForProviderOverride, planReplicationOpsAgePrune, probeCodememViewerLiveness, probeCodememViewerLiveness as probeCodememViewerLiveness$1, probeRequiredNativeRuntime, projectMatchesFilter, promptPackArtifactFingerprint, pruneReplicationOpsUntilCaughtUp, rawEventsGate, readCodememConfigFile, readCodememConfigFileAtPath, readCoordinatorSyncConfig, readImportPayload, recordAutomaticRecall, recordPromptPackArtifacts, recordPromptPackTerminal, recordRetrievalSurface, renderUnifiedDiff, replayBatchExtraction, replayBatchExtractionWithTierRouting, requestJson, resolveCodememConfigPath, resolveComparablePath, resolveDbPath, resolveHookProject, resolveProject, resolveProjectRoot, resolveRetrievalSession, retryRawEventFailures, runSyncDaemon, runSyncPass, scanSecretsRetroactive, schema, scoreExtractionBenchmarkOutput, setPeerProjectFilter, syncPassPreflight, tryUpdateRetrievalDelivery, updatePeerAddresses, vacuumDatabase, viewerUrl } from "@codemem/core";
+import { ContextOnlyReplayError, DEDUP_KEY_BACKFILL_JOB, DEFAULT_COORDINATOR_DB_PATH, DedupKeyBackfillRunner, DeviceIdentityError, EXTRACTION_BENCHMARK_QUALITY_WEIGHTS, MUTATING_TOOL_NAMES, MemoryStore, ObserverClient, ObserverOutputError, ObserverOutputTransportError, PROMPT_TRANSPORT_PROTOCOL_RANGE, REF_BACKFILL_JOB, RawEventIngestValidationError, RawEventSweeper, RefBackfillRunner, SCOPE_BACKFILL_JOB, SESSION_CONTEXT_BACKFILL_JOB, SUMMARY_DEDUP_BACKFILL_JOB, ScopeBackfillRunner, SessionContextBackfillRunner, SummaryDedupBackfillRunner, SyncRetentionRunner, TRUSTED_HOOK_MAPPER_OPTIONS, VERSION, VectorModelMigrationRunner, aiBackfillStructuredContent, applyBootstrapSnapshot, applyDistillRule, arePromptTransportProtocolRangesCompatible, atomicReplaceConfigFile, backfillMemoryDedupKeys, backfillNarrativeFromBody, backfillTagsText, backfillVectors, buildBaseUrl, buildDirectPeerAuthHeaders, buildDistillReport, buildPiFlushSignalFromEvent, buildRawEventEnvelopeFromHook, buildRawEventEnvelopeFromPiEvent, buildViewerIdentityTarget, classifyPromptTransportFailure, clonePromptPackAttempt, collectOperationalStatus, compareMemoryRoleReports, connect, connectReadOnly, coordinatorCreateGroupAction, coordinatorCreateInviteAction, coordinatorCreateScopeAction, coordinatorDisableDeviceAction, coordinatorEnrollDeviceAction, coordinatorGrantScopeMembershipAction, coordinatorImportInviteAction, coordinatorListBootstrapGrantsAction, coordinatorListDevicesAction, coordinatorListGroupsAction, coordinatorListJoinRequestsAction, coordinatorListScopeMembershipsAction, coordinatorListScopesAction, coordinatorRemoveDeviceAction, coordinatorRenameDeviceAction, coordinatorReviewJoinRequestAction, coordinatorRevokeBootstrapGrantAction, coordinatorRevokeScopeMembershipAction, coordinatorUpdateScopeAction, createBetterSqliteCoordinatorApp, deactivateLowSignalMemories, deactivateLowSignalObservations, dedupNearDuplicateMemories, describePiObserverStatus, detectInstallKind, draftDistillRule, ensureDeviceIdentity, ensureSchemaBootstrapped, estimateExtractionModelCost, exportMemories, extractApplyPatchPaths, extractionReplayObserverIdentity, fetchAllSnapshotPages, fingerprintPublicKey, flushRawEvents, formatHostPort, getAttributionDiagnostics, getCodememConfigPath, getExtractionBenchmarkProfile, getExtractionModelPricing, getInjectionEvalScenarioPack, getInjectionEvalScenarioPrompts, getMaintenanceJob, getMemoryArtifactReport, getMemoryRoleReport, getRawEventRelinkPlan, getRawEventRelinkReport, getRawEventStatus, getSemanticIndexDiagnostics, getSessionExtractionEval, getSessionExtractionEvalScenario, getUpdateStatus, getWorkspaceCodememConfigPath, hasPendingDedupKeyBackfill, hasPendingRefBackfill, hasPendingScopeBackfill, hasPendingSessionContextBackfill, hasPendingSummaryDedupBackfill, hasUnsyncedSharedMemoryChanges, importMemories, ingestRawEvents, initDatabase, isAutomaticRecallMeasurement, isEmbeddingDisabled, isExplicitUpdateInstallEligible, isReleaseVersionForChannel, judgeDistillReport, listMaintenanceJobs, listPerPeerScopeSyncState, listRetentionScopeIds, loadObserverConfig, loadPublicKey, loadSqliteVec, mdnsEnabled, mutateCodememConfigFile, normalizePromptTransportProtocolRange, observerBaseUrlForProviderOverride, planReplicationOpsAgePrune, probeCodememViewerLiveness, probeCodememViewerLiveness as probeCodememViewerLiveness$1, probeRequiredNativeRuntime, projectMatchesFilter, promptPackArtifactFingerprint, pruneReplicationOpsUntilCaughtUp, rawEventsGate, readCodememConfigFile, readCodememConfigFileAtPath, readCoordinatorSyncConfig, readImportPayload, recordAutomaticRecall, recordPromptPackArtifacts, recordPromptPackTerminal, recordRetrievalSurface, renderUnifiedDiff, replayBatchExtraction, replayBatchExtractionWithTierRouting, requestJson, resolveCodememConfigPath, resolveComparablePath, resolveDbPath, resolveHookProject, resolvePiAgentDir, resolvePiObserverConfig, resolveProject, resolveProjectRoot, resolveRetrievalSession, retryRawEventFailures, runSyncDaemon, runSyncPass, scanSecretsRetroactive, schema, scoreExtractionBenchmarkOutput, setPeerProjectFilter, syncPassPreflight, tryUpdateRetrievalDelivery, updatePeerAddresses, vacuumDatabase, viewerUrl, writeCodememConfigFile } from "@codemem/core";
 import { Command, Option } from "commander";
 import omelette from "omelette";
 import { randomInt, randomUUID } from "node:crypto";
@@ -4949,6 +4949,11 @@ function forgetMemoryAction(idStr, opts) {
 		store.close();
 	}
 }
+function parseConfidence(value) {
+	const n = value == null || value.trim() === "" ? .5 : Number(value);
+	if (!Number.isFinite(n)) return .5;
+	return Math.min(1, Math.max(0, n));
+}
 function rollbackManualMemory(store, sessionId, memoryId) {
 	store.db.transaction(() => {
 		const row = store.db.prepare("SELECT import_key FROM memory_items WHERE id = ?").get(memoryId);
@@ -4974,7 +4979,7 @@ async function rememberMemoryAction(opts) {
 			toolVersion: "manual",
 			metadata: { manual: true }
 		});
-		const memId = store.remember(sessionId, opts.kind, opts.title, opts.body, .5, opts.tags);
+		const memId = store.remember(sessionId, opts.kind, opts.title, opts.body, parseConfidence(opts.confidence), opts.tags);
 		if (!store.get(memId)) {
 			await store.flushPendingVectorWrites();
 			rollbackManualMemory(store, sessionId, memId);
@@ -5017,7 +5022,7 @@ function createForgetMemoryCommand() {
 	return cmd;
 }
 function createRememberMemoryCommand() {
-	const cmd = new Command("remember").configureHelp(helpStyle).description("Manually add a memory item").requiredOption("-k, --kind <kind>", "memory kind (discovery, decision, feature, bugfix, etc.)").requiredOption("-t, --title <title>", "memory title").requiredOption("-b, --body <body>", "memory body text").option("--tags <tags...>", "tags (space-separated)").option("--project <project>", "project name (defaults to git repo root)");
+	const cmd = new Command("remember").configureHelp(helpStyle).description("Manually add a memory item").requiredOption("-k, --kind <kind>", "memory kind (discovery, decision, feature, bugfix, etc.)").requiredOption("-t, --title <title>", "memory title").requiredOption("-b, --body <body>", "memory body text").option("--tags <tags...>", "tags (space-separated)").option("--project <project>", "project name (defaults to git repo root)").option("--confidence <n>", "confidence 0-1", "0.5");
 	addDbOption(cmd);
 	addJsonOption(cmd);
 	cmd.action(rememberMemoryAction);
@@ -8401,6 +8406,16 @@ function writeJsonConfig(path, data, { createBackup = true } = {}) {
 	atomicReplaceConfigFile(path, output, configMetadata(path));
 	return true;
 }
+/**
+* Write JSON config after copying any existing file to `<path>.codemem.bak`.
+* Backup failure is non-fatal (write still proceeds).
+*/
+function writeJsonConfigWithBackup(path, data) {
+	if (existsSync(path)) try {
+		copyFileSync(path, `${path}.codemem.bak`);
+	} catch {}
+	writeJsonConfig(path, data);
+}
 //#endregion
 //#region src/commands/setup.ts
 /**
@@ -8412,6 +8427,8 @@ function writeJsonConfig(path, data, { createBackup = true } = {}) {
 * 1. Adds "@codemem/opencode-plugin" to the plugin array in ~/.config/opencode/opencode.jsonc
 * 2. Adds/updates the MCP entry in ~/.config/opencode/opencode.jsonc
 * 3. For Claude Code: installs MCP config and guides marketplace plugin install
+* 4. For Codex: MCP + hooks via CODEX_HOME
+* 5. For pi: packages entry + observer derivation + optional MCP adapter surface
 *
 * Designed to be safe to run repeatedly (idempotent unless --force).
 */
@@ -8425,6 +8442,23 @@ function claudeConfigDir() {
 function codexConfigDir() {
 	return process.env.CODEX_HOME?.trim() || join(homedir(), ".codex");
 }
+/** Resolve the pi agent directory, honoring PI_CODING_AGENT_DIR. */
+function piConfigDir() {
+	return resolvePiAgentDir();
+}
+/** npm packages: entry prefix for the pi extension (version is appended). */
+var PI_EXTENSION_NPM_NAME = "@codemem/pi-extension";
+var PI_EXTENSION_NPM_PREFIX = `npm:${PI_EXTENSION_NPM_NAME}@`;
+var PI_MCP_ADAPTER_MARKER = "pi-mcp-adapter";
+/** Codemem stdio MCP server entry written to pi's mcp.json under --pi-mcp. */
+var PI_MCP_CODEMEM_ENTRY = {
+	command: "npx",
+	args: [
+		"-y",
+		"codemem",
+		"mcp"
+	]
+};
 /** Remove legacy copied plugin JS file from ~/.config/opencode/plugins/codemem.js */
 function migrateLegacyOpencodePlugin() {
 	const legacyPlugin = join(opencodeConfigDir(), "plugins", "codemem.js");
@@ -8949,38 +8983,374 @@ function installCodex(force, options = {}) {
 	ok = installCodexHooks(codexHome, force, hookBase) && ok;
 	return ok;
 }
-var setupCommand = new Command("setup").configureHelp(helpStyle).description("Install codemem plugin + MCP config for OpenCode and Claude Code").option("--force", "overwrite existing installations").option("--opencode-only", "only install for OpenCode").option("--claude-only", "only install for Claude Code").option("--codex-only", "only install for Codex").action((opts) => {
-	p.intro(`codemem setup v${VERSION}`);
-	const force = opts.force ?? false;
+/** Build the packages: entry for the pi extension (npm pin or local path). */
+function buildPiExtensionPackageSpec(extensionPath, version = VERSION) {
+	if (extensionPath?.trim()) {
+		const raw = extensionPath.trim();
+		return isAbsolute(raw) ? raw : resolve(raw);
+	}
+	return `${PI_EXTENSION_NPM_PREFIX}${version}`;
+}
+/** True when a packages: entry refers to @codemem/pi-extension (any version/path form). */
+function isPiExtensionPackageEntry(entry) {
+	if (typeof entry !== "string" || !entry.trim()) return false;
+	const value = entry.trim();
+	if (value === `npm:${PI_EXTENSION_NPM_NAME}` || value.startsWith(PI_EXTENSION_NPM_PREFIX)) return true;
+	if (value === PI_EXTENSION_NPM_NAME || value.startsWith(`${PI_EXTENSION_NPM_NAME}@`)) return true;
+	return /(?:^|[/\\])@codemem[/\\]pi-extension(?:[/\\]|$)/.test(value) || /(?:^|[/\\])packages[/\\]pi-extension(?:[/\\]|$)/.test(value);
+}
+/** True when entry is an npm: pin for @codemem/pi-extension (versioned or bare). */
+function isNpmPiExtensionPackageEntry(entry) {
+	if (typeof entry !== "string" || !entry.trim()) return false;
+	const value = entry.trim();
+	return value === `npm:${PI_EXTENSION_NPM_NAME}` || value.startsWith(PI_EXTENSION_NPM_PREFIX);
+}
+function piBinaryOnPath() {
+	try {
+		const out = execFileSync(process.platform === "win32" ? "where" : "which", ["pi"], { encoding: "utf-8" });
+		return Boolean(out.split(/\r?\n/).map((line) => line.trim()).find(Boolean));
+	} catch {
+		return false;
+	}
+}
+/** True when path exists and has non-whitespace content (real install marker). */
+function isNonEmptyFile(path) {
+	try {
+		if (!existsSync(path)) return false;
+		return readFileSync(path, "utf-8").trim().length > 0;
+	} catch {
+		return false;
+	}
+}
+/**
+* Detect pi via `pi` on PATH, or an agent dir that contains real install
+* markers (non-empty settings.json or auth.json). A bare empty ~/.pi/agent
+* directory must not count — users (and tests) create that path incidentally.
+* Honors PI_CODING_AGENT_DIR.
+*/
+function isPiDetected() {
+	if (piBinaryOnPath()) return true;
+	const dir = piConfigDir();
+	if (!existsSync(dir)) return false;
+	return isNonEmptyFile(join(dir, "settings.json")) || isNonEmptyFile(join(dir, "auth.json"));
+}
+/** Detect pi-mcp-adapter via packages: entry or extensions/ directory name. */
+function isPiMcpAdapterDetected(piDir = piConfigDir()) {
+	const settingsPath = join(piDir, "settings.json");
+	if (existsSync(settingsPath)) try {
+		const packages = loadJsoncConfig(settingsPath).packages;
+		if (Array.isArray(packages)) {
+			if (packages.some((entry) => typeof entry === "string" && entry.includes(PI_MCP_ADAPTER_MARKER))) return true;
+		}
+	} catch {}
+	const extensionsDir = join(piDir, "extensions");
+	if (!existsSync(extensionsDir)) return false;
+	try {
+		return readdirSync(extensionsDir, { withFileTypes: true }).some((entry) => {
+			if (!entry.isDirectory() && !entry.isSymbolicLink()) return false;
+			const name = entry.name.toLowerCase();
+			return name.includes("mcp-adapter") || name.includes("mcp_adapter");
+		});
+	} catch {
+		return false;
+	}
+}
+function readPiJsoncOrAbort(path, untouchedHint) {
+	if (!existsSync(path)) return {};
+	try {
+		return loadJsoncConfig(path);
+	} catch (err) {
+		p.log.error(`Failed to parse ${path}: ${err instanceof Error ? err.message : String(err)}`);
+		p.log.info(`Leaving ${path} untouched. ${untouchedHint}`);
+		return null;
+	}
+}
+function shouldKeepExistingPiExtension(existing, force, desired) {
+	if (existing == null || force) return false;
+	return !(isNpmPiExtensionPackageEntry(existing) && existing !== desired);
+}
+function writePiExtensionSettings(piDir, settingsPath, settings, existing, desired) {
+	try {
+		mkdirSync(piDir, { recursive: true });
+		writeJsonConfigWithBackup(settingsPath, settings);
+		p.log.success(existing != null ? `Pi extension package updated: ${desired}` : `Pi extension package added: ${desired}`);
+	} catch (err) {
+		p.log.error(`Failed to write ${settingsPath}: ${err instanceof Error ? err.message : String(err)}`);
+		return false;
+	}
+	return true;
+}
+/**
+* Idempotently ensure the pi extension packages: entry is present in
+* settings.json. Backs up before write; aborts on parse failure.
+*/
+function installPiExtensionPackage(piDir, force, extensionPath) {
+	const settingsPath = join(piDir, "settings.json");
+	const desired = buildPiExtensionPackageSpec(extensionPath);
+	const settings = readPiJsoncOrAbort(settingsPath, "Fix or remove the file, then re-run `codemem setup --pi-only`.");
+	if (settings == null) return false;
+	const list = Array.isArray(settings.packages) ? [...settings.packages] : [];
+	const existing = list.find((entry) => isPiExtensionPackageEntry(entry));
+	if (shouldKeepExistingPiExtension(existing, force, desired)) {
+		p.log.info(`Pi extension package already configured in ${settingsPath}`);
+		return true;
+	}
+	settings.packages = [...list.filter((entry) => !isPiExtensionPackageEntry(entry)), desired];
+	return writePiExtensionSettings(piDir, settingsPath, settings, existing, desired);
+}
+function isOfficialObserverBaseUrl(url) {
+	const normalized = url.trim().replace(/\/+$/, "").toLowerCase();
+	return normalized === "https://api.openai.com" || normalized === "https://api.openai.com/v1" || normalized === "https://api.anthropic.com" || normalized === "https://api.anthropic.com/v1";
+}
+function nextPiToolsBlock(existing, toolsMode) {
+	const piBlock = { ...existing.pi != null && typeof existing.pi === "object" && !Array.isArray(existing.pi) ? { ...existing.pi } : {} };
+	if (toolsMode === "mcp-adapter") {
+		if (piBlock.tools_mode === "mcp-adapter") return {
+			piBlock,
+			toolsModeUpdated: false
+		};
+		piBlock.tools_mode = "mcp-adapter";
+		return {
+			piBlock,
+			toolsModeUpdated: true
+		};
+	}
+	if (piBlock.tools_mode == null || piBlock.tools_mode === "") {
+		piBlock.tools_mode = "native";
+		return {
+			piBlock,
+			toolsModeUpdated: true
+		};
+	}
+	return {
+		piBlock,
+		toolsModeUpdated: false
+	};
+}
+function applyResolvedPiObserver(resolved, needs, next, updated) {
+	if (needs.envProvider || needs.envModel) return;
+	const explicitProvider = needs.fileProvider;
+	if (explicitProvider) {
+		if (needs.model && explicitProvider.toLowerCase() === String(resolved.provider).toLowerCase()) {
+			next.observer_model = resolved.model;
+			updated.push("observer_model");
+		}
+		return;
+	}
+	if (needs.fileModel) return;
+	next.observer_provider = resolved.provider;
+	updated.push("observer_provider");
+	next.observer_model = resolved.model;
+	updated.push("observer_model");
+	const customBaseUrl = resolved.baseUrl && !isOfficialObserverBaseUrl(resolved.baseUrl) ? resolved.baseUrl : void 0;
+	if (needs.baseUrl && customBaseUrl) {
+		next.observer_base_url = customBaseUrl;
+		updated.push("observer_base_url");
+	}
+	if (needs.useResponses && customBaseUrl) {
+		next.observer_openai_use_responses = resolved.openAIUseResponses;
+		updated.push("observer_openai_use_responses");
+	}
+}
+function unsetObserverSlots(existing) {
+	const envProvider = process.env.CODEMEM_OBSERVER_PROVIDER?.trim();
+	const envModel = process.env.CODEMEM_OBSERVER_MODEL?.trim();
+	const envBaseUrl = process.env.CODEMEM_OBSERVER_BASE_URL?.trim();
+	const envUseResponses = process.env.CODEMEM_OBSERVER_OPENAI_USE_RESPONSES?.trim();
+	const fileProvider = typeof existing.observer_provider === "string" ? existing.observer_provider.trim() : "";
+	const fileModel = typeof existing.observer_model === "string" ? existing.observer_model.trim() : "";
+	const fileBaseUrl = typeof existing.observer_base_url === "string" ? existing.observer_base_url.trim() : "";
+	return {
+		provider: !envProvider && !fileProvider,
+		model: !envModel && !fileModel,
+		baseUrl: !envBaseUrl && !fileBaseUrl,
+		useResponses: !envUseResponses && existing.observer_openai_use_responses == null,
+		fileProvider,
+		fileModel,
+		envProvider,
+		envModel
+	};
+}
+function fillUnsetObserverFromPi(existing, next, updated) {
+	const needs = unsetObserverSlots(existing);
+	if (!(needs.provider || needs.model || needs.baseUrl || needs.useResponses)) {
+		p.log.info("Existing codemem observer config left unchanged");
+		return;
+	}
+	const resolved = resolvePiObserverConfig();
+	p.log.info(`Pi observer: ${describePiObserverStatus(resolved)}`);
+	if (resolved.ok) {
+		applyResolvedPiObserver(resolved, needs, next, updated);
+		return;
+	}
+	if (!needs.fileProvider && !needs.fileModel && !needs.envProvider && !needs.envModel) p.log.info("Extraction model left unconfigured — set observer_provider/observer_model when ready.");
+}
+function writeUpdatedCodememConfig(next, updated) {
+	if (updated.length === 0) return true;
+	try {
+		const saved = writeCodememConfigFile(next);
+		p.log.success(`Codemem config updated (${updated.join(", ")}): ${saved}`);
+	} catch (err) {
+		p.log.error(`Failed to write codemem config: ${err instanceof Error ? err.message : String(err)}`);
+		return false;
+	}
+	return true;
+}
+/**
+* Derive unset observer_* keys from pi config and ensure pi.tools_mode default.
+* Never persists API keys or tokens from pi auth.
+*/
+function wirePiCodememConfig(opts) {
+	let existing;
+	try {
+		existing = readCodememConfigFile();
+	} catch (err) {
+		p.log.error(`Failed to read codemem config: ${err instanceof Error ? err.message : String(err)}`);
+		return false;
+	}
+	const next = { ...existing };
+	const updated = [];
+	const { piBlock, toolsModeUpdated } = nextPiToolsBlock(existing, opts.toolsMode);
+	if (toolsModeUpdated) updated.push("pi.tools_mode");
+	next.pi = piBlock;
+	fillUnsetObserverFromPi(existing, next, updated);
+	return writeUpdatedCodememConfig(next, updated);
+}
+/**
+* Opt-in MCP surface for pi via pi-mcp-adapter. Writes mcp.json only when the
+* adapter is detected; otherwise explains the prerequisite and writes nothing.
+*/
+function installPiMcp(piDir, force) {
+	if (!isPiMcpAdapterDetected(piDir)) {
+		p.log.warn("MCP in pi requires the pi-mcp-adapter package. Install it (e.g. `pi install npm:pi-mcp-adapter`), then re-run `codemem setup --pi-only --pi-mcp`.");
+		return {
+			ok: true,
+			wrote: false,
+			adapterPresent: false
+		};
+	}
+	const mcpPath = join(piDir, "mcp.json");
+	let mcp;
+	if (existsSync(mcpPath)) try {
+		mcp = loadJsoncConfig(mcpPath);
+	} catch (err) {
+		p.log.error(`Failed to parse ${mcpPath}: ${err instanceof Error ? err.message : String(err)}`);
+		p.log.info(`Leaving ${mcpPath} untouched. Fix or remove the file, then re-run with --pi-mcp.`);
+		return {
+			ok: false,
+			wrote: false,
+			adapterPresent: true
+		};
+	}
+	else mcp = {};
+	let servers = mcp.mcpServers;
+	if (servers == null || typeof servers !== "object" || Array.isArray(servers)) servers = {};
+	if ("codemem" in servers && !force) {
+		p.log.info(`Pi MCP entry already exists in ${mcpPath}`);
+		return {
+			ok: true,
+			wrote: false,
+			adapterPresent: true
+		};
+	}
+	servers.codemem = { ...PI_MCP_CODEMEM_ENTRY };
+	mcp.mcpServers = servers;
+	try {
+		mkdirSync(piDir, { recursive: true });
+		writeJsonConfigWithBackup(mcpPath, mcp);
+		p.log.success(`Pi MCP entry installed: ${mcpPath}`);
+	} catch (err) {
+		p.log.error(`Failed to write ${mcpPath}: ${err instanceof Error ? err.message : String(err)}`);
+		return {
+			ok: false,
+			wrote: false,
+			adapterPresent: true
+		};
+	}
+	return {
+		ok: true,
+		wrote: true,
+		adapterPresent: true
+	};
+}
+/**
+* Configure pi: packages entry, observer derivation, optional MCP adapter surface.
+* Idempotent; honors PI_CODING_AGENT_DIR. Returns true on success.
+*/
+function installPi(options = {}) {
+	const force = options.force ?? false;
+	const piDir = piConfigDir();
+	try {
+		mkdirSync(piDir, { recursive: true });
+	} catch (err) {
+		p.log.error(`Failed to create pi agent dir ${piDir}: ${err instanceof Error ? err.message : String(err)}`);
+		return false;
+	}
+	p.log.info(`Pi agent directory: ${piDir}`);
+	if (!installPiExtensionPackage(piDir, force, options.piExtensionPath)) return false;
 	let ok = true;
-	const onlyFlag = Boolean(opts.opencodeOnly || opts.claudeOnly || opts.codexOnly);
-	const doOpencode = opts.opencodeOnly || !onlyFlag;
-	const doClaude = opts.claudeOnly || !onlyFlag;
-	const doCodex = opts.codexOnly || !onlyFlag && existsSync(codexConfigDir());
-	if (doOpencode) {
-		p.log.step("Installing OpenCode plugin...");
-		const pluginResult = installPlugin(force);
-		ok = pluginResult.ok && ok;
-		p.log.step("Installing OpenCode MCP config...");
-		ok = installMcp(force, { createBackup: !pluginResult.wrote }) && ok;
-	}
-	if (doClaude) {
-		p.log.step("Installing Claude Code MCP config...");
-		ok = installClaudeMcp(force) && ok;
-	}
-	if (doCodex) {
-		p.log.step("Configuring Codex (MCP + hooks)...");
-		ok = installCodex(force) && ok;
-		p.log.info("Codex next steps:");
-		p.log.info("  - Restart Codex to load the new configuration");
-		p.log.info("  - On first run, approve the one-time prompt to trust the codemem hooks");
-		p.log.info("  - MCP recall works immediately (no trust prompt required)");
-	}
-	if (ok) p.outro("Setup complete — restart your editor to load the plugin");
-	else {
-		p.outro("Setup completed with warnings");
-		process.exitCode = 1;
+	let toolsMode = "native";
+	if (options.piMcp) {
+		const mcpResult = installPiMcp(piDir, force);
+		ok = mcpResult.ok && ok;
+		if (mcpResult.adapterPresent && mcpResult.ok) toolsMode = "mcp-adapter";
+	}
+	ok = wirePiCodememConfig({ toolsMode }) && ok;
+	if (ok) {
+		p.log.info("Pi next steps:");
+		p.log.info("  - Start (or restart) pi to load the extension package");
+		if (!options.piMcp) p.log.info("  - Optional MCP surface: install pi-mcp-adapter, then re-run with --pi-mcp");
+		p.log.info("  - To disable: remove the @codemem/pi-extension entry from pi settings.json packages");
+	}
+	return ok;
+}
+function maybeInstallPi(opts, onlyFlag, force) {
+	if (!(opts.piOnly || !onlyFlag && isPiDetected())) return true;
+	p.log.step("Configuring pi (extension package + observer)...");
+	return installPi({
+		force,
+		piMcp: opts.piMcp,
+		piExtensionPath: opts.piExtensionPath
+	});
+}
+function installOpencodeSetup(force) {
+	p.log.step("Installing OpenCode plugin...");
+	const pluginResult = installPlugin(force);
+	p.log.step("Installing OpenCode MCP config...");
+	return pluginResult.ok && installMcp(force, { createBackup: !pluginResult.wrote });
+}
+function installClaudeSetup(force) {
+	p.log.step("Installing Claude Code MCP config...");
+	return installClaudeMcp(force);
+}
+function installCodexSetup(force) {
+	p.log.step("Configuring Codex (MCP + hooks)...");
+	const ok = installCodex(force);
+	p.log.info("Codex next steps:");
+	p.log.info("  - Restart Codex to load the new configuration");
+	p.log.info("  - On first run, approve the one-time prompt to trust the codemem hooks");
+	p.log.info("  - MCP recall works immediately (no trust prompt required)");
+	return ok;
+}
+function runSetupInstalls(opts, force) {
+	const onlyFlag = Boolean(opts.opencodeOnly || opts.claudeOnly || opts.codexOnly || opts.piOnly);
+	let ok = true;
+	if (opts.opencodeOnly || !onlyFlag) ok = installOpencodeSetup(force) && ok;
+	if (opts.claudeOnly || !onlyFlag) ok = installClaudeSetup(force) && ok;
+	if (opts.codexOnly || !onlyFlag && existsSync(codexConfigDir())) ok = installCodexSetup(force) && ok;
+	return maybeInstallPi(opts, onlyFlag, force) && ok;
+}
+function runSetup(opts) {
+	p.intro(`codemem setup v${VERSION}`);
+	if (runSetupInstalls(opts, opts.force ?? false)) {
+		p.outro("Setup complete — restart your editor to load the plugin");
+		return;
 	}
+	p.outro("Setup completed with warnings");
+	process.exitCode = 1;
+}
+var setupCommand = new Command("setup").configureHelp(helpStyle).description("Install codemem plugin + MCP config for OpenCode, Claude Code, Codex, and pi").option("--force", "overwrite existing installations").option("--opencode-only", "only install for OpenCode").option("--claude-only", "only install for Claude Code").option("--codex-only", "only install for Codex").option("--pi-only", "only install for pi").option("--pi-mcp", "opt into pi MCP adapter surface (requires pi-mcp-adapter)").option("--pi-extension-path <path>", "dev: write a local path packages entry instead of the npm pin").action((opts) => {
+	runSetup(opts);
 });
 //#endregion
 //#region src/commands/stats.ts