@orquentra/codebuddy 2.4.1 → 2.6.0

package/dist/{chunk-5OWZVZSO.js → chunk-SYWJG5NA.js}

@@ -1,157 +1,10 @@
+import {
+  countTokens,
+  generateEntityId
+} from "./chunk-XBJQKCLU.js";
 import {
   writeAtomic
-} from "./chunk-K63EOA4B.js";
-
-// src/planner/budget.ts
-import { Tiktoken } from "js-tiktoken/lite";
-import cl100kBase from "js-tiktoken/ranks/cl100k_base";
-
-// src/providers/models.ts
-var EMBEDDING_DIMENSIONS = 384;
-var DEFAULT_EMBEDDING_MODELS = [
-  {
-    id: "sentence-transformers/all-MiniLM-L6-v2",
-    role: "primary",
-    kind: "embedding",
-    dimensions: EMBEDDING_DIMENSIONS,
-    dailyCallBudget: 1e3,
-    supportsBatch: true
-  },
-  {
-    id: "BAAI/bge-small-en-v1.5",
-    role: "fallback",
-    kind: "embedding",
-    dimensions: EMBEDDING_DIMENSIONS,
-    dailyCallBudget: 1e3,
-    supportsBatch: true
-  }
-];
-var DEFAULT_LLM_MODELS = [
-  {
-    id: "meta-llama/Llama-3.2-3B-Instruct",
-    role: "primary",
-    kind: "llm",
-    contextWindow: 131072,
-    dailyCallBudget: 300
-  },
-  {
-    id: "Qwen/Qwen2.5-7B-Instruct",
-    role: "fallback",
-    kind: "llm",
-    contextWindow: 32768,
-    dailyCallBudget: 300
-  },
-  {
-    id: "mistralai/Mistral-7B-Instruct-v0.3",
-    role: "last_resort",
-    kind: "llm",
-    contextWindow: 32768,
-    dailyCallBudget: 300
-  }
-];
-var MODEL_REGISTRY = [...DEFAULT_EMBEDDING_MODELS, ...DEFAULT_LLM_MODELS];
-function getModelDefinition(modelId) {
-  return MODEL_REGISTRY.find((model) => model.id === modelId);
-}
-function getModelContextWindow(modelId) {
-  return getModelDefinition(modelId)?.contextWindow;
-}
-function assertCompatibleEmbeddingModels(models) {
-  const mismatchedModel = models.find((model) => model.dimensions !== EMBEDDING_DIMENSIONS);
-  if (mismatchedModel) {
-    throw new Error(
-      `Embedding model ${mismatchedModel.id} has ${mismatchedModel.dimensions ?? "unknown"} dimensions; expected ${EMBEDDING_DIMENSIONS}.`
-    );
-  }
-}
-
-// src/planner/budget.ts
-var encoder = null;
-function getEncoder() {
-  if (!encoder) {
-    encoder = new Tiktoken(cl100kBase);
-  }
-  return encoder;
-}
-var PIECE = new RegExp(cl100kBase.pat_str, "gu");
-var SPECIAL_TOKENS = Object.keys(cl100kBase.special_tokens);
-var MAX_CACHED_PIECES = 2e5;
-var pieceTokens = /* @__PURE__ */ new Map();
-function countTokens(text) {
-  if (!text) return 0;
-  try {
-    if (SPECIAL_TOKENS.some((token) => text.includes(token))) throw new Error("special token");
-    let total = 0;
-    for (const [piece] of text.matchAll(PIECE)) {
-      let count = pieceTokens.get(piece);
-      if (count === void 0) {
-        count = getEncoder().encode(piece).length;
-        if (pieceTokens.size >= MAX_CACHED_PIECES) pieceTokens.clear();
-        pieceTokens.set(piece, count);
-      }
-      total += count;
-    }
-    return total;
-  } catch {
-    return Math.max(1, Math.ceil(text.length / 4));
-  }
-}
-function clampBudgetForCallerModel(requestedBudget, callerModel) {
-  if (!callerModel) {
-    return { budget: requestedBudget, clamped: false };
-  }
-  const ctx = getModelContextWindow(callerModel);
-  if (!ctx) {
-    return { budget: requestedBudget, clamped: false };
-  }
-  const headroom = Math.floor(ctx * 0.75);
-  if (requestedBudget > headroom) {
-    return {
-      budget: headroom,
-      clamped: true,
-      reason: `budget ${requestedBudget} exceeds 75% of ${callerModel} context window (${ctx}); clamped to ${headroom}.`
-    };
-  }
-  return { budget: requestedBudget, clamped: false };
-}
-
-// src/core/ids.ts
-import { createHash, randomBytes } from "crypto";
-var CROCKFORD = "0123456789ABCDEFGHJKMNPQRSTVWXYZ";
-function generateUlid(now = Date.now()) {
-  let timePart = "";
-  let ts = now;
-  for (let i = 0; i < 10; i++) {
-    timePart = (CROCKFORD[ts % 32] ?? "0") + timePart;
-    ts = Math.floor(ts / 32);
-  }
-  const bytes = randomBytes(10);
-  let randPart = "";
-  for (let i = 0; i < 16; i++) {
-    const bitOffset = i * 5;
-    const byteIndex = Math.floor(bitOffset / 8);
-    const bitInByte = bitOffset % 8;
-    const hi = bytes[byteIndex] ?? 0;
-    const lo = bytes[byteIndex + 1] ?? 0;
-    const combined = (hi << 8 | lo) >>> 0;
-    const shift = 16 - bitInByte - 5;
-    const value = combined >>> shift & 31;
-    randPart += CROCKFORD[value] ?? "0";
-  }
-  return timePart + randPart;
-}
-function generateSessionId(now) {
-  return `sess_${generateUlid(now).toLowerCase()}`;
-}
-function generateEntityId(prefix, now) {
-  return `${prefix}_${generateUlid(now).toLowerCase()}`;
-}
-function computeContentHash(namespace, sessionId, content) {
-  return createHash("sha256").update(`${namespace}${sessionId}${content}`).digest("hex");
-}
-function computeFactHash(namespace, content) {
-  return createHash("sha256").update(`${namespace}fact${content}`).digest("hex");
-}
+} from "./chunk-776EQXDV.js";
 
 // src/savings/tokens.ts
 function estimateTokens(value) {
@@ -162,7 +15,7 @@ function estimateTokens(value) {
 }
 
 // src/context/session-ledger.ts
-import { createHash as createHash2 } from "crypto";
+import { createHash } from "crypto";
 import { mkdir, readdir, readFile, rm, stat } from "fs/promises";
 import { join, resolve } from "path";
 var VERSION = 2;
@@ -406,7 +259,7 @@ function empty(now) {
   return { version: VERSION, updatedAt: now.toISOString(), records: {} };
 }
 function digest(text) {
-  return createHash2("sha256").update(text).digest("hex");
+  return createHash("sha256").update(text).digest("hex");
 }
 
 // src/context/tests.ts
@@ -458,19 +311,9 @@ async function detectTestCommand(repositoryRoot) {
 }
 
 export {
-  EMBEDDING_DIMENSIONS,
-  DEFAULT_EMBEDDING_MODELS,
-  DEFAULT_LLM_MODELS,
-  assertCompatibleEmbeddingModels,
-  countTokens,
-  clampBudgetForCallerModel,
-  generateSessionId,
-  generateEntityId,
-  computeContentHash,
-  computeFactHash,
   estimateTokens,
   SessionCapsuleLedger,
   findCoveringTest,
   detectTestCommand
 };
-//# sourceMappingURL=chunk-5OWZVZSO.js.map
+//# sourceMappingURL=chunk-SYWJG5NA.js.map

package/dist/chunk-SYWJG5NA.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/savings/tokens.ts","../src/context/session-ledger.ts","../src/context/tests.ts"],"sourcesContent":["import { countTokens } from \"../planner/budget.js\";\n\n/**\n * The one place token counts are computed. Strings use the same deterministic\n * cl100k tokenizer as planner budgets; objects are measured by their JSON\n * serialization, which matches how they travel over MCP.\n */\n\nexport function estimateTokens(value: unknown): number {\n  if (value === undefined || value === null) return 0;\n  const text = typeof value === \"string\" ? value : JSON.stringify(value);\n  if (!text) return 0;\n  return countTokens(text);\n}\n","import { createHash } from \"node:crypto\";\nimport { mkdir, readdir, readFile, rm, stat } from \"node:fs/promises\";\nimport { join, resolve } from \"node:path\";\nimport { generateEntityId } from \"../core/ids.js\";\nimport { writeAtomic } from \"../core/markdown-store-fs.js\";\nimport { estimateTokens } from \"../savings/tokens.js\";\n\nconst VERSION = 2;\nconst MAX_CAPSULES = 128;\nconst MAX_AGE_MS = 7 * 24 * 60 * 60 * 1000;\nconst MAX_PENDING = 8;\nconst PENDING_MAX_AGE_MS = 24 * 60 * 60 * 1000;\nconst STALE_LOCK_MS = 60_000;\n/** Acknowledged versions kept per section, so switching back to a task reuses them. */\nconst MAX_ACKNOWLEDGED_PER_KEY = 8;\n\ntype LedgerRecord = { hash: string; lastSentAt: string };\n/** A delivery sent to a client that has not yet confirmed receipt. */\ntype PendingDelivery = { createdAt: string; records: Record<string, string> };\n/** One acknowledged version of an `exchange` section: content hash and when it was sent. */\ntype AcknowledgedVersion = { hash: string; sentAt: string };\ntype LedgerFile = {\n  version: 2;\n  updatedAt: string;\n  /** `deliver`/`markSent` capsules (hooks): one current hash per key. */\n  records: Record<string, LedgerRecord>;\n  pending?: Record<string, PendingDelivery>;\n  /** `exchange` sections (MCP): every recently acknowledged version, newest first. */\n  acknowledged?: Record<string, AcknowledgedVersion[]>;\n};\n\nexport type ContextCapsule = { key: string; text: string };\nexport type CapsuleDelivery = {\n  /** Full text of new or changed capsules, in input order. */\n  lines: string[];\n  /** Keys of capsules already sent this session and not resent. */\n  unchanged: string[];\n  full: number;\n  referenced: number;\n  sentTokens: number;\n  avoidedTokens: number;\n};\n\nexport type SectionHash = { key: string; hash: string };\nexport type AcknowledgedExchange = {\n  deliveryId: string;\n  /** Keys whose content the client confirmed receiving and that have not changed. */\n  unchanged: string[];\n  ack: \"none\" | \"accepted\" | \"unknown\";\n};\n\nexport class SessionCapsuleLedger {\n  private readonly directory: string;\n\n  constructor(repositoryRoot = process.cwd()) {\n    this.directory = join(resolve(repositoryRoot), \".codebuddy\", \"cache\", \"session\");\n  }\n\n  async deliver(\n    sessionId: string,\n    capsules: ContextCapsule[],\n    now = new Date(),\n  ): Promise<CapsuleDelivery> {\n    await mkdir(this.directory, { recursive: true, mode: 0o700 });\n    const release = await this.acquire(sessionId);\n    try {\n      const path = this.path(sessionId);\n      const ledger = await this.read(path, now);\n      const lines: string[] = [];\n      const unchanged: string[] = [];\n      let full = 0;\n      let referenced = 0;\n      let sentTokens = 0;\n      let avoidedTokens = 0;\n\n      for (const capsule of capsules) {\n        const hash = digest(capsule.text);\n        const existing = ledger.records[capsule.key];\n        if (existing?.hash === hash) {\n          unchanged.push(capsule.key);\n          referenced++;\n          avoidedTokens += estimateTokens(capsule.text);\n        } else {\n          lines.push(capsule.text);\n          full++;\n          sentTokens += estimateTokens(capsule.text);\n        }\n        ledger.records[capsule.key] = { hash, lastSentAt: now.toISOString() };\n      }\n\n      await this.write(path, ledger, now);\n      return { lines, unchanged, full, referenced, sentTokens, avoidedTokens };\n    } finally {\n      await release();\n    }\n  }\n\n  /** Keys of capsules whose current text was already recorded as sent this session. */\n  async seen(\n    sessionId: string,\n    capsules: ContextCapsule[],\n    now = new Date(),\n  ): Promise<Set<string>> {\n    const ledger = await this.read(this.path(sessionId), now);\n    return new Set(\n      capsules\n        .filter((capsule) => ledger.records[capsule.key]?.hash === digest(capsule.text))\n        .map((capsule) => capsule.key),\n    );\n  }\n\n  /**\n   * Record capsules as delivered. Call only after their text actually reached\n   * the client, so capsules that were clipped or failed to write stay unseen\n   * and are delivered by a later prompt.\n   */\n  async markSent(sessionId: string, capsules: ContextCapsule[], now = new Date()): Promise<void> {\n    if (capsules.length === 0) return;\n    await mkdir(this.directory, { recursive: true, mode: 0o700 });\n    const release = await this.acquire(sessionId);\n    try {\n      const path = this.path(sessionId);\n      const ledger = await this.read(path, now);\n      for (const capsule of capsules) {\n        ledger.records[capsule.key] = { hash: digest(capsule.text), lastSentAt: now.toISOString() };\n      }\n      await this.write(path, ledger, now);\n    } finally {\n      await release();\n    }\n  }\n\n  /**\n   * Acknowledged delivery for clients that can lose context (MCP). Unlike\n   * `deliver`, sending marks nothing as seen: the sent hashes are held as a\n   * pending delivery until the client echoes its `deliveryId` back as `ack` on\n   * a later call. Only acknowledged, unchanged sections sent within `ttlMs` are\n   * reported unchanged, so a dropped response or a compacted client context\n   * fails safe to a full resend.\n   *\n   * A section keeps its last few acknowledged versions (by content hash), not\n   * just the latest, so returning to an earlier task in the same session does\n   * not resend what the client already holds. `full` means the client lost its\n   * context: every acknowledged version and unconfirmed delivery is forgotten.\n   */\n  async exchange(\n    sessionId: string,\n    sections: SectionHash[],\n    options: {\n      ack?: string;\n      ttlMs: number;\n      /** The client lost its context: nothing acknowledged before this call counts. */\n      full?: boolean;\n      now?: Date;\n      /** Finalize the response before recording hashes of the actual sent sections. */\n      prepare?: (exchange: AcknowledgedExchange) => SectionHash[];\n    },\n  ): Promise<AcknowledgedExchange> {\n    const now = options.now ?? new Date();\n    await mkdir(this.directory, { recursive: true, mode: 0o700 });\n    const release = await this.acquire(sessionId);\n    try {\n      const path = this.path(sessionId);\n      const ledger = await this.read(path, now);\n      let pending = ledger.pending ?? {};\n      let acknowledged = ledger.acknowledged ?? {};\n\n      let ack: AcknowledgedExchange[\"ack\"] = \"none\";\n      if (options.ack !== undefined) {\n        const confirmed = pending[options.ack];\n        if (confirmed) {\n          for (const [key, hash] of Object.entries(confirmed.records)) {\n            acknowledged[key] = [\n              { hash, sentAt: confirmed.createdAt },\n              ...(acknowledged[key] ?? []).filter((version) => version.hash !== hash),\n            ]\n              .sort((left, right) => right.sentAt.localeCompare(left.sentAt))\n              .slice(0, MAX_ACKNOWLEDGED_PER_KEY);\n          }\n          delete pending[options.ack];\n          ack = \"accepted\";\n        } else {\n          ack = \"unknown\";\n        }\n      }\n      if (options.full) {\n        acknowledged = {};\n        pending = {};\n      }\n\n      const unchanged: string[] = [];\n      const sent: Record<string, string> = {};\n      for (const section of sections) {\n        const held = acknowledged[section.key]?.some((version) => {\n          const age = now.getTime() - Date.parse(version.sentAt);\n          return version.hash === section.hash && age >= 0 && age <= options.ttlMs;\n        });\n        if (held) {\n          unchanged.push(section.key);\n        } else {\n          sent[section.key] = section.hash;\n        }\n      }\n\n      const deliveryId = generateEntityId(\"dlv\", now.getTime());\n      const exchange = { deliveryId, unchanged, ack };\n      const delivered = options.prepare?.(exchange);\n      pending[deliveryId] = {\n        createdAt: now.toISOString(),\n        records: delivered\n          ? Object.fromEntries(delivered.map(({ key, hash }) => [key, hash]))\n          : sent,\n      };\n      ledger.pending = Object.fromEntries(\n        Object.entries(pending)\n          .filter(([, item]) => now.getTime() - Date.parse(item.createdAt) <= PENDING_MAX_AGE_MS)\n          .sort((left, right) => right[1].createdAt.localeCompare(left[1].createdAt))\n          .slice(0, MAX_PENDING),\n      );\n      ledger.acknowledged = Object.fromEntries(\n        Object.entries(acknowledged)\n          .map(([key, versions]): [string, AcknowledgedVersion[]] => [\n            key,\n            versions.filter((version) => now.getTime() - Date.parse(version.sentAt) <= MAX_AGE_MS),\n          ])\n          .filter(([, versions]) => versions.length > 0)\n          .slice(0, MAX_CAPSULES),\n      );\n      await this.write(path, ledger, now);\n      return { deliveryId, unchanged, ack };\n    } finally {\n      await release();\n    }\n  }\n\n  /** Forget everything sent in a session, e.g. after the client compacted its context. */\n  async reset(sessionId: string): Promise<void> {\n    await mkdir(this.directory, { recursive: true, mode: 0o700 });\n    const release = await this.acquire(sessionId);\n    try {\n      await rm(this.path(sessionId), { force: true });\n    } finally {\n      await release();\n    }\n  }\n\n  /** Remove expired disposable state; locked sessions are left for the next pass. */\n  async pruneExpired(now = new Date()): Promise<number> {\n    let removed = 0;\n    for (const entry of await readdir(this.directory).catch(() => [])) {\n      if (!entry.endsWith(\".json\")) continue;\n      const path = join(this.directory, entry);\n      try {\n        await stat(`${path}.lock`);\n        continue;\n      } catch {\n        // No active lock.\n      }\n      try {\n        const metadata = await stat(path);\n        if (now.getTime() - metadata.mtimeMs > MAX_AGE_MS) {\n          await rm(path, { force: true });\n          removed++;\n        }\n      } catch {\n        // Another cleanup may have removed it already.\n      }\n    }\n    return removed;\n  }\n\n  private async write(path: string, ledger: LedgerFile, now: Date): Promise<void> {\n    ledger.updatedAt = now.toISOString();\n    ledger.records = Object.fromEntries(\n      Object.entries(ledger.records)\n        .sort((left, right) => right[1].lastSentAt.localeCompare(left[1].lastSentAt))\n        .slice(0, MAX_CAPSULES),\n    );\n    await writeAtomic(path, `${JSON.stringify(ledger, null, 2)}\\n`);\n  }\n\n  private async read(path: string, now: Date): Promise<LedgerFile> {\n    try {\n      const parsed = JSON.parse(await readFile(path, \"utf8\")) as LedgerFile;\n      if (parsed.version !== VERSION || typeof parsed.records !== \"object\") return empty(now);\n      const records = Object.fromEntries(\n        Object.entries(parsed.records).filter(([, record]) => {\n          const age = now.getTime() - Date.parse(record.lastSentAt);\n          return Number.isFinite(age) && age <= MAX_AGE_MS;\n        }),\n      );\n      const pending =\n        parsed.pending && typeof parsed.pending === \"object\" ? parsed.pending : undefined;\n      const acknowledged =\n        parsed.acknowledged && typeof parsed.acknowledged === \"object\"\n          ? Object.fromEntries(\n              Object.entries(parsed.acknowledged).filter(([, versions]) => Array.isArray(versions)),\n            )\n          : undefined;\n      return {\n        version: VERSION,\n        updatedAt: parsed.updatedAt,\n        records,\n        ...(pending ? { pending } : {}),\n        ...(acknowledged ? { acknowledged } : {}),\n      };\n    } catch {\n      return empty(now);\n    }\n  }\n\n  private path(sessionId: string): string {\n    // The full identity participates in the key. Legacy sanitized filenames\n    // are intentionally ignored, so a fresh session cannot inherit collisions.\n    return join(this.directory, `${digest(sessionId)}.json`);\n  }\n\n  private async acquire(sessionId: string): Promise<() => Promise<void>> {\n    const lock = `${this.path(sessionId)}.lock`;\n    for (let attempt = 0; ; attempt++) {\n      try {\n        await mkdir(lock);\n        return async () => rm(lock, { recursive: true, force: true });\n      } catch (error) {\n        if ((error as NodeJS.ErrnoException).code !== \"EEXIST\" || attempt >= 200) throw error;\n        try {\n          const age = Date.now() - (await stat(lock)).mtimeMs;\n          if (age > STALE_LOCK_MS) {\n            await rm(lock, { recursive: true, force: true });\n            continue;\n          }\n        } catch {\n          // The owner may have released the lock between checks.\n        }\n        await new Promise((resolveWait) => setTimeout(resolveWait, 10));\n      }\n    }\n  }\n}\n\nfunction empty(now: Date): LedgerFile {\n  return { version: VERSION, updatedAt: now.toISOString(), records: {} };\n}\n\nfunction digest(text: string): string {\n  return createHash(\"sha256\").update(text).digest(\"hex\");\n}\n","import { access, readFile } from \"node:fs/promises\";\nimport { basename, dirname, extname, join } from \"node:path\";\n\n/**\n * Which test covers a file and how this repo runs its tests, from naming\n * conventions and project markers only. Agents otherwise spend turns finding\n * both; a wrong guess costs one failed command, so first match wins.\n */\nexport function findCoveringTest(path: string, paths: ReadonlySet<string>): string | null {\n  const ext = extname(path);\n  const stem = basename(path, ext);\n  const dir = dirname(path);\n  const join2 = (...parts: string[]) => parts.filter((part) => part && part !== \".\").join(\"/\");\n  // `src/a/x.ts` mirrors to `test/a/x.test.ts`: drop a leading source root.\n  const mirrored = dir.replace(/^(src|lib|app)(\\/|$)/, \"\");\n  const candidates =\n    ext === \".py\"\n      ? [\n          join2(dir, `test_${stem}.py`),\n          join2(\"tests\", `test_${stem}.py`),\n          join2(dir, \"tests\", `test_${stem}.py`),\n          join2(dir, `${stem}_test.py`),\n        ]\n      : ext === \".go\"\n        ? [join2(dir, `${stem}_test.go`)]\n        : [\n            join2(dir, `${stem}.test${ext}`),\n            join2(dir, `${stem}.spec${ext}`),\n            join2(dir, \"__tests__\", `${stem}.test${ext}`),\n            join2(dir, \"__tests__\", `${stem}${ext}`),\n            join2(\"test\", mirrored, `${stem}.test${ext}`),\n            join2(\"tests\", mirrored, `${stem}.test${ext}`),\n          ];\n  return candidates.find((candidate) => paths.has(candidate)) ?? null;\n}\n\nconst NPM_PLACEHOLDER = /no test specified/;\n\nexport async function detectTestCommand(repositoryRoot: string): Promise<string | null> {\n  const exists = (file: string) =>\n    access(join(repositoryRoot, file)).then(\n      () => true,\n      () => false,\n    );\n  try {\n    const pkg = JSON.parse(await readFile(join(repositoryRoot, \"package.json\"), \"utf8\")) as {\n      scripts?: Record<string, unknown>;\n    };\n    const script = pkg.scripts?.test;\n    if (typeof script === \"string\" && script.trim() && !NPM_PLACEHOLDER.test(script)) {\n      if (await exists(\"pnpm-lock.yaml\")) return \"pnpm test\";\n      if (await exists(\"yarn.lock\")) return \"yarn test\";\n      return \"npm test\";\n    }\n  } catch {\n    // No or unreadable package.json: try other ecosystems.\n  }\n  for (const marker of [\"pyproject.toml\", \"pytest.ini\", \"setup.cfg\"]) {\n    if (await exists(marker)) return \"pytest\";\n  }\n  if (await exists(\"go.mod\")) return \"go test ./...\";\n  if (await exists(\"Cargo.toml\")) return \"cargo test\";\n  return null;\n}\n"],"mappings":";;;;;;;;;AAQO,SAAS,eAAe,OAAwB;AACrD,MAAI,UAAU,UAAa,UAAU,KAAM,QAAO;AAClD,QAAM,OAAO,OAAO,UAAU,WAAW,QAAQ,KAAK,UAAU,KAAK;AACrE,MAAI,CAAC,KAAM,QAAO;AAClB,SAAO,YAAY,IAAI;AACzB;;;ACbA,SAAS,kBAAkB;AAC3B,SAAS,OAAO,SAAS,UAAU,IAAI,YAAY;AACnD,SAAS,MAAM,eAAe;AAK9B,IAAM,UAAU;AAChB,IAAM,eAAe;AACrB,IAAM,aAAa,IAAI,KAAK,KAAK,KAAK;AACtC,IAAM,cAAc;AACpB,IAAM,qBAAqB,KAAK,KAAK,KAAK;AAC1C,IAAM,gBAAgB;AAEtB,IAAM,2BAA2B;AAqC1B,IAAM,uBAAN,MAA2B;AAAA,EACf;AAAA,EAEjB,YAAY,iBAAiB,QAAQ,IAAI,GAAG;AAC1C,SAAK,YAAY,KAAK,QAAQ,cAAc,GAAG,cAAc,SAAS,SAAS;AAAA,EACjF;AAAA,EAEA,MAAM,QACJ,WACA,UACA,MAAM,oBAAI,KAAK,GACW;AAC1B,UAAM,MAAM,KAAK,WAAW,EAAE,WAAW,MAAM,MAAM,IAAM,CAAC;AAC5D,UAAM,UAAU,MAAM,KAAK,QAAQ,SAAS;AAC5C,QAAI;AACF,YAAM,OAAO,KAAK,KAAK,SAAS;AAChC,YAAM,SAAS,MAAM,KAAK,KAAK,MAAM,GAAG;AACxC,YAAM,QAAkB,CAAC;AACzB,YAAM,YAAsB,CAAC;AAC7B,UAAI,OAAO;AACX,UAAI,aAAa;AACjB,UAAI,aAAa;AACjB,UAAI,gBAAgB;AAEpB,iBAAW,WAAW,UAAU;AAC9B,cAAM,OAAO,OAAO,QAAQ,IAAI;AAChC,cAAM,WAAW,OAAO,QAAQ,QAAQ,GAAG;AAC3C,YAAI,UAAU,SAAS,MAAM;AAC3B,oBAAU,KAAK,QAAQ,GAAG;AAC1B;AACA,2BAAiB,eAAe,QAAQ,IAAI;AAAA,QAC9C,OAAO;AACL,gBAAM,KAAK,QAAQ,IAAI;AACvB;AACA,wBAAc,eAAe,QAAQ,IAAI;AAAA,QAC3C;AACA,eAAO,QAAQ,QAAQ,GAAG,IAAI,EAAE,MAAM,YAAY,IAAI,YAAY,EAAE;AAAA,MACtE;AAEA,YAAM,KAAK,MAAM,MAAM,QAAQ,GAAG;AAClC,aAAO,EAAE,OAAO,WAAW,MAAM,YAAY,YAAY,cAAc;AAAA,IACzE,UAAE;AACA,YAAM,QAAQ;AAAA,IAChB;AAAA,EACF;AAAA;AAAA,EAGA,MAAM,KACJ,WACA,UACA,MAAM,oBAAI,KAAK,GACO;AACtB,UAAM,SAAS,MAAM,KAAK,KAAK,KAAK,KAAK,SAAS,GAAG,GAAG;AACxD,WAAO,IAAI;AAAA,MACT,SACG,OAAO,CAAC,YAAY,OAAO,QAAQ,QAAQ,GAAG,GAAG,SAAS,OAAO,QAAQ,IAAI,CAAC,EAC9E,IAAI,CAAC,YAAY,QAAQ,GAAG;AAAA,IACjC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,SAAS,WAAmB,UAA4B,MAAM,oBAAI,KAAK,GAAkB;AAC7F,QAAI,SAAS,WAAW,EAAG;AAC3B,UAAM,MAAM,KAAK,WAAW,EAAE,WAAW,MAAM,MAAM,IAAM,CAAC;AAC5D,UAAM,UAAU,MAAM,KAAK,QAAQ,SAAS;AAC5C,QAAI;AACF,YAAM,OAAO,KAAK,KAAK,SAAS;AAChC,YAAM,SAAS,MAAM,KAAK,KAAK,MAAM,GAAG;AACxC,iBAAW,WAAW,UAAU;AAC9B,eAAO,QAAQ,QAAQ,GAAG,IAAI,EAAE,MAAM,OAAO,QAAQ,IAAI,GAAG,YAAY,IAAI,YAAY,EAAE;AAAA,MAC5F;AACA,YAAM,KAAK,MAAM,MAAM,QAAQ,GAAG;AAAA,IACpC,UAAE;AACA,YAAM,QAAQ;AAAA,IAChB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAeA,MAAM,SACJ,WACA,UACA,SAS+B;AAC/B,UAAM,MAAM,QAAQ,OAAO,oBAAI,KAAK;AACpC,UAAM,MAAM,KAAK,WAAW,EAAE,WAAW,MAAM,MAAM,IAAM,CAAC;AAC5D,UAAM,UAAU,MAAM,KAAK,QAAQ,SAAS;AAC5C,QAAI;AACF,YAAM,OAAO,KAAK,KAAK,SAAS;AAChC,YAAM,SAAS,MAAM,KAAK,KAAK,MAAM,GAAG;AACxC,UAAI,UAAU,OAAO,WAAW,CAAC;AACjC,UAAI,eAAe,OAAO,gBAAgB,CAAC;AAE3C,UAAI,MAAmC;AACvC,UAAI,QAAQ,QAAQ,QAAW;AAC7B,cAAM,YAAY,QAAQ,QAAQ,GAAG;AACrC,YAAI,WAAW;AACb,qBAAW,CAAC,KAAK,IAAI,KAAK,OAAO,QAAQ,UAAU,OAAO,GAAG;AAC3D,yBAAa,GAAG,IAAI;AAAA,cAClB,EAAE,MAAM,QAAQ,UAAU,UAAU;AAAA,cACpC,IAAI,aAAa,GAAG,KAAK,CAAC,GAAG,OAAO,CAAC,YAAY,QAAQ,SAAS,IAAI;AAAA,YACxE,EACG,KAAK,CAAC,MAAM,UAAU,MAAM,OAAO,cAAc,KAAK,MAAM,CAAC,EAC7D,MAAM,GAAG,wBAAwB;AAAA,UACtC;AACA,iBAAO,QAAQ,QAAQ,GAAG;AAC1B,gBAAM;AAAA,QACR,OAAO;AACL,gBAAM;AAAA,QACR;AAAA,MACF;AACA,UAAI,QAAQ,MAAM;AAChB,uBAAe,CAAC;AAChB,kBAAU,CAAC;AAAA,MACb;AAEA,YAAM,YAAsB,CAAC;AAC7B,YAAM,OAA+B,CAAC;AACtC,iBAAW,WAAW,UAAU;AAC9B,cAAM,OAAO,aAAa,QAAQ,GAAG,GAAG,KAAK,CAAC,YAAY;AACxD,gBAAM,MAAM,IAAI,QAAQ,IAAI,KAAK,MAAM,QAAQ,MAAM;AACrD,iBAAO,QAAQ,SAAS,QAAQ,QAAQ,OAAO,KAAK,OAAO,QAAQ;AAAA,QACrE,CAAC;AACD,YAAI,MAAM;AACR,oBAAU,KAAK,QAAQ,GAAG;AAAA,QAC5B,OAAO;AACL,eAAK,QAAQ,GAAG,IAAI,QAAQ;AAAA,QAC9B;AAAA,MACF;AAEA,YAAM,aAAa,iBAAiB,OAAO,IAAI,QAAQ,CAAC;AACxD,YAAM,WAAW,EAAE,YAAY,WAAW,IAAI;AAC9C,YAAM,YAAY,QAAQ,UAAU,QAAQ;AAC5C,cAAQ,UAAU,IAAI;AAAA,QACpB,WAAW,IAAI,YAAY;AAAA,QAC3B,SAAS,YACL,OAAO,YAAY,UAAU,IAAI,CAAC,EAAE,KAAK,KAAK,MAAM,CAAC,KAAK,IAAI,CAAC,CAAC,IAChE;AAAA,MACN;AACA,aAAO,UAAU,OAAO;AAAA,QACtB,OAAO,QAAQ,OAAO,EACnB,OAAO,CAAC,CAAC,EAAE,IAAI,MAAM,IAAI,QAAQ,IAAI,KAAK,MAAM,KAAK,SAAS,KAAK,kBAAkB,EACrF,KAAK,CAAC,MAAM,UAAU,MAAM,CAAC,EAAE,UAAU,cAAc,KAAK,CAAC,EAAE,SAAS,CAAC,EACzE,MAAM,GAAG,WAAW;AAAA,MACzB;AACA,aAAO,eAAe,OAAO;AAAA,QAC3B,OAAO,QAAQ,YAAY,EACxB,IAAI,CAAC,CAAC,KAAK,QAAQ,MAAuC;AAAA,UACzD;AAAA,UACA,SAAS,OAAO,CAAC,YAAY,IAAI,QAAQ,IAAI,KAAK,MAAM,QAAQ,MAAM,KAAK,UAAU;AAAA,QACvF,CAAC,EACA,OAAO,CAAC,CAAC,EAAE,QAAQ,MAAM,SAAS,SAAS,CAAC,EAC5C,MAAM,GAAG,YAAY;AAAA,MAC1B;AACA,YAAM,KAAK,MAAM,MAAM,QAAQ,GAAG;AAClC,aAAO,EAAE,YAAY,WAAW,IAAI;AAAA,IACtC,UAAE;AACA,YAAM,QAAQ;AAAA,IAChB;AAAA,EACF;AAAA;AAAA,EAGA,MAAM,MAAM,WAAkC;AAC5C,UAAM,MAAM,KAAK,WAAW,EAAE,WAAW,MAAM,MAAM,IAAM,CAAC;AAC5D,UAAM,UAAU,MAAM,KAAK,QAAQ,SAAS;AAC5C,QAAI;AACF,YAAM,GAAG,KAAK,KAAK,SAAS,GAAG,EAAE,OAAO,KAAK,CAAC;AAAA,IAChD,UAAE;AACA,YAAM,QAAQ;AAAA,IAChB;AAAA,EACF;AAAA;AAAA,EAGA,MAAM,aAAa,MAAM,oBAAI,KAAK,GAAoB;AACpD,QAAI,UAAU;AACd,eAAW,SAAS,MAAM,QAAQ,KAAK,SAAS,EAAE,MAAM,MAAM,CAAC,CAAC,GAAG;AACjE,UAAI,CAAC,MAAM,SAAS,OAAO,EAAG;AAC9B,YAAM,OAAO,KAAK,KAAK,WAAW,KAAK;AACvC,UAAI;AACF,cAAM,KAAK,GAAG,IAAI,OAAO;AACzB;AAAA,MACF,QAAQ;AAAA,MAER;AACA,UAAI;AACF,cAAM,WAAW,MAAM,KAAK,IAAI;AAChC,YAAI,IAAI,QAAQ,IAAI,SAAS,UAAU,YAAY;AACjD,gBAAM,GAAG,MAAM,EAAE,OAAO,KAAK,CAAC;AAC9B;AAAA,QACF;AAAA,MACF,QAAQ;AAAA,MAER;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA,EAEA,MAAc,MAAM,MAAc,QAAoB,KAA0B;AAC9E,WAAO,YAAY,IAAI,YAAY;AACnC,WAAO,UAAU,OAAO;AAAA,MACtB,OAAO,QAAQ,OAAO,OAAO,EAC1B,KAAK,CAAC,MAAM,UAAU,MAAM,CAAC,EAAE,WAAW,cAAc,KAAK,CAAC,EAAE,UAAU,CAAC,EAC3E,MAAM,GAAG,YAAY;AAAA,IAC1B;AACA,UAAM,YAAY,MAAM,GAAG,KAAK,UAAU,QAAQ,MAAM,CAAC,CAAC;AAAA,CAAI;AAAA,EAChE;AAAA,EAEA,MAAc,KAAK,MAAc,KAAgC;AAC/D,QAAI;AACF,YAAM,SAAS,KAAK,MAAM,MAAM,SAAS,MAAM,MAAM,CAAC;AACtD,UAAI,OAAO,YAAY,WAAW,OAAO,OAAO,YAAY,SAAU,QAAO,MAAM,GAAG;AACtF,YAAM,UAAU,OAAO;AAAA,QACrB,OAAO,QAAQ,OAAO,OAAO,EAAE,OAAO,CAAC,CAAC,EAAE,MAAM,MAAM;AACpD,gBAAM,MAAM,IAAI,QAAQ,IAAI,KAAK,MAAM,OAAO,UAAU;AACxD,iBAAO,OAAO,SAAS,GAAG,KAAK,OAAO;AAAA,QACxC,CAAC;AAAA,MACH;AACA,YAAM,UACJ,OAAO,WAAW,OAAO,OAAO,YAAY,WAAW,OAAO,UAAU;AAC1E,YAAM,eACJ,OAAO,gBAAgB,OAAO,OAAO,iBAAiB,WAClD,OAAO;AAAA,QACL,OAAO,QAAQ,OAAO,YAAY,EAAE,OAAO,CAAC,CAAC,EAAE,QAAQ,MAAM,MAAM,QAAQ,QAAQ,CAAC;AAAA,MACtF,IACA;AACN,aAAO;AAAA,QACL,SAAS;AAAA,QACT,WAAW,OAAO;AAAA,QAClB;AAAA,QACA,GAAI,UAAU,EAAE,QAAQ,IAAI,CAAC;AAAA,QAC7B,GAAI,eAAe,EAAE,aAAa,IAAI,CAAC;AAAA,MACzC;AAAA,IACF,QAAQ;AACN,aAAO,MAAM,GAAG;AAAA,IAClB;AAAA,EACF;AAAA,EAEQ,KAAK,WAA2B;AAGtC,WAAO,KAAK,KAAK,WAAW,GAAG,OAAO,SAAS,CAAC,OAAO;AAAA,EACzD;AAAA,EAEA,MAAc,QAAQ,WAAiD;AACrE,UAAM,OAAO,GAAG,KAAK,KAAK,SAAS,CAAC;AACpC,aAAS,UAAU,KAAK,WAAW;AACjC,UAAI;AACF,cAAM,MAAM,IAAI;AAChB,eAAO,YAAY,GAAG,MAAM,EAAE,WAAW,MAAM,OAAO,KAAK,CAAC;AAAA,MAC9D,SAAS,OAAO;AACd,YAAK,MAAgC,SAAS,YAAY,WAAW,IAAK,OAAM;AAChF,YAAI;AACF,gBAAM,MAAM,KAAK,IAAI,KAAK,MAAM,KAAK,IAAI,GAAG;AAC5C,cAAI,MAAM,eAAe;AACvB,kBAAM,GAAG,MAAM,EAAE,WAAW,MAAM,OAAO,KAAK,CAAC;AAC/C;AAAA,UACF;AAAA,QACF,QAAQ;AAAA,QAER;AACA,cAAM,IAAI,QAAQ,CAAC,gBAAgB,WAAW,aAAa,EAAE,CAAC;AAAA,MAChE;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,MAAM,KAAuB;AACpC,SAAO,EAAE,SAAS,SAAS,WAAW,IAAI,YAAY,GAAG,SAAS,CAAC,EAAE;AACvE;AAEA,SAAS,OAAO,MAAsB;AACpC,SAAO,WAAW,QAAQ,EAAE,OAAO,IAAI,EAAE,OAAO,KAAK;AACvD;;;AC1VA,SAAS,QAAQ,YAAAA,iBAAgB;AACjC,SAAS,UAAU,SAAS,SAAS,QAAAC,aAAY;AAO1C,SAAS,iBAAiB,MAAc,OAA2C;AACxF,QAAM,MAAM,QAAQ,IAAI;AACxB,QAAM,OAAO,SAAS,MAAM,GAAG;AAC/B,QAAM,MAAM,QAAQ,IAAI;AACxB,QAAMC,SAAQ,IAAI,UAAoB,MAAM,OAAO,CAAC,SAAS,QAAQ,SAAS,GAAG,EAAE,KAAK,GAAG;AAE3F,QAAM,WAAW,IAAI,QAAQ,wBAAwB,EAAE;AACvD,QAAM,aACJ,QAAQ,QACJ;AAAA,IACEA,OAAM,KAAK,QAAQ,IAAI,KAAK;AAAA,IAC5BA,OAAM,SAAS,QAAQ,IAAI,KAAK;AAAA,IAChCA,OAAM,KAAK,SAAS,QAAQ,IAAI,KAAK;AAAA,IACrCA,OAAM,KAAK,GAAG,IAAI,UAAU;AAAA,EAC9B,IACA,QAAQ,QACN,CAACA,OAAM,KAAK,GAAG,IAAI,UAAU,CAAC,IAC9B;AAAA,IACEA,OAAM,KAAK,GAAG,IAAI,QAAQ,GAAG,EAAE;AAAA,IAC/BA,OAAM,KAAK,GAAG,IAAI,QAAQ,GAAG,EAAE;AAAA,IAC/BA,OAAM,KAAK,aAAa,GAAG,IAAI,QAAQ,GAAG,EAAE;AAAA,IAC5CA,OAAM,KAAK,aAAa,GAAG,IAAI,GAAG,GAAG,EAAE;AAAA,IACvCA,OAAM,QAAQ,UAAU,GAAG,IAAI,QAAQ,GAAG,EAAE;AAAA,IAC5CA,OAAM,SAAS,UAAU,GAAG,IAAI,QAAQ,GAAG,EAAE;AAAA,EAC/C;AACR,SAAO,WAAW,KAAK,CAAC,cAAc,MAAM,IAAI,SAAS,CAAC,KAAK;AACjE;AAEA,IAAM,kBAAkB;AAExB,eAAsB,kBAAkB,gBAAgD;AACtF,QAAM,SAAS,CAAC,SACd,OAAOD,MAAK,gBAAgB,IAAI,CAAC,EAAE;AAAA,IACjC,MAAM;AAAA,IACN,MAAM;AAAA,EACR;AACF,MAAI;AACF,UAAM,MAAM,KAAK,MAAM,MAAMD,UAASC,MAAK,gBAAgB,cAAc,GAAG,MAAM,CAAC;AAGnF,UAAM,SAAS,IAAI,SAAS;AAC5B,QAAI,OAAO,WAAW,YAAY,OAAO,KAAK,KAAK,CAAC,gBAAgB,KAAK,MAAM,GAAG;AAChF,UAAI,MAAM,OAAO,gBAAgB,EAAG,QAAO;AAC3C,UAAI,MAAM,OAAO,WAAW,EAAG,QAAO;AACtC,aAAO;AAAA,IACT;AAAA,EACF,QAAQ;AAAA,EAER;AACA,aAAW,UAAU,CAAC,kBAAkB,cAAc,WAAW,GAAG;AAClE,QAAI,MAAM,OAAO,MAAM,EAAG,QAAO;AAAA,EACnC;AACA,MAAI,MAAM,OAAO,QAAQ,EAAG,QAAO;AACnC,MAAI,MAAM,OAAO,YAAY,EAAG,QAAO;AACvC,SAAO;AACT;","names":["readFile","join","join2"]}

