@cortexkit/pi-magic-context 0.49.20 → 0.49.21

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

@@ -30,7 +30,7 @@ import {
   PER_HARNESS_MODEL_KEYS,
   ConfigProfilesSchema,
   MagicContextConfigSchema
-} from "./index-bxmf4dhr.js";
+} from "./index-ccanzg8p.js";
 import {
   MERGED_REASONING_PARTS_PREFIX,
   decodeMergedReasoningParts,
@@ -67,7 +67,7 @@ import {
   visitRawSessionMessages2,
   readRawSessionMessageById2,
   readSessionChunk2
-} from "./index-zhwwp8zz.js";
+} from "./index-gqjxy2p7.js";
 import {
   isRecord,
   providerVisibleMessage,
@@ -81,7 +81,7 @@ import {
   getTagNumberByMessageId2,
   getTagsBySession2,
   getTagById2
-} from "./index-bttctq07.js";
+} from "./index-n8e5e9fp.js";
 import {
   estimateTokens
 } from "./index-my0dn413.js";
@@ -90,7 +90,7 @@ import {
   Database2,
   withSqliteBackgroundWriter2,
   withoutSqliteTransformPass2
-} from "./index-hws5f0z9.js";
+} from "./index-2qtx0q6h.js";
 import {
   stableStringify,
   closeQuietly,
@@ -109,21 +109,21 @@ import {
   V2StoreReader,
   deleteSessionScopedRows,
   getLastIndexedOrdinal2
-} from "./index-vdy273xw.js";
+} from "./index-wgmhk91m.js";
 import {
   resolveOpenCodeDbPath,
   assertOpenCodeStoreGeneration
 } from "./index-8976mezv.js";
 import {
   noteRawSessionSummaryRowWritten2
-} from "./index-rfyvax95.js";
+} from "./index-at6py2k3.js";
 import {
   SubcClient,
   connectionFileExists,
   SYNAPSE_DEFAULT_MODEL,
   toSynapseLaneDescriptor,
   SynapseEmbeddingProvider
-} from "./index-jxjr7kz9.js";
+} from "./index-gwxaz0z5.js";
 import {
   invalidateAutoEmbedSession,
   drainBackgroundBatches,
@@ -136,10 +136,10 @@ import {
   unregisterProjectEmbedding2,
   getProjectEmbeddingSnapshot2,
   embedTextForProject2
-} from "./index-v2epyzbm.js";
+} from "./index-w1twp2kp.js";
 import {
   getErrorMessage
-} from "./index-mtvvhcw4.js";
+} from "./index-dbgg6hsf.js";
 import {
   V2_MEMORY_CATEGORIES,
   CATEGORY_PRIORITY,
@@ -153,13 +153,13 @@ import {
   expandWorkspaceIdentitySetWithAliases,
   resolveStoredPathWorkspaceIdentity,
   storedPathBelongsToWorkspace
-} from "./index-5vgt1zeh.js";
+} from "./index-3c3ztwgc.js";
 import {
   directoryHasGitMetadata,
   resolveProjectIdentityForSession,
   normalizeStoredProjectPath,
   storedPathBelongsToIdentity
-} from "./index-p63nwn5v.js";
+} from "./index-hxp6q5f8.js";
 import {
   getNoteByIdInScope,
   getNotes,
@@ -175,7 +175,7 @@ import {
   resolveMemoriesByIdsForSearch2,
   unifiedSearch2,
   formatSearchResults2
-} from "./index-af9t637y.js";
+} from "./index-ftvm2acc.js";
 import {
   computeCueContentHash,
   getMuralCueState,
@@ -191,22 +191,22 @@ import {
   supersededMemory2,
   mergeMemoryStats2,
   archiveMemory2
-} from "./index-dzem6gn9.js";
+} from "./index-29nw7n1r.js";
 import {
   saveEmbedding,
   invalidateMemory2
 } from "./index-yhzhdp3m.js";
 import {
   preflightLocalEmbeddingPooling
-} from "./index-e19qtjj0.js";
+} from "./index-wqpjhx1p.js";
 import {
   isMarkerCandidateBasisCurrent2,
   invalidateMarkerCandidateBasis2,
   readMarkerCandidateBasis2
-} from "./index-6kb2xpw7.js";
+} from "./index-6wdwv3a6.js";
 import {
   renderSearchArchive2
-} from "./index-tb3d9trm.js";
+} from "./index-st4gwt64.js";
 import {
   computeNormalizedHash
 } from "./index-9x1f96r8.js";
