@athenaintel/react 0.43.0 → 0.48.0

package/dist/agent-runtime/src/answer-label.js

@@ -0,0 +1,137 @@
+import { getExternalStoreMessages } from "@assistant-ui/core";
+const ATHENA_ANSWERED_BY_KEY = "answered_by";
+const ATHENA_ASSISTANT_NAME = "Athena";
+const ASSISTANT_AGENT_PREFIX = "assistant:";
+const PERSON_PRINCIPAL_PREFIX = "user_";
+const UNNAMED_AGENT_NAME = "Agent";
+const LABEL_SEPARATOR = " · ";
+const NO_ANSWERS = Object.freeze([]);
+const isRecord = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
+const optionalId = (value) => typeof value === "string" && value.length > 0 ? value : null;
+const optionalName = (value) => {
+  if (typeof value !== "string") return null;
+  const trimmed = value.trim();
+  return trimmed.length > 0 ? trimmed : null;
+};
+function readAnsweredBy(value) {
+  if (!isRecord(value)) return null;
+  const agent = optionalId(value.agent);
+  if (agent === null) return null;
+  return {
+    agent,
+    agent_name: optionalName(value.agent_name),
+    for_id: optionalId(value.for_id),
+    for_name: optionalName(value.for_name)
+  };
+}
+const ANSWER_ROW_RUN_ID_PREFIX = "answered_by:";
+function answerRowRunId(agent, forId) {
+  return `${ANSWER_ROW_RUN_ID_PREFIX}${agent}|${forId ?? ""}`;
+}
+function isSplitAnswerRow(message) {
+  var _a;
+  const runId = message.role === "assistant" ? (_a = message.metadata) == null ? void 0 : _a.runId : void 0;
+  return typeof runId === "string" && runId.startsWith(ANSWER_ROW_RUN_ID_PREFIX);
+}
+const isSameAnswerer = (a, b) => a.agent === b.agent && a.for_id === b.for_id;
+const isPersonPrincipal = (id) => id !== null && id.startsWith(PERSON_PRINCIPAL_PREFIX);
+function readLangChainAnswer(source) {
+  var _a;
+  if ((source == null ? void 0 : source.type) !== "ai") return null;
+  const bag = (_a = source.additional_kwargs) == null ? void 0 : _a.metadata;
+  return isRecord(bag) ? readAnsweredBy(bag[ATHENA_ANSWERED_BY_KEY]) : null;
+}
+function collectMessageAnswers(message) {
+  var _a, _b;
+  if (message.role !== "assistant") return NO_ANSWERS;
+  const sources = getExternalStoreMessages(message);
+  if (sources.length === 0) {
+    const joined = readAnsweredBy((_b = (_a = message.metadata) == null ? void 0 : _a.custom) == null ? void 0 : _b[ATHENA_ANSWERED_BY_KEY]);
+    return joined === null ? NO_ANSWERS : [joined];
+  }
+  let answers = null;
+  for (const source of sources) {
+    const answer = readLangChainAnswer(source);
+    if (answer === null || (answers == null ? void 0 : answers.some((seen) => isSameAnswerer(seen, answer)))) continue;
+    answers ?? (answers = []);
+    answers.push(answer);
+  }
+  return answers ?? NO_ANSWERS;
+}
+const answersByMessage = /* @__PURE__ */ new WeakMap();
+function readMessageAnswers(message) {
+  const cached = answersByMessage.get(message);
+  if (cached !== void 0) return cached;
+  const answers = collectMessageAnswers(message);
+  answersByMessage.set(message, answers);
+  return answers;
+}
+function threadShowsAnswerLabels(messages) {
+  let first = null;
+  for (const message of messages) {
+    if (isSplitAnswerRow(message)) return true;
+    for (const answer of readMessageAnswers(message)) {
+      if (first === null) {
+        first = answer;
+      } else if (!isSameAnswerer(first, answer)) {
+        return true;
+      }
+    }
+  }
+  return false;
+}
+function showsAnswerLabel({
+  answers,
+  threadShowsLabels,
+  viewerId
+}) {
+  if (answers.length === 0) return false;
+  if (threadShowsLabels) return true;
+  return viewerId !== null && answers.some((answer) => isPersonPrincipal(answer.for_id) && answer.for_id !== viewerId);
+}
+function answererName(answer) {
+  if (answer.agent_name !== null) return answer.agent_name;
+  return answer.agent.startsWith(ASSISTANT_AGENT_PREFIX) ? ATHENA_ASSISTANT_NAME : UNNAMED_AGENT_NAME;
+}
+function forWhomName(answer, viewerId) {
+  if (answer.for_id !== null && answer.for_id === viewerId) return "you";
+  if (answer.for_name !== null && isPersonPrincipal(answer.for_id)) return answer.for_name;
+  return null;
+}
+function joinNames(names) {
+  if (names.length <= 1) return names[0] ?? "";
+  return `${names.slice(0, -1).join(", ")} and ${names[names.length - 1]}`;
+}
+function answerLabelText({
+  answers,
+  viewerId
+}) {
+  if (answers.length === 0) return null;
+  const groups = [];
+  for (const answer of answers) {
+    const name = answererName(answer);
+    const forWhom = forWhomName(answer, viewerId);
+    const group = groups.find((candidate) => candidate.forId === answer.for_id);
+    if (group === void 0) {
+      groups.push({ forId: answer.for_id, names: [name], forWhom });
+      continue;
+    }
+    if (!group.names.includes(name)) group.names.push(name);
+    group.forWhom ?? (group.forWhom = forWhom);
+  }
+  return groups.map(
+    ({ names, forWhom }) => forWhom === null ? joinNames(names) : `${joinNames(names)}${LABEL_SEPARATOR}for ${forWhom}`
+  ).join(", ");
+}
+export {
+  ATHENA_ANSWERED_BY_KEY,
+  ATHENA_ASSISTANT_NAME,
+  answerLabelText,
+  answerRowRunId,
+  readAnsweredBy,
+  readLangChainAnswer,
+  readMessageAnswers,
+  showsAnswerLabel,
+  threadShowsAnswerLabels
+};
+//# sourceMappingURL=answer-label.js.map

