@paymanai/payman-ask-sdk 4.0.21 → 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';
@@ -709,8 +710,7 @@ function buildRequestBody(config, userMessage, sessionId, options) {
     sessionAttributes,
     analysisMode: options?.analysisMode,
     locale: resolveLocale(config.locale),
-    timezone: resolveTimezone(config.timezone),
-    ...options?.attachments?.length ? { attachments: options.attachments } : {}
+    timezone: resolveTimezone(config.timezone)
   };
 }
 function resolveLocale(configured) {
@@ -755,7 +755,6 @@ function buildResolveImagesUrl(config) {
   const basePath = normalizedEndpointPath.endsWith("/stream") ? normalizedEndpointPath.slice(0, -"/stream".length) : normalizedEndpointPath;
   return `${config.api.baseUrl}${basePath}/resolve-image-urls`;
 }
-var NGROK_SKIP_BROWSER_WARNING = "ngrok-skip-browser-warning";
 function buildRequestHeaders(config) {
   const headers = {
     ...config.api.headers
@@ -763,9 +762,6 @@ function buildRequestHeaders(config) {
   if (config.api.authToken) {
     headers.Authorization = `Bearer ${config.api.authToken}`;
   }
-  if (!headers[NGROK_SKIP_BROWSER_WARNING]) {
-    headers[NGROK_SKIP_BROWSER_WARNING] = "true";
-  }
   return headers;
 }
 var RAG_IMAGE_REGEX = /\/api\/rag\/chunks\/[^"'\s]+\/image/;
@@ -1046,81 +1042,6 @@ function createCancelledMessageUpdate(steps, currentMessage) {
     currentMessage: currentMessage || "Thinking..."
   };
 }
-var DEFAULT_SIGNED_URL_ENDPOINT = "/api/files/signed-url";
-function fileExtension(filename) {
-  const ext = filename.split(".").pop()?.toLowerCase().replace(/^\./, "");
-  return ext && ext.length > 0 ? ext : "bin";
-}
-function resolveMimeType(file) {
-  if (file.type && file.type.trim().length > 0) return file.type;
-  const ext = fileExtension(file.name);
-  const byExt = {
-    pdf: "application/pdf",
-    png: "image/png",
-    jpg: "image/jpeg",
-    jpeg: "image/jpeg",
-    gif: "image/gif",
-    webp: "image/webp"
-  };
-  return byExt[ext] ?? "application/octet-stream";
-}
-function buildSignedUrlEndpoint(config, ext) {
-  const endpoint = config.api.signedUrlEndpoint || DEFAULT_SIGNED_URL_ENDPOINT;
-  const queryParams = new URLSearchParams({ extn: ext.replace(/^\./, "") });
-  return `${config.api.baseUrl}${endpoint}?${queryParams.toString()}`;
-}
-async function requestSignedUrl(config, ext, signal) {
-  const url = buildSignedUrlEndpoint(config, ext);
-  const headers = buildRequestHeaders(config);
-  const response = await fetch(url, {
-    method: "GET",
-    headers,
-    signal
-  });
-  if (!response.ok) {
-    const errorText = await response.text().catch(() => "");
-    throw new Error(
-      errorText ? `Failed to get upload URL (${response.status}): ${errorText}` : `Failed to get upload URL (${response.status})`
-    );
-  }
-  const data = await response.json();
-  if (!data.key || !data.url) {
-    throw new Error("Signed URL response missing key or url");
-  }
-  return { key: data.key, url: data.url };
-}
-async function uploadToSignedUrl(signedUrl, file, mimeType, signal) {
-  const response = await fetch(signedUrl, {
-    method: "PUT",
-    headers: {
-      "Content-Type": mimeType,
-      "x-ms-blob-type": "BlockBlob"
-    },
-    body: file,
-    signal
-  });
-  if (!response.ok) {
-    const errorText = await response.text().catch(() => "");
-    throw new Error(
-      errorText ? `Failed to upload file (${response.status}): ${errorText}` : `Failed to upload file (${response.status})`
-    );
-  }
-}
-async function uploadAttachment(config, file, signal) {
-  const ext = fileExtension(file.name);
-  const mimeType = resolveMimeType(file);
-  const { key, url } = await requestSignedUrl(config, ext, signal);
-  await uploadToSignedUrl(url, file, mimeType, signal);
-  return {
-    tempKey: key,
-    filename: file.name,
-    mimeType
-  };
-}
-async function uploadAttachments(config, files, signal) {
-  const uploads = files.map((file) => uploadAttachment(config, file, signal));
-  return Promise.all(uploads);
-}
 var UserActionStaleError = class extends Error {
   constructor(userActionId, message = "User action is no longer actionable") {
     super(message);
@@ -1157,6 +1078,9 @@ async function cancelUserAction(config, userActionId) {
 async function resendUserAction(config, userActionId) {
   return sendUserActionRequest(config, userActionId, "resend");
 }
+async function expireUserAction(config, userActionId) {
+  return sendUserActionRequest(config, userActionId, "expired");
+}
 var EMPTY_USER_ACTION_STATE = { prompts: [], notifications: [] };
 function upsertPrompt(prompts, req) {
   const active = { ...req, status: "pending" };
@@ -1185,32 +1109,12 @@ function getStoredOrInitialMessages(config) {
 function getSessionIdFromMessages(messages) {
   return messages.find((message) => message.sessionId)?.sessionId;
 }
-var IMAGE_EXTENSIONS = /* @__PURE__ */ new Set(["jpg", "jpeg", "png", "gif", "webp", "avif"]);
-function attachmentKindFromFile(file) {
-  const ext = file.name.split(".").pop()?.toLowerCase() ?? "";
-  if (IMAGE_EXTENSIONS.has(ext) || file.type.startsWith("image/")) return "image";
-  return "file";
-}
-function buildMessageAttachments(files) {
-  return files.map((file, index) => {
-    const kind = attachmentKindFromFile(file);
-    return {
-      id: `att-${Date.now()}-${index}`,
-      filename: file.name,
-      mimeType: file.type || "application/octet-stream",
-      // Blob URL for all local files so PDFs/docs stay previewable after send.
-      previewUrl: URL.createObjectURL(file),
-      kind
-    };
-  });
-}
 function useChatV2(config, callbacks = {}) {
   const [messages, setMessages] = useState(() => getStoredOrInitialMessages(config));
   const [isWaitingForResponse, setIsWaitingForResponse] = useState(() => {
     if (!config.userId) return false;
     return activeStreamStore.get(config.userId)?.isWaiting ?? false;
   });
-  const [isUploadingAttachments, setIsUploadingAttachments] = useState(false);
   const sessionIdRef = useRef(
     getSessionIdFromMessages(getStoredOrInitialMessages(config)) ?? config.initialSessionId ?? void 0
   );
@@ -1285,45 +1189,21 @@ function useChatV2(config, callbacks = {}) {
   );
   const sendMessage = useCallback(
     async (userMessage, options) => {
-      const trimmedMessage = userMessage.trim();
-      const files = options?.files ?? [];
-      const hasPreuploadedAttachments = (options?.attachments?.length ?? 0) > 0;
-      if (!trimmedMessage && files.length === 0 && !hasPreuploadedAttachments) return;
-      let streamAttachments = options?.attachments;
-      if (!streamAttachments && files.length > 0) {
-        setIsUploadingAttachments(true);
-        try {
-          streamAttachments = await uploadAttachments(configRef.current, files);
-        } catch (error) {
-          if (error.name !== "AbortError") {
-            callbacksRef.current.onError?.(error);
-          }
-          throw error;
-        } finally {
-          setIsUploadingAttachments(false);
-        }
-      }
+      if (!userMessage.trim()) return;
       if (!sessionIdRef.current && configRef.current.autoGenerateSessionId !== false) {
         sessionIdRef.current = generateId();
         callbacksRef.current.onSessionIdChange?.(sessionIdRef.current);
       }
-      const messageAttachments = files.length > 0 ? buildMessageAttachments(files) : streamAttachments?.length ? streamAttachments.map((attachment, index) => ({
-        id: `att-${Date.now()}-${index}`,
-        filename: attachment.filename,
-        mimeType: attachment.mimeType,
-        kind: attachment.mimeType.startsWith("image/") ? "image" : "file"
-      })) : void 0;
       const userMessageId = `user-${Date.now()}`;
       const userMsg = {
         id: userMessageId,
         sessionId: sessionIdRef.current,
         role: "user",
-        content: trimmedMessage,
-        timestamp: (/* @__PURE__ */ new Date()).toISOString(),
-        attachments: messageAttachments
+        content: userMessage,
+        timestamp: (/* @__PURE__ */ new Date()).toISOString()
       };
       setMessages((prev) => [...prev, userMsg]);
-      callbacksRef.current.onMessageSent?.(trimmedMessage);
+      callbacksRef.current.onMessageSent?.(userMessage);
       setIsWaitingForResponse(true);
       callbacksRef.current.onStreamStart?.();
       const streamingId = `assistant-${Date.now()}`;
@@ -1350,14 +1230,11 @@ function useChatV2(config, callbacks = {}) {
         activeStreamStore.start(userId, abortController, initialMessages);
       }
       const newSessionId = await startStream(
-        trimmedMessage,
+        userMessage,
         streamingId,
         sessionIdRef.current,
         abortController,
-        {
-          analysisMode: options?.analysisMode,
-          attachments: streamAttachments
-        }
+        options
       );
       const finalStreamUserId = streamUserIdRef.current ?? userId;
       if (finalStreamUserId) {
@@ -1495,6 +1372,19 @@ function useChatV2(config, callbacks = {}) {
     },
     [setPromptStatus]
   );
+  const expireUserAction2 = useCallback(
+    async (userActionId) => {
+      setPromptStatus(userActionId, "expired");
+      try {
+        await expireUserAction(configRef.current, userActionId);
+      } catch (error) {
+        if (error instanceof UserActionStaleError) return;
+        callbacksRef.current.onError?.(error);
+        throw error;
+      }
+    },
+    [setPromptStatus]
+  );
   const dismissNotification = useCallback((id) => {
     setUserActionState((prev) => ({
       ...prev,
@@ -1535,18 +1425,6 @@ function useChatV2(config, callbacks = {}) {
     setMessages(config.initialMessages);
     sessionIdRef.current = getSessionIdFromMessages(config.initialMessages) ?? config.initialSessionId;
   }, [config.initialMessages, config.initialSessionId, config.userId]);
-  const hydratedSessionIdRef = useRef(void 0);
-  useEffect(() => {
-    if (config.userId) return;
-    if (!config.initialMessages?.length) return;
-    const sessionId = config.initialSessionId;
-    if (hydratedSessionIdRef.current === sessionId && messagesRef.current.length > 0) {
-      return;
-    }
-    hydratedSessionIdRef.current = sessionId;
-    setMessages(config.initialMessages);
-    sessionIdRef.current = getSessionIdFromMessages(config.initialMessages) ?? sessionId;
-  }, [config.initialMessages, config.initialSessionId, config.userId]);
   useEffect(() => {
     const prevUserId = prevUserIdRef.current;
     prevUserIdRef.current = config.userId;
@@ -1581,12 +1459,12 @@ function useChatV2(config, callbacks = {}) {
     getSessionId,
     getMessages,
     isWaitingForResponse,
-    isUploadingAttachments,
     sessionId: sessionIdRef.current,
     userActionState,
     submitUserAction: submitUserAction2,
     cancelUserAction: cancelUserAction2,
     resendUserAction: resendUserAction2,
+    expireUserAction: expireUserAction2,
     dismissNotification
   };
 }
@@ -3501,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 }),
@@ -3541,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(
@@ -4668,18 +4630,28 @@ function useExpiryCountdown(prompt) {
       setSecondsLeft(void 0);
       return;
     }
-    setSecondsLeft(initial);
-    const id = window.setInterval(() => {
-      setSecondsLeft((s) => {
-        if (s === void 0) return s;
-        if (s <= 1) {
-          window.clearInterval(id);
-          return 0;
-        }
-        return s - 1;
-      });
-    }, 1e3);
-    return () => window.clearInterval(id);
+    const deadlineMs = Date.now() + initial * 1e3;
+    let intervalId;
+    const remaining = () => Math.max(0, Math.ceil((deadlineMs - Date.now()) / 1e3));
+    const sync = () => {
+      const next = remaining();
+      setSecondsLeft(next);
+      if (next === 0 && intervalId !== void 0) {
+        window.clearInterval(intervalId);
+        intervalId = void 0;
+      }
+    };
+    sync();
+    intervalId = window.setInterval(sync, 1e3);
+    document.addEventListener("visibilitychange", sync);
+    window.addEventListener("focus", sync);
+    window.addEventListener("pageshow", sync);
+    return () => {
+      if (intervalId !== void 0) window.clearInterval(intervalId);
+      document.removeEventListener("visibilitychange", sync);
+      window.removeEventListener("focus", sync);
+      window.removeEventListener("pageshow", sync);
+    };
   }, [prompt.userActionId, prompt.subAction, initial]);
   const expired = initial !== void 0 && secondsLeft === 0;
   return [secondsLeft, expired];
@@ -6236,7 +6208,7 @@ var PaymanChatInner = forwardRef(function PaymanChatInner2({
   const submitUserAction2 = chat.submitUserAction ?? NOOP_ASYNC;
   const cancelUserAction2 = chat.cancelUserAction ?? NOOP_ASYNC;
   const resendUserAction2 = chat.resendUserAction ?? NOOP_ASYNC;
-  const expireUserAction = chat.expireUserAction ?? NOOP_ASYNC;
+  const expireUserAction2 = chat.expireUserAction ?? NOOP_ASYNC;
   const dismissNotification = chat.dismissNotification ?? NOOP;
   const isUserActionSupported = typeof chat.submitUserAction === "function" && typeof chat.cancelUserAction === "function" && typeof chat.resendUserAction === "function";
   const {
@@ -6637,7 +6609,7 @@ var PaymanChatInner = forwardRef(function PaymanChatInner2({
                   onSubmitUserAction: isUserActionSupported ? submitUserAction2 : void 0,
                   onCancelUserAction: isUserActionSupported ? cancelUserAction2 : void 0,
                   onResendUserAction: isUserActionSupported ? resendUserAction2 : void 0,
-                  onExpireUserAction: isUserActionSupported ? expireUserAction : void 0,
+                  onExpireUserAction: isUserActionSupported ? expireUserAction2 : void 0,
                   onDismissNotification: dismissNotification,
                   onSubmitFeedback: handleSubmitFeedback
                 }