@cortexkit/pi-magic-context 0.49.18 → 0.49.20

package/dist/{index-xxn1tpnv.js → index-jt0g3hn0.js}

@@ -17,7 +17,7 @@ import {
   getMagicContextStorageDir,
   log2,
   sessionLog2
-} from "./index-7bqrfyny.js";
+} from "./index-fgs78jyd.js";
 import {
   piModelRefToCanonical,
   modelRefLookupOrder,
@@ -54,6 +54,7 @@ import {
   serializeStrippedPlaceholderState,
   THINKING_BINDING_RECOVERY_FROZEN_PREFIX,
   THINKING_BINDING_STRIP_ORDER_END_MARKER,
+  COMPACTION_OFF_THINKING_FROZEN_PREFIX,
   parseProcessedImageMarkers,
   serializeProcessedImageMarkers,
   getPendingPiCompactionMarkerState,
@@ -66,7 +67,7 @@ import {
   visitRawSessionMessages2,
   readRawSessionMessageById2,
   readSessionChunk2
-} from "./index-r9f4f6s1.js";
+} from "./index-zhwwp8zz.js";
 import {
   isRecord,
   providerVisibleMessage,
@@ -80,25 +81,25 @@ import {
   getTagNumberByMessageId2,
   getTagsBySession2,
   getTagById2
-} from "./index-rhprywn6.js";
+} from "./index-bttctq07.js";
 import {
   estimateTokens
-} from "./index-1kcfwk9c.js";
+} from "./index-my0dn413.js";
 import {
   inRootAsyncContext2,
   Database2,
   withSqliteBackgroundWriter2,
   withoutSqliteTransformPass2
-} from "./index-rrwv9y2k.js";
+} from "./index-hws5f0z9.js";
 import {
   stableStringify,
   closeQuietly,
   getTableColumnNames
-} from "./index-jrap43gc.js";
+} from "./index-w1avzkdt.js";
 import {
   piHarnessKindFromExecutable,
   isPidAlive
-} from "./index-apgwdby5.js";
+} from "./index-n4wyxcgr.js";
 import {
   BoundedSessionMap,
   clearLkgFenceProcessState,
@@ -108,21 +109,21 @@ import {
   V2StoreReader,
   deleteSessionScopedRows,
   getLastIndexedOrdinal2
-} from "./index-fvjthct0.js";
+} from "./index-vdy273xw.js";
 import {
   resolveOpenCodeDbPath,
   assertOpenCodeStoreGeneration
 } from "./index-8976mezv.js";
 import {
   noteRawSessionSummaryRowWritten2
-} from "./index-vktahpdj.js";
+} from "./index-rfyvax95.js";
 import {
   SubcClient,
   connectionFileExists,
   SYNAPSE_DEFAULT_MODEL,
   toSynapseLaneDescriptor,
   SynapseEmbeddingProvider
-} from "./index-d8gc0112.js";
+} from "./index-jxjr7kz9.js";
 import {
   invalidateAutoEmbedSession,
   drainBackgroundBatches,
@@ -135,10 +136,10 @@ import {
   unregisterProjectEmbedding2,
   getProjectEmbeddingSnapshot2,
   embedTextForProject2
-} from "./index-vmcgew7x.js";
+} from "./index-v2epyzbm.js";
 import {
   getErrorMessage
-} from "./index-gp928w10.js";
+} from "./index-mtvvhcw4.js";
 import {
   V2_MEMORY_CATEGORIES,
   CATEGORY_PRIORITY,
@@ -152,13 +153,13 @@ import {
   expandWorkspaceIdentitySetWithAliases,
   resolveStoredPathWorkspaceIdentity,
   storedPathBelongsToWorkspace
-} from "./index-gdjv1nnd.js";
+} from "./index-5vgt1zeh.js";
 import {
   directoryHasGitMetadata,
   resolveProjectIdentityForSession,
   normalizeStoredProjectPath,
   storedPathBelongsToIdentity
-} from "./index-g7p75667.js";
+} from "./index-p63nwn5v.js";
 import {
   getNoteByIdInScope,
   getNotes,
@@ -174,7 +175,7 @@ import {
   resolveMemoriesByIdsForSearch2,
   unifiedSearch2,
   formatSearchResults2
-} from "./index-d43hmn31.js";
+} from "./index-af9t637y.js";
 import {
   computeCueContentHash,
   getMuralCueState,
@@ -190,25 +191,28 @@ import {
   supersededMemory2,
   mergeMemoryStats2,
   archiveMemory2
-} from "./index-e1rza8gp.js";
+} from "./index-dzem6gn9.js";
 import {
   saveEmbedding,
   invalidateMemory2
 } from "./index-yhzhdp3m.js";