package/dist/agent-runtime/src/answer-label.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"answer-label.js","sources":["../../../../agent-runtime/src/answer-label.ts"],"sourcesContent":["import { getExternalStoreMessages, type ThreadMessage } from '@assistant-ui/core';\nimport type { LangChainMessage } from '@assistant-ui/react-langgraph';\n\n/**\n * `metadata.custom` key naming the agent that answered an AI message and whom\n * its run worked for (identity v7 D13: \"each answer is labelled with its\n * agent, 'for <author>'\").\n *\n * The backend stamps it on every answer a deep-agent run's model produces\n * (`agora/.../conversation_agents/answer_label.py`), under the\n * `additional_kwargs.metadata` bag the converter exposes as\n * `message.metadata.custom`, and mirrors it onto the live wire once the\n * model's turn lands. Only the server writes it. Answers written before the\n * stamp existed carry none, and get no label.\n */\nexport const ATHENA_ANSWERED_BY_KEY = 'answered_by';\n\n/** The agent that answered and whom its run worked for, as the backend stamps them. */\nexport interface DeepAgentAnsweredBy {\n  /** `collab_agent:<asset id>` for a team agent, `assistant:<principal id>` for someone's own assistant. */\n  agent: string;\n  /** The collab agent's name when it answered; `null` for an assistant, which clients name. */\n  agent_name: string | null;\n  /** Whom the run worked for: a person (`user_…`), an agent's or automation's own run, or nobody (`null`). */\n  for_id: string | null;\n  /** That person's display name when it answered, else `null`. */\n  for_name: string | null;\n}\n\n/** What a label calls someone's own assistant: the managed default agent's product name. */\nexport const ATHENA_ASSISTANT_NAME = 'Athena';\n\nconst ASSISTANT_AGENT_PREFIX = 'assistant:';\nconst PERSON_PRINCIPAL_PREFIX = 'user_';\nconst UNNAMED_AGENT_NAME = 'Agent';\nconst LABEL_SEPARATOR = ' · ';\nconst NO_ANSWERS: readonly DeepAgentAnsweredBy[] = Object.freeze([]);\n\nconst isRecord = (value: unknown): value is Record<string, unknown> =>\n  typeof value === 'object' && value !== null && !Array.isArray(value);\n\nconst optionalId = (value: unknown): string | null =>\n  typeof value === 'string' && value.length > 0 ? value : null;\n\nconst optionalName = (value: unknown): string | null => {\n  if (typeof value !== 'string') return null;\n  const trimmed = value.trim();\n  return trimmed.length > 0 ? trimmed : null;\n};\n\n/**\n * Parse an `answered_by` stamp off a wire value.\n *\n * A stamp without a string `agent` is no stamp at all; absent or malformed\n * optional fields read as `null`, so a partly malformed stamp still names its\n * agent and never names a person it can't.\n */\nexport function readAnsweredBy(value: unknown): DeepAgentAnsweredBy | null {\n  if (!isRecord(value)) return null;\n  const agent = optionalId(value.agent);\n  if (agent === null) return null;\n  return {\n    agent,\n    agent_name: optionalName(value.agent_name),\n    for_id: optionalId(value.for_id),\n    for_name: optionalName(value.for_name),\n  };\n}\n\n/**\n * The run id an answer's row carries when the deep-agent conversion split a\n * turn several answerers wrote into one row per answerer (`answer-rows.ts`):\n * one per agent and principal.\n */\nconst ANSWER_ROW_RUN_ID_PREFIX = 'answered_by:';\n\n/** The row run id of an answer by `agent` for `forId` (see `withAnswerRows`). */\nexport function answerRowRunId(agent: string, forId: string | null): string {\n  return `${ANSWER_ROW_RUN_ID_PREFIX}${agent}|${forId ?? ''}`;\n}\n\n/**\n * Whether a converted row is one of a turn the conversion split by answerer.\n * Reads `metadata` optionally, as `collectMessageAnswers` does: a message\n * another runtime (or a test) builds may carry none.\n */\nfunction isSplitAnswerRow(message: ThreadMessage): boolean {\n  const runId = message.role === 'assistant' ? message.metadata?.runId : undefined;\n  return typeof runId === 'string' && runId.startsWith(ANSWER_ROW_RUN_ID_PREFIX);\n}\n\nconst isSameAnswerer = (a: DeepAgentAnsweredBy, b: DeepAgentAnsweredBy): boolean =>\n  a.agent === b.agent && a.for_id === b.for_id;\n\nconst isPersonPrincipal = (id: string | null): id is string =>\n  id !== null && id.startsWith(PERSON_PRINCIPAL_PREFIX);\n\n/** The `answered_by` stamp an AI message as the wire carries it holds; `null` for any other message. */\nexport function readLangChainAnswer(\n  source: LangChainMessage | undefined\n): DeepAgentAnsweredBy | null {\n  if (source?.type !== 'ai') return null;\n  const bag = source.additional_kwargs?.metadata;\n  return isRecord(bag) ? readAnsweredBy(bag[ATHENA_ANSWERED_BY_KEY]) : null;\n}\n\nfunction collectMessageAnswers(message: ThreadMessage): readonly DeepAgentAnsweredBy[] {\n  if (message.role !== 'assistant') return NO_ANSWERS;\n  const sources = getExternalStoreMessages<LangChainMessage>(message);\n  if (sources.length === 0) {\n    const joined = readAnsweredBy(message.metadata?.custom?.[ATHENA_ANSWERED_BY_KEY]);\n    return joined === null ? NO_ANSWERS : [joined];\n  }\n  let answers: DeepAgentAnsweredBy[] | null = null;\n  for (const source of sources) {\n    const answer = readLangChainAnswer(source);\n    if (answer === null || answers?.some((seen) => isSameAnswerer(seen, answer))) continue;\n    answers ??= [];\n    answers.push(answer);\n  }\n  return answers ?? NO_ANSWERS;\n}\n\nconst answersByMessage = new WeakMap<ThreadMessage, readonly DeepAgentAnsweredBy[]>();\n\n/**\n * The answers one thread message holds: each agent that answered in it and\n * whom it worked for, in the order they answered, one entry per agent and\n * principal.\n *\n * Read off the message's bound source messages, not its joined\n * `metadata.custom`: assistant-ui joins consecutive AI messages into one\n * turn, and its joined metadata keeps only the last stamp. The deep-agent\n * conversion gives each agent that answers one message in turn (D13 several\n * agents, one message) a row of its own (`answer-rows.ts`), so a row\n * normally holds one answer; a conversion without that split (another\n * pipeline, a stamp that never landed) still names every agent of the row.\n * A message with no bound sources (another runtime) falls back to the joined\n * value. Cached per message object: a settled message keeps its object\n * across a streamed token, so a thread-wide scan per token reads only the\n * message that changed.\n */\nexport function readMessageAnswers(message: ThreadMessage): readonly DeepAgentAnsweredBy[] {\n  const cached = answersByMessage.get(message);\n  if (cached !== undefined) return cached;\n  const answers = collectMessageAnswers(message);\n  answersByMessage.set(message, answers);\n  return answers;\n}\n\n/**\n * The thread-wide half of the label decision: whether the transcript's\n * answers name two or more agents, or two or more principals they worked\n * for. Then every answer is labelled; a 1:1 chat with one agent working for\n * one person looks as it did before labels. Unstamped answers count for\n * nothing.\n *\n * It decides over the messages the client holds (on a windowed wire, the\n * pages loaded so far), as the human-author avatars do: a page of older\n * history that brings another agent or person turns the labels on. Each\n * label still names its own answer's stamp, so loading history never changes\n * what an answer is attributed to, only whether the line shows.\n *\n * A turn the conversion split into one row per answerer turns the labels on\n * too, before the next agent's stamp lands: while several agents answer one\n * message, the host's record already says a second agent is answering, so\n * the first agent's answer is labelled from the moment the second one's row\n * appears.\n */\nexport function threadShowsAnswerLabels(messages: readonly ThreadMessage[]): boolean {\n  let first: DeepAgentAnsweredBy | null = null;\n  for (const message of messages) {\n    if (isSplitAnswerRow(message)) return true;\n    for (const answer of readMessageAnswers(message)) {\n      if (first === null) {\n        first = answer;\n      } else if (!isSameAnswerer(first, answer)) {\n        return true;\n      }\n    }\n  }\n  return false;\n}\n\n/**\n * Whether a message's answers are labelled: always once the thread's answers\n * name several agents or principals (`threadShowsLabels`), and otherwise when\n * one of them worked for a person other than the viewer. Without a known\n * viewer that last rule is off, so a 1:1 chat never flashes labels while the\n * viewer loads.\n */\nexport function showsAnswerLabel({\n  answers,\n  threadShowsLabels,\n  viewerId,\n}: {\n  answers: readonly DeepAgentAnsweredBy[];\n  threadShowsLabels: boolean;\n  viewerId: string | null;\n}): boolean {\n  if (answers.length === 0) return false;\n  if (threadShowsLabels) return true;\n  return (\n    viewerId !== null &&\n    answers.some((answer) => isPersonPrincipal(answer.for_id) && answer.for_id !== viewerId)\n  );\n}\n\nfunction answererName(answer: DeepAgentAnsweredBy): string {\n  if (answer.agent_name !== null) return answer.agent_name;\n  return answer.agent.startsWith(ASSISTANT_AGENT_PREFIX)\n    ? ATHENA_ASSISTANT_NAME\n    : UNNAMED_AGENT_NAME;\n}\n\nfunction forWhomName(answer: DeepAgentAnsweredBy, viewerId: string | null): string | null {\n  if (answer.for_id !== null && answer.for_id === viewerId) return 'you';\n  if (answer.for_name !== null && isPersonPrincipal(answer.for_id)) return answer.for_name;\n  return null;\n}\n\nfunction joinNames(names: readonly string[]): string {\n  if (names.length <= 1) return names[0] ?? '';\n  return `${names.slice(0, -1).join(', ')} and ${names[names.length - 1]}`;\n}\n\n/**\n * The label's words: who answered, then whom for. \"Finance · for you\",\n * \"Athena · for Priya Shah\", \"Finance\" (a run that worked for nobody, or for\n * an agent's own run). The agent is the collab agent's name, else\n * {@link ATHENA_ASSISTANT_NAME} for someone's own assistant, else \"Agent\".\n * Several agents answering for the same principal read \"Finance and Legal ·\n * for you\"; answers for different principals are listed apart, even when the\n * people share a name (\"Finance · for Alex, Legal · for Alex\"). `null` when\n * the message holds no answers.\n */\nexport function answerLabelText({\n  answers,\n  viewerId,\n}: {\n  answers: readonly DeepAgentAnsweredBy[];\n  viewerId: string | null;\n}): string | null {\n  if (answers.length === 0) return null;\n  // One group per principal (nobody is one too), never per displayed name: two\n  // people who share a name stay two recipients.\n  const groups: { forId: string | null; names: string[]; forWhom: string | null }[] = [];\n  for (const answer of answers) {\n    const name = answererName(answer);\n    const forWhom = forWhomName(answer, viewerId);\n    const group = groups.find((candidate) => candidate.forId === answer.for_id);\n    if (group === undefined) {\n      groups.push({ forId: answer.for_id, names: [name], forWhom });\n      continue;\n    }\n    if (!group.names.includes(name)) group.names.push(name);\n    // A recipient one answer left unnamed takes the name a later answer for\n    // the same principal carries. The first name found wins: answers stamped\n    // under two names for one id (a rename between runs) read as the first.\n    group.forWhom ??= forWhom;\n  }\n  return groups\n    .map(({ names, forWhom }) =>\n      forWhom === null ? joinNames(names) : `${joinNames(names)}${LABEL_SEPARATOR}for ${forWhom}`\n    )\n    .join(', ');\n}\n"],"names":[],"mappings":";AAeO,MAAM,yBAAyB;AAe/B,MAAM,wBAAwB;AAErC,MAAM,yBAAyB;AAC/B,MAAM,0BAA0B;AAChC,MAAM,qBAAqB;AAC3B,MAAM,kBAAkB;AACxB,MAAM,aAA6C,OAAO,OAAO,EAAE;AAEnE,MAAM,WAAW,CAAC,UAChB,OAAO,UAAU,YAAY,UAAU,QAAQ,CAAC,MAAM,QAAQ,KAAK;AAErE,MAAM,aAAa,CAAC,UAClB,OAAO,UAAU,YAAY,MAAM,SAAS,IAAI,QAAQ;AAE1D,MAAM,eAAe,CAAC,UAAkC;AACtD,MAAI,OAAO,UAAU,SAAU,QAAO;AACtC,QAAM,UAAU,MAAM,KAAA;AACtB,SAAO,QAAQ,SAAS,IAAI,UAAU;AACxC;AASO,SAAS,eAAe,OAA4C;AACzE,MAAI,CAAC,SAAS,KAAK,EAAG,QAAO;AAC7B,QAAM,QAAQ,WAAW,MAAM,KAAK;AACpC,MAAI,UAAU,KAAM,QAAO;AAC3B,SAAO;AAAA,IACL;AAAA,IACA,YAAY,aAAa,MAAM,UAAU;AAAA,IACzC,QAAQ,WAAW,MAAM,MAAM;AAAA,IAC/B,UAAU,aAAa,MAAM,QAAQ;AAAA,EAAA;AAEzC;AAOA,MAAM,2BAA2B;AAG1B,SAAS,eAAe,OAAe,OAA8B;AAC1E,SAAO,GAAG,wBAAwB,GAAG,KAAK,IAAI,SAAS,EAAE;AAC3D;AAOA,SAAS,iBAAiB,SAAiC;;AACzD,QAAM,QAAQ,QAAQ,SAAS,eAAc,aAAQ,aAAR,mBAAkB,QAAQ;AACvE,SAAO,OAAO,UAAU,YAAY,MAAM,WAAW,wBAAwB;AAC/E;AAEA,MAAM,iBAAiB,CAAC,GAAwB,MAC9C,EAAE,UAAU,EAAE,SAAS,EAAE,WAAW,EAAE;AAExC,MAAM,oBAAoB,CAAC,OACzB,OAAO,QAAQ,GAAG,WAAW,uBAAuB;AAG/C,SAAS,oBACd,QAC4B;;AAC5B,OAAI,iCAAQ,UAAS,KAAM,QAAO;AAClC,QAAM,OAAM,YAAO,sBAAP,mBAA0B;AACtC,SAAO,SAAS,GAAG,IAAI,eAAe,IAAI,sBAAsB,CAAC,IAAI;AACvE;AAEA,SAAS,sBAAsB,SAAwD;;AACrF,MAAI,QAAQ,SAAS,YAAa,QAAO;AACzC,QAAM,UAAU,yBAA2C,OAAO;AAClE,MAAI,QAAQ,WAAW,GAAG;AACxB,UAAM,SAAS,gBAAe,mBAAQ,aAAR,mBAAkB,WAAlB,mBAA2B,uBAAuB;AAChF,WAAO,WAAW,OAAO,aAAa,CAAC,MAAM;AAAA,EAC/C;AACA,MAAI,UAAwC;AAC5C,aAAW,UAAU,SAAS;AAC5B,UAAM,SAAS,oBAAoB,MAAM;AACzC,QAAI,WAAW,SAAQ,mCAAS,KAAK,CAAC,SAAS,eAAe,MAAM,MAAM,IAAI;AAC9E,0BAAY,CAAA;AACZ,YAAQ,KAAK,MAAM;AAAA,EACrB;AACA,SAAO,WAAW;AACpB;AAEA,MAAM,uCAAuB,QAAA;AAmBtB,SAAS,mBAAmB,SAAwD;AACzF,QAAM,SAAS,iBAAiB,IAAI,OAAO;AAC3C,MAAI,WAAW,OAAW,QAAO;AACjC,QAAM,UAAU,sBAAsB,OAAO;AAC7C,mBAAiB,IAAI,SAAS,OAAO;AACrC,SAAO;AACT;AAqBO,SAAS,wBAAwB,UAA6C;AACnF,MAAI,QAAoC;AACxC,aAAW,WAAW,UAAU;AAC9B,QAAI,iBAAiB,OAAO,EAAG,QAAO;AACtC,eAAW,UAAU,mBAAmB,OAAO,GAAG;AAChD,UAAI,UAAU,MAAM;AAClB,gBAAQ;AAAA,MACV,WAAW,CAAC,eAAe,OAAO,MAAM,GAAG;AACzC,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AACA,SAAO;AACT;AASO,SAAS,iBAAiB;AAAA,EAC/B;AAAA,EACA;AAAA,EACA;AACF,GAIY;AACV,MAAI,QAAQ,WAAW,EAAG,QAAO;AACjC,MAAI,kBAAmB,QAAO;AAC9B,SACE,aAAa,QACb,QAAQ,KAAK,CAAC,WAAW,kBAAkB,OAAO,MAAM,KAAK,OAAO,WAAW,QAAQ;AAE3F;AAEA,SAAS,aAAa,QAAqC;AACzD,MAAI,OAAO,eAAe,KAAM,QAAO,OAAO;AAC9C,SAAO,OAAO,MAAM,WAAW,sBAAsB,IACjD,wBACA;AACN;AAEA,SAAS,YAAY,QAA6B,UAAwC;AACxF,MAAI,OAAO,WAAW,QAAQ,OAAO,WAAW,SAAU,QAAO;AACjE,MAAI,OAAO,aAAa,QAAQ,kBAAkB,OAAO,MAAM,UAAU,OAAO;AAChF,SAAO;AACT;AAEA,SAAS,UAAU,OAAkC;AACnD,MAAI,MAAM,UAAU,EAAG,QAAO,MAAM,CAAC,KAAK;AAC1C,SAAO,GAAG,MAAM,MAAM,GAAG,EAAE,EAAE,KAAK,IAAI,CAAC,QAAQ,MAAM,MAAM,SAAS,CAAC,CAAC;AACxE;AAYO,SAAS,gBAAgB;AAAA,EAC9B;AAAA,EACA;AACF,GAGkB;AAChB,MAAI,QAAQ,WAAW,EAAG,QAAO;AAGjC,QAAM,SAA8E,CAAA;AACpF,aAAW,UAAU,SAAS;AAC5B,UAAM,OAAO,aAAa,MAAM;AAChC,UAAM,UAAU,YAAY,QAAQ,QAAQ;AAC5C,UAAM,QAAQ,OAAO,KAAK,CAAC,cAAc,UAAU,UAAU,OAAO,MAAM;AAC1E,QAAI,UAAU,QAAW;AACvB,aAAO,KAAK,EAAE,OAAO,OAAO,QAAQ,OAAO,CAAC,IAAI,GAAG,SAAS;AAC5D;AAAA,IACF;AACA,QAAI,CAAC,MAAM,MAAM,SAAS,IAAI,EAAG,OAAM,MAAM,KAAK,IAAI;AAItD,UAAM,YAAN,MAAM,UAAY;AAAA,EACpB;AACA,SAAO,OACJ;AAAA,IAAI,CAAC,EAAE,OAAO,QAAA,MACb,YAAY,OAAO,UAAU,KAAK,IAAI,GAAG,UAAU,KAAK,CAAC,GAAG,eAAe,OAAO,OAAO;AAAA,EAAA,EAE1F,KAAK,IAAI;AACd;"}