package/dist/chunk-U4HAJLIB.js

@@ -0,0 +1,91 @@
+import {
+  LOCAL_EMBEDDING_MODEL
+} from "./chunk-JPHJBCZB.js";
+
+// src/providers/local.ts
+import { homedir } from "os";
+import { join } from "path";
+var MAX_INPUT_CHARS = 8e3;
+var transformers = null;
+var extractor = null;
+function modelCacheDir() {
+  return process.env.CODEBUDDY_MODEL_CACHE ?? join(homedir(), ".cache", "codebuddy", "models");
+}
+function loadTransformers() {
+  transformers ??= import("@huggingface/transformers").then((module) => {
+    module.env.cacheDir = modelCacheDir();
+    module.env.logLevel = module.LogLevel.ERROR;
+    return module;
+  });
+  transformers.catch(() => {
+    transformers = null;
+  });
+  return transformers;
+}
+var LocalModelProvider = class {
+  space = {
+    model: LOCAL_EMBEDDING_MODEL.id,
+    revision: LOCAL_EMBEDDING_MODEL.revision,
+    schema: LOCAL_EMBEDDING_MODEL.schema,
+    dimensions: LOCAL_EMBEDDING_MODEL.dimensions
+  };
+  /** `request.model` is ignored: one provider is one embedding space. */
+  async embed(request) {
+    const inputs = (typeof request.input === "string" ? [request.input] : request.input).map(
+      (text) => text.slice(0, MAX_INPUT_CHARS)
+    );
+    extractor ??= loadTransformers().then(
+      async (module) => await module.pipeline("feature-extraction", this.space.model, {
+        revision: this.space.revision,
+        dtype: "q8"
+      })
+    );
+    extractor.catch(() => {
+      extractor = null;
+    });
+    const extract = await extractor;
+    const output = await extract(inputs, { pooling: "cls", normalize: true });
+    return { ...this.space, vectors: output.tolist() };
+  }
+  async generateText() {
+    throw new Error("The local provider only embeds; text generation uses the BYOK LLM layer.");
+  }
+};
+var rerankers = /* @__PURE__ */ new Map();
+async function rerankLocally(model, query, documents) {
+  if (documents.length === 0) return [];
+  let pending = rerankers.get(model.id);
+  if (!pending) {
+    pending = loadTransformers().then(async (module) => {
+      const options = model.revision ? { revision: model.revision } : {};
+      const [tokenizer2, classifier2] = await Promise.all([
+        module.AutoTokenizer.from_pretrained(model.id, options),
+        module.AutoModelForSequenceClassification.from_pretrained(model.id, {
+          ...options,
+          dtype: "q8"
+        })
+      ]);
+      return { tokenizer: tokenizer2, model: classifier2 };
+    });
+    rerankers.set(model.id, pending);
+    pending.catch(() => rerankers.delete(model.id));
+  }
+  const { tokenizer, model: classifier } = await pending;
+  const inputs = tokenizer(
+    documents.map(() => query.slice(0, MAX_INPUT_CHARS)),
+    {
+      text_pair: documents.map((document) => document.slice(0, MAX_INPUT_CHARS)),
+      padding: true,
+      truncation: true
+    }
+  );
+  const { logits } = await classifier(inputs);
+  return logits.tolist().map((row) => row[0] ?? Number.NEGATIVE_INFINITY);
+}
+
+export {
+  modelCacheDir,
+  LocalModelProvider,
+  rerankLocally
+};
+//# sourceMappingURL=chunk-U4HAJLIB.js.map

