@epam/ai-dial-quotations 1.1.0-dev.405 → 1.1.0-dev.411

package/context/CitationCardContext.d.ts

@@ -3,11 +3,11 @@ import { useCitationCard } from '../utils/useCitationCard';
 export type CitationCardHook = ReturnType<typeof useCitationCard>;
 /** Provider that supplies `CitationCardHook` state to a citation subtree. */
 export declare const CitationCardProvider: import('react').Provider<{
-    openPopup: (sourceUrl: string) => void;
+    openPopup: (groupKey: string) => void;
     closePopup: () => void;
-    setActiveIndex: (sourceUrl: string, index: number) => void;
-    isOpen: (sourceUrl: string) => boolean;
-    getActiveIndex: (sourceUrl: string) => number;
+    setActiveIndex: (groupKey: string, index: number) => void;
+    isOpen: (groupKey: string) => boolean;
+    getActiveIndex: (groupKey: string) => number;
 } | null>;
 /** Returns the nearest `CitationCardHook` from context. Throws when used outside a `CitationCardProvider`. */
 export declare const useCitationCardContext: () => CitationCardHook;

package/hooks/useCitationMarkdownComponents/useCitationMarkdownComponents.d.ts

@@ -16,25 +16,32 @@ export interface UseCitationMarkdownComponentsCallbacks {
     };
 }
 /**
- * Builds react-markdown component overrides that inject citation markers into
- * rendered paragraph text at the character offsets stored in each annotation
+ * Builds react-markdown component overrides that inject citation markers
+ * into rendered assistant message content: a `cit` element override for
+ * `<cit data-id="…">` tags (rendered as real elements once `rehype-raw` and
+ * the host's `rehype-sanitize` allowlist parse them — see
+ * `MarkdownRenderer`'s `baseRehypePlugins`), and `p`/`li` overrides that
+ * inject markers at the character offsets of every other annotation
  * group's primary selector.
  *
- * The `p` and `li` component overrides are stable references — they only
- * change when `groups` transitions between empty and non-empty. Citation card
- * open/close state is provided via `CitationCardContext` so that state changes
- * do not recreate the component functions, preventing ReactMarkdown from
- * unmounting and remounting the paragraph subtree on every interaction.
+ * Citation card open/close state is provided via `CitationCardContext` so
+ * that state changes do not recreate the component functions, preventing
+ * ReactMarkdown from unmounting and remounting the paragraph subtree on
+ * every interaction.
  *
- * Returns both the pre-processed content string (with sentinel placeholders
- * injected at the right offsets) and the `Components` map to pass to the
- * markdown renderer.
+ * Returns both the pre-processed content string and the `Components` map to
+ * pass to the markdown renderer. While `isStreaming` is true, complete
+ * supported `<cit data-id="…"></cit>` elements are hidden from
+ * `processedContent`; incomplete or otherwise unsupported `cit` markup is
+ * escaped and remains visible as literal text. Citation pills for supported
+ * `<cit>` elements only appear once the message has finished streaming,
+ * matching every other citation family.
  *
  * `isCompactTypography` drops the paragraph class one type-scale step, matching
  * `COMPACT_MARKDOWN_CLASS_NAMES` so cited and uncited paragraphs stay the same
  * size.
  */