package/dist/agent-runtime/src/answer-rows.cjs

@@ -0,0 +1,180 @@
+"use strict";
+Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
+const answerLabel = require("./answer-label.cjs");
+const ADDRESSED_TURN_STATE_KEY = "addressedTurn";
+const ADDRESSED_TURN_RECORD_VERSION = 1;
+const MAX_ADDRESSED_AGENTS = 8;
+const AGENT_KEY_PATTERN = /^(collab_agent|assistant):\S+$/;
+const isRecord = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
+const isAgentKey = (value) => typeof value === "string" && AGENT_KEY_PATTERN.test(value);
+function readOutcome(value) {
+  if (!isRecord(value)) return null;
+  const { outcome } = value;
+  return {
+    agent: String(value.agent),
+    outcome: outcome === "answered" || outcome === "refused" ? outcome : "failed",
+    code: typeof value.code === "string" && value.code.length > 0 ? value.code : null
+  };
+}
+function readDeepAgentAddressedTurn(value) {
+  if (!isRecord(value) || value.v !== ADDRESSED_TURN_RECORD_VERSION) return null;
+  const { author, author_id: authorId, agents, position, anchor, outcomes, started } = value;
+  if (typeof author !== "string" || author.length === 0) return null;
+  if (typeof authorId !== "string" || authorId.length === 0) return null;
+  if (started !== void 0 && typeof started !== "boolean") return null;
+  if (!Array.isArray(agents) || agents.length < 2 || agents.length > MAX_ADDRESSED_AGENTS || !agents.every(isAgentKey)) {
+    return null;
+  }
+  if (typeof position !== "number" || !Number.isInteger(position)) return null;
+  if (position < 0 || position >= agents.length) return null;
+  if (anchor !== null && anchor !== void 0 && (typeof anchor !== "string" || !anchor)) {
+    return null;
+  }
+  if (!Array.isArray(outcomes)) return null;
+  const finished = outcomes.map(readOutcome);
+  if (finished.some((outcome) => outcome === null)) return null;
+  return {
+    authorId,
+    agents,
+    position,
+    anchor: anchor ?? null,
+    outcomes: finished.filter((outcome) => outcome !== null)
+  };
+}
+const ANSWER_ROW_KEY = Symbol.for("@athenaintel/agent-runtime/answer-row");
+function readAnswerRowKey(message) {
+  const key = Reflect.get(message, ANSWER_ROW_KEY);
+  return typeof key === "string" ? key : void 0;
+}
+const keyedMessages = /* @__PURE__ */ new WeakMap();
+function withAnswerRowKey(message, key) {
+  const cached = keyedMessages.get(message);
+  if ((cached == null ? void 0 : cached.key) === key) return cached.message;
+  const keyed = { ...message };
+  Reflect.set(keyed, ANSWER_ROW_KEY, key);
+  keyedMessages.set(message, { key, message: keyed });
+  return keyed;
+}
+const stampKeys = /* @__PURE__ */ new WeakMap();
+function stampedAnswererKey(message) {
+  const cached = stampKeys.get(message);
+  if (cached !== void 0) return cached;
+  const answer = answerLabel.readLangChainAnswer(message);
+  const key = answer === null ? null : answerLabel.answerRowRunId(answer.agent, answer.for_id);
+  stampKeys.set(message, key);
+  return key;
+}
+const NEXT_AGENT_PENDING_MESSAGE_ID = "__athena_deep_agent_pending_next_agent__";
+const nextAgentPlaceholders = /* @__PURE__ */ new Map();
+function nextAgentPlaceholder(key) {
+  const cached = nextAgentPlaceholders.get(key);
+  if (cached !== void 0) return cached;
+  const placeholder = {
+    type: "ai",
+    content: [],
+    id: NEXT_AGENT_PENDING_MESSAGE_ID,
+    status: { type: "running" }
+  };
+  Reflect.set(placeholder, ANSWER_ROW_KEY, key);
+  Object.freeze(placeholder);
+  if (nextAgentPlaceholders.size >= 64) nextAgentPlaceholders.clear();
+  nextAgentPlaceholders.set(key, placeholder);
+  return placeholder;
+}
+const isTurnBoundary = (message) => message.type === "human" || message.type === "system";
+function answeringAgent({
+  messages,
+  addressedTurn
+}) {
+  if (addressedTurn === null || addressedTurn.position === 0 || addressedTurn.anchor === null) {
+    return null;
+  }
+  const agent = addressedTurn.agents[addressedTurn.position];
+  if (agent === void 0) return null;
+  for (let index = messages.length - 1; index >= 0; index -= 1) {
+    const message = messages[index];
+    if (message === void 0) continue;
+    if (message.id === addressedTurn.anchor) {
+      return { key: answerLabel.answerRowRunId(agent, addressedTurn.authorId), anchorIndex: index };
+    }
+    if (isTurnBoundary(message)) return null;
+  }
+  return null;
+}
+function turnKeys({
+  messages,
+  start,
+  end,
+  answering
+}) {
+  var _a, _b;
+  const keys = [];
+  let first = null;
+  let several = false;
+  let answeringStamp = null;
+  for (let index = start; index < end; index += 1) {
+    const message = messages[index];
+    if ((message == null ? void 0 : message.type) !== "ai") {
+      keys.push(null);
+      continue;
+    }
+    const answeringKey = answering !== null && index > answering.anchorIndex ? answering.key : null;
+    let key = stampedAnswererKey(message);
+    if (key !== null && answeringKey !== null) answeringStamp = key;
+    else if (key === null && answeringKey !== null) key = answeringStamp ?? answeringKey;
+    keys.push(key);
+    if (key === null) continue;
+    if (first === null) first = key;
+    else if (key !== first) several = true;
+  }
+  if (!several) return null;
+  let previous = null;
+  for (let offset = 0; offset < keys.length; offset += 1) {
+    if (((_a = messages[start + offset]) == null ? void 0 : _a.type) !== "ai") continue;
+    const key = keys[offset] ?? null;
+    if (key === null) keys[offset] = previous;
+    else previous = key;
+  }
+  let next = null;
+  for (let offset = keys.length - 1; offset >= 0; offset -= 1) {
+    if (((_b = messages[start + offset]) == null ? void 0 : _b.type) !== "ai") continue;
+    const key = keys[offset] ?? null;
+    if (key === null) keys[offset] = next;
+    else next = key;
+  }
+  return keys;
+}
+function withAnswerRows({
+  messages,
+  addressedTurn,
+  isRunning
+}) {
+  const answering = answeringAgent({ messages, addressedTurn });
+  const pending = answering !== null && isRunning && !messages.slice(answering.anchorIndex + 1).some((message) => message.type === "ai");
+  const sources = pending ? [...messages, nextAgentPlaceholder(answering.key)] : messages;
+  let keyed = null;
+  let start = 0;
+  for (let index = 0; index <= sources.length; index += 1) {
+    const message = sources[index];
+    if (message !== void 0 && !isTurnBoundary(message)) continue;
+    if (index - start > 1) {
+      const keys = turnKeys({ messages: sources, start, end: index, answering });
+      if (keys !== null) {
+        for (const [offset, key] of keys.entries()) {
+          const source = sources[start + offset];
+          if (key === null || source === void 0 || readAnswerRowKey(source) === key) continue;
+          keyed ?? (keyed = [...sources]);
+          keyed[start + offset] = withAnswerRowKey(source, key);
+        }
+      }
+    }
+    start = index + 1;
+  }
+  return keyed ?? sources;
+}
+exports.ADDRESSED_TURN_STATE_KEY = ADDRESSED_TURN_STATE_KEY;
+exports.NEXT_AGENT_PENDING_MESSAGE_ID = NEXT_AGENT_PENDING_MESSAGE_ID;
+exports.readAnswerRowKey = readAnswerRowKey;
+exports.readDeepAgentAddressedTurn = readDeepAgentAddressedTurn;
+exports.withAnswerRows = withAnswerRows;
+//# sourceMappingURL=answer-rows.cjs.map

