omp-fabric 0.85.0 → 1.0.2

package/dist/chunks/{chunk-SXMWV447.js → chunk-DOKANNFC.js}

@@ -6,13 +6,13 @@ import {
 } from "./chunk-HDJBUXDX.js";
 import {
   NESTED_TOOL_CALL_ID_PREFIX
-} from "./chunk-GKOFZ76R.js";
+} from "./chunk-2LGLDM34.js";
 import {
   THINKING_DIGEST_CUSTOM_TYPE,
   buildThinkingDigest,
   snapshotHandoffSession,
   thinkingTransferPolicy
-} from "./chunk-OPTP4PSE.js";
+} from "./chunk-YPYVUGON.js";
 import {
   isFabricThinking
 } from "./chunk-JSBTB3P6.js";
@@ -589,6 +589,207 @@ var runFabricHandoffAtBoundary = async (controller, runner, extension, pending,
   }
 };
 
+// src/speculation/partial-json.ts
+var CODE_KEY_PATTERN = /"code"\s*:\s*"/;
+var ESCAPES = {
+  '"': '"',
+  "\\": "\\",
+  "/": "/",
+  b: "\b",
+  f: "\f",
+  n: "\n",
+  r: "\r",
+  t: "	"
+};
+var PartialCodeFieldExtractor = class {
+  #raw = "";
+  #codeStart = -1;
+  #cursor = 0;
+  #decoded = "";
+  #complete = false;
+  #failed = false;
+  #maxBytes;
+  constructor(maxBytes) {
+    this.#maxBytes = maxBytes;
+  }
+  get complete() {
+    return this.#complete;
+  }
+  /** Decoded `code` content so far, or undefined before the key appears. */
+  get code() {
+    return this.#codeStart === -1 ? void 0 : this.#decoded;
+  }
+  push(delta) {
+    if (this.#complete || this.#failed) return;
+    if (this.#raw.length + delta.length > this.#maxBytes) {
+      this.#failed = true;
+      return;
+    }
+    this.#raw += delta;
+    if (this.#codeStart === -1) {
+      const match = CODE_KEY_PATTERN.exec(this.#raw);
+      if (!match) return;
+      this.#codeStart = match.index + match[0].length;
+      this.#cursor = this.#codeStart;
+    }
+    const raw = this.#raw;
+    while (this.#cursor < raw.length) {
+      const ch = raw[this.#cursor];
+      if (ch === '"') {
+        this.#complete = true;
+        this.#cursor = raw.length;
+        return;
+      }
+      if (ch === "\\") {
+        if (this.#cursor + 1 >= raw.length) return;
+        const esc = raw[this.#cursor + 1];
+        if (esc === "u") {
+          if (this.#cursor + 5 >= raw.length) return;
+          const hex = raw.slice(this.#cursor + 2, this.#cursor + 6);
+          if (!/^[0-9a-fA-F]{4}$/.test(hex)) {
+            this.#failed = true;
+            return;
+          }
+          this.#decoded += String.fromCharCode(parseInt(hex, 16));
+          this.#cursor += 6;
+          continue;
+        }
+        const mapped = ESCAPES[esc];
+        if (mapped === void 0) {
+          this.#failed = true;
+          return;
+        }
+        this.#decoded += mapped;
+        this.#cursor += 2;
+        continue;
+      }
+      if (ch < " ") {
+        this.#failed = true;
+        return;
+      }
+      this.#decoded += ch;
+      this.#cursor += 1;
+    }
+  }
+};
+
+// src/speculation/stream-tap.ts
+var toolCallBlock = (partial, contentIndex) => {
+  const content = partial?.content;
+  if (!Array.isArray(content)) return {};
+  const block = content[contentIndex];
+  if (!block || typeof block !== "object") return {};
+  const record = block;
+  if (record.type !== "toolCall") return {};
+  return {
+    ...typeof record.name === "string" && record.name ? { name: record.name } : {},
+    ...typeof record.id === "string" && record.id ? { id: record.id } : {}
+  };
+};
+var streamedToolCallName = (event) => {
+  const assistantEvent = event.assistantMessageEvent;
+  if (assistantEvent.contentIndex === void 0) return void 0;
+  const source = assistantEvent.type === "toolcall_end" ? assistantEvent.toolCall : assistantEvent.partial;
+  return toolCallBlock(source, assistantEvent.contentIndex).name;
+};
+var PARSE_INTERVAL_MS = 50;
+var FabricSpeculationStreamTap = class {
+  #options;
+  #streams = /* @__PURE__ */ new Map();
+  // The scanner depends on the TypeScript compiler; the factory arrives through
+  // a lazy dynamic import so session startup never pays for it. Streams opened
+  // before the factory lands are caught up in full (extractors buffer the
+  // whole decoded prefix, so no candidate is lost, only delayed).
+  #createScanner;
+  #lastParseAt = 0;
+  constructor(options) {
+    this.#options = options;
+  }
+  setScannerFactory(factory) {
+    this.#createScanner = factory;
+    for (const stream of this.#streams.values()) {
+      if (stream.scanner) continue;
+      stream.scanner = factory();
+      const code = stream.extractor.code;
+      if (!stream.isFabricExec || code === void 0) continue;
+      try {
+        for (const candidate of stream.scanner.push(code)) {
+          if (this.#options.isEligible(candidate.ref)) {
+            this.#pendingCatchUp.push({ stream, candidate });
+          }
+        }
+      } catch {
+      }
+    }
+  }
+  #pendingCatchUp = [];
+  /** Drain candidates recovered while the scanner module loaded. */
+  flushCatchUp(context) {
+    const pending = this.#pendingCatchUp.splice(0);
+    for (const { stream, candidate } of pending) {
+      this.#options.launch(stream.toolCallId, candidate, context);
+    }
+  }
+  /** New assistant message: content indices restart. */
+  reset() {
+    this.#streams.clear();
+  }
+  handleMessageUpdate(event, context) {
+    try {
+      if (!this.#options.enabled()) return;
+      if (this.#pendingCatchUp.length > 0) this.flushCatchUp(context);
+      const assistantEvent = event.assistantMessageEvent;
+      if (assistantEvent.type === "toolcall_start") {
+        const block = toolCallBlock(assistantEvent.partial, assistantEvent.contentIndex);
+        this.#streams.set(assistantEvent.contentIndex, {
+          toolCallId: block.id ?? `index-${assistantEvent.contentIndex}`,
+          isFabricExec: block.name !== void 0 ? block.name === "fabric_exec" : true,
+          extractor: new PartialCodeFieldExtractor(this.#options.maxBufferBytes()),
+          ...this.#createScanner ? { scanner: this.#createScanner() } : {}
+        });
+        return;
+      }
+      if (assistantEvent.type === "toolcall_delta") {
+        const stream = this.#streams.get(assistantEvent.contentIndex);
+        if (!stream) return;
+        const block = toolCallBlock(assistantEvent.partial, assistantEvent.contentIndex);
+        if (block.id) stream.toolCallId = block.id;
+        if (block.name !== void 0) stream.isFabricExec = block.name === "fabric_exec";
+        stream.extractor.push(assistantEvent.delta);
+        const code = stream.extractor.code;
+        if (!stream.isFabricExec || code === void 0) return;
+        this.#scan(stream, code, context, false);
+        return;
+      }
+      if (assistantEvent.type === "toolcall_end") {
+        const stream = this.#streams.get(assistantEvent.contentIndex);
+        if (!stream) return;
+        this.#streams.delete(assistantEvent.contentIndex);
+        const block = toolCallBlock(assistantEvent.toolCall, assistantEvent.contentIndex);
+        if (block.id) stream.toolCallId = block.id;
+        if (block.name !== void 0) stream.isFabricExec = block.name === "fabric_exec";
+        const code = stream.extractor.code;
+        if (stream.isFabricExec && code !== void 0) {
+          this.#scan(stream, code, context, true);
+        }
+      }
+    } catch {
+    }
+  }
+  #scan(stream, code, context, force) {
+    const now = Date.now();
+    if (!force && now - this.#lastParseAt < PARSE_INTERVAL_MS) return;
+    this.#lastParseAt = now;
+    if (!stream.scanner) return;
+    const candidates = stream.scanner.push(code);
+    for (const candidate of candidates) {
+      if (this.#options.isEligible(candidate.ref)) {
+        this.#options.launch(stream.toolCallId, candidate, context);
+      }
+    }
+  }
+};
+
 // src/core/skill-block.ts
 var escapeXml = (str) => str.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;").replace(/"/g, "&quot;").replace(/'/g, "&apos;");
 var formatSkillsForPrompt = (skills) => {
@@ -2425,6 +2626,8 @@ export {
   claimFabricFsDriftHandoff,
   settleInPlacePrewalk,
   runFabricHandoffAtBoundary,
+  streamedToolCallName,
+  FabricSpeculationStreamTap,
   formatSkillsForPrompt,
   expandSkillDirMarkersInSkillBlock,
   expandSkillDirMarkersForRead,
@@ -2442,4 +2645,4 @@ export {
   resolveFabricIdentity,
   MainAgentController
 };
-//# sourceMappingURL=chunk-SXMWV447.js.map
+//# sourceMappingURL=chunk-DOKANNFC.js.map