@athenaintel/react 0.57.2 → 0.60.0

package/dist/agent-runtime/src/lane-follow.cjs

@@ -0,0 +1,289 @@
+"use strict";
+Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
+const store = require("@assistant-ui/store");
+const require$$0 = require("react");
+const runconfigContract = require("./generated/runconfig-contract.cjs");
+const messageWindow = require("./message-window.cjs");
+const protocol = require("./protocol.cjs");
+const runningToolPlaceholder = require("./running-tool-placeholder.cjs");
+const statewireTransport = require("./statewire-transport.cjs");
+const threadExtras = require("./thread-extras.cjs");
+const isRecord = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
+const isCount = (value) => typeof value === "number" && Number.isInteger(value) && value >= 0;
+const stepId = (step) => {
+  const id = isRecord(step) ? step.id : void 0;
+  return typeof id === "string" ? id : void 0;
+};
+function laneFollowRequest(cardId) {
+  return {
+    ns: runconfigContract.DEEP_AGENT_MAIN_NS,
+    before: 1,
+    at: cardId,
+    follow: { before: runconfigContract.DEEP_AGENT_MAX_FOLLOWED_STEPS }
+  };
+}
+function isLaneFollowRequest(request) {
+  return request.follow !== void 0 && request.at !== void 0;
+}
+const NO_FOLLOWS = { requests: [], lanes: /* @__PURE__ */ new Map() };
+function followRequestOf(window) {
+  const { ns, before, at, follow, chain } = window;
+  if (ns !== runconfigContract.DEEP_AGENT_MAIN_NS || typeof at !== "string" || !isRecord(follow)) return null;
+  if (!Array.isArray(chain) || chain.length === 0) return null;
+  if (!isCount(before) || !isCount(follow.before)) return null;
+  return { ns: runconfigContract.DEEP_AGENT_MAIN_NS, before, at, follow: { before: follow.before } };
+}
+function readDeepAgentLaneFollows(mounted) {
+  var _a;
+  if (mounted === void 0) return NO_FOLLOWS;
+  let windows;
+  const steps = /* @__PURE__ */ new Map();
+  for (const document of mounted.values()) {
+    const key = document.key;
+    if (document.protocol === runconfigContract.DEEP_AGENT_INTEREST_PROTOCOL && document.main === true) {
+      const value = document.value;
+      if (isRecord(value) && Array.isArray(value.windows)) windows = value.windows;
+      continue;
+    }
+    if (document.protocol !== statewireTransport.DEEP_AGENT_STATEWIRE_PROTOCOL || key === null || key.length !== 3) {
+      continue;
+    }
+    const [, ns, id] = key;
+    if (typeof ns !== "string" || ns === runconfigContract.DEEP_AGENT_MAIN_NS || typeof id !== "string") continue;
+    let lane = steps.get(ns);
+    if (lane === void 0) {
+      lane = /* @__PURE__ */ new Map();
+      steps.set(ns, lane);
+    }
+    lane.set(id, document.value);
+  }
+  if (windows === void 0) return NO_FOLLOWS;
+  const requests = [];
+  const lanes = /* @__PURE__ */ new Map();
+  for (const window of windows) {
+    if (!isRecord(window)) continue;
+    const request = followRequestOf(window);
+    if (request !== null) requests.push(request);
+    if (!isRecord(window.follows)) continue;
+    for (const [ns, follow] of Object.entries(window.follows)) {
+      if (!isRecord(follow) || !Array.isArray(follow.chain)) continue;
+      const toolCallId = follow[runconfigContract.DEEP_AGENT_FOLLOW_TOOL_CALL_ID_KEY];
+      if (typeof toolCallId !== "string") continue;
+      const lane = steps.get(ns);
+      const ordered = [];
+      for (const entry of follow.chain) {
+        const id = stepId(entry);
+        const step = id === void 0 ? void 0 : lane == null ? void 0 : lane.get(id);
+        if (step !== void 0) ordered.push(step);
+      }
+      if (ordered.length > (((_a = lanes.get(toolCallId)) == null ? void 0 : _a.length) ?? 0)) lanes.set(toolCallId, ordered);
+    }
+  }
+  return requests.length === 0 && lanes.size === 0 ? NO_FOLLOWS : { requests, lanes };
+}
+function mergeFollowedLane(carried, followed, cutTurns = 0) {
+  const position = /* @__PURE__ */ new Map();
+  for (const [index, step] of followed.entries()) {
+    const id = stepId(step);
+    if (id !== void 0) position.set(id, index);
+  }
+  const newest = carried.slice(cutTurns);
+  const turns = carried.slice(0, cutTurns).filter((turn) => {
+    const id = stepId(turn);
+    return id === void 0 || !position.has(id);
+  });
+  let first = followed.length;
+  let last = -1;
+  for (const step of newest) {
+    const id = stepId(step);
+    const at = id === void 0 ? void 0 : position.get(id);
+    if (at === void 0) continue;
+    first = Math.min(first, at);
+    last = Math.max(last, at);
+  }
+  if (last < 0) {
+    return carried.length === 0 ? followed : [...turns, ...followed, ...newest];
+  }
+  return [...turns, ...followed.slice(0, first), ...newest, ...followed.slice(last + 1)];
+}
+function cutTurnsOf(message, carried) {
+  const marker = message[runconfigContract.DEEP_AGENT_TRUNCATED_LANE_KEY];
+  return isRecord(marker) ? Math.max(0, carried.length - runconfigContract.DEEP_AGENT_RUNNING_LANE_WIRE_MESSAGES) : 0;
+}
+const sameItems = (a, b) => a.length === b.length && a.every((item, index) => item === b[index]);
+const followedCards = /* @__PURE__ */ new WeakMap();
+function withFollowedLane(message, lanes) {
+  if (message.type !== "tool" || !runningToolPlaceholder.isRunningToolPlaceholder(message)) return message;
+  const steps = lanes.get(message.tool_call_id);
+  if (steps === void 0 || steps.length === 0) return message;
+  const kept = followedCards.get(message);
+  if (kept !== void 0 && sameItems(kept.steps, steps)) return kept.card;
+  const artifact = isRecord(message.artifact) ? message.artifact : {};
+  const state = isRecord(artifact.subgraph_state) ? artifact.subgraph_state : {};
+  const carried = Array.isArray(state.messages) ? state.messages : [];
+  const card = {
+    ...message,
+    artifact: {
+      ...artifact,
+      subgraph_state: {
+        ...state,
+        messages: mergeFollowedLane(carried, steps, cutTurnsOf(message, carried))
+      }
+    }
+  };
+  followedCards.set(message, { steps, card });
+  return card;
+}
+function readDeepAgentFollowedWindow(mounted) {
+  const window = messageWindow.readDeepAgentMessageWindow(mounted);
+  if (window === null) return null;
+  const follows = readDeepAgentLaneFollows(mounted);
+  if (follows === NO_FOLLOWS) return window;
+  return {
+    ...window,
+    requests: [...window.requests, ...follows.requests],
+    messages: follows.lanes.size === 0 ? window.messages : window.messages.map((message) => withFollowedLane(message, follows.lanes))
+  };
+}
+function createDeepAgentLaneFollowRegistry() {
+  const holds = /* @__PURE__ */ new Map();
+  const listeners = /* @__PURE__ */ new Set();
+  let snapshot = /* @__PURE__ */ new Set();
+  const publish = () => {
+    snapshot = new Set(holds.keys());
+    for (const listener of listeners) listener();
+  };
+  return {
+    follow(toolCallId) {
+      const held = holds.get(toolCallId) ?? 0;
+      holds.set(toolCallId, held + 1);
+      if (held === 0) publish();
+      let released = false;
+      return () => {
+        if (released) return;
+        released = true;
+        const left = (holds.get(toolCallId) ?? 1) - 1;
+        if (left > 0) {
+          holds.set(toolCallId, left);
+          return;
+        }
+        holds.delete(toolCallId);
+        publish();
+      };
+    },
+    subscribe(listener) {
+      listeners.add(listener);
+      return () => {
+        listeners.delete(listener);
+      };
+    },
+    getSnapshot: () => snapshot
+  };
+}
+function useDeepAgentLaneFollow(registry, toolCallId, active) {
+  require$$0.useEffect(() => {
+    if (registry === null || !active) return void 0;
+    return registry.follow(toolCallId);
+  }, [registry, toolCallId, active]);
+}
+function laneFollowRequestsFor(window, followed) {
+  if (followed.size === 0) return [];
+  const requests = [];
+  for (const message of window.messages) {
+    if (message.type !== "tool" || !followed.has(message.tool_call_id)) continue;
+    if (typeof message.id !== "string" || !runningToolPlaceholder.isRunningToolPlaceholder(message)) continue;
+    requests.push(laneFollowRequest(message.id));
+  }
+  return requests;
+}
+const followKey = (request) => {
+  var _a;
+  return `${request.at}:${(_a = request.follow) == null ? void 0 : _a.before}`;
+};
+const sameLaneFollows = (held, wanted) => held.length === wanted.length && wanted.every((request) => held.some((have) => followKey(have) === followKey(request)));
+const rejectionCode = (error) => {
+  const failure = isRecord(error) ? error.failure : void 0;
+  return isRecord(failure) && failure.fate === "rejected" && typeof failure.code === "string" ? failure.code : void 0;
+};
+const noopSubscribe = () => () => {
+};
+function useDeepAgentLaneFollowSync({
+  registry,
+  paused = false,
+  onError
+}) {
+  const { invoke, connection, documents } = store.useAuiState((s) => {
+    if (!threadExtras.hasStatewireThreadExtras(s.thread.extras)) {
+      throw new Error("useDeepAgentLaneFollowSync requires a statewire thread");
+    }
+    return s.thread.extras;
+  });
+  const hostVersion = store.useAuiState((s) => {
+    var _a;
+    const extras = s.thread.extras;
+    return ((_a = protocol.readDeepAgentProtocol(extras == null ? void 0 : extras.state)) == null ? void 0 : _a.version) ?? 0;
+  });
+  const mounted = require$$0.useSyncExternalStore(
+    (documents == null ? void 0 : documents.subscribe) ?? noopSubscribe,
+    () => documents == null ? void 0 : documents.list(),
+    () => documents == null ? void 0 : documents.list()
+  );
+  const window = require$$0.useMemo(() => readDeepAgentFollowedWindow(mounted), [mounted]);
+  const followed = require$$0.useSyncExternalStore(
+    registry.subscribe,
+    registry.getSnapshot,
+    registry.getSnapshot
+  );
+  const [settled, setSettled] = require$$0.useState(0);
+  const inflightRef = require$$0.useRef(false);
+  const aliveRef = require$$0.useRef(true);
+  const unservedRef = require$$0.useRef(null);
+  const lastStatedRef = require$$0.useRef(null);
+  const onErrorRef = require$$0.useRef(onError);
+  onErrorRef.current = onError;
+  require$$0.useEffect(() => {
+    aliveRef.current = true;
+    return () => {
+      aliveRef.current = false;
+    };
+  }, []);
+  require$$0.useEffect(() => {
+    if (window === null || paused || inflightRef.current) return;
+    if (connection.status !== "connected") return;
+    const served = hostVersion >= runconfigContract.DEEP_AGENT_LANE_FOLLOW_PROTOCOL_VERSION && unservedRef.current !== connection;
+    const others = window.requests.filter((request) => !isLaneFollowRequest(request));
+    const held = window.requests.filter(isLaneFollowRequest);
+    const wanted = served ? laneFollowRequestsFor(window, followed).slice(
+      0,
+      Math.max(0, runconfigContract.DEEP_AGENT_MAX_INTERESTS - others.length)
+    ) : [];
+    if (sameLaneFollows(held, wanted)) return;
+    const requests = [...others, ...wanted];
+    const key = JSON.stringify(requests);
+    const last = lastStatedRef.current;
+    if (last !== null && last.connection === connection && last.key === key) return;
+    lastStatedRef.current = { connection, key };
+    inflightRef.current = true;
+    invoke(runconfigContract.DEEP_AGENT_INTEREST_PROTOCOL, "set", [requests]).catch((error) => {
+      var _a;
+      const code = rejectionCode(error);
+      if (code === "invalid-interest" && wanted.length > 0) {
+        unservedRef.current = connection;
+        return;
+      }
+      if (code !== "unknown-message") (_a = onErrorRef.current) == null ? void 0 : _a.call(onErrorRef, error);
+    }).finally(() => {
+      inflightRef.current = false;
+      if (aliveRef.current) setSettled((count) => count + 1);
+    });
+  }, [window, followed, hostVersion, connection, invoke, paused, settled]);
+}
+exports.createDeepAgentLaneFollowRegistry = createDeepAgentLaneFollowRegistry;
+exports.isLaneFollowRequest = isLaneFollowRequest;
+exports.laneFollowRequest = laneFollowRequest;
+exports.mergeFollowedLane = mergeFollowedLane;
+exports.readDeepAgentFollowedWindow = readDeepAgentFollowedWindow;
+exports.readDeepAgentLaneFollows = readDeepAgentLaneFollows;
+exports.useDeepAgentLaneFollow = useDeepAgentLaneFollow;
+exports.useDeepAgentLaneFollowSync = useDeepAgentLaneFollowSync;
+//# sourceMappingURL=lane-follow.cjs.map