+import {
+  preflightLocalEmbeddingPooling
+} from "./index-e19qtjj0.js";
 import {
   isMarkerCandidateBasisCurrent2,
   invalidateMarkerCandidateBasis2,
   readMarkerCandidateBasis2
-} from "./index-qpq6w67c.js";
+} from "./index-6kb2xpw7.js";
 import {
   renderSearchArchive2
-} from "./index-d21pdr8v.js";
+} from "./index-tb3d9trm.js";
 import {
   computeNormalizedHash
 } from "./index-9x1f96r8.js";
 import {
   logSlowWriteTransaction
-} from "./index-d4dt19md.js";
+} from "./index-hqw0rt9k.js";
 import {
   ensureStorageDirectorySync,
   writeStorageFileAtomicSync
@@ -5221,11 +5225,15 @@ function clonePiContentDecisions(raw, filter) {
       ])}`);
       continue;
     }
-    if (entry.startsWith(THINKING_BINDING_RECOVERY_FROZEN_PREFIX)) {
-      const messageId = entry.slice(THINKING_BINDING_RECOVERY_FROZEN_PREFIX.length);
+    const thinkingPrefix = [
+      THINKING_BINDING_RECOVERY_FROZEN_PREFIX,
+      COMPACTION_OFF_THINKING_FROZEN_PREFIX
+    ].find((prefix) => entry.startsWith(prefix));
+    if (thinkingPrefix) {
+      const messageId = entry.slice(thinkingPrefix.length);
       if (messageId.length === 0 || !filter.includeMessageId(messageId))
         continue;
-      copied.push(`${THINKING_BINDING_RECOVERY_FROZEN_PREFIX}${mapMessageId(filter, messageId) ?? messageId}`);
+      copied.push(`${thinkingPrefix}${mapMessageId(filter, messageId) ?? messageId}`);
       continue;
     }
     if (entry.startsWith(MERGED_REASONING_PARTS_PREFIX) || entry.startsWith("pi-content-replay-"))
@@ -9927,7 +9935,7 @@ function unregisterPiProjectEmbeddings(db) {
   registeredIdentitiesByDatabase.delete(db);
   registrationFingerprintsByDatabase.get(db)?.clear();
 }
-async function ensureProjectRegisteredFromPiDirectory(directory, db) {
+async function ensureProjectRegisteredFromPiDirectory(directory, db, options = {}) {
   const detailed = loadPiConfigDetailed({ cwd: directory });
   const projectIdentity = resolveProjectIdentityForSession(directory, detailed.config.allow_home_project);
   if (!projectIdentity)
@@ -9958,6 +9966,9 @@ async function ensureProjectRegisteredFromPiDirectory(directory, db) {
     memoryEnabled: detailed.config.memory.enabled,
     gitCommitEnabled: detailed.config.memory.git_commit_indexing.enabled
   };
+  if (options.preflightLocalPooling && routing.primary.provider === "local") {
+    await preflightLocalEmbeddingPooling(routing.primary.model ?? "Xenova/all-MiniLM-L6-v2");
+  }
   let registered = registeredIdentitiesByDatabase.get(db);
   if (!registered) {
     registered = new Set;
@@ -12712,7 +12723,7 @@ function createCtxReduceTool(deps) {
       params = unwrapImitatedReducedArgs(params, ["drop"], { drop: "string" });
       const sessionId = ctx.sessionManager.getSessionId();
       if (!params.drop) {
-        return err4("Error: 'drop' must be provided.");
+        return err4(`Error: 'drop' must be provided as a string of tag numbers, for example {"drop": "3-5,8"}.`);
       }
       let dropIds = [];
       try {
@@ -12876,7 +12887,7 @@ function createCtxSearchTool(deps) {
     label: "Magic Context: Search",
     description: CTX_SEARCH_DESCRIPTION,
     parameters: ParamsSchema4,
-    async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
+    async execute(_toolCallId, params, signal, _onUpdate, ctx) {
       params = unwrapImitatedReducedArgs(params, ["query"], {
         query: "string",
         limit: "number",
@@ -12972,6 +12983,7 @@ function createCtxSearchTool(deps) {
         gitRepositoryAvailable: directoryHasGitMetadata(ctx.cwd),
         explicitSearch: true,
         rerankConnectionFile: deps.rerankConnectionFile,
+        signal,
         ...dateRange
       });
       return {
@@ -13516,6 +13528,11 @@ function createTodowriteTool() {
 
 // src/tools/index.ts
 var CTX_MEMORY_TOOL_NAME = "ctx_memory";
+function setOmpEssentialLoadMode(definition) {
+  if (resolvePiHarnessKind() !== "omp")
+    return definition;
+  return Object.assign(definition, { loadMode: "essential" });
+}
 function syncCtxMemoryToolEnabled(pi, memoryEnabled) {
   const activeTools = pi.getActiveTools();
   const isActive = activeTools.includes(CTX_MEMORY_TOOL_NAME);
@@ -13534,11 +13551,11 @@ function registerMagicContextTools(pi, opts) {
   const surfaceTool = (definition) => {
     const parameters = structuredClone(definition.parameters);
     applyJsonSchemaParameterDescriptions(definition.name, parameters, registration.preset, smartNotesAvailable ? opts.smartNotesEnabled : false);
-    return throwReturnedToolErrors({
+    return setOmpEssentialLoadMode(throwReturnedToolErrors({
       ...definition,
       parameters,
       description: definition.name === "ctx_note" && !smartNotesAvailable ? withoutSmartNoteText(registration.preset === "light" ? CTX_NOTE_LIGHT_DESCRIPTION : definition.description) : registration.descriptionFor(definition.name, definition.description)
-    });
+    }));
   };
   pi.registerTool(surfaceTool(createCtxSearchTool({
     db: opts.db,
@@ -13560,7 +13577,7 @@ function registerMagicContextTools(pi, opts) {
     };
     pi.registerTool(surfaceTool(createCtxMemoryTool(memoryDeps)));
     if (opts.allowDreamerActions === true) {
-      pi.registerTool(throwReturnedToolErrors(createCtxMemoryListTool(memoryDeps)));
+      pi.registerTool(setOmpEssentialLoadMode(throwReturnedToolErrors(createCtxMemoryListTool(memoryDeps))));
     }
   }
   if (!opts.sessionScopedToolsDisabled) {

package/dist/{index-d8gc0112.js → index-jxjr7kz9.js}

@@ -1,15 +1,15 @@
 import {
   estimateTokens
-} from "./index-1kcfwk9c.js";
+} from "./index-my0dn413.js";
 import {
   getHarness2
 } from "./index-0h2mkab2.js";
 import {
   log2
-} from "./index-7bqrfyny.js";
+} from "./index-fgs78jyd.js";
 import {
   inRootAsyncContext2
-} from "./index-rrwv9y2k.js";
+} from "./index-hws5f0z9.js";
 
 // ../plugin/src/features/magic-context/memory/embedding-synapse.ts
 import { createHash } from "node:crypto";

package/dist/{index-4bv5bttd.js → index-kqb3d8bs.js}

@@ -3,18 +3,21 @@ import {
   tagRevisionCountSql,
   tagRevisionKey,
   TAG_INSERT_CONFLICT_ROWS
-} from "./index-jrap43gc.js";
+} from "./index-w1avzkdt.js";
 import {
   getTagsBySession2,
   getTagsByNumbers2
-} from "./index-rhprywn6.js";
+} from "./index-bttctq07.js";
 import {
   inRootAsyncContext2,
   getSqliteDatabasePath2
-} from "./index-rrwv9y2k.js";
+} from "./index-hws5f0z9.js";
 import {
   createFailClosedBlockingError
-} from "./index-gp928w10.js";
+} from "./index-mtvvhcw4.js";
+import {
+  createWorkerShutdown
+} from "./index-54wmg4rj.js";
 
 // src/tag-snapshot-worker-client-pi.ts
 import { setImmediate as yieldImmediate } from "node:timers/promises";
@@ -228,6 +231,7 @@ function startPiTagSnapshotWorker(request, options = {}) {
   const started = performance.now();
   const consumer = createPiHydrationPacketConsumer(request, options.cached, options.byteLimit);
   const worker = inRootAsyncContext2(() => new Worker(entry, { workerData: request }));
+  const shutdown = createWorkerShutdown(worker, "pi-tag-snapshot", 1000);
   let exited = false;
   let settled = false;
   let consumption = Promise.resolve();
@@ -292,7 +296,7 @@ function startPiTagSnapshotWorker(request, options = {}) {
     clearTimeout(timer);
     rejectResult(error);
     if (!exited)
-      worker.terminate().catch(() => {});
+      shutdown.close();
   }
   const timer = inRootAsyncContext2(() => setTimeout(() => abort(new PiTagHydrationError("job deadline exceeded", "deadline")), options.deadlineMs ?? 1e4));
   timer.unref();

package/dist/{index-gp928w10.js → index-mtvvhcw4.js}

@@ -5,7 +5,7 @@ import {
   importPluginModule,
   writeForwardedLogLine2,
   log2
-} from "./index-7bqrfyny.js";
+} from "./index-fgs78jyd.js";
 import {
   inRootAsyncContext2,
   registerSlowWriteReporter2,
@@ -16,10 +16,10 @@ import {
   withoutSqliteTransformPass2,
   isTransientSqliteError2,
   withPrivilegedWriter2
-} from "./index-rrwv9y2k.js";
+} from "./index-hws5f0z9.js";
 import {
   scheduleAfterBootQuiet
-} from "./index-g7p75667.js";
+} from "./index-p63nwn5v.js";
 import {
   getHarness2,
   harnessOwnsOpenCodeStore2
@@ -34,13 +34,13 @@ import {
   parseRpcPortFile,
   inspectProcessesAsync,
   inspectWindowsProcessesSync
-} from "./index-apgwdby5.js";
+} from "./index-n4wyxcgr.js";
 import {
   configureContextDatabasePragmas2
 } from "./index-fapenpch.js";
 import {
   startMessageFtsRowidMapBackfill2
-} from "./index-yygsh681.js";
+} from "./index-rgfb753z.js";
 import {
   LKG_SLOT_CHUNKS_DDL,
   SESSION_REPLAY_DECISIONS_DDL,
@@ -50,13 +50,13 @@ import {
   hasPendingMigrations,
   getMainThreadMigrationBodyCount,
   runMigrations
-} from "./index-gdjv1nnd.js";
+} from "./index-5vgt1zeh.js";
 import {
   closeQuietly,
   ensureColumn,
   installV95PerfSchema,
   installTagIdentityRevisionTrigger
-} from "./index-jrap43gc.js";
+} from "./index-w1avzkdt.js";
 import {
   resolveOpenCodeDbPath,
   hasV1MessageTables,
@@ -65,7 +65,7 @@ import {
 } from "./index-8976mezv.js";
 import {
   logSlowWriteTransaction
-} from "./index-d4dt19md.js";
+} from "./index-hqw0rt9k.js";
 import {
   shouldEnforcePrivateStoragePermissions,
   ensureStorageDirectorySync,
@@ -1305,8 +1305,8 @@ function finishDatabaseOpen(db, dbPath, explicitDbPath, latestSupportedVersion)
       state.running = startMessageFtsRowidMapBackfill2(db, signal).then(async () => {
         if (signal.aborted || !readsOpenCodeStore)
           return;
-        const { readRawSessionMessagePage, readRawSessionMessages } = await importPluginModule(() => import("./read-session-chunk-chepqmw5.js"));
-        const { startMessageTimeBackfill } = await importPluginModule(() => import("./message-time-backfill-ssshbdy8.js"));
+        const { readRawSessionMessagePage, readRawSessionMessages } = await importPluginModule(() => import("./read-session-chunk-7g4zvpd8.js"));
+        const { startMessageTimeBackfill } = await importPluginModule(() => import("./message-time-backfill-kc10yrcj.js"));
         if (signal.aborted)
           return;
         await startMessageTimeBackfill(db, Object.assign(readRawSessionMessages, {

package/dist/{index-1kcfwk9c.js → index-my0dn413.js}

@@ -1,6 +1,6 @@
 import {
   log2
-} from "./index-7bqrfyny.js";
+} from "./index-fgs78jyd.js";
 
 // ../plugin/src/hooks/magic-context/read-session-formatting.ts
 import { existsSync, readFileSync, realpathSync, statSync } from "node:fs";

package/dist/{index-apgwdby5.js → index-n4wyxcgr.js}

@@ -1,6 +1,6 @@
 import {
   log2
-} from "./index-7bqrfyny.js";
+} from "./index-fgs78jyd.js";
 
 // ../plugin/src/shared/rpc-utils.ts
 import { execFile, execFileSync } from "node:child_process";

package/dist/{index-g7p75667.js → index-p63nwn5v.js}

@@ -1,10 +1,10 @@
 import {
   getMagicContextStorageDir,
   log2
-} from "./index-7bqrfyny.js";
+} from "./index-fgs78jyd.js";
 import {
   inRootAsyncContext2
-} from "./index-rrwv9y2k.js";
+} from "./index-hws5f0z9.js";
 import {
   projectDirectoryKey
 } from "./index-wrr9rp9a.js";

package/dist/{index-svgkv4g2.js → index-qerzchrs.js}

@@ -3,10 +3,10 @@ import {
 } from "./index-0h2mkab2.js";
 import {
   isUserHomeDirectory
-} from "./index-g7p75667.js";
+} from "./index-p63nwn5v.js";
 import {
   logSlowWriteTransaction
-} from "./index-d4dt19md.js";
+} from "./index-hqw0rt9k.js";
 
 // ../plugin/src/features/magic-context/session-project-storage.ts
 var SESSION_CHUNK_REPAIR_BATCH_SIZE = 100;

package/dist/{index-vktahpdj.js → index-rfyvax95.js}

@@ -2,13 +2,16 @@ import {
   inRootAsyncContext2,
   getSqliteDatabasePath2,
   releaseIndexOwner2
-} from "./index-rrwv9y2k.js";
+} from "./index-hws5f0z9.js";
 import {
   registerExitAbort
 } from "./index-399p688g.js";
 import {
   log2
-} from "./index-7bqrfyny.js";
+} from "./index-fgs78jyd.js";
+import {
+  createWorkerShutdown
+} from "./index-54wmg4rj.js";
 
 // ../plugin/src/hooks/magic-context/read-session-raw.ts
 import { isMainThread } from "node:worker_threads";
@@ -17,6 +20,7 @@ import { isMainThread } from "node:worker_threads";
 import { Worker } from "node:worker_threads";
 var ACCEPT_TIMEOUT_MS = 2000;
 var RESULT_TIMEOUT_MS = 10 * 60000;
+var WARMUP_CLOSE_DEADLINE_MS = 2000;
 var FAILURE_BACKOFF_MS = 5 * 60000;
 var inFlight = new Map;
 var connectionIds = new WeakMap;
@@ -45,6 +49,7 @@ function runSessionWarmupWorker(input, options) {
   return new Promise((resolve) => {
     const openCancellation = new Int32Array(new SharedArrayBuffer(Int32Array.BYTES_PER_ELEMENT));
     let worker;
+    let shutdown;
     let deadline;
     let settled = false;
     const finish = (result) => {
@@ -56,7 +61,7 @@ function runSessionWarmupWorker(input, options) {
       options.signal?.removeEventListener("abort", abort);
       if (worker) {
         activeWorkers.delete(worker);
-        const stopped = worker.terminate();
+        const stopped = shutdown?.close() ?? worker.terminate();
         worker.unref();
         stopped.finally(() => {
           releaseIndexOwner2(input);
@@ -82,6 +87,7 @@ function runSessionWarmupWorker(input, options) {
       worker = inRootAsyncContext2(() => new Worker(options.entry ?? defaultEntry(), {
         workerData: { ...input, openCancellation: openCancellation.buffer }
       }));
+      shutdown = createWorkerShutdown(worker, "raw-ordinal-warmup", WARMUP_CLOSE_DEADLINE_MS);
     } catch (error) {
       finish({ failure: "start-failed", detail: String(error) });
       return;

package/dist/{index-yygsh681.js → index-rgfb753z.js}

@@ -2,13 +2,13 @@ import {
   getSqliteDatabasePath2,
   withSqliteBackgroundWriter2,
   withoutSqliteTransformPass2
-} from "./index-rrwv9y2k.js";
+} from "./index-hws5f0z9.js";
 import {
   runSessionWarmupWorker
-} from "./index-vktahpdj.js";
+} from "./index-rfyvax95.js";
 import {
   logSlowWriteTransaction
-} from "./index-d4dt19md.js";
+} from "./index-hqw0rt9k.js";
 
 // ../plugin/src/features/magic-context/message-fts-rowid-map.ts
 import { createHash } from "node:crypto";

package/dist/{index-d21pdr8v.js → index-tb3d9trm.js}

@@ -3,10 +3,10 @@ import {
 } from "./index-1cvyaw9a.js";
 import {
   estimateTokens
-} from "./index-1kcfwk9c.js";
+} from "./index-my0dn413.js";
 import {
   readRawSessionMessageIdOrdinalsForRange2
-} from "./index-r9f4f6s1.js";
+} from "./index-zhwwp8zz.js";
 
 // ../plugin/src/tools/ctx-expand/search-archive.ts
 var archiveReadsOnThisThread = 0;