@paymanai/payman-ask-sdk 4.0.22 → 4.0.23

package/dist/index.mjs

@@ -1,3 +1,4 @@
+import * as React from 'react';
 import { createContext, forwardRef, useCallback, useRef, useState, useMemo, useEffect, useImperativeHandle, useLayoutEffect, useContext } from 'react';
 import { AnimatePresence, motion } from 'framer-motion';
 import { clsx } from 'clsx';
@@ -3378,9 +3379,88 @@ function MarkdownImageV2({
     }
   ) });
 }
-function buildComponents(onImageClick, isResolvingRef) {
+var RAG_SOURCE_CLASS = "payman-v2-rag-source";
+var RAG_FOLLOW_UP_CLASS = "payman-v2-rag-follow-up";
+var SOURCE_PATTERN = /^Sources?:\s*\S/i;
+var MAX_FOLLOW_UP_LENGTH = 320;
+var HR_PATTERN = /^-{3,}$|^_{3,}$|^\*{3,}$/;
+function normalizeRagParagraphText(text) {
+  return text.replace(/\s+/g, " ").trim();
+}
+function isRagSourceParagraph(text) {
+  return SOURCE_PATTERN.test(normalizeRagParagraphText(text));
+}
+function isRagFollowUpCandidate(text) {
+  const normalized = normalizeRagParagraphText(text);
+  return normalized.length > 0 && normalized.length <= MAX_FOLLOW_UP_LENGTH && normalized.endsWith("?");
+}
+function extractRagParagraphTexts(content) {
+  const paragraphs = [];
+  const currentParagraph = [];
+  const flushParagraph = () => {
+    const paragraph = normalizeRagParagraphText(currentParagraph.join(" "));
+    currentParagraph.length = 0;
+    if (paragraph) paragraphs.push(paragraph);
+  };
+  content.replace(/\\n/g, "\n").split(/\r?\n/).forEach((line) => {
+    const trimmedLine = line.trim();
+    if (!trimmedLine) {
+      flushParagraph();
+      return;
+    }
+    if (HR_PATTERN.test(trimmedLine)) {
+      flushParagraph();
+      return;
+    }
+    currentParagraph.push(trimmedLine);
+  });
+  flushParagraph();
+  return paragraphs;
+}
+function getRagAnswerParagraphRoles(paragraphs) {
+  const normalized = paragraphs.map(normalizeRagParagraphText);
+  const roles = normalized.map(() => null);
+  const sourceIndexes = normalized.map((paragraph, index) => isRagSourceParagraph(paragraph) ? index : -1).filter((index) => index >= 0);
+  for (const sourceIndex of sourceIndexes) {
+    roles[sourceIndex] = "source";
+    for (const candidateIndex of [sourceIndex - 1, sourceIndex + 1]) {
+      if (candidateIndex >= 0 && candidateIndex < normalized.length && roles[candidateIndex] == null && isRagFollowUpCandidate(normalized[candidateIndex] ?? "")) {
+        roles[candidateIndex] = "follow-up";
+      }
+    }
+  }
+  return roles;
+}
+function getRagAnswerParagraphRole(paragraph, paragraphs) {
+  const normalizedParagraph = normalizeRagParagraphText(paragraph);
+  if (isRagSourceParagraph(normalizedParagraph)) return "source";
+  const roles = getRagAnswerParagraphRoles(paragraphs);
+  return paragraphs.map(normalizeRagParagraphText).some((candidate, index) => {
+    return candidate === normalizedParagraph && roles[index] === "follow-up";
+  }) ? "follow-up" : null;
+}
+function ragAnswerRoleClassName(role) {
+  if (role === "source") return RAG_SOURCE_CLASS;
+  if (role === "follow-up") return RAG_FOLLOW_UP_CLASS;
+  return void 0;
+}
+function reactNodeText(node) {
+  if (typeof node === "string" || typeof node === "number") return String(node);
+  if (Array.isArray(node)) return node.map(reactNodeText).join("");
+  if (React.isValidElement(node)) {
+    return reactNodeText(node.props.children);
+  }
+  return "";
+}
+function buildComponents(onImageClick, isResolvingRef, ragParagraphsRef) {
   return {
-    p: ({ children }) => /* @__PURE__ */ jsx("p", { children }),
+    p: ({ children }) => {
+      const role = getRagAnswerParagraphRole(
+        reactNodeText(children),
+        ragParagraphsRef?.current ?? []
+      );
+      return /* @__PURE__ */ jsx("p", { className: ragAnswerRoleClassName(role), children });
+    },
     code: ({ children }) => /* @__PURE__ */ jsx("code", { children }),
     pre: ({ children }) => /* @__PURE__ */ jsx("div", { className: "payman-v2-markdown-pre", children: /* @__PURE__ */ jsx("pre", { children }) }),
     ul: ({ children }) => /* @__PURE__ */ jsx("ul", { children }),
@@ -3418,8 +3498,13 @@ function MarkdownRendererV2({
 }) {
   const isResolvingRef = useRef(isResolvingImages);
   isResolvingRef.current = isResolvingImages;
+  const ragParagraphsRef = useRef([]);
+  ragParagraphsRef.current = useMemo(
+    () => extractRagParagraphTexts(content),
+    [content]
+  );
   const components = useMemo(
-    () => buildComponents(onImageClick, isResolvingRef),
+    () => buildComponents(onImageClick, isResolvingRef, ragParagraphsRef),
     [onImageClick]
   );
   return /* @__PURE__ */ jsx(