package/dist/agent-runtime/src/lane-follow.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"lane-follow.cjs","sources":["../../../../agent-runtime/src/lane-follow.ts"],"sourcesContent":["/**\n * Followed subagent lanes — the client half of agora's `follow` interest\n * (`message_window.follow_windows`).\n *\n * A running `task` card carries only its lane's newest steps (eight, with\n * `athena_lane_truncated` counting the rest). A request that adds\n * `follow: {before}` also covers the newest `before` steps of every lane a card\n * in its window names: the resolution gains `follows`\n * (`{ns: {chain, more, toolCallId}}`) and each step is its own `athena`\n * document, keyed `[threadId, ns, stepId]`.\n *\n * A lane is followed through a request anchored at its card alone\n * ({@link laneFollowRequest}), so following one lane pulls in no other and has\n * the attach's whole step allowance. {@link readDeepAgentFollowedWindow} reads\n * the followed steps back onto the card, so the card's tool-call part carries\n * the whole live lane as its nested thread; {@link useDeepAgentLaneFollowSync}\n * keeps the attach's follow requests to the running lanes a surface shows\n * ({@link createDeepAgentLaneFollowRegistry}). It follows only on a host whose\n * `protocol` lane advertises `DEEP_AGENT_LANE_FOLLOW_PROTOCOL_VERSION`; a host\n * that refuses `follow` anyway (`invalid-interest`, one predating it) keeps the\n * attach's windows, and the cards keep the steps their own documents carry.\n *\n * Surfaces opt in. A surface that follows nothing ships none of this and sees\n * the wire it saw before.\n */\nimport type { LangChainMessage } from '@assistant-ui/react-langgraph';\nimport type { StatewireThread } from '@assistant-ui/react-statewire';\nimport { useAuiState } from '@assistant-ui/store';\nimport { useEffect, useMemo, useRef, useState, useSyncExternalStore } from 'react';\nimport {\n  DEEP_AGENT_FOLLOW_TOOL_CALL_ID_KEY,\n  DEEP_AGENT_INTEREST_PROTOCOL,\n  DEEP_AGENT_LANE_FOLLOW_PROTOCOL_VERSION,\n  DEEP_AGENT_MAIN_NS,\n  DEEP_AGENT_MAX_FOLLOWED_STEPS,\n  DEEP_AGENT_MAX_INTERESTS,\n  DEEP_AGENT_RUNNING_LANE_WIRE_MESSAGES,\n  DEEP_AGENT_TRUNCATED_LANE_KEY,\n} from './generated/runconfig-contract';\nimport {\n  type DeepAgentInterestRequest,\n  type DeepAgentMessageWindow,\n  readDeepAgentMessageWindow,\n} from './message-window';\nimport { readDeepAgentProtocol } from './protocol';\nimport { isRunningToolPlaceholder } from './running-tool-placeholder';\nimport { DEEP_AGENT_STATEWIRE_PROTOCOL } from './statewire-transport';\nimport { hasStatewireThreadExtras } from './thread-extras';\n\ntype MountedDocuments = Parameters<typeof readDeepAgentMessageWindow>[0];\n\nconst isRecord = (value: unknown): value is Record<string, unknown> =>\n  typeof value === 'object' && value !== null && !Array.isArray(value);\n\nconst isCount = (value: unknown): value is number =>\n  typeof value === 'number' && Number.isInteger(value) && value >= 0;\n\nconst stepId = (step: unknown): string | undefined => {\n  const id = isRecord(step) ? step.id : undefined;\n  return typeof id === 'string' ? id : undefined;\n};\n\n/** A request following the lane one `task` card holds. */\nexport interface DeepAgentLaneFollowRequest extends DeepAgentInterestRequest {\n  readonly at: string;\n  readonly follow: { readonly before: number };\n}\n\n/**\n * The request following the lane of the running `task` card `cardId`: the card\n * alone (`before: 1`), so its window names no other card's lane, with as many\n * of its lane's newest steps as one attach follows.\n */\nexport function laneFollowRequest(cardId: string): DeepAgentLaneFollowRequest {\n  return {\n    ns: DEEP_AGENT_MAIN_NS,\n    before: 1,\n    at: cardId,\n    follow: { before: DEEP_AGENT_MAX_FOLLOWED_STEPS },\n  };\n}\n\nexport function isLaneFollowRequest(\n  request: DeepAgentInterestRequest\n): request is DeepAgentLaneFollowRequest {\n  return request.follow !== undefined && request.at !== undefined;\n}\n\n/** What an attach follows, read off its mounted documents. */\nexport interface DeepAgentLaneFollows {\n  /** The follow requests the attach holds whose card is still in the transcript. */\n  readonly requests: readonly DeepAgentLaneFollowRequest[];\n  /** Each followed lane's mounted steps, oldest first, by the `task` call whose card holds it. */\n  readonly lanes: ReadonlyMap<string, readonly LangChainMessage[]>;\n}\n\nconst NO_FOLLOWS: DeepAgentLaneFollows = { requests: [], lanes: new Map() };\n\n/**\n * A follow request as the host echoes it, or `null` for any other window. A\n * card that left the transcript (its result replaced the running placeholder)\n * leaves its window empty in place: there is nothing left to follow.\n */\nfunction followRequestOf(window: Record<string, unknown>): DeepAgentLaneFollowRequest | null {\n  const { ns, before, at, follow, chain } = window;\n  if (ns !== DEEP_AGENT_MAIN_NS || typeof at !== 'string' || !isRecord(follow)) return null;\n  if (!Array.isArray(chain) || chain.length === 0) return null;\n  if (!isCount(before) || !isCount(follow.before)) return null;\n  return { ns: DEEP_AGENT_MAIN_NS, before, at, follow: { before: follow.before } };\n}\n\n/**\n * The attach's follow requests and followed lanes. A step whose document is\n * not mounted is left out; a lane two windows follow is read from the deeper\n * one. A malformed window or step reads as absent.\n */\nexport function readDeepAgentLaneFollows(mounted: MountedDocuments): DeepAgentLaneFollows {\n  if (mounted === undefined) return NO_FOLLOWS;\n  let windows: readonly unknown[] | undefined;\n  const steps = new Map<string, Map<string, LangChainMessage>>();\n  for (const document of mounted.values()) {\n    const key = document.key;\n    if (document.protocol === DEEP_AGENT_INTEREST_PROTOCOL && document.main === true) {\n      const value = document.value;\n      if (isRecord(value) && Array.isArray(value.windows)) windows = value.windows;\n      continue;\n    }\n    if (document.protocol !== DEEP_AGENT_STATEWIRE_PROTOCOL || key === null || key.length !== 3) {\n      continue;\n    }\n    const [, ns, id] = key;\n    if (typeof ns !== 'string' || ns === DEEP_AGENT_MAIN_NS || typeof id !== 'string') continue;\n    let lane = steps.get(ns);\n    if (lane === undefined) {\n      lane = new Map();\n      steps.set(ns, lane);\n    }\n    lane.set(id, document.value as LangChainMessage);\n  }\n  if (windows === undefined) return NO_FOLLOWS;\n  const requests: DeepAgentLaneFollowRequest[] = [];\n  const lanes = new Map<string, readonly LangChainMessage[]>();\n  for (const window of windows) {\n    if (!isRecord(window)) continue;\n    const request = followRequestOf(window);\n    if (request !== null) requests.push(request);\n    if (!isRecord(window.follows)) continue;\n    for (const [ns, follow] of Object.entries(window.follows)) {\n      if (!isRecord(follow) || !Array.isArray(follow.chain)) continue;\n      const toolCallId = follow[DEEP_AGENT_FOLLOW_TOOL_CALL_ID_KEY];\n      if (typeof toolCallId !== 'string') continue;\n      const lane = steps.get(ns);\n      const ordered: LangChainMessage[] = [];\n      for (const entry of follow.chain) {\n        const id = stepId(entry);\n        const step = id === undefined ? undefined : lane?.get(id);\n        if (step !== undefined) ordered.push(step);\n      }\n      if (ordered.length > (lanes.get(toolCallId)?.length ?? 0)) lanes.set(toolCallId, ordered);\n    }\n  }\n  return requests.length === 0 && lanes.size === 0 ? NO_FOLLOWS : { requests, lanes };\n}\n\n/**\n * A running card's transcript with its followed lane read in. A cut card\n * carries its lane's newest steps whole, led by its first `cutTurns` entries:\n * the older assistant turns that requested results among them, cut to those\n * calls. The merge is the followed steps older than the card's newest, then\n * the card's newest steps, then any followed step newer than all of them. The\n * card's newest copies are the fuller ones (a lane step's document leaves every\n * inline image a placeholder), and only the card carries a step without an id,\n * which no follow addresses; the follow holds the cut turns whole, with the\n * steps between them and the newest, and a cut turn it does not hold leads.\n */\nexport function mergeFollowedLane(\n  carried: readonly unknown[],\n  followed: readonly LangChainMessage[],\n  cutTurns = 0\n): readonly unknown[] {\n  const position = new Map<string, number>();\n  for (const [index, step] of followed.entries()) {\n    const id = stepId(step);\n    if (id !== undefined) position.set(id, index);\n  }\n  const newest = carried.slice(cutTurns);\n  const turns = carried.slice(0, cutTurns).filter((turn) => {\n    const id = stepId(turn);\n    return id === undefined || !position.has(id);\n  });\n  let first = followed.length;\n  let last = -1;\n  for (const step of newest) {\n    const id = stepId(step);\n    const at = id === undefined ? undefined : position.get(id);\n    if (at === undefined) continue;\n    first = Math.min(first, at);\n    last = Math.max(last, at);\n  }\n  if (last < 0) {\n    return carried.length === 0 ? followed : [...turns, ...followed, ...newest];\n  }\n  return [...turns, ...followed.slice(0, first), ...newest, ...followed.slice(last + 1)];\n}\n\n/** How many of a running card's carried entries are cut assistant turns ahead of its newest steps. */\nfunction cutTurnsOf(message: LangChainMessage, carried: readonly unknown[]): number {\n  const marker = (message as Record<string, unknown>)[DEEP_AGENT_TRUNCATED_LANE_KEY];\n  return isRecord(marker) ? Math.max(0, carried.length - DEEP_AGENT_RUNNING_LANE_WIRE_MESSAGES) : 0;\n}\n\nconst sameItems = (a: readonly unknown[], b: readonly unknown[]): boolean =>\n  a.length === b.length && a.every((item, index) => item === b[index]);\n\n/** Each card read with its followed lane, by the card object, while its steps stand. */\nconst followedCards = new WeakMap<\n  object,\n  { readonly steps: readonly LangChainMessage[]; readonly card: LangChainMessage }\n>();\n\nfunction withFollowedLane(\n  message: LangChainMessage,\n  lanes: ReadonlyMap<string, readonly LangChainMessage[]>\n): LangChainMessage {\n  if (message.type !== 'tool' || !isRunningToolPlaceholder(message)) return message;\n  const steps = lanes.get(message.tool_call_id);\n  if (steps === undefined || steps.length === 0) return message;\n  const kept = followedCards.get(message);\n  if (kept !== undefined && sameItems(kept.steps, steps)) return kept.card;\n  const artifact: Record<string, unknown> = isRecord(message.artifact) ? message.artifact : {};\n  const state = isRecord(artifact.subgraph_state) ? artifact.subgraph_state : {};\n  const carried: readonly unknown[] = Array.isArray(state.messages) ? state.messages : [];\n  const card: LangChainMessage = {\n    ...message,\n    artifact: {\n      ...artifact,\n      subgraph_state: {\n        ...state,\n        messages: mergeFollowedLane(carried, steps, cutTurnsOf(message, carried)),\n      },\n    },\n  };\n  followedCards.set(message, { steps, card });\n  return card;\n}\n\n/**\n * {@link readDeepAgentMessageWindow} with the attach's followed lanes read in.\n * The follow requests ride at the end of `requests`, so a restatement built\n * from this window (the thread's head widening, a page of older history) keeps\n * them: the joined view never counts them as pages. Each running card a follow\n * covers carries its lane's followed steps as its transcript, kept as the same\n * object while its steps stand. Without follows, the window is\n * {@link readDeepAgentMessageWindow}'s own.\n */\nexport function readDeepAgentFollowedWindow(\n  mounted: MountedDocuments\n): DeepAgentMessageWindow | null {\n  const window = readDeepAgentMessageWindow(mounted);\n  if (window === null) return null;\n  const follows = readDeepAgentLaneFollows(mounted);\n  if (follows === NO_FOLLOWS) return window;\n  return {\n    ...window,\n    requests: [...window.requests, ...follows.requests],\n    messages:\n      follows.lanes.size === 0\n        ? window.messages\n        : window.messages.map((message) => withFollowedLane(message, follows.lanes)),\n  };\n}\n\n/**\n * The lanes a surface shows (an open transcript drawer's), by the `task` call\n * whose card holds each: what {@link useDeepAgentLaneFollowSync} follows.\n */\nexport interface DeepAgentLaneFollowRegistry {\n  /** Follow the lane of the `task` call `toolCallId` until the returned release is called. */\n  follow(toolCallId: string): () => void;\n  subscribe(listener: () => void): () => void;\n  /** The `task` calls whose lanes are followed; a new set on every change. */\n  getSnapshot(): ReadonlySet<string>;\n}\n\nexport function createDeepAgentLaneFollowRegistry(): DeepAgentLaneFollowRegistry {\n  const holds = new Map<string, number>();\n  const listeners = new Set<() => void>();\n  let snapshot: ReadonlySet<string> = new Set();\n  const publish = () => {\n    snapshot = new Set(holds.keys());\n    for (const listener of listeners) listener();\n  };\n  return {\n    follow(toolCallId) {\n      const held = holds.get(toolCallId) ?? 0;\n      holds.set(toolCallId, held + 1);\n      if (held === 0) publish();\n      let released = false;\n      return () => {\n        if (released) return;\n        released = true;\n        const left = (holds.get(toolCallId) ?? 1) - 1;\n        if (left > 0) {\n          holds.set(toolCallId, left);\n          return;\n        }\n        holds.delete(toolCallId);\n        publish();\n      };\n    },\n    subscribe(listener) {\n      listeners.add(listener);\n      return () => {\n        listeners.delete(listener);\n      };\n    },\n    getSnapshot: () => snapshot,\n  };\n}\n\n/** Follows the lane of `toolCallId` while `active` (a drawer open on a running lane). */\nexport function useDeepAgentLaneFollow(\n  registry: DeepAgentLaneFollowRegistry | null,\n  toolCallId: string,\n  active: boolean\n): void {\n  useEffect(() => {\n    if (registry === null || !active) return undefined;\n    return registry.follow(toolCallId);\n  }, [registry, toolCallId, active]);\n}\n\n/** The follow requests for the running cards of the followed lanes, in transcript order. */\nfunction laneFollowRequestsFor(\n  window: DeepAgentMessageWindow,\n  followed: ReadonlySet<string>\n): DeepAgentLaneFollowRequest[] {\n  if (followed.size === 0) return [];\n  const requests: DeepAgentLaneFollowRequest[] = [];\n  for (const message of window.messages) {\n    if (message.type !== 'tool' || !followed.has(message.tool_call_id)) continue;\n    if (typeof message.id !== 'string' || !isRunningToolPlaceholder(message)) continue;\n    requests.push(laneFollowRequest(message.id));\n  }\n  return requests;\n}\n\nconst followKey = (request: DeepAgentInterestRequest): string =>\n  `${request.at}:${request.follow?.before}`;\n\nconst sameLaneFollows = (\n  held: readonly DeepAgentInterestRequest[],\n  wanted: readonly DeepAgentInterestRequest[]\n): boolean =>\n  held.length === wanted.length &&\n  wanted.every((request) => held.some((have) => followKey(have) === followKey(request)));\n\n/** The code a host refused a statement with (`StatewireSendError.failure`), if any. */\nconst rejectionCode = (error: unknown): string | undefined => {\n  const failure = isRecord(error) ? error.failure : undefined;\n  return isRecord(failure) && failure.fate === 'rejected' && typeof failure.code === 'string'\n    ? failure.code\n    : undefined;\n};\n\nconst noopSubscribe = () => () => {};\n\nexport interface UseDeepAgentLaneFollowSyncOptions {\n  /** The lanes the surface shows. */\n  registry: DeepAgentLaneFollowRegistry;\n  /**\n   * Another statement on the interest lane is in flight (a page of older\n   * history): wait for it, so neither statement drops what the other adds.\n   */\n  paused?: boolean;\n  /**\n   * A follow statement failed. A host that does not serve `follow` is not a\n   * failure: the cards keep the lanes their own documents carry.\n   */\n  onError?: ((error: unknown) => void) | undefined;\n}\n\n/**\n * Keeps the attach's follow requests to the running lanes the registry holds,\n * once per change, on a host whose `protocol` lane advertises at least\n * `DEEP_AGENT_LANE_FOLLOW_PROTOCOL_VERSION`. A statement restates the\n * attach's other requests as the window holds them, with the follows after\n * them, inside the host's interest cap; a statement elsewhere that left a\n * follow out (a page of older history) is answered by stating it again.\n *\n * Never a retry loop: a statement is not sent again unchanged on the same\n * connection. A host that refuses `follow` (`invalid-interest`, one that\n * predates it) keeps the attach's windows, and nothing is followed on that\n * connection again; a card that left the transcript first (`unknown-message`)\n * is not followed until the transcript moves.\n */\nexport function useDeepAgentLaneFollowSync({\n  registry,\n  paused = false,\n  onError,\n}: UseDeepAgentLaneFollowSyncOptions): void {\n  const { invoke, connection, documents } = useAuiState((s) => {\n    if (!hasStatewireThreadExtras(s.thread.extras)) {\n      throw new Error('useDeepAgentLaneFollowSync requires a statewire thread');\n    }\n    return s.thread.extras as StatewireThread.Extras;\n  });\n  const hostVersion = useAuiState((s) => {\n    const extras = s.thread.extras as Partial<StatewireThread.Extras> | undefined;\n    return readDeepAgentProtocol(extras?.state)?.version ?? 0;\n  });\n  const mounted = useSyncExternalStore(\n    documents?.subscribe ?? noopSubscribe,\n    () => documents?.list(),\n    () => documents?.list()\n  );\n  const window = useMemo(() => readDeepAgentFollowedWindow(mounted), [mounted]);\n  const followed = useSyncExternalStore(\n    registry.subscribe,\n    registry.getSnapshot,\n    registry.getSnapshot\n  );\n  const [settled, setSettled] = useState(0);\n  const inflightRef = useRef(false);\n  const aliveRef = useRef(true);\n  // The connection whose host refused `follow`, and the last statement sent on a connection.\n  const unservedRef = useRef<unknown>(null);\n  const lastStatedRef = useRef<{ connection: unknown; key: string } | null>(null);\n  const onErrorRef = useRef(onError);\n  onErrorRef.current = onError;\n\n  useEffect(() => {\n    aliveRef.current = true;\n    return () => {\n      aliveRef.current = false;\n    };\n  }, []);\n\n  // biome-ignore lint/correctness/useExhaustiveDependencies: `settled` re-runs the check once a statement settles.\n  useEffect(() => {\n    if (window === null || paused || inflightRef.current) return;\n    if (connection.status !== 'connected') return;\n    const served =\n      hostVersion >= DEEP_AGENT_LANE_FOLLOW_PROTOCOL_VERSION && unservedRef.current !== connection;\n    const others = window.requests.filter((request) => !isLaneFollowRequest(request));\n    const held = window.requests.filter(isLaneFollowRequest);\n    const wanted = served\n      ? laneFollowRequestsFor(window, followed).slice(\n          0,\n          Math.max(0, DEEP_AGENT_MAX_INTERESTS - others.length)\n        )\n      : [];\n    if (sameLaneFollows(held, wanted)) return;\n    const requests = [...others, ...wanted];\n    const key = JSON.stringify(requests);\n    const last = lastStatedRef.current;\n    if (last !== null && last.connection === connection && last.key === key) return;\n    lastStatedRef.current = { connection, key };\n    inflightRef.current = true;\n    invoke(DEEP_AGENT_INTEREST_PROTOCOL, 'set', [requests])\n      .catch((error: unknown) => {\n        const code = rejectionCode(error);\n        if (code === 'invalid-interest' && wanted.length > 0) {\n          unservedRef.current = connection;\n          return;\n        }\n        if (code !== 'unknown-message') onErrorRef.current?.(error);\n      })\n      .finally(() => {\n        inflightRef.current = false;\n        if (aliveRef.current) setSettled((count) => count + 1);\n      });\n  }, [window, followed, hostVersion, connection, invoke, paused, settled]);\n}\n"],"names":["DEEP_AGENT_MAIN_NS","DEEP_AGENT_MAX_FOLLOWED_STEPS","DEEP_AGENT_INTEREST_PROTOCOL","DEEP_AGENT_STATEWIRE_PROTOCOL","DEEP_AGENT_FOLLOW_TOOL_CALL_ID_KEY","DEEP_AGENT_TRUNCATED_LANE_KEY","DEEP_AGENT_RUNNING_LANE_WIRE_MESSAGES","isRunningToolPlaceholder","readDeepAgentMessageWindow","useEffect","useAuiState","hasStatewireThreadExtras","readDeepAgentProtocol","useSyncExternalStore","useMemo","useState","useRef","DEEP_AGENT_LANE_FOLLOW_PROTOCOL_VERSION","DEEP_AGENT_MAX_INTERESTS"],"mappings":";;;;;;;;;;AAmDA,MAAM,WAAW,CAAC,UAChB,OAAO,UAAU,YAAY,UAAU,QAAQ,CAAC,MAAM,QAAQ,KAAK;AAErE,MAAM,UAAU,CAAC,UACf,OAAO,UAAU,YAAY,OAAO,UAAU,KAAK,KAAK,SAAS;AAEnE,MAAM,SAAS,CAAC,SAAsC;AACpD,QAAM,KAAK,SAAS,IAAI,IAAI,KAAK,KAAK;AACtC,SAAO,OAAO,OAAO,WAAW,KAAK;AACvC;AAaO,SAAS,kBAAkB,QAA4C;AAC5E,SAAO;AAAA,IACL,IAAIA,kBAAAA;AAAAA,IACJ,QAAQ;AAAA,IACR,IAAI;AAAA,IACJ,QAAQ,EAAE,QAAQC,kBAAAA,8BAAA;AAAA,EAA8B;AAEpD;AAEO,SAAS,oBACd,SACuC;AACvC,SAAO,QAAQ,WAAW,UAAa,QAAQ,OAAO;AACxD;AAUA,MAAM,aAAmC,EAAE,UAAU,CAAA,GAAI,OAAO,oBAAI,MAAI;AAOxE,SAAS,gBAAgB,QAAoE;AAC3F,QAAM,EAAE,IAAI,QAAQ,IAAI,QAAQ,UAAU;AAC1C,MAAI,OAAOD,kBAAAA,sBAAsB,OAAO,OAAO,YAAY,CAAC,SAAS,MAAM,EAAG,QAAO;AACrF,MAAI,CAAC,MAAM,QAAQ,KAAK,KAAK,MAAM,WAAW,EAAG,QAAO;AACxD,MAAI,CAAC,QAAQ,MAAM,KAAK,CAAC,QAAQ,OAAO,MAAM,EAAG,QAAO;AACxD,SAAO,EAAE,IAAIA,kBAAAA,oBAAoB,QAAQ,IAAI,QAAQ,EAAE,QAAQ,OAAO,SAAO;AAC/E;AAOO,SAAS,yBAAyB,SAAiD;;AACxF,MAAI,YAAY,OAAW,QAAO;AAClC,MAAI;AACJ,QAAM,4BAAY,IAAA;AAClB,aAAW,YAAY,QAAQ,UAAU;AACvC,UAAM,MAAM,SAAS;AACrB,QAAI,SAAS,aAAaE,kBAAAA,gCAAgC,SAAS,SAAS,MAAM;AAChF,YAAM,QAAQ,SAAS;AACvB,UAAI,SAAS,KAAK,KAAK,MAAM,QAAQ,MAAM,OAAO,EAAG,WAAU,MAAM;AACrE;AAAA,IACF;AACA,QAAI,SAAS,aAAaC,oDAAiC,QAAQ,QAAQ,IAAI,WAAW,GAAG;AAC3F;AAAA,IACF;AACA,UAAM,CAAA,EAAG,IAAI,EAAE,IAAI;AACnB,QAAI,OAAO,OAAO,YAAY,OAAOH,kBAAAA,sBAAsB,OAAO,OAAO,SAAU;AACnF,QAAI,OAAO,MAAM,IAAI,EAAE;AACvB,QAAI,SAAS,QAAW;AACtB,iCAAW,IAAA;AACX,YAAM,IAAI,IAAI,IAAI;AAAA,IACpB;AACA,SAAK,IAAI,IAAI,SAAS,KAAyB;AAAA,EACjD;AACA,MAAI,YAAY,OAAW,QAAO;AAClC,QAAM,WAAyC,CAAA;AAC/C,QAAM,4BAAY,IAAA;AAClB,aAAW,UAAU,SAAS;AAC5B,QAAI,CAAC,SAAS,MAAM,EAAG;AACvB,UAAM,UAAU,gBAAgB,MAAM;AACtC,QAAI,YAAY,KAAM,UAAS,KAAK,OAAO;AAC3C,QAAI,CAAC,SAAS,OAAO,OAAO,EAAG;AAC/B,eAAW,CAAC,IAAI,MAAM,KAAK,OAAO,QAAQ,OAAO,OAAO,GAAG;AACzD,UAAI,CAAC,SAAS,MAAM,KAAK,CAAC,MAAM,QAAQ,OAAO,KAAK,EAAG;AACvD,YAAM,aAAa,OAAOI,oDAAkC;AAC5D,UAAI,OAAO,eAAe,SAAU;AACpC,YAAM,OAAO,MAAM,IAAI,EAAE;AACzB,YAAM,UAA8B,CAAA;AACpC,iBAAW,SAAS,OAAO,OAAO;AAChC,cAAM,KAAK,OAAO,KAAK;AACvB,cAAM,OAAO,OAAO,SAAY,SAAY,6BAAM,IAAI;AACtD,YAAI,SAAS,OAAW,SAAQ,KAAK,IAAI;AAAA,MAC3C;AACA,UAAI,QAAQ,YAAU,WAAM,IAAI,UAAU,MAApB,mBAAuB,WAAU,GAAI,OAAM,IAAI,YAAY,OAAO;AAAA,IAC1F;AAAA,EACF;AACA,SAAO,SAAS,WAAW,KAAK,MAAM,SAAS,IAAI,aAAa,EAAE,UAAU,MAAA;AAC9E;AAaO,SAAS,kBACd,SACA,UACA,WAAW,GACS;AACpB,QAAM,+BAAe,IAAA;AACrB,aAAW,CAAC,OAAO,IAAI,KAAK,SAAS,WAAW;AAC9C,UAAM,KAAK,OAAO,IAAI;AACtB,QAAI,OAAO,OAAW,UAAS,IAAI,IAAI,KAAK;AAAA,EAC9C;AACA,QAAM,SAAS,QAAQ,MAAM,QAAQ;AACrC,QAAM,QAAQ,QAAQ,MAAM,GAAG,QAAQ,EAAE,OAAO,CAAC,SAAS;AACxD,UAAM,KAAK,OAAO,IAAI;AACtB,WAAO,OAAO,UAAa,CAAC,SAAS,IAAI,EAAE;AAAA,EAC7C,CAAC;AACD,MAAI,QAAQ,SAAS;AACrB,MAAI,OAAO;AACX,aAAW,QAAQ,QAAQ;AACzB,UAAM,KAAK,OAAO,IAAI;AACtB,UAAM,KAAK,OAAO,SAAY,SAAY,SAAS,IAAI,EAAE;AACzD,QAAI,OAAO,OAAW;AACtB,YAAQ,KAAK,IAAI,OAAO,EAAE;AAC1B,WAAO,KAAK,IAAI,MAAM,EAAE;AAAA,EAC1B;AACA,MAAI,OAAO,GAAG;AACZ,WAAO,QAAQ,WAAW,IAAI,WAAW,CAAC,GAAG,OAAO,GAAG,UAAU,GAAG,MAAM;AAAA,EAC5E;AACA,SAAO,CAAC,GAAG,OAAO,GAAG,SAAS,MAAM,GAAG,KAAK,GAAG,GAAG,QAAQ,GAAG,SAAS,MAAM,OAAO,CAAC,CAAC;AACvF;AAGA,SAAS,WAAW,SAA2B,SAAqC;AAClF,QAAM,SAAU,QAAoCC,+CAA6B;AACjF,SAAO,SAAS,MAAM,IAAI,KAAK,IAAI,GAAG,QAAQ,SAASC,kBAAAA,qCAAqC,IAAI;AAClG;AAEA,MAAM,YAAY,CAAC,GAAuB,MACxC,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,CAAC,MAAM,UAAU,SAAS,EAAE,KAAK,CAAC;AAGrE,MAAM,oCAAoB,QAAA;AAK1B,SAAS,iBACP,SACA,OACkB;AAClB,MAAI,QAAQ,SAAS,UAAU,CAACC,uBAAAA,yBAAyB,OAAO,EAAG,QAAO;AAC1E,QAAM,QAAQ,MAAM,IAAI,QAAQ,YAAY;AAC5C,MAAI,UAAU,UAAa,MAAM,WAAW,EAAG,QAAO;AACtD,QAAM,OAAO,cAAc,IAAI,OAAO;AACtC,MAAI,SAAS,UAAa,UAAU,KAAK,OAAO,KAAK,UAAU,KAAK;AACpE,QAAM,WAAoC,SAAS,QAAQ,QAAQ,IAAI,QAAQ,WAAW,CAAA;AAC1F,QAAM,QAAQ,SAAS,SAAS,cAAc,IAAI,SAAS,iBAAiB,CAAA;AAC5E,QAAM,UAA8B,MAAM,QAAQ,MAAM,QAAQ,IAAI,MAAM,WAAW,CAAA;AACrF,QAAM,OAAyB;AAAA,IAC7B,GAAG;AAAA,IACH,UAAU;AAAA,MACR,GAAG;AAAA,MACH,gBAAgB;AAAA,QACd,GAAG;AAAA,QACH,UAAU,kBAAkB,SAAS,OAAO,WAAW,SAAS,OAAO,CAAC;AAAA,MAAA;AAAA,IAC1E;AAAA,EACF;AAEF,gBAAc,IAAI,SAAS,EAAE,OAAO,MAAM;AAC1C,SAAO;AACT;AAWO,SAAS,4BACd,SAC+B;AAC/B,QAAM,SAASC,cAAAA,2BAA2B,OAAO;AACjD,MAAI,WAAW,KAAM,QAAO;AAC5B,QAAM,UAAU,yBAAyB,OAAO;AAChD,MAAI,YAAY,WAAY,QAAO;AACnC,SAAO;AAAA,IACL,GAAG;AAAA,IACH,UAAU,CAAC,GAAG,OAAO,UAAU,GAAG,QAAQ,QAAQ;AAAA,IAClD,UACE,QAAQ,MAAM,SAAS,IACnB,OAAO,WACP,OAAO,SAAS,IAAI,CAAC,YAAY,iBAAiB,SAAS,QAAQ,KAAK,CAAC;AAAA,EAAA;AAEnF;AAcO,SAAS,oCAAiE;AAC/E,QAAM,4BAAY,IAAA;AAClB,QAAM,gCAAgB,IAAA;AACtB,MAAI,+BAAoC,IAAA;AACxC,QAAM,UAAU,MAAM;AACpB,eAAW,IAAI,IAAI,MAAM,KAAA,CAAM;AAC/B,eAAW,YAAY,UAAW,UAAA;AAAA,EACpC;AACA,SAAO;AAAA,IACL,OAAO,YAAY;AACjB,YAAM,OAAO,MAAM,IAAI,UAAU,KAAK;AACtC,YAAM,IAAI,YAAY,OAAO,CAAC;AAC9B,UAAI,SAAS,EAAG,SAAA;AAChB,UAAI,WAAW;AACf,aAAO,MAAM;AACX,YAAI,SAAU;AACd,mBAAW;AACX,cAAM,QAAQ,MAAM,IAAI,UAAU,KAAK,KAAK;AAC5C,YAAI,OAAO,GAAG;AACZ,gBAAM,IAAI,YAAY,IAAI;AAC1B;AAAA,QACF;AACA,cAAM,OAAO,UAAU;AACvB,gBAAA;AAAA,MACF;AAAA,IACF;AAAA,IACA,UAAU,UAAU;AAClB,gBAAU,IAAI,QAAQ;AACtB,aAAO,MAAM;AACX,kBAAU,OAAO,QAAQ;AAAA,MAC3B;AAAA,IACF;AAAA,IACA,aAAa,MAAM;AAAA,EAAA;AAEvB;AAGO,SAAS,uBACd,UACA,YACA,QACM;AACNC,aAAAA,UAAU,MAAM;AACd,QAAI,aAAa,QAAQ,CAAC,OAAQ,QAAO;AACzC,WAAO,SAAS,OAAO,UAAU;AAAA,EACnC,GAAG,CAAC,UAAU,YAAY,MAAM,CAAC;AACnC;AAGA,SAAS,sBACP,QACA,UAC8B;AAC9B,MAAI,SAAS,SAAS,EAAG,QAAO,CAAA;AAChC,QAAM,WAAyC,CAAA;AAC/C,aAAW,WAAW,OAAO,UAAU;AACrC,QAAI,QAAQ,SAAS,UAAU,CAAC,SAAS,IAAI,QAAQ,YAAY,EAAG;AACpE,QAAI,OAAO,QAAQ,OAAO,YAAY,CAACF,uBAAAA,yBAAyB,OAAO,EAAG;AAC1E,aAAS,KAAK,kBAAkB,QAAQ,EAAE,CAAC;AAAA,EAC7C;AACA,SAAO;AACT;AAEA,MAAM,YAAY,CAAC,YAAA;;AACjB,YAAG,QAAQ,EAAE,KAAI,aAAQ,WAAR,mBAAgB,MAAM;AAAA;AAEzC,MAAM,kBAAkB,CACtB,MACA,WAEA,KAAK,WAAW,OAAO,UACvB,OAAO,MAAM,CAAC,YAAY,KAAK,KAAK,CAAC,SAAS,UAAU,IAAI,MAAM,UAAU,OAAO,CAAC,CAAC;AAGvF,MAAM,gBAAgB,CAAC,UAAuC;AAC5D,QAAM,UAAU,SAAS,KAAK,IAAI,MAAM,UAAU;AAClD,SAAO,SAAS,OAAO,KAAK,QAAQ,SAAS,cAAc,OAAO,QAAQ,SAAS,WAC/E,QAAQ,OACR;AACN;AAEA,MAAM,gBAAgB,MAAM,MAAM;AAAC;AA+B5B,SAAS,2BAA2B;AAAA,EACzC;AAAA,EACA,SAAS;AAAA,EACT;AACF,GAA4C;AAC1C,QAAM,EAAE,QAAQ,YAAY,cAAcG,MAAAA,YAAY,CAAC,MAAM;AAC3D,QAAI,CAACC,aAAAA,yBAAyB,EAAE,OAAO,MAAM,GAAG;AAC9C,YAAM,IAAI,MAAM,wDAAwD;AAAA,IAC1E;AACA,WAAO,EAAE,OAAO;AAAA,EAClB,CAAC;AACD,QAAM,cAAcD,kBAAY,CAAC,MAAM;;AACrC,UAAM,SAAS,EAAE,OAAO;AACxB,aAAOE,cAAAA,sBAAsB,iCAAQ,KAAK,MAAnCA,mBAAsC,YAAW;AAAA,EAC1D,CAAC;AACD,QAAM,UAAUC,WAAAA;AAAAA,KACd,uCAAW,cAAa;AAAA,IACxB,MAAM,uCAAW;AAAA,IACjB,MAAM,uCAAW;AAAA,EAAK;AAExB,QAAM,SAASC,WAAAA,QAAQ,MAAM,4BAA4B,OAAO,GAAG,CAAC,OAAO,CAAC;AAC5E,QAAM,WAAWD,WAAAA;AAAAA,IACf,SAAS;AAAA,IACT,SAAS;AAAA,IACT,SAAS;AAAA,EAAA;AAEX,QAAM,CAAC,SAAS,UAAU,IAAIE,WAAAA,SAAS,CAAC;AACxC,QAAM,cAAcC,WAAAA,OAAO,KAAK;AAChC,QAAM,WAAWA,WAAAA,OAAO,IAAI;AAE5B,QAAM,cAAcA,WAAAA,OAAgB,IAAI;AACxC,QAAM,gBAAgBA,WAAAA,OAAoD,IAAI;AAC9E,QAAM,aAAaA,WAAAA,OAAO,OAAO;AACjC,aAAW,UAAU;AAErBP,aAAAA,UAAU,MAAM;AACd,aAAS,UAAU;AACnB,WAAO,MAAM;AACX,eAAS,UAAU;AAAA,IACrB;AAAA,EACF,GAAG,CAAA,CAAE;AAGLA,aAAAA,UAAU,MAAM;AACd,QAAI,WAAW,QAAQ,UAAU,YAAY,QAAS;AACtD,QAAI,WAAW,WAAW,YAAa;AACvC,UAAM,SACJ,eAAeQ,kBAAAA,2CAA2C,YAAY,YAAY;AACpF,UAAM,SAAS,OAAO,SAAS,OAAO,CAAC,YAAY,CAAC,oBAAoB,OAAO,CAAC;AAChF,UAAM,OAAO,OAAO,SAAS,OAAO,mBAAmB;AACvD,UAAM,SAAS,SACX,sBAAsB,QAAQ,QAAQ,EAAE;AAAA,MACtC;AAAA,MACA,KAAK,IAAI,GAAGC,kBAAAA,2BAA2B,OAAO,MAAM;AAAA,IAAA,IAEtD,CAAA;AACJ,QAAI,gBAAgB,MAAM,MAAM,EAAG;AACnC,UAAM,WAAW,CAAC,GAAG,QAAQ,GAAG,MAAM;AACtC,UAAM,MAAM,KAAK,UAAU,QAAQ;AACnC,UAAM,OAAO,cAAc;AAC3B,QAAI,SAAS,QAAQ,KAAK,eAAe,cAAc,KAAK,QAAQ,IAAK;AACzE,kBAAc,UAAU,EAAE,YAAY,IAAA;AACtC,gBAAY,UAAU;AACtB,WAAOhB,kBAAAA,8BAA8B,OAAO,CAAC,QAAQ,CAAC,EACnD,MAAM,CAAC,UAAmB;;AACzB,YAAM,OAAO,cAAc,KAAK;AAChC,UAAI,SAAS,sBAAsB,OAAO,SAAS,GAAG;AACpD,oBAAY,UAAU;AACtB;AAAA,MACF;AACA,UAAI,SAAS,kBAAmB,kBAAW,YAAX,oCAAqB;AAAA,IACvD,CAAC,EACA,QAAQ,MAAM;AACb,kBAAY,UAAU;AACtB,UAAI,SAAS,QAAS,YAAW,CAAC,UAAU,QAAQ,CAAC;AAAA,IACvD,CAAC;AAAA,EACL,GAAG,CAAC,QAAQ,UAAU,aAAa,YAAY,QAAQ,QAAQ,OAAO,CAAC;AACzE;;;;;;;;;"}