package/dist/chunk-U4HAJLIB.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/providers/local.ts"],"sourcesContent":["import { homedir } from \"node:os\";\nimport { join } from \"node:path\";\nimport type {\n  EmbeddingRequest,\n  EmbeddingResponse,\n  ModelProvider,\n  TextGenerationResponse,\n} from \"./adapter.js\";\nimport type { EmbeddingSpace } from \"./embedding-space.js\";\nimport { LOCAL_EMBEDDING_MODEL } from \"./models.js\";\n\n/**\n * Local Hugging Face models through transformers.js (ONNX Runtime, CPU). The\n * weights download once into `~/.cache/codebuddy/models` (override with\n * `CODEBUDDY_MODEL_CACHE`); afterwards nothing leaves the machine and no token\n * is needed. `@huggingface/transformers` is an optional dependency, loaded on\n * first use, so a missing install only disables semantic ranking.\n */\n\ntype Transformers = typeof import(\"@huggingface/transformers\");\n/** The slice of transformers.js' FeatureExtractionPipeline used here. */\ntype Extractor = (\n  texts: string[],\n  options: { pooling: \"cls\"; normalize: boolean },\n) => Promise<{ tolist(): number[][] }>;\n\n/** Hard cap on a single input; ModernBERT reads 8k tokens, memories are far shorter. */\nconst MAX_INPUT_CHARS = 8_000;\n\nlet transformers: Promise<Transformers> | null = null;\nlet extractor: Promise<Extractor> | null = null;\n\nexport function modelCacheDir(): string {\n  return process.env.CODEBUDDY_MODEL_CACHE ?? join(homedir(), \".cache\", \"codebuddy\", \"models\");\n}\n\nfunction loadTransformers(): Promise<Transformers> {\n  transformers ??= import(\"@huggingface/transformers\").then((module) => {\n    module.env.cacheDir = modelCacheDir();\n    // Info/debug logs go to console.log, i.e. stdout: the MCP JSON-RPC channel\n    // under `codebuddy serve`. Errors only (they use stderr), here and in ONNX Runtime.\n    module.env.logLevel = module.LogLevel.ERROR;\n    return module;\n  });\n  // A failed import (optional dependency missing) is retried on the next call.\n  transformers.catch(() => {\n    transformers = null;\n  });\n  return transformers;\n}\n\nexport class LocalModelProvider implements ModelProvider {\n  readonly space: EmbeddingSpace = {\n    model: LOCAL_EMBEDDING_MODEL.id,\n    revision: LOCAL_EMBEDDING_MODEL.revision,\n    schema: LOCAL_EMBEDDING_MODEL.schema,\n    dimensions: LOCAL_EMBEDDING_MODEL.dimensions,\n  };\n\n  /** `request.model` is ignored: one provider is one embedding space. */\n  async embed(request: EmbeddingRequest): Promise<EmbeddingResponse> {\n    const inputs = (typeof request.input === \"string\" ? [request.input] : request.input).map(\n      (text) => text.slice(0, MAX_INPUT_CHARS),\n    );\n    extractor ??= loadTransformers().then(\n      async (module) =>\n        (await module.pipeline(\"feature-extraction\", this.space.model, {\n          revision: this.space.revision,\n          dtype: \"q8\",\n        })) as unknown as Extractor,\n    );\n    // A failed load (no network on first use) is retried on the next call.\n    extractor.catch(() => {\n      extractor = null;\n    });\n    const extract = await extractor;\n    // Granite r2 is trained with CLS pooling; L2-normalized so cosine = dot.\n    const output = await extract(inputs, { pooling: \"cls\", normalize: true });\n    return { ...this.space, vectors: output.tolist() };\n  }\n\n  async generateText(): Promise<TextGenerationResponse> {\n    throw new Error(\"The local provider only embeds; text generation uses the BYOK LLM layer.\");\n  }\n}\n\ntype Reranker = {\n  tokenizer: (\n    texts: string[],\n    options: { text_pair: string[]; padding: boolean; truncation: boolean },\n  ) => unknown;\n  model: (inputs: unknown) => Promise<{ logits: { tolist(): number[][] } }>;\n};\nconst rerankers = new Map<string, Promise<Reranker>>();\n\n/**\n * Local cross-encoder logits for (query, document) pairs, in input order.\n * Higher is more relevant; values order documents for one query and are not\n * calibrated across queries, so callers order by them and never threshold.\n */\nexport async function rerankLocally(\n  model: { id: string; revision?: string },\n  query: string,\n  documents: string[],\n): Promise<number[]> {\n  if (documents.length === 0) return [];\n  let pending = rerankers.get(model.id);\n  if (!pending) {\n    pending = loadTransformers().then(async (module) => {\n      const options = model.revision ? { revision: model.revision } : {};\n      const [tokenizer, classifier] = await Promise.all([\n        module.AutoTokenizer.from_pretrained(model.id, options),\n        module.AutoModelForSequenceClassification.from_pretrained(model.id, {\n          ...options,\n          dtype: \"q8\",\n        }),\n      ]);\n      return { tokenizer, model: classifier } as unknown as Reranker;\n    });\n    rerankers.set(model.id, pending);\n    pending.catch(() => rerankers.delete(model.id));\n  }\n  const { tokenizer, model: classifier } = await pending;\n  const inputs = tokenizer(\n    documents.map(() => query.slice(0, MAX_INPUT_CHARS)),\n    {\n      text_pair: documents.map((document) => document.slice(0, MAX_INPUT_CHARS)),\n      padding: true,\n      truncation: true,\n    },\n  );\n  const { logits } = await classifier(inputs);\n  return logits.tolist().map((row) => row[0] ?? Number.NEGATIVE_INFINITY);\n}\n"],"mappings":";;;;;AAAA,SAAS,eAAe;AACxB,SAAS,YAAY;AA0BrB,IAAM,kBAAkB;AAExB,IAAI,eAA6C;AACjD,IAAI,YAAuC;AAEpC,SAAS,gBAAwB;AACtC,SAAO,QAAQ,IAAI,yBAAyB,KAAK,QAAQ,GAAG,UAAU,aAAa,QAAQ;AAC7F;AAEA,SAAS,mBAA0C;AACjD,mBAAiB,OAAO,2BAA2B,EAAE,KAAK,CAAC,WAAW;AACpE,WAAO,IAAI,WAAW,cAAc;AAGpC,WAAO,IAAI,WAAW,OAAO,SAAS;AACtC,WAAO;AAAA,EACT,CAAC;AAED,eAAa,MAAM,MAAM;AACvB,mBAAe;AAAA,EACjB,CAAC;AACD,SAAO;AACT;AAEO,IAAM,qBAAN,MAAkD;AAAA,EAC9C,QAAwB;AAAA,IAC/B,OAAO,sBAAsB;AAAA,IAC7B,UAAU,sBAAsB;AAAA,IAChC,QAAQ,sBAAsB;AAAA,IAC9B,YAAY,sBAAsB;AAAA,EACpC;AAAA;AAAA,EAGA,MAAM,MAAM,SAAuD;AACjE,UAAM,UAAU,OAAO,QAAQ,UAAU,WAAW,CAAC,QAAQ,KAAK,IAAI,QAAQ,OAAO;AAAA,MACnF,CAAC,SAAS,KAAK,MAAM,GAAG,eAAe;AAAA,IACzC;AACA,kBAAc,iBAAiB,EAAE;AAAA,MAC/B,OAAO,WACJ,MAAM,OAAO,SAAS,sBAAsB,KAAK,MAAM,OAAO;AAAA,QAC7D,UAAU,KAAK,MAAM;AAAA,QACrB,OAAO;AAAA,MACT,CAAC;AAAA,IACL;AAEA,cAAU,MAAM,MAAM;AACpB,kBAAY;AAAA,IACd,CAAC;AACD,UAAM,UAAU,MAAM;AAEtB,UAAM,SAAS,MAAM,QAAQ,QAAQ,EAAE,SAAS,OAAO,WAAW,KAAK,CAAC;AACxE,WAAO,EAAE,GAAG,KAAK,OAAO,SAAS,OAAO,OAAO,EAAE;AAAA,EACnD;AAAA,EAEA,MAAM,eAAgD;AACpD,UAAM,IAAI,MAAM,0EAA0E;AAAA,EAC5F;AACF;AASA,IAAM,YAAY,oBAAI,IAA+B;AAOrD,eAAsB,cACpB,OACA,OACA,WACmB;AACnB,MAAI,UAAU,WAAW,EAAG,QAAO,CAAC;AACpC,MAAI,UAAU,UAAU,IAAI,MAAM,EAAE;AACpC,MAAI,CAAC,SAAS;AACZ,cAAU,iBAAiB,EAAE,KAAK,OAAO,WAAW;AAClD,YAAM,UAAU,MAAM,WAAW,EAAE,UAAU,MAAM,SAAS,IAAI,CAAC;AACjE,YAAM,CAACA,YAAWC,WAAU,IAAI,MAAM,QAAQ,IAAI;AAAA,QAChD,OAAO,cAAc,gBAAgB,MAAM,IAAI,OAAO;AAAA,QACtD,OAAO,mCAAmC,gBAAgB,MAAM,IAAI;AAAA,UAClE,GAAG;AAAA,UACH,OAAO;AAAA,QACT,CAAC;AAAA,MACH,CAAC;AACD,aAAO,EAAE,WAAAD,YAAW,OAAOC,YAAW;AAAA,IACxC,CAAC;AACD,cAAU,IAAI,MAAM,IAAI,OAAO;AAC/B,YAAQ,MAAM,MAAM,UAAU,OAAO,MAAM,EAAE,CAAC;AAAA,EAChD;AACA,QAAM,EAAE,WAAW,OAAO,WAAW,IAAI,MAAM;AAC/C,QAAM,SAAS;AAAA,IACb,UAAU,IAAI,MAAM,MAAM,MAAM,GAAG,eAAe,CAAC;AAAA,IACnD;AAAA,MACE,WAAW,UAAU,IAAI,CAAC,aAAa,SAAS,MAAM,GAAG,eAAe,CAAC;AAAA,MACzE,SAAS;AAAA,MACT,YAAY;AAAA,IACd;AAAA,EACF;AACA,QAAM,EAAE,OAAO,IAAI,MAAM,WAAW,MAAM;AAC1C,SAAO,OAAO,OAAO,EAAE,IAAI,CAAC,QAAQ,IAAI,CAAC,KAAK,OAAO,iBAAiB;AACxE;","names":["tokenizer","classifier"]}

package/dist/{chunk-IE2OUO54.js → chunk-UFZD3JAA.js}

@@ -1,9 +1,9 @@
 import {
   redactLlmKeys
-} from "./chunk-ITGRANYW.js";
+} from "./chunk-6MKEIDX5.js";
 import {
   writeAtomic
-} from "./chunk-K63EOA4B.js";
+} from "./chunk-776EQXDV.js";
 
 // src/llm/budget.ts
 import { mkdir, readFile } from "fs/promises";
@@ -196,4 +196,4 @@ export {
   coolingDown,
   chatJson
 };
-//# sourceMappingURL=chunk-IE2OUO54.js.map
+//# sourceMappingURL=chunk-UFZD3JAA.js.map