@@ -1946,6 +1946,17 @@ function stripUnsafeProjectConfigFields(projectRaw) {
       warnings.push(`Ignoring ${field} from project config (security: daemon routing and developer-only embedding traffic are user-level settings).`);
     }
   }
+  if ("rerank" in projectRaw) {
+    const rerank = projectRaw.rerank;
+    const off = isPlainObject(rerank) && rerank.provider === "off";
+    const removed = !isPlainObject(rerank) || Object.keys(rerank).some((key) => key !== "provider") || !off;
+    if (off)
+      projectRaw.rerank = { provider: "off" };
+    else
+      delete projectRaw.rerank;
+    if (removed)
+      warnings.push("Ignoring rerank settings from project config (security: projects may only set rerank.provider to off; hosted routing, credentials and limits are user-level only).");
+  }
   const embedding = projectRaw.embedding;
   if (isPlainObject(embedding)) {
     const removed = [];
@@ -12966,6 +12977,7 @@ function createCtxSearchTool(deps) {
           };
         }
       }
+      let rerankSkipped = "";
       const results = await unifiedSearch2(deps.db, sessionId, projectIdentity, query, {
         limit: normalizeLimit2(params.limit),
         memoryEnabled,
@@ -12983,14 +12995,19 @@ function createCtxSearchTool(deps) {
         gitRepositoryAvailable: directoryHasGitMetadata(ctx.cwd),
         explicitSearch: true,
         rerankConnectionFile: deps.rerankConnectionFile,
+        rerankConfig: deps.rerankConfig,
         signal,
+        onRerankSkipped: (line) => {
+          rerankSkipped = line;
+        },
         ...dateRange
       });
       return {
         content: [
           {
             type: "text",
-            text: formatSearchResults2(query, results, sessionId, diagnostics)
+            text: formatSearchResults2(query, results, sessionId, diagnostics) + (rerankSkipped ? `
+${rerankSkipped}` : "")
           }
         ],
         details: undefined
@@ -13564,6 +13581,7 @@ function registerMagicContextTools(pi, opts) {
     embeddingEnabled: opts.embeddingEnabled,
     gitCommitsEnabled: opts.gitCommitsEnabled,
     rerankConnectionFile: opts.rerankConnectionFile,
+    rerankConfig: opts.rerankConfig,
     resolveProjectIdentity
   })));
   if (opts.memoryToolEnabled !== false) {

package/dist/{index-vdy273xw.js → index-wgmhk91m.js}

@@ -18,7 +18,7 @@ import {
   prepareCachedStatement2,
   withSqliteBackgroundWriter2,
   withoutSqliteTransformPass2
-} from "./index-hws5f0z9.js";
+} from "./index-2qtx0q6h.js";
 import {
   removeSystemReminders,
   removeNestedBlocks,
@@ -31,10 +31,10 @@ import {
   messageFtsOrdinalRangeIsMapped2,
   getMessageFtsRowidMapBackfillProgress2,
   deleteUnmappedMessageFtsRows2
-} from "./index-rgfb753z.js";
+} from "./index-579m1751.js";
 import {
   deleteChunkEmbedBackoffForSession2
-} from "./index-5n5zxs8e.js";
+} from "./index-vx6f7wff.js";
 import {
   closeQuietly,
   TAG_TOMBSTONE_OWNER_PREFIX,
@@ -46,7 +46,7 @@ import {
 } from "./index-w1avzkdt.js";
 import {
   cleanUserText2
-} from "./index-zhwwp8zz.js";
+} from "./index-gqjxy2p7.js";
 import {
   sameProjectDirectory
 } from "./index-wrr9rp9a.js";
@@ -791,6 +791,12 @@ class V2StoreReader {
       return typeof row?.count === "number" ? row.count : 0;
     });
   }
