@matthewfl/pi-contemplator 0.1.9 → 0.1.11

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@matthewfl/pi-contemplator",
-  "version": "0.1.9",
+  "version": "0.1.11",
   "description": "A Pi extension that keeps long-running agentic sessions on track with background memory, contemplation, and structural review.",
   "type": "module",
   "license": "MIT",

package/src/agents/summarizer/agent.ts

@@ -27,7 +27,7 @@ import { estimateStringTokens } from "../../tokens.js";
 import { createRecallAgentTool } from "../../tools/recall-observation.js";
 import { createSearchMemoriesAgentTool } from "../../tools/search-memories.js";
 import { logAgentStreamError } from "../stream-errors.js";
-import { SUMMARIZER_CONTINUE, SUMMARIZER_SYSTEM } from "./prompts.js";
+import { summarizerContinue, SUMMARIZER_SYSTEM } from "./prompts.js";
 import {
 	renderSummarizerMemory,
 	sampleSummarizerMemories,
@@ -568,7 +568,7 @@ export async function runSummarizer(args: RunSummarizerArgs): Promise<Summarizer
 
 	try {
 		await runOnce("The preceding summarize call and receipt are an illustrative example only. Its placeholder ids are not real and it did not create a summary. Now inspect the actual records and use tools to register safe compression, or call done if none is warranted.", false);
-		for (let invocation = 1; !completedWithDone && invocation < SUMMARIZER_MAX_INVOCATIONS; invocation++) await runOnce(SUMMARIZER_CONTINUE, true);
+		for (let invocation = 1; !completedWithDone && invocation < SUMMARIZER_MAX_INVOCATIONS; invocation++) await runOnce(summarizerContinue(drafts.size, invocation), true);
 	} catch (error) {
 		debugLog("summarizer.error", { error: error instanceof Error ? error.message : String(error), acceptedSummaries: drafts.size });
 		if (drafts.size === 0) return { completed: false, reviewedUpToId: coversUpToId, sample };

package/src/agents/summarizer/prompts.ts

@@ -24,7 +24,7 @@ Citations and retrieval:
 - Cite sources inline with square brackets: [aaaaaaaaaaaa, bbbbbbbbbbbb]. Square brackets are only for citations.
 - A future agent can recall citations for full paths, commands, errors, logs, and intermediate results. Keep those details inline only when they are needed to understand or use the summary; otherwise preserve the conclusion and a useful retrieval cue.
 - A summary must stand alone and cite at least two newly consumable provided memories.
-- Do not count tokens or laboriously audit ids. Call the tool early: it validates ids and compression and explains any rejection.
+- Do not count tokens, laboriously audit ids, or draft summaries in prose. Call summarize directly with in-progress summaries: the tool validates ids and compression and explains any rejection. If a recorded summary needs revision, correct it afterward with fix_summary.
 
 Examples:
 - BAD: "The test command was run several times [aaaaaaaaaaaa, bbbbbbbbbbbb]."
@@ -38,9 +38,13 @@ Tools:
 - summarize records one or more summaries and can mark inspected memories keep_verbatim for this run. Read its receipt: it identifies every source removed from the visible pool. A rejected candidate changes nothing; correct it or leave the sources verbatim.
 - fix_summary corrects or removes only a summary created in this run.
 - search_memories and recall are for concrete evidence suggested by the provided records, not for hunting unrelated history to compress.
-- Prose does not change memory. Use tool calls to register decisions.
+- Prose does not change memory. DO NOT DRAFT summaries in text. Directly record in-progress summaries with summarize; use fix_summary afterward if they need revision.
 - Call done alone after all safe work is recorded. If no safe summary is warranted, call done immediately.
 
 Prefer faithful useful compression over both distortion and indefinite accumulation. Under pressure, make progress on old low-value clusters first; treat durable valuable records and user intent as the last things to compress.`;
 
-export const SUMMARIZER_CONTINUE = "IMPORTANT!!!! CALL summarize TOOL NOW TO RECORD ANY SUMMARIES YOU HAVE DECIDED, OR CALL done IF NO SAFE SUMMARY IS WARRANTED. DO NOT DESCRIBE THE ACTION IN PROSE—USE A TOOL NOW.";
+export function summarizerContinue(recordedSummaries: number, reminderNumber: number): string {
+	const count = Math.max(0, Math.floor(recordedSummaries));
+	const thinkingMinutes = Math.max(1, Math.floor(reminderNumber)) * 20;
+	return `IMPORTANT!!!! YOU HAVE BEEN THINKING FOR ${thinkingMinutes} MINUTES. CALL A TOOL NOW. DO NOT DRAFT OR WRITE SUMMARIES IN THE MAIN TEXT. DIRECTLY RECORD IN-PROGRESS SUMMARIES USING summarize; IF THERE IS A PROBLEM, REVISE THEM LATER USING fix_summary. THERE ${count === 1 ? "IS" : "ARE"} CURRENTLY ${count} RECORDED ${count === 1 ? "SUMMARY" : "SUMMARIES"}${count === 0 ? "; NOTHING HAS BEEN SUMMARIZED YET" : ""}. IF YOU ALREADY WROTE SUMMARIES IN THE MAIN TEXT, RECORD THEM USING summarize NOW. IF NO SAFE SUMMARY IS WARRANTED, CALL done.`;
+}

package/src/commands/settings.ts

@@ -58,8 +58,16 @@ export function observerInputCapLabel(runtime: Runtime, contextWindow: number |
 	return `${cap.toLocaleString()} tokens (fallback default; model context unavailable)`;
 }
 
+function memoryWorkerModelSettingLabel(runtime: Runtime, worker: "observer" | "summarizer"): string {
+	const key = worker === "observer" ? "observerModel" : "summarizerModel";
+	const settings = runtime.getSessionSettings();
+	if (hasOverride(settings, key)) return modelLabel(runtime.config[key]);
+	const inherited = runtime.config[key] ?? runtime.config.model;
+	return `${modelLabel(inherited)} (default)`;
+}
+
 function observerContextWindow(runtime: Runtime, ctx: ExtensionContext): number | undefined {
-	const configured = runtime.config.model;
+	const configured = runtime.configuredMemoryWorkerModel("observer");
 	const registry = ctx.modelRegistry as unknown as ModelRegistryLike;
 	const model = configured ? registry.find?.(configured.provider, configured.id) ?? ctx.model : ctx.model;
 	return (model as { contextWindow?: number } | undefined)?.contextWindow;
@@ -140,11 +148,25 @@ class FilterableModelSelector extends Container implements Focusable {
 	}
 }
 
-async function chooseModel(ctx: ExtensionContext, current: ConfiguredModel | undefined, title: string): Promise<ConfiguredModel | null | undefined> {
+export async function modelsForSettingsSelector(ctx: ExtensionContext): Promise<ConfiguredModel[]> {
+	// Pi resolves enabledModels (and --models) into this session-scoped list.
+	// Mirroring it keeps our worker pickers consistent with the built-in model
+	// picker instead of unexpectedly exposing the entire provider catalogue.
+	if (ctx.scopedModels.length > 0) {
+		return ctx.scopedModels.map(({ model, thinkingLevel }) => ({
+			provider: model.provider,
+			id: model.id,
+			...(thinkingLevel === undefined ? {} : { thinking: thinkingLevel }),
+		}));
+	}
 	const registry = ctx.modelRegistry as unknown as ModelRegistryLike;
 	await registry.refresh?.();
 	const available = registry.getAvailable();
-	const models = available.length > 0 ? available : registry.getAll();
+	return (available.length > 0 ? available : registry.getAll()).map(({ provider, id }) => ({ provider, id }));
+}
+
+async function chooseModel(ctx: ExtensionContext, current: ConfiguredModel | undefined, title: string): Promise<ConfiguredModel | null | undefined> {
+	const models = await modelsForSettingsSelector(ctx);
 	if (models.length === 0) {
 		ctx.ui.notify("No configured models are available.", "warning");
 		return undefined;
@@ -218,6 +240,7 @@ export function registerSettingsCommand(pi: ExtensionAPI, runtime: Runtime): voi
 				`Contemplator new-observation trigger (count): ${scalarLabel(runtime, "contemplatorMinNewObservations")}`,
 				`Contemplator response spacing (count): ${scalarLabel(runtime, "contemplatorMinTurns")}`,
 				`Summarizer enabled: ${scalarLabel(runtime, "summarizerEnabled")}`,
+				`Summarizer model: ${memoryWorkerModelSettingLabel(runtime, "summarizer")}`,
 				`New memory pool protection budget (tokens): ${scalarLabel(runtime, "newMemoryPoolMaxTokens")}`,
 				`Old memory pool target (tokens, advisory): ${scalarLabel(runtime, "oldMemoryPoolTargetTokens")}`,
 				`Summarizer old-pool retrigger growth (tokens): ${scalarLabel(runtime, "summarizerRetriggerTokens")}`,
@@ -225,7 +248,7 @@ export function registerSettingsCommand(pi: ExtensionAPI, runtime: Runtime): voi
 				`Structural reviewer enabled: ${scalarLabel(runtime, "reviewerEnabled")}`,
 				`Structural reviewer model: ${hasOverride(settings, "reviewerModel") ? modelLabel(runtime.config.reviewerModel) : `${modelLabel(runtime.getDefaultConfig().reviewerModel)} (default)`}`,
 				`Observe source during compaction: ${scalarLabel(runtime, "compactionObserverEnabled")}`,
-				`Observer and summarizer model: ${hasOverride(settings, "model") ? modelLabel(runtime.config.model) : `${modelLabel(runtime.getDefaultConfig().model)} (default)`}`,
+				`Observer model: ${memoryWorkerModelSettingLabel(runtime, "observer")}`,
 				`Observer source backlog trigger (tokens): ${scalarLabel(runtime, "observeAfterTokens")}`,
 				`Observer input cap: ${observerInputCapLabel(runtime, observerContextWindow(runtime, ctx))}`,
 				`Observer and summarizer max rounds: ${scalarLabel(runtime, "agentMaxTurns")}`,
@@ -251,9 +274,12 @@ export function registerSettingsCommand(pi: ExtensionAPI, runtime: Runtime): voi
 			} else if (choice.startsWith("Structural reviewer model:")) {
 				const model = await chooseModel(ctx, runtime.config.reviewerModel, "Structural reviewer model");
 				if (model !== undefined) appendSettings(pi, runtime, { reviewerModel: model });
-			} else if (choice.startsWith("Observer and summarizer model:")) {
-				const model = await chooseModel(ctx, runtime.config.model, "Observer and summarizer model");
-				if (model !== undefined) appendSettings(pi, runtime, { model });
+			} else if (choice.startsWith("Observer model:")) {
+				const model = await chooseModel(ctx, runtime.config.observerModel ?? runtime.config.model, "Observer model");
+				if (model !== undefined) appendSettings(pi, runtime, { observerModel: model });
+			} else if (choice.startsWith("Summarizer model:")) {
+				const model = await chooseModel(ctx, runtime.config.summarizerModel ?? runtime.config.model, "Summarizer model");
+				if (model !== undefined) appendSettings(pi, runtime, { summarizerModel: model });
 			} else if (choice.startsWith("Automatic compaction threshold mode:")) {
 				const mode = await ctx.ui.select("Automatic compaction threshold mode", ["calibrated", "ratio"]);
 				if (mode === "calibrated" || mode === "ratio") appendSettings(pi, runtime, { compactAfterTokensMode: mode });

package/src/commands/status.ts

@@ -40,6 +40,10 @@ function formatRunAge(timestamp: number): string {
 	return `${new Date(timestamp).toISOString()} (${formatDuration(Math.max(0, Date.now() - timestamp))} ago)`;
 }
 
+function configuredModelLabel(model: { provider: string; id: string } | null): string {
+	return model ? `${model.provider}/${model.id}` : "current session model";
+}
+
 function contemplatorWaitingLabel(waitingFor: Runtime["contemplatorState"]["waitingFor"]): string {
 	switch (waitingFor) {
 		case "observer": return "waiting for observer backlog";
@@ -127,6 +131,8 @@ export function registerStatusCommand(pi: ExtensionAPI, runtime: Runtime): void
 				`Old memory pool:         ~${pools.oldTokens.toLocaleString()} / ${runtime.config.oldMemoryPoolTargetTokens.toLocaleString()} advisory target tokens (${pct(pools.oldTokens, runtime.config.oldMemoryPoolTargetTokens)}%)`,
 				`Summary pool:            ~${visibleSummaryTokens.toLocaleString()} visible tokens`,
 				`Summarizer:              ${runtime.config.summarizerEnabled === false ? "disabled" : "enabled"}; retrigger after +${runtime.config.summarizerRetriggerTokens.toLocaleString()} old-pool tokens / sample above ~${summarizerSamplingTokens.toLocaleString()} tokens`,
+				`Summarizer model:        ${configuredModelLabel(runtime.configuredMemoryWorkerModel("summarizer"))}`,
+				`Observer model:          ${configuredModelLabel(runtime.configuredMemoryWorkerModel("observer"))}`,
 				`Cumulative agent time:   ${formatDuration(agentActiveTimeMs(entries))}`,
 				`Observe source during compaction: ${runtime.config.compactionObserverEnabled === false ? "disabled" : "enabled"}`,
 				`Contemplator:             ${runtime.config.contemplatorEnabled ? "enabled" : "disabled"}`,

package/src/config.ts

@@ -46,7 +46,12 @@ export interface Config {
 	/** Advisory token target for older summarizer-eligible memory. */
 	oldMemoryPoolTargetTokens: number;
 	agentMaxTurns: number;
+	/** Legacy shared fallback for observer/summarizer model selection. */
 	model?: ConfiguredModel;
+	/** Optional model override used only by the observer. */
+	observerModel?: ConfiguredModel;
+	/** Optional model override used only by the summarizer. */
+	summarizerModel?: ConfiguredModel;
 	showWorkerNotifications: boolean;
 	passive: boolean;
 	/** Run the asynchronous observer when a compaction begins. */
@@ -231,6 +236,10 @@ function normalizeSettingsConfig(value: Record<string, unknown>): Partial<Config
 	if (typeof value.debugLog === "boolean") normalized.debugLog = value.debugLog;
 	const model = normalizeModel(value.model);
 	if (model) normalized.model = model;
+	const observerModel = normalizeModel(value.observerModel);
+	if (observerModel) normalized.observerModel = observerModel;
+	const summarizerModel = normalizeModel(value.summarizerModel);
+	if (summarizerModel) normalized.summarizerModel = summarizerModel;
 	const contemplatorModel = normalizeModel(value.contemplatorModel);
 	if (contemplatorModel) normalized.contemplatorModel = contemplatorModel;
 	const reviewerModel = normalizeModel(value.reviewerModel);

package/src/hooks/consolidation-trigger.ts

@@ -72,6 +72,12 @@ function shouldNotifyWorker(runtime: Runtime, ctx: ConsolidationCtx): boolean {
 	return runtime.config.showWorkerNotifications && ctx.hasUI;
 }
 
+function configuredMemoryWorkerModel(runtime: Runtime, worker: "observer" | "summarizer") {
+	return typeof runtime.configuredMemoryWorkerModel === "function"
+		? runtime.configuredMemoryWorkerModel(worker)
+		: runtime.config[worker === "observer" ? "observerModel" : "summarizerModel"] ?? runtime.config.model ?? null;
+}
+
 function makeModelResolver(runtime: Runtime, ctx: ConsolidationCtx): (stage: "observer") => Promise<ResolvedModel | undefined> {
 	let cached: ResolveResult | undefined;
 	return async (stage) => {
@@ -80,6 +86,7 @@ function makeModelResolver(runtime: Runtime, ctx: ConsolidationCtx): (stage: "ob
 			modelRegistry: ctx.modelRegistry,
 			hasUI: ctx.hasUI,
 			ui: ctx.ui,
+			configuredModel: configuredMemoryWorkerModel(runtime, "observer"),
 		});
 		if (cached.ok) {
 			runtime.resolveFailureNotified = false;
@@ -368,7 +375,13 @@ export function scheduleSummarizer(pi: ExtensionAPI, runtime: Runtime, ctx: Cons
 		runtime.lastSummarizerStartedAt = startedAt;
 		runtime.lastSummarizerRun = { startedAt, status: "running", messages: [] };
 		try {
-			const resolved = await runtime.resolveModel({ model: ctx.model, modelRegistry: ctx.modelRegistry, hasUI: ctx.hasUI, ui: ctx.ui });
+			const resolved = await runtime.resolveModel({
+				model: ctx.model,
+				modelRegistry: ctx.modelRegistry,
+				hasUI: ctx.hasUI,
+				ui: ctx.ui,
+				configuredModel: configuredMemoryWorkerModel(runtime, "summarizer"),
+			});
 			if (!resolved.ok) {
 				debugLog("summarizer.model_unavailable", { reason: resolved.reason });
 				runtime.lastSummarizerRun = { startedAt, status: "failed", messages: [], error: resolved.reason };
@@ -394,7 +407,7 @@ export function scheduleSummarizer(pi: ExtensionAPI, runtime: Runtime, ctx: Cons
 				newPoolMaxTokens: runtime.config.newMemoryPoolMaxTokens,
 				samplingThresholdTokens: runtime.config.summarizerSamplingThresholdTokens,
 				maxTurns: runtime.config.agentMaxTurns,
-				thinkingLevel: runtime.config.model?.thinking ?? "minimal",
+				thinkingLevel: runtime.config.summarizerModel?.thinking ?? runtime.config.model?.thinking ?? "minimal",
 				recordUsage: (usage) => runtime.recordAgentUsage(usage),
 				onMessages: (messages) => {
 					watchdog.progress();
@@ -561,7 +574,7 @@ async function runObserverStage(
 			chunk,
 			allowedSourceEntryIds: sourceEntryIds,
 			maxTurns: runtime.config.agentMaxTurns,
-			thinkingLevel: runtime.config.model?.thinking ?? "low",
+			thinkingLevel: runtime.config.observerModel?.thinking ?? runtime.config.model?.thinking ?? "low",
 			recordUsage: (usage) => runtime.recordAgentUsage(usage),
 			onProgress: observerWatchdog.progress,
 			onMessages: (messages) => {

package/src/runtime.ts

@@ -31,6 +31,8 @@ export type SessionSettings = Partial<Pick<Config,
 >> & {
 	/** null explicitly means use the configured/session model. */
 	model?: ConfiguredModel | null;
+	observerModel?: ConfiguredModel | null;
+	summarizerModel?: ConfiguredModel | null;
 	contemplatorModel?: ConfiguredModel | null;
 	reviewerModel?: ConfiguredModel | null;
 };
@@ -148,6 +150,10 @@ export function computeSessionSettings(entries: readonly unknown[]): SessionSett
 		if (typeof data.compactAfterTokensRatio === "number" && data.compactAfterTokensRatio > 0 && data.compactAfterTokensRatio < 1) restored.compactAfterTokensRatio = data.compactAfterTokensRatio;
 		if (data.model === null) restored.model = null;
 		else if (isConfiguredModel(data.model)) restored.model = data.model;
+		if (data.observerModel === null) restored.observerModel = null;
+		else if (isConfiguredModel(data.observerModel)) restored.observerModel = data.observerModel;
+		if (data.summarizerModel === null) restored.summarizerModel = null;
+		else if (isConfiguredModel(data.summarizerModel)) restored.summarizerModel = data.summarizerModel;
 		if (data.contemplatorModel === null) restored.contemplatorModel = null;
 		else if (isConfiguredModel(data.contemplatorModel)) restored.contemplatorModel = data.contemplatorModel;
 		if (data.reviewerModel === null) restored.reviewerModel = null;
@@ -240,11 +246,13 @@ export class Runtime {
 	}
 
 	private applySessionSettings(): void {
-		const { model, contemplatorModel, reviewerModel, ...scalarSettings } = this.sessionSettings;
+		const { model, observerModel, summarizerModel, contemplatorModel, reviewerModel, ...scalarSettings } = this.sessionSettings;
 		this.config = {
 			...this.baseConfig,
 			...scalarSettings,
 			...(model === undefined ? {} : { model: model ?? undefined }),
+			...(observerModel === undefined ? {} : { observerModel: observerModel ?? undefined }),
+			...(summarizerModel === undefined ? {} : { summarizerModel: summarizerModel ?? undefined }),
 			...(contemplatorModel === undefined ? {} : { contemplatorModel: contemplatorModel ?? undefined }),
 			...(reviewerModel === undefined ? {} : { reviewerModel: reviewerModel ?? undefined }),
 		};
@@ -263,6 +271,14 @@ export class Runtime {
 		return { ...this.baseConfig };
 	}
 
+	/** Resolve an observer/summarizer override, retaining explicit session-model selection. */
+	configuredMemoryWorkerModel(worker: "observer" | "summarizer"): ConfiguredModel | null {
+		const key = worker === "observer" ? "observerModel" : "summarizerModel";
+		const sessionValue = this.sessionSettings[key];
+		if (sessionValue === null) return null;
+		return this.config[key] ?? this.config.model ?? null;
+	}
+
 	advanceContextGeneration(): void {
 		this.contextGeneration++;
 		// A session switch (or reload/shutdown) invalidates any in-flight or pending