-export declare const useCitationMarkdownComponents: (content: string, groups: AnnotationGroup[], callbacks: UseCitationMarkdownComponentsCallbacks, isCompactTypography?: boolean) => {
+export declare const useCitationMarkdownComponents: (content: string, groups: AnnotationGroup[], callbacks: UseCitationMarkdownComponentsCallbacks, isStreaming?: boolean, isCompactTypography?: boolean) => {
     processedContent: string;
     markdownComponents: Components;
 };

package/hooks/useCitationMarkdownComponents/useCitationMarkdownComponents.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useCitationMarkdownComponents.d.ts","sourceRoot":"","sources":["../../../src/hooks/useCitationMarkdownComponents/useCitationMarkdownComponents.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAgB,KAAK,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAE1E,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,4CAA4C,CAAC;AAErF,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,gDAAgD,CAAC;AAK3F,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,yCAAyC,CAAC;AAE/E,6FAA6F;AAC7F,MAAM,WAAW,sCAAsC;IACrD,4GAA4G;IAC5G,SAAS,CAAC,UAAU,EAAE,UAAU,EAAE,KAAK,EAAE,eAAe,GAAG,IAAI,CAAC;IAChE,yEAAyE;IACzE,eAAe,CAAC,UAAU,EAAE,UAAU,GAAG,IAAI,CAAC;IAC9C,4FAA4F;IAC5F,WAAW,CAAC,KAAK,EAAE,eAAe,GAAG;QACnC,UAAU,EAAE,kBAAkB,CAAC;QAC/B,YAAY,EAAE,oBAAoB,CAAC;KACpC,CAAC;CACH;AAED;;;;;;;;;;;;;;;;;;GAkBG;AACH,eAAO,MAAM,6BAA6B,GACxC,SAAS,MAAM,EACf,QAAQ,eAAe,EAAE,EACzB,WAAW,sCAAsC,EACjD,6BAA2B,KAC1B;IAAE,gBAAgB,EAAE,MAAM,CAAC;IAAC,kBAAkB,EAAE,UAAU,CAAA;CAsD5D,CAAC"}
+{"version":3,"file":"useCitationMarkdownComponents.d.ts","sourceRoot":"","sources":["../../../src/hooks/useCitationMarkdownComponents/useCitationMarkdownComponents.tsx"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,UAAU,EAEhB,MAAM,2BAA2B,CAAC;AAEnC,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,4CAA4C,CAAC;AAErF,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,gDAAgD,CAAC;AAO3F,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,yCAAyC,CAAC;AAE/E,6FAA6F;AAC7F,MAAM,WAAW,sCAAsC;IACrD,4GAA4G;IAC5G,SAAS,CAAC,UAAU,EAAE,UAAU,EAAE,KAAK,EAAE,eAAe,GAAG,IAAI,CAAC;IAChE,yEAAyE;IACzE,eAAe,CAAC,UAAU,EAAE,UAAU,GAAG,IAAI,CAAC;IAC9C,4FAA4F;IAC5F,WAAW,CAAC,KAAK,EAAE,eAAe,GAAG;QACnC,UAAU,EAAE,kBAAkB,CAAC;QAC/B,YAAY,EAAE,oBAAoB,CAAC;KACpC,CAAC;CACH;AA2BD;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,eAAO,MAAM,6BAA6B,GACxC,SAAS,MAAM,EACf,QAAQ,eAAe,EAAE,EACzB,WAAW,sCAAsC,EACjD,qBAAmB,EACnB,6BAA2B,KAC1B;IAAE,gBAAgB,EAAE,MAAM,CAAC;IAAC,kBAAkB,EAAE,UAAU,CAAA;CA8F5D,CAAC"}

package/index.d.ts

@@ -10,10 +10,10 @@ export { useAnnotations } from './utils/useAnnotations';
 export { useCitationCard } from './utils/useCitationCard';
 export { useCitationMarkdownComponents } from './hooks/useCitationMarkdownComponents/useCitationMarkdownComponents';
 export type { UseCitationMarkdownComponentsCallbacks } from './hooks/useCitationMarkdownComponents/useCitationMarkdownComponents';
-export { groupAnnotationsBySource } from './utils/group-annotations-by-source';
+export { groupAnnotations, groupAnnotationsByCitId, groupAnnotationsBySource, } from './utils/group-annotations-by-source';
 export type { AnnotationGroup } from './utils/group-annotations-by-source';
 export { annotationsToPdfHighlights, annotationHighlightId, normalizeRawAnnotations, resolveMessageAnnotations, } from './utils/annotation';
-export { injectCitationSentinels, replaceSentinelsInChildren, } from './utils/citation-injection';
+export { injectCitationSentinels, replaceSentinelsInChildren, stripCitTagsWhileStreaming, } from './utils/citation-injection';
 export { isReferenceOnlyAttachment, parsePdfPageReference, getReferenceAttachmentGroups, } from './utils/reference-attachment';
 export type { PdfPageReference } from './utils/reference-attachment';
 //# sourceMappingURL=index.d.ts.map

package/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,wCAAwC,CAAC;AACtE,YAAY,EACV,kBAAkB,EAClB,iBAAiB,EACjB,sBAAsB,GACvB,MAAM,wCAAwC,CAAC;AAEhD,OAAO,EAAE,cAAc,EAAE,MAAM,4CAA4C,CAAC;AAC5E,YAAY,EACV,oBAAoB,EACpB,mBAAmB,GACpB,MAAM,4CAA4C,CAAC;AAEpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,gDAAgD,CAAC;AAClF,YAAY,EAAE,qBAAqB,EAAE,MAAM,gDAAgD,CAAC;AAG5F,OAAO,EACL,oBAAoB,EACpB,sBAAsB,GACvB,MAAM,+BAA+B,CAAC;AACvC,YAAY,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AAGtE,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,6BAA6B,EAAE,MAAM,qEAAqE,CAAC;AACpH,YAAY,EAAE,sCAAsC,EAAE,MAAM,qEAAqE,CAAC;AAGlI,OAAO,EAAE,wBAAwB,EAAE,MAAM,qCAAqC,CAAC;AAC/E,YAAY,EAAE,eAAe,EAAE,MAAM,qCAAqC,CAAC;AAE3E,OAAO,EACL,0BAA0B,EAC1B,qBAAqB,EACrB,uBAAuB,EACvB,yBAAyB,GAC1B,MAAM,oBAAoB,CAAC;AAE5B,OAAO,EACL,uBAAuB,EACvB,0BAA0B,GAC3B,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EACL,yBAAyB,EACzB,qBAAqB,EACrB,4BAA4B,GAC7B,MAAM,8BAA8B,CAAC;AACtC,YAAY,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,wCAAwC,CAAC;AACtE,YAAY,EACV,kBAAkB,EAClB,iBAAiB,EACjB,sBAAsB,GACvB,MAAM,wCAAwC,CAAC;AAEhD,OAAO,EAAE,cAAc,EAAE,MAAM,4CAA4C,CAAC;AAC5E,YAAY,EACV,oBAAoB,EACpB,mBAAmB,GACpB,MAAM,4CAA4C,CAAC;AAEpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,gDAAgD,CAAC;AAClF,YAAY,EAAE,qBAAqB,EAAE,MAAM,gDAAgD,CAAC;AAG5F,OAAO,EACL,oBAAoB,EACpB,sBAAsB,GACvB,MAAM,+BAA+B,CAAC;AACvC,YAAY,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AAGtE,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,6BAA6B,EAAE,MAAM,qEAAqE,CAAC;AACpH,YAAY,EAAE,sCAAsC,EAAE,MAAM,qEAAqE,CAAC;AAGlI,OAAO,EACL,gBAAgB,EAChB,uBAAuB,EACvB,wBAAwB,GACzB,MAAM,qCAAqC,CAAC;AAC7C,YAAY,EAAE,eAAe,EAAE,MAAM,qCAAqC,CAAC;AAE3E,OAAO,EACL,0BAA0B,EAC1B,qBAAqB,EACrB,uBAAuB,EACvB,yBAAyB,GAC1B,MAAM,oBAAoB,CAAC;AAE5B,OAAO,EACL,uBAAuB,EACvB,0BAA0B,EAC1B,0BAA0B,GAC3B,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EACL,yBAAyB,EACzB,qBAAqB,EACrB,4BAA4B,GAC7B,MAAM,8BAA8B,CAAC;AACtC,YAAY,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC"}

package/index.js

@@ -1,166 +1,175 @@
-import { MIMEType as e, MarkdownRenderer as t, buildCssVars as n, mergeClasses as r } from "@epam/ai-dial-chat-shared";
-import { DIAL_KIT_ICON_STROKE as i, ElementSize as a, EllipsisTooltip as o, GhostIconButton as s, NeutralButton as c, PrimaryButton as l, Tooltip as u } from "@epam/ai-dial-ui-kit";
-import { IconChevronLeft as d, IconChevronRight as f } from "@tabler/icons-react";
-import { createContext as p, memo as m, useCallback as h, useContext as g, useMemo as _, useState as v } from "react";
-import { jsx as y, jsxs as b } from "react/jsx-runtime";
-var x = {
+import { MIMEType as e, MarkdownRenderer as t, buildCssVars as n, inferMimeTypeFromPath as r, mergeClasses as i } from "@epam/ai-dial-chat-shared";
+import { DIAL_KIT_ICON_STROKE as a, ElementSize as o, EllipsisTooltip as s, GhostIconButton as c, NeutralButton as l, PrimaryButton as u, Tooltip as d } from "@epam/ai-dial-ui-kit";
+import { IconChevronLeft as f, IconChevronRight as p } from "@tabler/icons-react";
+import { createContext as m, memo as h, useCallback as g, useContext as _, useMemo as v, useState as y } from "react";
+import { Fragment as b, jsx as x, jsxs as S } from "react/jsx-runtime";
+var C = {
 	card: "_card_1kph1_1",
 	quote: "_quote_1kph1_5",
 	title: "_title_1kph1_9",
 	switcher: "_switcher_1kph1_13",
 	sourceName: "_sourceName_1kph1_17"
-}, S = ({ group: c, activeIndex: u, onIndexChange: p, onPreview: m, onOpenInBrowser: h, headerIcon: g, labels: _, typography: v, colors: S }) => {
-	let C = c.annotations.length, w = c.annotations[u] ?? c.primaryAnnotation, T = C > 1, E = c.primaryAnnotation.body?.source?.attachment?.type, D = m == null || E === e.HTML || E === e.XHTML, O = v?.sourceNameClassName ?? "dial-tiny-text", k = v?.titleClassName ?? "dial-body-semi-text", A = v?.quoteClassName ?? "dial-small-text", j = v?.quoteStrongClassName ?? "dial-small-semi-text", M = v?.switcherClassName ?? "dial-tiny-text", N = n({
-		"--cc-card-bg": S?.cardBackground,
-		"--cc-quote-text": S?.quoteText,
-		"--cc-title-text": S?.titleText,
-		"--cc-switcher-text": S?.switcherText,
-		"--cc-source-name-text": S?.sourceNameText
+}, w = ({ group: r, activeIndex: l, onIndexChange: d, onPreview: m, onOpenInBrowser: h, headerIcon: g, labels: _, typography: v, colors: y }) => {
+	let b = r.annotations.length, w = r.annotations[l] ?? r.primaryAnnotation, T = b > 1, E = r.primaryAnnotation.body?.source?.attachment?.type, D = m == null || E === e.HTML || E === e.XHTML, O = v?.sourceNameClassName ?? "dial-tiny-text", k = v?.titleClassName ?? "dial-body-semi-text", A = v?.quoteClassName ?? "dial-small-text", j = v?.quoteStrongClassName ?? "dial-small-semi-text", M = v?.switcherClassName ?? "dial-tiny-text", N = n({
+		"--cc-card-bg": y?.cardBackground,
+		"--cc-quote-text": y?.quoteText,
+		"--cc-title-text": y?.titleText,
+		"--cc-switcher-text": y?.switcherText,
+		"--cc-source-name-text": y?.sourceNameText
 	});
-	return /* @__PURE__ */ b("div", {
+	return /* @__PURE__ */ S("div", {
 		role: "dialog",
 		"aria-modal": "true",
 		"aria-label": _.ariaLabel,
 		style: N,
-		className: r("flex w-[400px] flex-col gap-3 rounded-lg p-4 shadow-lg", x.card),
+		className: i("flex w-[400px] flex-col gap-3 rounded-lg p-4 shadow-lg", C.card),
 		children: [
-			/* @__PURE__ */ b("div", {
+			/* @__PURE__ */ S("div", {
 				className: "flex items-center justify-between gap-2",
-				children: [/* @__PURE__ */ b("div", {
+				children: [/* @__PURE__ */ S("div", {
 					className: "flex min-w-0 items-center gap-1",
-					children: [g, /* @__PURE__ */ y(o, {
-						text: c.sourceName,
-						className: r(O, "min-w-0", x.sourceName)
+					children: [g, /* @__PURE__ */ x(s, {
+						text: r.sourceName,
+						className: i(O, "min-w-0", C.sourceName)
 					})]
-				}), T && /* @__PURE__ */ b("div", {
+				}), T && /* @__PURE__ */ S("div", {
 					className: "flex shrink-0 items-center gap-1",
 					children: [
-						/* @__PURE__ */ y(s, {
-							icon: /* @__PURE__ */ y(d, {
+						/* @__PURE__ */ x(c, {
+							icon: /* @__PURE__ */ x(f, {
 								size: 14,
 								className: "rtl:scale-x-[-1]",
-								stroke: i
+								stroke: a
 							}),
-							size: a.Small,
+							size: o.Small,
 							"aria-label": _.previousCitation,
-							onClick: () => p((u - 1 + C) % C)
+							onClick: () => d((l - 1 + b) % b)
 						}),
-						/* @__PURE__ */ y("span", {
-							className: r(M, x.switcher),
-							children: _.formatSwitcherText(u + 1, C)
+						/* @__PURE__ */ x("span", {
+							className: i(M, C.switcher),
+							children: _.formatSwitcherText(l + 1, b)
 						}),
-						/* @__PURE__ */ y(s, {
-							icon: /* @__PURE__ */ y(f, {
+						/* @__PURE__ */ x(c, {
+							icon: /* @__PURE__ */ x(p, {
 								size: 14,
 								className: "rtl:scale-x-[-1]",
-								stroke: i
+								stroke: a
 							}),
-							size: a.Small,
+							size: o.Small,
 							"aria-label": _.nextCitation,
-							onClick: () => p((u + 1) % C)
+							onClick: () => d((l + 1) % b)
 						})
 					]
 				})]
 			}),
-			(w.body?.title || w.body?.quote || T) && /* @__PURE__ */ b("div", {
+			(w.body?.title || w.body?.quote || T) && /* @__PURE__ */ S("div", {
 				className: "flex flex-col gap-3",
-				children: [w.body?.title && /* @__PURE__ */ y("p", {
-					className: r(k, x.title, "break-words"),
+				children: [w.body?.title && /* @__PURE__ */ x("p", {
+					className: i(k, C.title, "break-words"),
 					children: w.body.title
-				}), (w.body?.quote || T) && /* @__PURE__ */ y("div", {
-					className: r(A, x.quote, "line-clamp-6 break-words", T && "min-h-[3lh]"),
-					children: w.body?.quote && /* @__PURE__ */ y(t, {
+				}), (w.body?.quote || T) && /* @__PURE__ */ x("div", {
+					className: i(A, C.quote, "line-clamp-6 break-words", T && "min-h-[3lh]"),
+					children: w.body?.quote && /* @__PURE__ */ x(t, {
 						content: w.body.quote,
 						classNames: {
-							p: r(A, x.quote),
-							ul: r(A, "ps-3"),
-							ol: r(A, "ps-3"),
+							p: i(A, C.quote),
+							ul: i(A, "ps-3"),
+							ol: i(A, "ps-3"),
 							strong: j
 						}
 					})
 				})]
 			}),
-			/* @__PURE__ */ b("div", {
+			/* @__PURE__ */ S("div", {
 				className: "flex justify-start gap-2",
-				children: [m && /* @__PURE__ */ y(l, {
+				children: [m && /* @__PURE__ */ x(u, {
 					label: _.preview,
-					size: a.Small,
+					size: o.Small,
 					onClick: () => m(w)
-				}), /* @__PURE__ */ y(l, {
+				}), /* @__PURE__ */ x(u, {
 					label: D ? _.openInBrowser : _.download,
-					size: a.Small,
+					size: o.Small,
 					onClick: () => h(w)
 				})]
 			})
 		]
 	});
 };
-m(S);
+h(w);
 //#endregion
 //#region src/components/CitationMarker/CitationMarker.tsx
-var C = ({ annotationCount: e, onOpen: t, icon: n, labels: r, labelClassName: i = "dial-caption-text" }) => {
-	let o = /* @__PURE__ */ b("span", {
+var T = ({ annotationCount: e, onOpen: t, icon: n, labels: r, labelClassName: i = "dial-caption-text" }) => {
+	let a = /* @__PURE__ */ S("span", {
 		className: "flex items-center gap-1",
-		children: [n, /* @__PURE__ */ y("span", {
+		children: [n, /* @__PURE__ */ x("span", {
 			className: i,
 			children: e > 1 ? r.labelWithOverflow : r.label
 		})]
 	});
-	return /* @__PURE__ */ y(c, {
-		size: a.Small,
-		label: o,
+	return /* @__PURE__ */ x(l, {
+		size: o.Small,
+		label: a,
 		"aria-label": r.ariaLabel,
 		onClick: t
 	});
 };
-m(C);
+h(T);
 //#endregion
 //#region src/context/CitationCardContext.tsx
-var w = p(null), T = w.Provider, E = () => {
-	let e = g(w);
+var E = m(null), D = E.Provider, O = () => {
+	let e = _(E);
 	if (e == null) throw Error("useCitationCardContext must be used within a CitationCardProvider");
 	return e;
-}, D = { content: "_content_xzs0h_1" }, O = ({ group: e, onPreview: t, onOpenInBrowser: n, icon: i, headerIcon: a, cardLabels: o, markerLabels: s, cardTypography: c, markerLabelClassName: l }) => {
-	let d = E(), f = d.isOpen(e.sourceUrl), p = d.getActiveIndex(e.sourceUrl), m = h((e) => {
-		e || d.closePopup();
-	}, [d]), g = _(() => t ? (e) => {
-		t(e), d.closePopup();
-	} : void 0, [t, d]);
-	return /* @__PURE__ */ y(u, {
+}, k = { content: "_content_xzs0h_1" }, A = ({ group: e, onPreview: t, onOpenInBrowser: n, icon: r, headerIcon: a, cardLabels: o, markerLabels: s, cardTypography: c, markerLabelClassName: l }) => {
+	let u = O(), f = u.isOpen(e.groupKey), p = u.getActiveIndex(e.groupKey), m = g((e) => {
+		e || u.closePopup();
+	}, [u]), h = v(() => t ? (e) => {
+		t(e), u.closePopup();
+	} : void 0, [t, u]);
+	return /* @__PURE__ */ x(d, {
 		open: f,
 		onOpenChange: m,
 		triggerClassName: "ms-1 inline-flex align-middle",
-		contentClassName: r("!p-0 !bg-transparent !border-0 !shadow-none !max-w-none !rounded-none", D.content),
-		tooltip: /* @__PURE__ */ y(S, {
+		contentClassName: i("!p-0 !bg-transparent !border-0 !shadow-none !max-w-none !rounded-none", k.content),
+		tooltip: /* @__PURE__ */ x(w, {
 			group: e,
 			activeIndex: p,
-			onIndexChange: (t) => d.setActiveIndex(e.sourceUrl, t),
-			onPreview: g,
+			onIndexChange: (t) => u.setActiveIndex(e.groupKey, t),
+			onPreview: h,
 			onOpenInBrowser: n,
 			headerIcon: a,
 			labels: o,
 			typography: c
 		}),
-		children: /* @__PURE__ */ y(C, {
+		children: /* @__PURE__ */ x(T, {
 			sourceName: e.sourceName,
 			annotationCount: e.annotations.length,
-			onOpen: () => d.openPopup(e.sourceUrl),
-			icon: i,
+			onOpen: () => u.openPopup(e.groupKey),
+			icon: r,
 			labels: s,
 			labelClassName: l
 		})
 	});
 };
-m(O);
+h(A);
 //#endregion
 //#region src/utils/annotation.ts
-var k = {
+var j = {
 	backgroundColor: "transparent",
 	borderColor: "var(--stroke-accent, #1D4ED8)",
 	borderWidth: "2px",
 	opacity: .5,
 	hoverOpacity: .5
-}, A = (e) => e.flatMap((e, t) => {
+}, M = {
+	docx: "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
+	xlsx: "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
+	pptx: "application/vnd.openxmlformats-officedocument.presentationml.presentation"
+}, N = (e) => {
+	let t = r(e);
+	if (t != null) return t;
+	let n = e.split(/[?#]/)[0], i = n.lastIndexOf(".");
+	if (i !== -1) return M[n.slice(i + 1).toLowerCase()];
+}, P = (e) => e.flatMap((e, t) => {
 	let n = e.body?.selector;
 	if (n == null) return [];
 	let r = (Array.isArray(n) ? n : [n]).flatMap((e) => {
@@ -177,74 +186,133 @@ var k = {
 	return r.length === 0 ? [] : [{
 		id: String(e.index ?? t),
 		bboxes: r,
-		style: k
+		style: j
 	}];
-}), j = (e, t) => String(e.index ?? t), M = (e) => typeof e == "number" ? e : null, N = (e) => {
+}), F = (e, t) => String(e.index ?? t), I = (e) => typeof e == "number" ? e : null, L = (e) => {
 	if (typeof e != "object" || !e) return null;
 	let t = e;
 	if (t.type !== "pdf_region") return null;
 	let n = t.bbox;
 	if (typeof n != "object" || !n) return null;
-	let r = n, i = M(r.left), a = M(r.top), o = M(r.width), s = M(r.height);
+	let r = n, i = I(r.left), a = I(r.top), o = I(r.width), s = I(r.height);
 	return i === null || a === null || o === null || s === null ? null : {
 		type: "pdf_bbox",
-		page: M(t.page) ?? 1,
+		page: I(t.page) ?? 1,
 		x1: i,
 		y1: a,
 		x2: i + o,
 		y2: a + s
 	};
-}, P = (t, n) => t.flatMap((t) => {
-	if (typeof t != "object" || !t) return [];
-	let r = t, i = r.target;
-	if (typeof i != "object" || !i) return [];
-	let a = i, o = a.source;
-	if (typeof o != "object" || !o) return [];
-	let s = M(o.attachment_index);
-	if (s === null) return [];
-	let c = n.find((e) => e.index === s);
-	if (c?.url == null) return [];
-	let l = N(a.selector);
-	if (l === null) return [];
-	let u = r.body, d = typeof u == "object" && u ? u : {}, f = typeof d.title == "string" ? d.title : void 0, p = typeof d.quote == "string" ? d.quote : void 0;
-	return [{
-		index: M(r.index) ?? void 0,
+}, ee = (t, n) => {
+	let r = t.target;
+	if (typeof r != "object" || !r) return null;
+	let i = r, a = i.source;
+	if (typeof a != "object" || !a) return null;
+	let o = I(a.attachment_index);
+	if (o === null) return null;
+	let s = n.find((e) => e.index === o);
+	if (s?.url == null) return null;
+	let c = L(i.selector);
+	if (c === null) return null;
+	let l = t.body, u = typeof l == "object" && l ? l : {}, d = typeof u.title == "string" ? u.title : void 0, f = typeof u.quote == "string" ? u.quote : void 0;
+	return {
+		index: I(t.index) ?? void 0,
 		body: {
-			title: f,
-			quote: p,
+			title: d,
+			quote: f,
 			source: {
 				type: "attachment",
 				attachment: {
-					type: c.type ?? e.PDF,
-					url: c.url,
-					title: c.title
+					type: s.type ?? e.PDF,
+					url: s.url,
+					title: s.title
 				}
 			},
-			selector: l
+			selector: c
 		}
-	}];
-}), F = (e) => {
+	};
+}, te = (t) => {
+	let n = t.target;
+	if (typeof n != "object" || !n) return null;
+	let r = n.selector;
+	if (typeof r != "object" || !r) return null;
+	let i = r;
+	if (i.type !== "html_tag" || typeof i.tag != "string" || typeof i.id != "string") return null;
+	let a = {
+		type: "html_tag",
+		tag: i.tag,
+		id: i.id
+	}, o = t.body;
+	if (typeof o != "object" || !o) return null;
+	let s = o, c = s.source;
+	if (typeof c != "object" || !c) return null;
+	let l = c;
+	if (l.type !== "attachment") return null;
+	let u = l.url;
+	if (typeof u != "string") return null;
+	let d = typeof s.title == "string" ? s.title : void 0, f = typeof s.quote == "string" ? s.quote : void 0;
+	return {
+		target: { selector: a },
+		body: {
+			title: d,
+			quote: f,
+			source: {
+				type: "attachment",
+				attachment: {
+					type: N(u) ?? e.PDF,
+					url: u,
+					title: d
+				}
+			}
+		}
+	};
+}, R = (e) => {
+	if (e.target?.selector?.type !== "html_tag") return e;
+	let t = e.body?.source?.attachment;
+	if (t?.url == null) return e;
+	let n = N(t.url);
+	return n == null || n === t.type ? e : {
+		...e,
+		body: {
+			...e.body,
+			source: {
+				...e.body?.source,
+				type: "attachment",
+				attachment: {
+					...t,
+					type: n
+				}
+			}
+		}
+	};
+}, z = (e, t) => e.flatMap((e) => {
+	if (typeof e != "object" || !e) return [];
+	let n = e, r = te(n);
+	if (r) return [r];
+	let i = ee(n, t);
+	return i ? [i] : [];
+}), B = (e) => {
 	let t = e.custom_content?.annotations;
-	if (t?.length) return t.filter((e) => e != null && e.body?.source?.attachment?.url != null);
+	if (t?.length) return t.filter((e) => e != null && e.body?.source?.attachment?.url != null).map(R);
 	let n = e.custom_fields;
 	if (typeof n != "object" || !n) return [];
 	let r = n.annotations;
-	return !Array.isArray(r) || r.length === 0 ? [] : P(r, e.custom_content?.attachments ?? []);
-}, I = (e, t) => _(() => t ? [] : F(e), [t, e]), L = {
-	openGroupSourceUrl: null,
+	return !Array.isArray(r) || r.length === 0 ? [] : z(r, e.custom_content?.attachments ?? []).map(R);
+}, ne = (e, t) => v(() => t ? [] : B(e), [t, e]), re = {
+	openGroupKey: null,
 	activeIndexByGroup: {}
-}, R = () => {
-	let [e, t] = v(L), n = h((e) => {
+}, ie = () => {
+	let [e, t] = y(re), n = g((e) => {
 		t((t) => ({
 			...t,
-			openGroupSourceUrl: e
+			openGroupKey: e
 		}));
-	}, []), r = h(() => {
+	}, []), r = g(() => {
 		t((e) => ({
 			...e,
-			openGroupSourceUrl: null
+			openGroupKey: null
 		}));
-	}, []), i = h((e, n) => {
+	}, []), i = g((e, n) => {
 		t((t) => ({
 			...t,
 			activeIndexByGroup: {
@@ -253,11 +321,11 @@ var k = {
 			}
 		}));
 	}, []);
-	return _(() => ({
+	return v(() => ({
 		openPopup: n,
 		closePopup: r,
 		setActiveIndex: i,
-		isOpen: (t) => e.openGroupSourceUrl === t,
+		isOpen: (t) => e.openGroupKey === t,
 		getActiveIndex: (t) => e.activeIndexByGroup[t] ?? 0
 	}), [
 		n,
@@ -265,8 +333,11 @@ var k = {
 		i,
 		e
 	]);
-}, z = /⟦C(\d+)⟧/g, B = (e, t) => {
-	let n = t.map((t, n) => {
+}, ae = /⟦C(\d+)⟧/g, V = (e, t) => {
+	let n = t.map((e, t) => ({
+		g: e,
+		idx: t
+	})).filter(({ g: e }) => e.primaryAnnotation.target?.selector?.type !== "html_tag").map(({ g: t, idx: n }) => {
 		let r = t.primaryAnnotation.target?.selector, i = typeof r == "object" && r && "type" in r && r.type === "text_character_range" ? r.end : e.length;
 		return {
 			idx: n,
@@ -275,9 +346,9 @@ var k = {
 	}).sort((e, t) => t.pos - e.pos), r = e;
 	for (let { idx: e, pos: t } of n) r = `${r.slice(0, t)}⟦C${e}⟧${r.slice(t)}`;
 	return r;
-}, V = (e, t) => {
+}, oe = /<cit\s+data-id=(?:"[^"]+"|'[^']+')\s*>\s*<\/cit>/g, se = /^<cit\s+data-id=(?:"[^"]+"|'[^']+')\s*>\s*<\/cit>$/, H = /<cit\s+data-id=(?:"[^"]+"|'[^']+')\s*>\s*<\/cit>|<\/?cit\b[^>]*>|<\/?cit\b[^>]*$/gi, U = (e) => e.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;"), W = (e) => e.replace(H, (e) => se.test(e) ? e : U(e)), G = (e) => W(e).replace(oe, ""), K = (e, t) => {
 	if (typeof e == "string") {
-		let n = e.split(z);
+		let n = e.split(ae);
 		if (n.length === 1) return e;
 		let r = [];
 		return n.forEach((e, n) => {
@@ -285,82 +356,125 @@ var k = {
 		}), r;
 	}
 	return Array.isArray(e) ? e.flatMap((e) => {
-		let n = V(e, t);
+		let n = K(e, t);
 		return Array.isArray(n) ? n : [n];
 	}) : e;
-}, H = (e, t, n, i = !1) => {
-	let { onPreview: a, onOpenInBrowser: o, buildLabels: s } = n;
+}, ce = (e) => {
+	let t = e.primaryAnnotation.target?.selector;
+	return t?.type === "html_tag" ? t.id : void 0;
+}, le = (e, t) => /* @__PURE__ */ S(b, { children: [
+	e == null ? "<cit>" : `<cit data-id="${e}">`,
+	t,
+	"</cit>"
+] }), ue = (e, t, n, r = !1, a = !1) => {
+	let { onPreview: o, onOpenInBrowser: s, buildLabels: c } = n, l = v(() => r ? G(e) : W(t.length > 0 ? V(e, t) : e), [
+		e,
+		t,
+		r
+	]), u = l.includes("<cit");
 	return {
-		processedContent: _(() => t.length > 0 ? B(e, t) : e, [e, t]),
-		markdownComponents: _(() => {
-			if (t.length === 0) return {};
-			let e = (e) => {
-				let n = t[e];
-				if (!n) return null;
-				let { cardLabels: r, markerLabels: i } = s(n);
-				return /* @__PURE__ */ y(O, {
-					group: n,
-					onPreview: (e) => a(e, n),
-					onOpenInBrowser: o,
-					cardLabels: r,
-					markerLabels: i
-				}, `citation-${n.sourceUrl}`);
-			};
-			return {
-				p: ({ children: t, ...n }) => /* @__PURE__ */ y("p", {
-					...n,
-					className: r(i ? "dial-small-paragraph-text" : "dial-body-paragraph-text", "mb-3 [overflow-wrap:anywhere] [text-wrap:pretty] last:mb-0"),
-					children: V(t, e)
+		processedContent: l,
+		markdownComponents: v(() => {
+			let e = /* @__PURE__ */ new Map();
+			for (let n of t) {
+				let t = ce(n);
+				t != null && e.set(t, n);
+			}
+			let n = (e) => {
+				let { cardLabels: t, markerLabels: n } = c(e);
+				return /* @__PURE__ */ x(A, {
+					group: e,
+					onPreview: (t) => o(t, e),
+					onOpenInBrowser: s,
+					cardLabels: t,
+					markerLabels: n
+				}, `citation-${e.groupKey}`);
+			}, r = (e) => {
+				let r = t[e];
+				return r ? n(r) : null;
+			}, l = { cit: (t) => {
+				let r = t["data-id"] == null ? void 0 : e.get(t["data-id"]);
+				return r ? n(r) : le(t["data-id"], t.children);
+			} };
+			return t.length === 0 ? u ? l : {} : {
+				...l,
+				p: ({ children: e, ...t }) => /* @__PURE__ */ x("p", {
+					...t,
+					className: i(a ? "dial-small-paragraph-text" : "dial-body-paragraph-text", "mb-3 [overflow-wrap:anywhere] [text-wrap:pretty] last:mb-0"),
+					children: K(e, r)
 				}),
-				li: ({ children: t, ...n }) => /* @__PURE__ */ y("li", {
-					...n,
+				li: ({ children: e, ...t }) => /* @__PURE__ */ x("li", {
+					...t,
 					className: "mb-1.5 last:mb-0",
-					children: V(t, e)
+					children: K(e, r)
 				})
 			};
 		}, [
 			t,
-			a,
 			o,
 			s,
-			i
+			c,
+			a,
+			u
 		])
 	};
-}, U = (e) => {
+}, q = (e) => {
 	try {
 		return decodeURIComponent(e);
 	} catch {
 		return e;
 	}
-}, W = (e) => {
+}, de = (e) => {
 	try {
 		let t = new URL(e);
-		return t.pathname.split("/").map((e) => U(e)).filter(Boolean).at(-1) ?? t.hostname;
+		return t.pathname.split("/").map((e) => q(e)).filter(Boolean).at(-1) ?? t.hostname;
 	} catch {
-		return e.split("/").map((e) => U(e)).filter(Boolean).at(-1) ?? e;
+		return e.split("/").map((e) => q(e)).filter(Boolean).at(-1) ?? e;
 	}
-}, G = (e) => {
+}, J = (e, t) => e.body?.source?.attachment?.title ?? de(t), Y = (e) => {
+	let t = e.target?.selector;
+	return t?.type === "html_tag" ? t : void 0;
+}, fe = (e) => Y(e) != null, X = (e) => {
 	let t = /* @__PURE__ */ new Map();
 	for (let n of e) {
+		if (fe(n)) continue;
 		let e = n.body?.source?.attachment?.url;
 		if (e == null) continue;
 		let r = t.get(e);
 		r ? r.annotations.push(n) : t.set(e, {
+			groupKey: e,
 			sourceUrl: e,
-			sourceName: n.body?.source?.attachment?.title ?? W(e),
+			sourceName: J(n, e),
+			annotations: [n],
+			primaryAnnotation: n
+		});
+	}
+	return Array.from(t.values());
+}, Z = (e) => {
+	let t = /* @__PURE__ */ new Map();
+	for (let n of e) {
+		let e = Y(n)?.id;
+		if (e == null) continue;
+		let r = n.body?.source?.attachment?.url;
+		if (r == null) continue;
+		let i = t.get(e);
+		i ? i.annotations.push(n) : t.set(e, {
+			groupKey: `cit:${e}`,
+			sourceUrl: r,
+			sourceName: J(n, r),
 			annotations: [n],
 			primaryAnnotation: n
 		});
 	}
 	return Array.from(t.values());
-}, K = (e) => e.url == null && e.reference_url != null, q = /^(.*\.pdf)(?:#page=(\d+))?$/i, J = (e) => {
-	let t = q.exec(e);
+}, pe = (e) => [...Z(e), ...X(e)], Q = (e) => e.url == null && e.reference_url != null, me = /^(.*\.pdf)(?:#page=(\d+))?$/i, $ = (e) => {
+	let t = me.exec(e);
 	return t ? {
 		baseUrl: t[1],
 		page: t[2] == null ? null : Number(t[2])
 	} : null;
-}, Y = (t) => G((t ?? []).filter(K).map((t) => {
-	let n = t.reference_url, r = J(n) != null;
+}, he = (t) => X((t ?? []).filter(Q).map((t) => {
+	let n = t.reference_url, r = $(n) != null;
 	return { body: {
 		quote: t.data,
 		source: {
@@ -374,4 +488,4 @@ var k = {
 	} };
 }));
 //#endregion
-export { S as CitationCard, T as CitationCardProvider, O as CitationDropdown, C as CitationMarker, j as annotationHighlightId, A as annotationsToPdfHighlights, Y as getReferenceAttachmentGroups, G as groupAnnotationsBySource, B as injectCitationSentinels, K as isReferenceOnlyAttachment, P as normalizeRawAnnotations, J as parsePdfPageReference, V as replaceSentinelsInChildren, F as resolveMessageAnnotations, I as useAnnotations, R as useCitationCard, E as useCitationCardContext, H as useCitationMarkdownComponents };
+export { w as CitationCard, D as CitationCardProvider, A as CitationDropdown, T as CitationMarker, F as annotationHighlightId, P as annotationsToPdfHighlights, he as getReferenceAttachmentGroups, pe as groupAnnotations, Z as groupAnnotationsByCitId, X as groupAnnotationsBySource, V as injectCitationSentinels, Q as isReferenceOnlyAttachment, z as normalizeRawAnnotations, $ as parsePdfPageReference, K as replaceSentinelsInChildren, B as resolveMessageAnnotations, G as stripCitTagsWhileStreaming, ne as useAnnotations, ie as useCitationCard, O as useCitationCardContext, ue as useCitationMarkdownComponents };

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@epam/ai-dial-quotations",
   "description": "Citation and annotation components, hooks, and utilities for AI DIAL conversations",
-  "version": "1.1.0-dev.405",
+  "version": "1.1.0-dev.411",
   "license": "Apache-2.0",
   "type": "module",
   "main": "./index.js",
@@ -19,7 +19,7 @@
     "react": "^19.0.0",
     "react-markdown": "^10.1.0",
     "@tabler/icons-react": "^3.0.0",
-    "@epam/ai-dial-chat-shared": "1.1.0-dev.405",
+    "@epam/ai-dial-chat-shared": "1.1.0-dev.411",
     "@epam/ai-dial-ui-kit": "*",
     "@epam/pdf-highlighter-kit": ">=0.0.14"
   },

package/utils/annotation.d.ts

@@ -13,13 +13,17 @@ export declare const annotationHighlightId: (annotation: Annotation, fallbackInd
  * Normalizes raw annotations from the API wire format (stored in
  * `message.custom_fields.annotations`) to the internal `Annotation` model.
  *
- * The API format uses `target.source.attachment_index` (integer) and a
- * `pdf_region` selector with `{ left, top, width, height }` coordinates.
- * The internal model uses `body.source.attachment.url` (resolved URL) and a
- * `pdf_bbox` selector with `{ x1, y1, x2, y2 }` coordinates.
+ * Recognizes two wire shapes:
+ * - `target.source.attachment_index` (integer) + a `pdf_region` selector
+ *   with `{ left, top, width, height }` coordinates, resolved against
+ *   `attachments` and converted to `body.source.attachment.url` + a
+ *   `pdf_bbox` selector with `{ x1, y1, x2, y2 }` coordinates.
+ * - `target.selector` of type `html_tag` (`{ tag, id }`) with a flat
+ *   `body.source = { type: 'attachment', url }` — normalized to
+ *   `body.source.attachment.url`/`.title` directly, with no attachment-list
+ *   lookup and no `index`.
  *
- * Annotations whose attachment index cannot be resolved or whose selector
- * is not a recognised PDF region are omitted from the result.
+ * Annotations matching neither shape are omitted from the result.
  */
 export declare const normalizeRawAnnotations: (rawAnnotations: unknown[], attachments: MessageAttachment[]) => Annotation[];
 /**

package/utils/annotation.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"annotation.d.ts","sourceRoot":"","sources":["../../src/utils/annotation.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,UAAU,EACV,OAAO,EACP,iBAAiB,EAElB,MAAM,2BAA2B,CAAC;AAEnC,OAAO,KAAK,EAEV,kBAAkB,EACnB,MAAM,2BAA2B,CAAC;AAUnC;;;;;GAKG;AACH,eAAO,MAAM,0BAA0B,GACrC,aAAa,UAAU,EAAE,KACxB,kBAAkB,EAoBjB,CAAC;AAEL,oHAAoH;AACpH,eAAO,MAAM,qBAAqB,GAChC,YAAY,UAAU,EACtB,eAAe,MAAM,KACpB,MAAmD,CAAC;AA+BvD;;;;;;;;;;;GAWG;AACH,eAAO,MAAM,uBAAuB,GAClC,gBAAgB,OAAO,EAAE,EACzB,aAAa,iBAAiB,EAAE,KAC/B,UAAU,EAmDT,CAAC;AAEL;;;;;;GAMG;AACH,eAAO,MAAM,yBAAyB,GAAI,SAAS,OAAO,KAAG,UAAU,EAkBtE,CAAC"}
+{"version":3,"file":"annotation.d.ts","sourceRoot":"","sources":["../../src/utils/annotation.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,UAAU,EAEV,OAAO,EACP,iBAAiB,EAElB,MAAM,2BAA2B,CAAC;AAEnC,OAAO,KAAK,EAEV,kBAAkB,EACnB,MAAM,2BAA2B,CAAC;AA2BnC;;;;;GAKG;AACH,eAAO,MAAM,0BAA0B,GACrC,aAAa,UAAU,EAAE,KACxB,kBAAkB,EAoBjB,CAAC;AAEL,oHAAoH;AACpH,eAAO,MAAM,qBAAqB,GAChC,YAAY,UAAU,EACtB,eAAe,MAAM,KACpB,MAAmD,CAAC;AAyKvD;;;;;;;;;;;;;;;GAeG;AACH,eAAO,MAAM,uBAAuB,GAClC,gBAAgB,OAAO,EAAE,EACzB,aAAa,iBAAiB,EAAE,KAC/B,UAAU,EAeT,CAAC;AAEL;;;;;;GAMG;AACH,eAAO,MAAM,yBAAyB,GAAI,SAAS,OAAO,KAAG,UAAU,EAoBtE,CAAC"}

package/utils/citation-injection.d.ts

@@ -2,10 +2,26 @@ import { ReactNode } from 'react';
 import { AnnotationGroup } from './group-annotations-by-source';
 /**
  * Inserts sentinel strings into `content` at the `end` character offsets of
- * each annotation group's primary selector. Offsets are processed in
- * descending order so earlier insertions do not shift later ones.
+ * each non-`html_tag` annotation group's primary selector (descending order
+ * so earlier insertions don't shift later ones). Sentinel indices are the
+ * group's position in the original flat `groups` array, so a caller doing
+ * `groups[idx]` still resolves correctly even though `html_tag` groups are
+ * skipped here — those render as real `<cit>` elements instead (see
+ * `useCitationMarkdownComponents`'s `cit` component override).
  */
 export declare const injectCitationSentinels: (content: string, groups: AnnotationGroup[]) => string;
+/**
+ * Leaves the one supported paired citation element parseable as HTML and
+ * escapes every other `cit` shape so Markdown renders the original markup as
+ * visible text. Unsupported markup is never interpreted as a citation.
+ */
+export declare const escapeUnsupportedCitTags: (content: string) => string;
+/**
+ * Hides supported paired citation elements while the message is streaming.
+ * Every other `cit` shape is escaped and remains visible as ordinary text,
+ * preventing an incomplete opening tag from swallowing the streamed suffix.
+ */
+export declare const stripCitTagsWhileStreaming: (content: string) => string;
 /**
  * Recursively walks a React child tree and replaces sentinel strings with
  * the result of `renderMarker(idx)`. Handles string children, arrays, and

package/utils/citation-injection.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"citation-injection.d.ts","sourceRoot":"","sources":["../../src/utils/citation-injection.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,OAAO,CAAC;AACvC,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAC;AAUrE;;;;GAIG;AACH,eAAO,MAAM,uBAAuB,GAClC,SAAS,MAAM,EACf,QAAQ,eAAe,EAAE,KACxB,MAoBF,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,0BAA0B,GACrC,UAAU,SAAS,EACnB,cAAc,CAAC,GAAG,EAAE,MAAM,KAAK,SAAS,KACvC,SA0BF,CAAC"}
+{"version":3,"file":"citation-injection.d.ts","sourceRoot":"","sources":["../../src/utils/citation-injection.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,OAAO,CAAC;AACvC,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAC;AAUrE;;;;;;;;GAQG;AACH,eAAO,MAAM,uBAAuB,GAClC,SAAS,MAAM,EACf,QAAQ,eAAe,EAAE,KACxB,MAwBF,CAAC;AAmBF;;;;GAIG;AACH,eAAO,MAAM,wBAAwB,GAAI,SAAS,MAAM,KAAG,MAGxD,CAAC;AAEJ;;;;GAIG;AACH,eAAO,MAAM,0BAA0B,GAAI,SAAS,MAAM,KAAG,MAE5D,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,0BAA0B,GACrC,UAAU,SAAS,EACnB,cAAc,CAAC,GAAG,EAAE,MAAM,KAAK,SAAS,KACvC,SA0BF,CAAC"}

package/utils/group-annotations-by-source.d.ts

@@ -1,7 +1,14 @@
 import { Annotation } from '@epam/ai-dial-chat-shared';
-/** A set of annotations that all cite the same source document. */
+/** A set of annotations rendered behind one inline citation marker. */
 export interface AnnotationGroup {
-    /** The attachment URL shared by all annotations in this group. */
+    /**
+     * This group's identity — used for open/active-index popup state and
+     * React keys. Equals `sourceUrl` for a `groupAnnotationsBySource` group;
+     * equals `` `cit:${id}` `` for a `groupAnnotationsByCitId` group, so two
+     * cit ids citing the same document never share popup state.
+     */
+    groupKey: string;
+    /** The cited attachment's URL — used for Preview/Download. */
     sourceUrl: string;
     /**
      * Human-readable name derived from the URL: the last non-empty path segment
@@ -13,6 +20,25 @@ export interface AnnotationGroup {
     /** The first annotation in the group — used for the inline citation marker position. */
     primaryAnnotation: Annotation;
 }
-/** Groups annotations by their source attachment URL. Annotations without a `body.source.attachment.url` are excluded. */
+/**
+ * Groups annotations by their source attachment URL. Annotations without a
+ * `body.source.attachment.url` are excluded, as are `html_tag`-selector
+ * annotations — those are grouped by tag id instead, see
+ * `groupAnnotationsByCitId`.
+ */
 export declare const groupAnnotationsBySource: (annotations: Annotation[]) => AnnotationGroup[];
+/**
+ * Groups `html_tag`-selector annotations by `target.selector.id`, one group
+ * per distinct tag id — never collapsing two different ids that cite the
+ * same source document into one group. Annotations without
+ * `target.selector.id` or without `body.source.attachment.url` are excluded.
+ */
+export declare const groupAnnotationsByCitId: (annotations: Annotation[]) => AnnotationGroup[];
+/**
+ * Partitions annotations by selector type and groups each family with its
+ * own algorithm: `html_tag` annotations via `groupAnnotationsByCitId`, every
+ * other annotation via `groupAnnotationsBySource`. Lets a host render
+ * citations without branching on selector type itself.
+ */
+export declare const groupAnnotations: (annotations: Annotation[]) => AnnotationGroup[];
 //# sourceMappingURL=group-annotations-by-source.d.ts.map

package/utils/group-annotations-by-source.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"group-annotations-by-source.d.ts","sourceRoot":"","sources":["../../src/utils/group-annotations-by-source.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAE5D,mEAAmE;AACnE,MAAM,WAAW,eAAe;IAC9B,kEAAkE;IAClE,SAAS,EAAE,MAAM,CAAC;IAClB;;;OAGG;IACH,UAAU,EAAE,MAAM,CAAC;IACnB,+DAA+D;IAC/D,WAAW,EAAE,UAAU,EAAE,CAAC;IAC1B,wFAAwF;IACxF,iBAAiB,EAAE,UAAU,CAAC;CAC/B;AA2BD,0HAA0H;AAC1H,eAAO,MAAM,wBAAwB,GACnC,aAAa,UAAU,EAAE,KACxB,eAAe,EAsBjB,CAAC"}
+{"version":3,"file":"group-annotations-by-source.d.ts","sourceRoot":"","sources":["../../src/utils/group-annotations-by-source.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAmB,MAAM,2BAA2B,CAAC;AAE7E,uEAAuE;AACvE,MAAM,WAAW,eAAe;IAC9B;;;;;OAKG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB,8DAA8D;IAC9D,SAAS,EAAE,MAAM,CAAC;IAClB;;;OAGG;IACH,UAAU,EAAE,MAAM,CAAC;IACnB,+DAA+D;IAC/D,WAAW,EAAE,UAAU,EAAE,CAAC;IAC1B,wFAAwF;IACxF,iBAAiB,EAAE,UAAU,CAAC;CAC/B;AAkDD;;;;;GAKG;AACH,eAAO,MAAM,wBAAwB,GACnC,aAAa,UAAU,EAAE,KACxB,eAAe,EAuBjB,CAAC;AAEF;;;;;GAKG;AACH,eAAO,MAAM,uBAAuB,GAClC,aAAa,UAAU,EAAE,KACxB,eAAe,EAwBjB,CAAC;AAEF;;;;;GAKG;AACH,eAAO,MAAM,gBAAgB,GAC3B,aAAa,UAAU,EAAE,KACxB,eAAe,EAGjB,CAAC"}

package/utils/useCitationCard.d.ts

@@ -1,12 +1,15 @@
 /**
  * Manages open/close state and per-group active annotation index for citation
- * cards within a single assistant message.
+ * cards within a single assistant message. State is keyed by
+ * `AnnotationGroup.groupKey`, not `sourceUrl` — two groups can share a
+ * `sourceUrl` (e.g. two `cit`-id groups citing the same document) while
+ * having independent popup/switcher state.
  */
 export declare const useCitationCard: () => {
-    openPopup: (sourceUrl: string) => void;
+    openPopup: (groupKey: string) => void;
     closePopup: () => void;
-    setActiveIndex: (sourceUrl: string, index: number) => void;
-    isOpen: (sourceUrl: string) => boolean;
-    getActiveIndex: (sourceUrl: string) => number;
+    setActiveIndex: (groupKey: string, index: number) => void;
+    isOpen: (groupKey: string) => boolean;
+    getActiveIndex: (groupKey: string) => number;
 };
 //# sourceMappingURL=useCitationCard.d.ts.map

package/utils/useCitationCard.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useCitationCard.d.ts","sourceRoot":"","sources":["../../src/utils/useCitationCard.ts"],"names":[],"mappings":"AAcA;;;GAGG;AACH,eAAO,MAAM,eAAe;2BAGgB,MAAM;;gCAQD,MAAM,SAAS,MAAM;wBAY5C,MAAM;gCACE,MAAM;CAKvC,CAAC"}
+{"version":3,"file":"useCitationCard.d.ts","sourceRoot":"","sources":["../../src/utils/useCitationCard.ts"],"names":[],"mappings":"AAcA;;;;;;GAMG;AACH,eAAO,MAAM,eAAe;0BAGe,MAAM;;+BAQD,MAAM,SAAS,MAAM;uBAY5C,MAAM;+BACE,MAAM;CAKtC,CAAC"}