+  messageCountAfterSequence(sessionID, after, through) {
+    const rawTypes = RAW_MESSAGE_TYPES.map(() => "?").join(", ");
+    const row = this.prepare(`SELECT COUNT(*) AS count FROM session_message
+             WHERE session_id = ? AND seq > ? AND seq <= ? AND type IN (${rawTypes})`).get(sessionID, after, through, ...RAW_MESSAGE_TYPES);
+    return typeof row?.count === "number" ? row.count : 0;
+  }
   sessionDirectoryPage(afterSessionID, limit) {
     const rows = this.prepare(`SELECT id, COALESCE(directory, '') AS directory FROM session_v2
                  WHERE id > ? ORDER BY id ASC LIMIT ?`).all(afterSessionID ?? "", limit);

package/dist/{index-e19qtjj0.js → index-wqpjhx1p.js}

@@ -1,9 +1,9 @@
 import {
   DEFAULT_LOCAL_EMBEDDING_MODEL
-} from "./index-bxmf4dhr.js";
+} from "./index-ccanzg8p.js";
 import {
   getSynapseLaneIdentity
-} from "./index-jxjr7kz9.js";
+} from "./index-gwxaz0z5.js";
 import {
   getHarness2
 } from "./index-0h2mkab2.js";
@@ -14,7 +14,7 @@ import {
 } from "./index-fgs78jyd.js";
 import {
   inRootAsyncContext2
-} from "./index-hws5f0z9.js";
+} from "./index-2qtx0q6h.js";
 import {
   computeNormalizedHash
 } from "./index-9x1f96r8.js";
@@ -725,16 +725,20 @@ class EmbeddingWorkerClient {
   entry;
   timeoutMs;
   prepareTelemetry;
+  idleMs;
   worker = null;
   startPromise = null;
   nextId = 1;
   pending = new Map;
   disposed = false;
-  constructor(data, entry = new URL(new URL(import.meta.url).pathname.endsWith(".ts") ? "./embedding-worker.ts" : "./embedding-worker.js", import.meta.url), timeoutMs = 5 * 60000, prepareTelemetry = prepareNativeOnnxTelemetry) {
+  activeRequests = 0;
+  idleTimer = null;
+  constructor(data, entry = new URL(new URL(import.meta.url).pathname.endsWith(".ts") ? "./embedding-worker.ts" : "./embedding-worker.js", import.meta.url), timeoutMs = 5 * 60000, prepareTelemetry = prepareNativeOnnxTelemetry, idleMs = 10 * 60000) {
     this.data = data;
     this.entry = entry;
     this.timeoutMs = timeoutMs;
     this.prepareTelemetry = prepareTelemetry;
+    this.idleMs = idleMs;
   }
   start() {
     if (this.startPromise)
@@ -800,6 +804,7 @@ class EmbeddingWorkerClient {
     const fail = (error) => {
       if (this.worker !== worker)
         return;
+      this.clearIdleTimer();
       this.worker = null;
       for (const request of this.pending.values()) {
         clearTimeout(request.timer);
@@ -824,6 +829,31 @@ class EmbeddingWorkerClient {
   async request(texts, signal) {
     if (signal?.aborted)
       return Promise.reject(new Error("embedding request aborted"));
+    this.activeRequests++;
+    this.clearIdleTimer();
+    try {
+      return await this.requestOnWorker(texts, signal);
+    } finally {
+      this.activeRequests--;
+      this.armIdleTimer();
+    }
+  }
+  clearIdleTimer() {
+    if (this.idleTimer)
+      clearTimeout(this.idleTimer);
+    this.idleTimer = null;
+  }
+  armIdleTimer() {
+    if (this.disposed || this.activeRequests !== 0 || !this.worker)
+      return;
+    this.clearIdleTimer();
+    this.idleTimer = inRootAsyncContext2(() => setTimeout(() => {
+      this.idleTimer = null;
+      this.stop(new Error("embedding worker idle"));
+    }, this.idleMs));
+    this.idleTimer.unref();
+  }
+  async requestOnWorker(texts, signal) {
     const worker = await this.start();
     if (this.disposed)
       throw new Error("embedding worker disposed");
@@ -875,6 +905,7 @@ class EmbeddingWorkerClient {
     });
   }
   async stop(error) {
+    this.clearIdleTimer();
     const worker = this.worker;
     this.worker = null;
     for (const request of this.pending.values()) {