package/dist/agent-runtime/src/lane-follow.js

@@ -0,0 +1,289 @@
+import { useAuiState } from "@assistant-ui/store";
+import { useSyncExternalStore, useMemo, useState, useRef, useEffect } from "react";
+import { DEEP_AGENT_LANE_FOLLOW_PROTOCOL_VERSION, DEEP_AGENT_MAX_INTERESTS, DEEP_AGENT_INTEREST_PROTOCOL, DEEP_AGENT_MAIN_NS, DEEP_AGENT_FOLLOW_TOOL_CALL_ID_KEY, DEEP_AGENT_MAX_FOLLOWED_STEPS, DEEP_AGENT_TRUNCATED_LANE_KEY, DEEP_AGENT_RUNNING_LANE_WIRE_MESSAGES } from "./generated/runconfig-contract.js";
+import { readDeepAgentMessageWindow } from "./message-window.js";
+import { readDeepAgentProtocol } from "./protocol.js";
+import { isRunningToolPlaceholder } from "./running-tool-placeholder.js";
+import { DEEP_AGENT_STATEWIRE_PROTOCOL } from "./statewire-transport.js";
+import { hasStatewireThreadExtras } from "./thread-extras.js";
+const isRecord = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
+const isCount = (value) => typeof value === "number" && Number.isInteger(value) && value >= 0;
+const stepId = (step) => {
+  const id = isRecord(step) ? step.id : void 0;
+  return typeof id === "string" ? id : void 0;
+};
+function laneFollowRequest(cardId) {
+  return {
+    ns: DEEP_AGENT_MAIN_NS,
+    before: 1,
+    at: cardId,
+    follow: { before: DEEP_AGENT_MAX_FOLLOWED_STEPS }
+  };
+}
+function isLaneFollowRequest(request) {
+  return request.follow !== void 0 && request.at !== void 0;
+}
+const NO_FOLLOWS = { requests: [], lanes: /* @__PURE__ */ new Map() };
+function followRequestOf(window) {
+  const { ns, before, at, follow, chain } = window;
+  if (ns !== DEEP_AGENT_MAIN_NS || typeof at !== "string" || !isRecord(follow)) return null;
+  if (!Array.isArray(chain) || chain.length === 0) return null;
+  if (!isCount(before) || !isCount(follow.before)) return null;
+  return { ns: DEEP_AGENT_MAIN_NS, before, at, follow: { before: follow.before } };
+}
+function readDeepAgentLaneFollows(mounted) {
+  var _a;
+  if (mounted === void 0) return NO_FOLLOWS;
+  let windows;
+  const steps = /* @__PURE__ */ new Map();
+  for (const document of mounted.values()) {
+    const key = document.key;
+    if (document.protocol === DEEP_AGENT_INTEREST_PROTOCOL && document.main === true) {
+      const value = document.value;
+      if (isRecord(value) && Array.isArray(value.windows)) windows = value.windows;
+      continue;
+    }
+    if (document.protocol !== DEEP_AGENT_STATEWIRE_PROTOCOL || key === null || key.length !== 3) {
+      continue;
+    }
+    const [, ns, id] = key;
+    if (typeof ns !== "string" || ns === DEEP_AGENT_MAIN_NS || typeof id !== "string") continue;
+    let lane = steps.get(ns);
+    if (lane === void 0) {
+      lane = /* @__PURE__ */ new Map();
+      steps.set(ns, lane);
+    }
+    lane.set(id, document.value);
+  }
+  if (windows === void 0) return NO_FOLLOWS;
+  const requests = [];
+  const lanes = /* @__PURE__ */ new Map();
+  for (const window of windows) {
+    if (!isRecord(window)) continue;
+    const request = followRequestOf(window);
+    if (request !== null) requests.push(request);
+    if (!isRecord(window.follows)) continue;
+    for (const [ns, follow] of Object.entries(window.follows)) {
+      if (!isRecord(follow) || !Array.isArray(follow.chain)) continue;
+      const toolCallId = follow[DEEP_AGENT_FOLLOW_TOOL_CALL_ID_KEY];
+      if (typeof toolCallId !== "string") continue;
+      const lane = steps.get(ns);
+      const ordered = [];
+      for (const entry of follow.chain) {
+        const id = stepId(entry);
+        const step = id === void 0 ? void 0 : lane == null ? void 0 : lane.get(id);
+        if (step !== void 0) ordered.push(step);
+      }
+      if (ordered.length > (((_a = lanes.get(toolCallId)) == null ? void 0 : _a.length) ?? 0)) lanes.set(toolCallId, ordered);
+    }
+  }
+  return requests.length === 0 && lanes.size === 0 ? NO_FOLLOWS : { requests, lanes };
+}
+function mergeFollowedLane(carried, followed, cutTurns = 0) {
+  const position = /* @__PURE__ */ new Map();
+  for (const [index, step] of followed.entries()) {
+    const id = stepId(step);
+    if (id !== void 0) position.set(id, index);
+  }
+  const newest = carried.slice(cutTurns);
+  const turns = carried.slice(0, cutTurns).filter((turn) => {
+    const id = stepId(turn);
+    return id === void 0 || !position.has(id);
+  });
+  let first = followed.length;
+  let last = -1;
+  for (const step of newest) {
+    const id = stepId(step);
+    const at = id === void 0 ? void 0 : position.get(id);
+    if (at === void 0) continue;
+    first = Math.min(first, at);
+    last = Math.max(last, at);
+  }
+  if (last < 0) {
+    return carried.length === 0 ? followed : [...turns, ...followed, ...newest];
+  }
+  return [...turns, ...followed.slice(0, first), ...newest, ...followed.slice(last + 1)];
+}
+function cutTurnsOf(message, carried) {
+  const marker = message[DEEP_AGENT_TRUNCATED_LANE_KEY];
+  return isRecord(marker) ? Math.max(0, carried.length - DEEP_AGENT_RUNNING_LANE_WIRE_MESSAGES) : 0;
+}
+const sameItems = (a, b) => a.length === b.length && a.every((item, index) => item === b[index]);
+const followedCards = /* @__PURE__ */ new WeakMap();
+function withFollowedLane(message, lanes) {
+  if (message.type !== "tool" || !isRunningToolPlaceholder(message)) return message;
+  const steps = lanes.get(message.tool_call_id);
+  if (steps === void 0 || steps.length === 0) return message;
+  const kept = followedCards.get(message);
+  if (kept !== void 0 && sameItems(kept.steps, steps)) return kept.card;
+  const artifact = isRecord(message.artifact) ? message.artifact : {};
+  const state = isRecord(artifact.subgraph_state) ? artifact.subgraph_state : {};
+  const carried = Array.isArray(state.messages) ? state.messages : [];
+  const card = {
+    ...message,
+    artifact: {
+      ...artifact,
+      subgraph_state: {
+        ...state,
+        messages: mergeFollowedLane(carried, steps, cutTurnsOf(message, carried))
+      }
+    }
+  };
+  followedCards.set(message, { steps, card });
+  return card;
+}
+function readDeepAgentFollowedWindow(mounted) {
+  const window = readDeepAgentMessageWindow(mounted);
+  if (window === null) return null;
+  const follows = readDeepAgentLaneFollows(mounted);
+  if (follows === NO_FOLLOWS) return window;
+  return {
+    ...window,
+    requests: [...window.requests, ...follows.requests],
+    messages: follows.lanes.size === 0 ? window.messages : window.messages.map((message) => withFollowedLane(message, follows.lanes))
+  };
+}
+function createDeepAgentLaneFollowRegistry() {
+  const holds = /* @__PURE__ */ new Map();
+  const listeners = /* @__PURE__ */ new Set();
+  let snapshot = /* @__PURE__ */ new Set();
+  const publish = () => {
+    snapshot = new Set(holds.keys());
+    for (const listener of listeners) listener();
+  };
+  return {
+    follow(toolCallId) {
+      const held = holds.get(toolCallId) ?? 0;
+      holds.set(toolCallId, held + 1);
+      if (held === 0) publish();
+      let released = false;
+      return () => {
+        if (released) return;
+        released = true;
+        const left = (holds.get(toolCallId) ?? 1) - 1;
+        if (left > 0) {
+          holds.set(toolCallId, left);
+          return;
+        }
+        holds.delete(toolCallId);
+        publish();
+      };
+    },
+    subscribe(listener) {
+      listeners.add(listener);
+      return () => {
+        listeners.delete(listener);
+      };
+    },
+    getSnapshot: () => snapshot
+  };
+}
+function useDeepAgentLaneFollow(registry, toolCallId, active) {
+  useEffect(() => {
+    if (registry === null || !active) return void 0;
+    return registry.follow(toolCallId);
+  }, [registry, toolCallId, active]);
+}
+function laneFollowRequestsFor(window, followed) {
+  if (followed.size === 0) return [];
+  const requests = [];
+  for (const message of window.messages) {
+    if (message.type !== "tool" || !followed.has(message.tool_call_id)) continue;
+    if (typeof message.id !== "string" || !isRunningToolPlaceholder(message)) continue;
+    requests.push(laneFollowRequest(message.id));
+  }
+  return requests;
+}
+const followKey = (request) => {
+  var _a;
+  return `${request.at}:${(_a = request.follow) == null ? void 0 : _a.before}`;
+};
+const sameLaneFollows = (held, wanted) => held.length === wanted.length && wanted.every((request) => held.some((have) => followKey(have) === followKey(request)));
+const rejectionCode = (error) => {
+  const failure = isRecord(error) ? error.failure : void 0;
+  return isRecord(failure) && failure.fate === "rejected" && typeof failure.code === "string" ? failure.code : void 0;
+};
+const noopSubscribe = () => () => {
+};
+function useDeepAgentLaneFollowSync({
+  registry,
+  paused = false,
+  onError
+}) {
+  const { invoke, connection, documents } = useAuiState((s) => {
+    if (!hasStatewireThreadExtras(s.thread.extras)) {
+      throw new Error("useDeepAgentLaneFollowSync requires a statewire thread");
+    }
+    return s.thread.extras;
+  });
+  const hostVersion = useAuiState((s) => {
+    var _a;
+    const extras = s.thread.extras;
+    return ((_a = readDeepAgentProtocol(extras == null ? void 0 : extras.state)) == null ? void 0 : _a.version) ?? 0;
+  });
+  const mounted = useSyncExternalStore(
+    (documents == null ? void 0 : documents.subscribe) ?? noopSubscribe,
+    () => documents == null ? void 0 : documents.list(),
+    () => documents == null ? void 0 : documents.list()
+  );
+  const window = useMemo(() => readDeepAgentFollowedWindow(mounted), [mounted]);
+  const followed = useSyncExternalStore(
+    registry.subscribe,
+    registry.getSnapshot,
+    registry.getSnapshot
+  );
+  const [settled, setSettled] = useState(0);
+  const inflightRef = useRef(false);
+  const aliveRef = useRef(true);
+  const unservedRef = useRef(null);
+  const lastStatedRef = useRef(null);
+  const onErrorRef = useRef(onError);
+  onErrorRef.current = onError;
+  useEffect(() => {
+    aliveRef.current = true;
+    return () => {
+      aliveRef.current = false;
+    };
+  }, []);
+  useEffect(() => {
+    if (window === null || paused || inflightRef.current) return;
+    if (connection.status !== "connected") return;
+    const served = hostVersion >= DEEP_AGENT_LANE_FOLLOW_PROTOCOL_VERSION && unservedRef.current !== connection;
+    const others = window.requests.filter((request) => !isLaneFollowRequest(request));
+    const held = window.requests.filter(isLaneFollowRequest);
+    const wanted = served ? laneFollowRequestsFor(window, followed).slice(
+      0,
+      Math.max(0, DEEP_AGENT_MAX_INTERESTS - others.length)
+    ) : [];
+    if (sameLaneFollows(held, wanted)) return;
+    const requests = [...others, ...wanted];
+    const key = JSON.stringify(requests);
+    const last = lastStatedRef.current;
+    if (last !== null && last.connection === connection && last.key === key) return;
+    lastStatedRef.current = { connection, key };
+    inflightRef.current = true;
+    invoke(DEEP_AGENT_INTEREST_PROTOCOL, "set", [requests]).catch((error) => {
+      var _a;
+      const code = rejectionCode(error);
+      if (code === "invalid-interest" && wanted.length > 0) {
+        unservedRef.current = connection;
+        return;
+      }
+      if (code !== "unknown-message") (_a = onErrorRef.current) == null ? void 0 : _a.call(onErrorRef, error);
+    }).finally(() => {
+      inflightRef.current = false;
+      if (aliveRef.current) setSettled((count) => count + 1);
+    });
+  }, [window, followed, hostVersion, connection, invoke, paused, settled]);
+}
+export {
+  createDeepAgentLaneFollowRegistry,
+  isLaneFollowRequest,
+  laneFollowRequest,
+  mergeFollowedLane,
+  readDeepAgentFollowedWindow,
+  readDeepAgentLaneFollows,
+  useDeepAgentLaneFollow,
+  useDeepAgentLaneFollowSync
+};
+//# sourceMappingURL=lane-follow.js.map