package/dist/agent-runtime/src/answer-rows.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"answer-rows.cjs","sources":["../../../../agent-runtime/src/answer-rows.ts"],"sourcesContent":["import type { LangChainMessage } from '@assistant-ui/react-langgraph';\nimport { answerRowRunId, readLangChainAnswer } from './answer-label';\n\n/**\n * One row per answering agent (identity v7 D13: several agents answer one\n * message, each in its own turn for its author).\n *\n * assistant-ui joins consecutive assistant messages into one row unless their\n * `metadata.runId` differs (`chunkExternalMessages` in `@assistant-ui/core`).\n * The agents that answer one message answer in turn, with nothing between\n * them, so without a run id per answer every agent's answer lands in one row.\n * This module gives each AI message the answerer it belongs to, which the\n * converter turns into `metadata.runId` (`convertDeepAgentLangChainMessage`):\n *\n * - a stamped answer: its `answered_by` stamp (agent and principal), written by\n *   agora when its model turn lands;\n * - an answer still streaming while several agents answer: the agent the\n *   host's record says is answering (`addressedTurn`, replicated statewire\n *   state), for every message after the message its turn began after. The\n *   stamp lands only once a model turn finishes, so reading the record is what\n *   keeps a next agent's answer in its own row from its first token instead of\n *   joining the previous agent's row and jumping out when its stamp lands;\n * - anything else (an answer written before the stamp existed, a reused one,\n *   a run marker): the answerer of the AI message before it in the same turn,\n *   else of the first one after it.\n *\n * Only a turn (the messages between two people's messages) whose answers come\n * from two or more answerers is given run ids: a 1:1 chat, a turn one agent\n * answered and every transcript written before the stamp keep their messages'\n * identity and convert exactly as before.\n */\n\n/** Statewire state key of the message several agents are answering (agora `several_agents.ADDRESSED_TURN_STATE_KEY`). */\nexport const ADDRESSED_TURN_STATE_KEY = 'addressedTurn';\n\n/** The record version this reader understands (agora `several_agents.RECORD_VERSION`). */\nconst ADDRESSED_TURN_RECORD_VERSION = 1;\n\n/** At most this many agents answer one message (agora `agent_refs.MAX_ADDRESSED_AGENTS`). */\nconst MAX_ADDRESSED_AGENTS = 8;\n\nconst AGENT_KEY_PATTERN = /^(collab_agent|assistant):\\S+$/;\n\n/** How one agent's turn of a message several agents answer ended. */\nexport interface DeepAgentAddressedOutcome {\n  /** The agent's key (`collab_agent:<asset id>`, `assistant:<principal id>`). */\n  agent: string;\n  outcome: 'answered' | 'failed' | 'refused';\n  /** A refusal's code (e.g. `conversation_history_recheck_required`); `null` otherwise. */\n  code: string | null;\n}\n\n/** The message several agents are answering, as the host's replicated record holds it. */\nexport interface DeepAgentAddressedTurn {\n  /** Whom every agent of the message answers for: the author's principal id. */\n  authorId: string;\n  /** The named agents' keys, in the order they answer. */\n  agents: readonly string[];\n  /** The index in {@link agents} of the agent answering now. */\n  position: number;\n  /** The message the answering agent's turn began after; `null` for the first agent. */\n  anchor: string | null;\n  /** Each finished agent's outcome, in order. */\n  outcomes: readonly DeepAgentAddressedOutcome[];\n}\n\nconst isRecord = (value: unknown): value is Record<string, unknown> =>\n  typeof value === 'object' && value !== null && !Array.isArray(value);\n\nconst isAgentKey = (value: unknown): value is string =>\n  typeof value === 'string' && AGENT_KEY_PATTERN.test(value);\n\n/** One finished agent's outcome, read as agora reads it: an outcome it doesn't know is a failure. */\nfunction readOutcome(value: unknown): DeepAgentAddressedOutcome | null {\n  if (!isRecord(value)) return null;\n  const { outcome } = value;\n  return {\n    agent: String(value.agent),\n    outcome: outcome === 'answered' || outcome === 'refused' ? outcome : 'failed',\n    code: typeof value.code === 'string' && value.code.length > 0 ? value.code : null,\n  };\n}\n\n/**\n * The record of the message several agents are answering, off the replicated\n * state value (`state.addressedTurn`); `null` while none is (agora clears it\n * once the last agent settled) and for anything this reader can't trust,\n * mirroring agora's own `read_record`: another version, a malformed field, an\n * agent key it can't read, a position outside the list.\n */\nexport function readDeepAgentAddressedTurn(value: unknown): DeepAgentAddressedTurn | null {\n  if (!isRecord(value) || value.v !== ADDRESSED_TURN_RECORD_VERSION) return null;\n  const { author, author_id: authorId, agents, position, anchor, outcomes, started } = value;\n  if (typeof author !== 'string' || author.length === 0) return null;\n  if (typeof authorId !== 'string' || authorId.length === 0) return null;\n  if (started !== undefined && typeof started !== 'boolean') return null;\n  if (\n    !Array.isArray(agents) ||\n    agents.length < 2 ||\n    agents.length > MAX_ADDRESSED_AGENTS ||\n    !agents.every(isAgentKey)\n  ) {\n    return null;\n  }\n  if (typeof position !== 'number' || !Number.isInteger(position)) return null;\n  if (position < 0 || position >= agents.length) return null;\n  if (anchor !== null && anchor !== undefined && (typeof anchor !== 'string' || !anchor)) {\n    return null;\n  }\n  if (!Array.isArray(outcomes)) return null;\n  const finished = outcomes.map(readOutcome);\n  if (finished.some((outcome) => outcome === null)) return null;\n  return {\n    authorId,\n    agents,\n    position,\n    anchor: anchor ?? null,\n    outcomes: finished.filter((outcome): outcome is DeepAgentAddressedOutcome => outcome !== null),\n  };\n}\n\n/** Where an AI message carries the answerer its row is keyed by (see {@link readAnswerRowKey}). */\nconst ANSWER_ROW_KEY = Symbol.for('@athenaintel/agent-runtime/answer-row');\n\n/**\n * The answerer an AI message's row is keyed by, as {@link withAnswerRows} put\n * it on the message; `undefined` for a message it left alone. The converter\n * passes it on as the row's `metadata.runId`.\n */\nexport function readAnswerRowKey(message: LangChainMessage): string | undefined {\n  const key: unknown = Reflect.get(message, ANSWER_ROW_KEY);\n  return typeof key === 'string' ? key : undefined;\n}\n\n/**\n * The keyed copy of each message, one per message and key: the conversion\n * caches by message identity, so an unchanged message must yield the same\n * keyed object on every pass, and a message whose key changed must yield a\n * new one (it converts again with its new run id).\n */\nconst keyedMessages = new WeakMap<LangChainMessage, { key: string; message: LangChainMessage }>();\n\nfunction withAnswerRowKey(message: LangChainMessage, key: string): LangChainMessage {\n  const cached = keyedMessages.get(message);\n  if (cached?.key === key) return cached.message;\n  const keyed: LangChainMessage = { ...message };\n  // A symbol never reaches JSON, so a keyed message sent back to agora (a\n  // branch, an edit) carries nothing it didn't receive.\n  Reflect.set(keyed, ANSWER_ROW_KEY, key);\n  keyedMessages.set(message, { key, message: keyed });\n  return keyed;\n}\n\n/**\n * Each AI message's stamped answerer key (`null` when unstamped), cached by\n * message object: the messages handed to {@link withAnswerRows} keep their\n * object while unchanged, so a streamed token reads only the stamp of the\n * message it changed.\n */\nconst stampKeys = new WeakMap<LangChainMessage, string | null>();\n\nfunction stampedAnswererKey(message: LangChainMessage): string | null {\n  const cached = stampKeys.get(message);\n  if (cached !== undefined) return cached;\n  const answer = readLangChainAnswer(message);\n  const key = answer === null ? null : answerRowRunId(answer.agent, answer.for_id);\n  stampKeys.set(message, key);\n  return key;\n}\n\n/** The id of the row a next agent of a message several agents answer shows before its first token. */\nexport const NEXT_AGENT_PENDING_MESSAGE_ID = '__athena_deep_agent_pending_next_agent__';\n\n/** The pending row for each answerer key: one object per key, so the row keeps its identity. */\nconst nextAgentPlaceholders = new Map<string, LangChainMessage>();\n\nfunction nextAgentPlaceholder(key: string): LangChainMessage {\n  const cached = nextAgentPlaceholders.get(key);\n  if (cached !== undefined) return cached;\n  // Empty content with an explicit running status: the thread's thinking\n  // status keys on a running assistant row with no parts, as the first\n  // agent's pending row does (`withPendingAssistantMessage`).\n  const placeholder: LangChainMessage = {\n    type: 'ai',\n    content: [],\n    id: NEXT_AGENT_PENDING_MESSAGE_ID,\n    status: { type: 'running' },\n  };\n  Reflect.set(placeholder, ANSWER_ROW_KEY, key);\n  Object.freeze(placeholder);\n  // Bounded by the answerers a client meets; each entry is a few strings.\n  if (nextAgentPlaceholders.size >= 64) nextAgentPlaceholders.clear();\n  nextAgentPlaceholders.set(key, placeholder);\n  return placeholder;\n}\n\nconst isTurnBoundary = (message: LangChainMessage): boolean =>\n  message.type === 'human' || message.type === 'system';\n\n/** The agent answering now, from the message its turn began after on. */\ninterface AnsweringAgent {\n  key: string;\n  anchorIndex: number;\n}\n\nfunction answeringAgent({\n  messages,\n  addressedTurn,\n}: {\n  messages: readonly LangChainMessage[];\n  addressedTurn: DeepAgentAddressedTurn | null;\n}): AnsweringAgent | null {\n  // The first agent answers the message itself: its run brings the message as\n  // input, so its answer starts its own row anyway.\n  if (addressedTurn === null || addressedTurn.position === 0 || addressedTurn.anchor === null) {\n    return null;\n  }\n  const agent = addressedTurn.agents[addressedTurn.position];\n  if (agent === undefined) return null;\n  // The anchor is the transcript's last message when the agent's turn began:\n  // read from the tail.\n  for (let index = messages.length - 1; index >= 0; index -= 1) {\n    const message = messages[index];\n    if (message === undefined) continue;\n    if (message.id === addressedTurn.anchor) {\n      return { key: answerRowRunId(agent, addressedTurn.authorId), anchorIndex: index };\n    }\n    // A person's message after the anchor means the record is stale.\n    if (isTurnBoundary(message)) return null;\n  }\n  return null;\n}\n\n/**\n * Key one turn's AI messages (`messages[start..end)`) by their answerer when\n * two or more answerers wrote them; `null` otherwise.\n */\nfunction turnKeys({\n  messages,\n  start,\n  end,\n  answering,\n}: {\n  messages: readonly LangChainMessage[];\n  start: number;\n  end: number;\n  answering: AnsweringAgent | null;\n}): (string | null)[] | null {\n  const keys: (string | null)[] = [];\n  let first: string | null = null;\n  let several = false;\n  // The answering agent's own stamp, once one of its answers carries it: its\n  // later messages follow that stamp, so a record and a stamp spelling the\n  // author apart never split one agent's answer in two.\n  let answeringStamp: string | null = null;\n  for (let index = start; index < end; index += 1) {\n    const message = messages[index];\n    if (message?.type !== 'ai') {\n      keys.push(null);\n      continue;\n    }\n    const answeringKey =\n      answering !== null && index > answering.anchorIndex ? answering.key : null;\n    let key = stampedAnswererKey(message);\n    if (key !== null && answeringKey !== null) answeringStamp = key;\n    else if (key === null && answeringKey !== null) key = answeringStamp ?? answeringKey;\n    keys.push(key);\n    if (key === null) continue;\n    if (first === null) first = key;\n    else if (key !== first) several = true;\n  }\n  if (!several) return null;\n  // An AI message nobody stamped joins the answerer before it in the turn,\n  // else the first one after it.\n  let previous: string | null = null;\n  for (let offset = 0; offset < keys.length; offset += 1) {\n    if (messages[start + offset]?.type !== 'ai') continue;\n    const key = keys[offset] ?? null;\n    if (key === null) keys[offset] = previous;\n    else previous = key;\n  }\n  let next: string | null = null;\n  for (let offset = keys.length - 1; offset >= 0; offset -= 1) {\n    if (messages[start + offset]?.type !== 'ai') continue;\n    const key = keys[offset] ?? null;\n    if (key === null) keys[offset] = next;\n    else next = key;\n  }\n  return keys;\n}\n\n/**\n * The messages with every answer of a turn several agents answered keyed by\n * its answerer (see the module comment), so each agent's answer converts into\n * a row of its own; the same array when no turn needs it.\n *\n * While a next agent is answering (`addressedTurn` past its first agent) and\n * the run is live but the agent hasn't sent its first token, its row shows as\n * pending (a running placeholder, {@link NEXT_AGENT_PENDING_MESSAGE_ID}), as the\n * first agent's does.\n *\n * O(the messages) per pass, with no allocation for a transcript nobody but one\n * answerer per turn wrote.\n */\nexport function withAnswerRows({\n  messages,\n  addressedTurn,\n  isRunning,\n}: {\n  messages: LangChainMessage[];\n  addressedTurn: DeepAgentAddressedTurn | null;\n  isRunning: boolean;\n}): LangChainMessage[] {\n  const answering = answeringAgent({ messages, addressedTurn });\n  const pending =\n    answering !== null &&\n    isRunning &&\n    !messages.slice(answering.anchorIndex + 1).some((message) => message.type === 'ai');\n  const sources = pending ? [...messages, nextAgentPlaceholder(answering.key)] : messages;\n  let keyed: LangChainMessage[] | null = null;\n  let start = 0;\n  for (let index = 0; index <= sources.length; index += 1) {\n    const message = sources[index];\n    if (message !== undefined && !isTurnBoundary(message)) continue;\n    if (index - start > 1) {\n      const keys = turnKeys({ messages: sources, start, end: index, answering });\n      if (keys !== null) {\n        for (const [offset, key] of keys.entries()) {\n          const source = sources[start + offset];\n          if (key === null || source === undefined || readAnswerRowKey(source) === key) continue;\n          keyed ??= [...sources];\n          keyed[start + offset] = withAnswerRowKey(source, key);\n        }\n      }\n    }\n    start = index + 1;\n  }\n  return keyed ?? sources;\n}\n"],"names":["readLangChainAnswer","answerRowRunId"],"mappings":";;;AAiCO,MAAM,2BAA2B;AAGxC,MAAM,gCAAgC;AAGtC,MAAM,uBAAuB;AAE7B,MAAM,oBAAoB;AAyB1B,MAAM,WAAW,CAAC,UAChB,OAAO,UAAU,YAAY,UAAU,QAAQ,CAAC,MAAM,QAAQ,KAAK;AAErE,MAAM,aAAa,CAAC,UAClB,OAAO,UAAU,YAAY,kBAAkB,KAAK,KAAK;AAG3D,SAAS,YAAY,OAAkD;AACrE,MAAI,CAAC,SAAS,KAAK,EAAG,QAAO;AAC7B,QAAM,EAAE,YAAY;AACpB,SAAO;AAAA,IACL,OAAO,OAAO,MAAM,KAAK;AAAA,IACzB,SAAS,YAAY,cAAc,YAAY,YAAY,UAAU;AAAA,IACrE,MAAM,OAAO,MAAM,SAAS,YAAY,MAAM,KAAK,SAAS,IAAI,MAAM,OAAO;AAAA,EAAA;AAEjF;AASO,SAAS,2BAA2B,OAA+C;AACxF,MAAI,CAAC,SAAS,KAAK,KAAK,MAAM,MAAM,8BAA+B,QAAO;AAC1E,QAAM,EAAE,QAAQ,WAAW,UAAU,QAAQ,UAAU,QAAQ,UAAU,QAAA,IAAY;AACrF,MAAI,OAAO,WAAW,YAAY,OAAO,WAAW,EAAG,QAAO;AAC9D,MAAI,OAAO,aAAa,YAAY,SAAS,WAAW,EAAG,QAAO;AAClE,MAAI,YAAY,UAAa,OAAO,YAAY,UAAW,QAAO;AAClE,MACE,CAAC,MAAM,QAAQ,MAAM,KACrB,OAAO,SAAS,KAChB,OAAO,SAAS,wBAChB,CAAC,OAAO,MAAM,UAAU,GACxB;AACA,WAAO;AAAA,EACT;AACA,MAAI,OAAO,aAAa,YAAY,CAAC,OAAO,UAAU,QAAQ,EAAG,QAAO;AACxE,MAAI,WAAW,KAAK,YAAY,OAAO,OAAQ,QAAO;AACtD,MAAI,WAAW,QAAQ,WAAW,WAAc,OAAO,WAAW,YAAY,CAAC,SAAS;AACtF,WAAO;AAAA,EACT;AACA,MAAI,CAAC,MAAM,QAAQ,QAAQ,EAAG,QAAO;AACrC,QAAM,WAAW,SAAS,IAAI,WAAW;AACzC,MAAI,SAAS,KAAK,CAAC,YAAY,YAAY,IAAI,EAAG,QAAO;AACzD,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA,QAAQ,UAAU;AAAA,IAClB,UAAU,SAAS,OAAO,CAAC,YAAkD,YAAY,IAAI;AAAA,EAAA;AAEjG;AAGA,MAAM,iBAAiB,OAAO,IAAI,uCAAuC;AAOlE,SAAS,iBAAiB,SAA+C;AAC9E,QAAM,MAAe,QAAQ,IAAI,SAAS,cAAc;AACxD,SAAO,OAAO,QAAQ,WAAW,MAAM;AACzC;AAQA,MAAM,oCAAoB,QAAA;AAE1B,SAAS,iBAAiB,SAA2B,KAA+B;AAClF,QAAM,SAAS,cAAc,IAAI,OAAO;AACxC,OAAI,iCAAQ,SAAQ,IAAK,QAAO,OAAO;AACvC,QAAM,QAA0B,EAAE,GAAG,QAAA;AAGrC,UAAQ,IAAI,OAAO,gBAAgB,GAAG;AACtC,gBAAc,IAAI,SAAS,EAAE,KAAK,SAAS,OAAO;AAClD,SAAO;AACT;AAQA,MAAM,gCAAgB,QAAA;AAEtB,SAAS,mBAAmB,SAA0C;AACpE,QAAM,SAAS,UAAU,IAAI,OAAO;AACpC,MAAI,WAAW,OAAW,QAAO;AACjC,QAAM,SAASA,YAAAA,oBAAoB,OAAO;AAC1C,QAAM,MAAM,WAAW,OAAO,OAAOC,YAAAA,eAAe,OAAO,OAAO,OAAO,MAAM;AAC/E,YAAU,IAAI,SAAS,GAAG;AAC1B,SAAO;AACT;AAGO,MAAM,gCAAgC;AAG7C,MAAM,4CAA4B,IAAA;AAElC,SAAS,qBAAqB,KAA+B;AAC3D,QAAM,SAAS,sBAAsB,IAAI,GAAG;AAC5C,MAAI,WAAW,OAAW,QAAO;AAIjC,QAAM,cAAgC;AAAA,IACpC,MAAM;AAAA,IACN,SAAS,CAAA;AAAA,IACT,IAAI;AAAA,IACJ,QAAQ,EAAE,MAAM,UAAA;AAAA,EAAU;AAE5B,UAAQ,IAAI,aAAa,gBAAgB,GAAG;AAC5C,SAAO,OAAO,WAAW;AAEzB,MAAI,sBAAsB,QAAQ,GAAI,uBAAsB,MAAA;AAC5D,wBAAsB,IAAI,KAAK,WAAW;AAC1C,SAAO;AACT;AAEA,MAAM,iBAAiB,CAAC,YACtB,QAAQ,SAAS,WAAW,QAAQ,SAAS;AAQ/C,SAAS,eAAe;AAAA,EACtB;AAAA,EACA;AACF,GAG0B;AAGxB,MAAI,kBAAkB,QAAQ,cAAc,aAAa,KAAK,cAAc,WAAW,MAAM;AAC3F,WAAO;AAAA,EACT;AACA,QAAM,QAAQ,cAAc,OAAO,cAAc,QAAQ;AACzD,MAAI,UAAU,OAAW,QAAO;AAGhC,WAAS,QAAQ,SAAS,SAAS,GAAG,SAAS,GAAG,SAAS,GAAG;AAC5D,UAAM,UAAU,SAAS,KAAK;AAC9B,QAAI,YAAY,OAAW;AAC3B,QAAI,QAAQ,OAAO,cAAc,QAAQ;AACvC,aAAO,EAAE,KAAKA,YAAAA,eAAe,OAAO,cAAc,QAAQ,GAAG,aAAa,MAAA;AAAA,IAC5E;AAEA,QAAI,eAAe,OAAO,EAAG,QAAO;AAAA,EACtC;AACA,SAAO;AACT;AAMA,SAAS,SAAS;AAAA,EAChB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAK6B;;AAC3B,QAAM,OAA0B,CAAA;AAChC,MAAI,QAAuB;AAC3B,MAAI,UAAU;AAId,MAAI,iBAAgC;AACpC,WAAS,QAAQ,OAAO,QAAQ,KAAK,SAAS,GAAG;AAC/C,UAAM,UAAU,SAAS,KAAK;AAC9B,SAAI,mCAAS,UAAS,MAAM;AAC1B,WAAK,KAAK,IAAI;AACd;AAAA,IACF;AACA,UAAM,eACJ,cAAc,QAAQ,QAAQ,UAAU,cAAc,UAAU,MAAM;AACxE,QAAI,MAAM,mBAAmB,OAAO;AACpC,QAAI,QAAQ,QAAQ,iBAAiB,KAAM,kBAAiB;AAAA,aACnD,QAAQ,QAAQ,iBAAiB,YAAY,kBAAkB;AACxE,SAAK,KAAK,GAAG;AACb,QAAI,QAAQ,KAAM;AAClB,QAAI,UAAU,KAAM,SAAQ;AAAA,aACnB,QAAQ,MAAO,WAAU;AAAA,EACpC;AACA,MAAI,CAAC,QAAS,QAAO;AAGrB,MAAI,WAA0B;AAC9B,WAAS,SAAS,GAAG,SAAS,KAAK,QAAQ,UAAU,GAAG;AACtD,UAAI,cAAS,QAAQ,MAAM,MAAvB,mBAA0B,UAAS,KAAM;AAC7C,UAAM,MAAM,KAAK,MAAM,KAAK;AAC5B,QAAI,QAAQ,KAAM,MAAK,MAAM,IAAI;AAAA,QAC5B,YAAW;AAAA,EAClB;AACA,MAAI,OAAsB;AAC1B,WAAS,SAAS,KAAK,SAAS,GAAG,UAAU,GAAG,UAAU,GAAG;AAC3D,UAAI,cAAS,QAAQ,MAAM,MAAvB,mBAA0B,UAAS,KAAM;AAC7C,UAAM,MAAM,KAAK,MAAM,KAAK;AAC5B,QAAI,QAAQ,KAAM,MAAK,MAAM,IAAI;AAAA,QAC5B,QAAO;AAAA,EACd;AACA,SAAO;AACT;AAeO,SAAS,eAAe;AAAA,EAC7B;AAAA,EACA;AAAA,EACA;AACF,GAIuB;AACrB,QAAM,YAAY,eAAe,EAAE,UAAU,eAAe;AAC5D,QAAM,UACJ,cAAc,QACd,aACA,CAAC,SAAS,MAAM,UAAU,cAAc,CAAC,EAAE,KAAK,CAAC,YAAY,QAAQ,SAAS,IAAI;AACpF,QAAM,UAAU,UAAU,CAAC,GAAG,UAAU,qBAAqB,UAAU,GAAG,CAAC,IAAI;AAC/E,MAAI,QAAmC;AACvC,MAAI,QAAQ;AACZ,WAAS,QAAQ,GAAG,SAAS,QAAQ,QAAQ,SAAS,GAAG;AACvD,UAAM,UAAU,QAAQ,KAAK;AAC7B,QAAI,YAAY,UAAa,CAAC,eAAe,OAAO,EAAG;AACvD,QAAI,QAAQ,QAAQ,GAAG;AACrB,YAAM,OAAO,SAAS,EAAE,UAAU,SAAS,OAAO,KAAK,OAAO,WAAW;AACzE,UAAI,SAAS,MAAM;AACjB,mBAAW,CAAC,QAAQ,GAAG,KAAK,KAAK,WAAW;AAC1C,gBAAM,SAAS,QAAQ,QAAQ,MAAM;AACrC,cAAI,QAAQ,QAAQ,WAAW,UAAa,iBAAiB,MAAM,MAAM,IAAK;AAC9E,4BAAU,CAAC,GAAG,OAAO;AACrB,gBAAM,QAAQ,MAAM,IAAI,iBAAiB,QAAQ,GAAG;AAAA,QACtD;AAAA,MACF;AAAA,IACF;AACA,YAAQ,QAAQ;AAAA,EAClB;AACA,SAAO,SAAS;AAClB;;;;;;"}

package/dist/agent-runtime/src/answer-rows.d.ts

@@ -0,0 +1,86 @@
+import type { LangChainMessage } from '@assistant-ui/react-langgraph';
+/**
+ * One row per answering agent (identity v7 D13: several agents answer one
+ * message, each in its own turn for its author).
+ *
+ * assistant-ui joins consecutive assistant messages into one row unless their
+ * `metadata.runId` differs (`chunkExternalMessages` in `@assistant-ui/core`).
+ * The agents that answer one message answer in turn, with nothing between
+ * them, so without a run id per answer every agent's answer lands in one row.
+ * This module gives each AI message the answerer it belongs to, which the
+ * converter turns into `metadata.runId` (`convertDeepAgentLangChainMessage`):
+ *
+ * - a stamped answer: its `answered_by` stamp (agent and principal), written by
+ *   agora when its model turn lands;
+ * - an answer still streaming while several agents answer: the agent the
+ *   host's record says is answering (`addressedTurn`, replicated statewire
+ *   state), for every message after the message its turn began after. The
+ *   stamp lands only once a model turn finishes, so reading the record is what
+ *   keeps a next agent's answer in its own row from its first token instead of
+ *   joining the previous agent's row and jumping out when its stamp lands;
+ * - anything else (an answer written before the stamp existed, a reused one,
+ *   a run marker): the answerer of the AI message before it in the same turn,
+ *   else of the first one after it.
+ *
+ * Only a turn (the messages between two people's messages) whose answers come
+ * from two or more answerers is given run ids: a 1:1 chat, a turn one agent
+ * answered and every transcript written before the stamp keep their messages'
+ * identity and convert exactly as before.
+ */
+/** Statewire state key of the message several agents are answering (agora `several_agents.ADDRESSED_TURN_STATE_KEY`). */
+export declare const ADDRESSED_TURN_STATE_KEY = "addressedTurn";
+/** How one agent's turn of a message several agents answer ended. */
+export interface DeepAgentAddressedOutcome {
+    /** The agent's key (`collab_agent:<asset id>`, `assistant:<principal id>`). */
+    agent: string;
+    outcome: 'answered' | 'failed' | 'refused';
+    /** A refusal's code (e.g. `conversation_history_recheck_required`); `null` otherwise. */
+    code: string | null;
+}
+/** The message several agents are answering, as the host's replicated record holds it. */
+export interface DeepAgentAddressedTurn {
+    /** Whom every agent of the message answers for: the author's principal id. */
+    authorId: string;
+    /** The named agents' keys, in the order they answer. */
+    agents: readonly string[];
+    /** The index in {@link agents} of the agent answering now. */
+    position: number;
+    /** The message the answering agent's turn began after; `null` for the first agent. */
+    anchor: string | null;
+    /** Each finished agent's outcome, in order. */
+    outcomes: readonly DeepAgentAddressedOutcome[];
+}
+/**
+ * The record of the message several agents are answering, off the replicated
+ * state value (`state.addressedTurn`); `null` while none is (agora clears it
+ * once the last agent settled) and for anything this reader can't trust,
+ * mirroring agora's own `read_record`: another version, a malformed field, an
+ * agent key it can't read, a position outside the list.
+ */
+export declare function readDeepAgentAddressedTurn(value: unknown): DeepAgentAddressedTurn | null;
+/**
+ * The answerer an AI message's row is keyed by, as {@link withAnswerRows} put
+ * it on the message; `undefined` for a message it left alone. The converter
+ * passes it on as the row's `metadata.runId`.
+ */
+export declare function readAnswerRowKey(message: LangChainMessage): string | undefined;
+/** The id of the row a next agent of a message several agents answer shows before its first token. */
+export declare const NEXT_AGENT_PENDING_MESSAGE_ID = "__athena_deep_agent_pending_next_agent__";
+/**
+ * The messages with every answer of a turn several agents answered keyed by
+ * its answerer (see the module comment), so each agent's answer converts into
+ * a row of its own; the same array when no turn needs it.
+ *
+ * While a next agent is answering (`addressedTurn` past its first agent) and
+ * the run is live but the agent hasn't sent its first token, its row shows as
+ * pending (a running placeholder, {@link NEXT_AGENT_PENDING_MESSAGE_ID}), as the
+ * first agent's does.
+ *
+ * O(the messages) per pass, with no allocation for a transcript nobody but one
+ * answerer per turn wrote.
+ */
+export declare function withAnswerRows({ messages, addressedTurn, isRunning, }: {
+    messages: LangChainMessage[];
+    addressedTurn: DeepAgentAddressedTurn | null;
+    isRunning: boolean;
+}): LangChainMessage[];

package/dist/agent-runtime/src/answer-rows.js

@@ -0,0 +1,180 @@
+import { answerRowRunId, readLangChainAnswer } from "./answer-label.js";
+const ADDRESSED_TURN_STATE_KEY = "addressedTurn";
+const ADDRESSED_TURN_RECORD_VERSION = 1;
+const MAX_ADDRESSED_AGENTS = 8;
+const AGENT_KEY_PATTERN = /^(collab_agent|assistant):\S+$/;
+const isRecord = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
+const isAgentKey = (value) => typeof value === "string" && AGENT_KEY_PATTERN.test(value);
+function readOutcome(value) {
+  if (!isRecord(value)) return null;
+  const { outcome } = value;
+  return {
+    agent: String(value.agent),
+    outcome: outcome === "answered" || outcome === "refused" ? outcome : "failed",
+    code: typeof value.code === "string" && value.code.length > 0 ? value.code : null
+  };
+}
+function readDeepAgentAddressedTurn(value) {
+  if (!isRecord(value) || value.v !== ADDRESSED_TURN_RECORD_VERSION) return null;
+  const { author, author_id: authorId, agents, position, anchor, outcomes, started } = value;
+  if (typeof author !== "string" || author.length === 0) return null;
+  if (typeof authorId !== "string" || authorId.length === 0) return null;
+  if (started !== void 0 && typeof started !== "boolean") return null;
+  if (!Array.isArray(agents) || agents.length < 2 || agents.length > MAX_ADDRESSED_AGENTS || !agents.every(isAgentKey)) {
+    return null;
+  }
+  if (typeof position !== "number" || !Number.isInteger(position)) return null;
+  if (position < 0 || position >= agents.length) return null;
+  if (anchor !== null && anchor !== void 0 && (typeof anchor !== "string" || !anchor)) {
+    return null;
+  }
+  if (!Array.isArray(outcomes)) return null;
+  const finished = outcomes.map(readOutcome);
+  if (finished.some((outcome) => outcome === null)) return null;
+  return {
+    authorId,
+    agents,
+    position,
+    anchor: anchor ?? null,
+    outcomes: finished.filter((outcome) => outcome !== null)
+  };
+}
+const ANSWER_ROW_KEY = Symbol.for("@athenaintel/agent-runtime/answer-row");
+function readAnswerRowKey(message) {
+  const key = Reflect.get(message, ANSWER_ROW_KEY);
+  return typeof key === "string" ? key : void 0;
+}
+const keyedMessages = /* @__PURE__ */ new WeakMap();
+function withAnswerRowKey(message, key) {
+  const cached = keyedMessages.get(message);
+  if ((cached == null ? void 0 : cached.key) === key) return cached.message;
+  const keyed = { ...message };
+  Reflect.set(keyed, ANSWER_ROW_KEY, key);
+  keyedMessages.set(message, { key, message: keyed });
+  return keyed;
+}
+const stampKeys = /* @__PURE__ */ new WeakMap();
+function stampedAnswererKey(message) {
+  const cached = stampKeys.get(message);
+  if (cached !== void 0) return cached;
+  const answer = readLangChainAnswer(message);
+  const key = answer === null ? null : answerRowRunId(answer.agent, answer.for_id);
+  stampKeys.set(message, key);
+  return key;
+}
+const NEXT_AGENT_PENDING_MESSAGE_ID = "__athena_deep_agent_pending_next_agent__";
+const nextAgentPlaceholders = /* @__PURE__ */ new Map();
+function nextAgentPlaceholder(key) {
+  const cached = nextAgentPlaceholders.get(key);
+  if (cached !== void 0) return cached;
+  const placeholder = {
+    type: "ai",
+    content: [],
+    id: NEXT_AGENT_PENDING_MESSAGE_ID,
+    status: { type: "running" }
+  };
+  Reflect.set(placeholder, ANSWER_ROW_KEY, key);
+  Object.freeze(placeholder);
+  if (nextAgentPlaceholders.size >= 64) nextAgentPlaceholders.clear();
+  nextAgentPlaceholders.set(key, placeholder);
+  return placeholder;
+}
+const isTurnBoundary = (message) => message.type === "human" || message.type === "system";
+function answeringAgent({
+  messages,
+  addressedTurn
+}) {
+  if (addressedTurn === null || addressedTurn.position === 0 || addressedTurn.anchor === null) {
+    return null;
+  }
+  const agent = addressedTurn.agents[addressedTurn.position];
+  if (agent === void 0) return null;
+  for (let index = messages.length - 1; index >= 0; index -= 1) {
+    const message = messages[index];
+    if (message === void 0) continue;
+    if (message.id === addressedTurn.anchor) {
+      return { key: answerRowRunId(agent, addressedTurn.authorId), anchorIndex: index };
+    }
+    if (isTurnBoundary(message)) return null;
+  }
+  return null;
+}
+function turnKeys({
+  messages,
+  start,
+  end,
+  answering
+}) {
+  var _a, _b;
+  const keys = [];
+  let first = null;
+  let several = false;
+  let answeringStamp = null;
+  for (let index = start; index < end; index += 1) {
+    const message = messages[index];
+    if ((message == null ? void 0 : message.type) !== "ai") {
+      keys.push(null);
+      continue;
+    }
+    const answeringKey = answering !== null && index > answering.anchorIndex ? answering.key : null;
+    let key = stampedAnswererKey(message);
+    if (key !== null && answeringKey !== null) answeringStamp = key;
+    else if (key === null && answeringKey !== null) key = answeringStamp ?? answeringKey;
+    keys.push(key);
+    if (key === null) continue;
+    if (first === null) first = key;
+    else if (key !== first) several = true;
+  }
+  if (!several) return null;
+  let previous = null;
+  for (let offset = 0; offset < keys.length; offset += 1) {
+    if (((_a = messages[start + offset]) == null ? void 0 : _a.type) !== "ai") continue;
+    const key = keys[offset] ?? null;
+    if (key === null) keys[offset] = previous;
+    else previous = key;
+  }
+  let next = null;
+  for (let offset = keys.length - 1; offset >= 0; offset -= 1) {
+    if (((_b = messages[start + offset]) == null ? void 0 : _b.type) !== "ai") continue;
+    const key = keys[offset] ?? null;
+    if (key === null) keys[offset] = next;
+    else next = key;
+  }
+  return keys;
+}
+function withAnswerRows({
+  messages,
+  addressedTurn,
+  isRunning
+}) {
+  const answering = answeringAgent({ messages, addressedTurn });
+  const pending = answering !== null && isRunning && !messages.slice(answering.anchorIndex + 1).some((message) => message.type === "ai");
+  const sources = pending ? [...messages, nextAgentPlaceholder(answering.key)] : messages;
+  let keyed = null;
+  let start = 0;
+  for (let index = 0; index <= sources.length; index += 1) {
+    const message = sources[index];
+    if (message !== void 0 && !isTurnBoundary(message)) continue;
+    if (index - start > 1) {
+      const keys = turnKeys({ messages: sources, start, end: index, answering });
+      if (keys !== null) {
+        for (const [offset, key] of keys.entries()) {
+          const source = sources[start + offset];
+          if (key === null || source === void 0 || readAnswerRowKey(source) === key) continue;
+          keyed ?? (keyed = [...sources]);
+          keyed[start + offset] = withAnswerRowKey(source, key);
+        }
+      }
+    }
+    start = index + 1;
+  }
+  return keyed ?? sources;
+}
+export {
+  ADDRESSED_TURN_STATE_KEY,
+  NEXT_AGENT_PENDING_MESSAGE_ID,
+  readAnswerRowKey,
+  readDeepAgentAddressedTurn,
+  withAnswerRows
+};
+//# sourceMappingURL=answer-rows.js.map