@axiom-lattice/react-sdk 2.2.1 → 2.2.2

package/dist/index.js

@@ -3367,6 +3367,13 @@ var WorkspaceContextProvider = ({
     },
     [client, workspaceId, projectId]
   );
+  const getFileDownloadUrl = (0, import_react12.useCallback)(
+    (filePath, assistantId) => {
+      if (!workspaceId || !projectId) return "";
+      return client.getFileDownloadUrl(workspaceId, projectId, filePath, assistantId);
+    },
+    [client, workspaceId, projectId]
+  );
   const getDirectFileUrl = (0, import_react12.useCallback)(
     (filePath) => {
       if (!tenantId || !workspaceId || !projectId) return "";
@@ -3423,6 +3430,7 @@ var WorkspaceContextProvider = ({
     uploadFileToFolder,
     uploadFileToProject,
     getFileViewUrl,
+    getFileDownloadUrl,
     getDirectFileUrl,
     shareFile,
     listShares,
@@ -19868,12 +19876,14 @@ var useStyles24 = (0, import_antd_style33.createStyles)(({ token, css }) => ({
 var MarkdownViewer = ({
   content,
   fileName,
+  downloadUrl,
   className,
   style
 }) => {
   const { styles, cx } = useStyles24();
   const [copied, setCopied] = (0, import_react75.useState)(false);
   const contentRef = (0, import_react75.useRef)(null);
+  const downloadFileName = fileName.split("/").pop() || fileName;
   (0, import_react75.useEffect)(() => {
     if (copied) {
       const timer = setTimeout(() => setCopied(false), 2e3);
@@ -19891,7 +19901,6 @@ var MarkdownViewer = ({
     const url = URL.createObjectURL(blob);
     const a = document.createElement("a");
     a.href = url;
-    const downloadFileName = fileName.split("/").pop() || fileName;
     a.download = downloadFileName;
     document.body.appendChild(a);
     a.click();
@@ -20015,7 +20024,10 @@ ${content}
         {
           type: "text",
           icon: /* @__PURE__ */ (0, import_jsx_runtime73.jsx)(import_icons14.DownloadOutlined, {}),
-          onClick: handleDownload,
+          "aria-label": "Download File",
+          href: downloadUrl,
+          download: downloadUrl ? downloadFileName : void 0,
+          onClick: downloadUrl ? void 0 : handleDownload,
           size: "small"
         }
       ) }),
@@ -20427,6 +20439,10 @@ var AttachmentsCard = ({
   const { config } = useLatticeChatShellContext();
   const workspaceCtx = (0, import_react77.useContext)(WorkspaceContext);
   const baseURL = config.baseURL;
+  const getSandboxFileUrl = (filePath) => {
+    const params = new URLSearchParams({ path: filePath });
+    return `${baseURL}/api/assistants/${assistantId}/threads/${threadId}/sandbox/downloadfile?${params.toString()}`;
+  };
   const getFileUrl = (item) => {
     if (typeof item.id === "string" && /^https?:\/\//i.test(item.id)) {
       return item.id;
@@ -20434,7 +20450,16 @@ var AttachmentsCard = ({
     if (workspaceCtx?.workspaceId && workspaceCtx?.projectId) {
       return workspaceCtx.getFileViewUrl(item.id, assistantId || void 0);
     }
-    return `${baseURL}/api/assistants/${assistantId}/threads/${threadId}/sandbox/downloadfile?path=${item.id}`;
+    return getSandboxFileUrl(item.id);
+  };
+  const getFileDownloadUrl = (item) => {
+    if (typeof item.id === "string" && /^https?:\/\//i.test(item.id)) {
+      return item.id;
+    }
+    if (workspaceCtx?.workspaceId && workspaceCtx?.projectId) {
+      return workspaceCtx.getFileDownloadUrl(item.id, assistantId || void 0);
+    }
+    return getSandboxFileUrl(item.id);
   };
   const { Text: Text77 } = import_antd56.Typography;
   const [showAll, setShowAll] = (0, import_react77.useState)(false);
@@ -20470,7 +20495,12 @@ var AttachmentsCard = ({
     if (getFileCategory(item.name) === "image") return;
     openSideApp({
       component_key: "attachments",
-      data: { file_id: item.id, message: "Preview: " + item.name, full_url: getFileUrl(item) },
+      data: {
+        file_id: item.id,
+        message: "Preview: " + item.name,
+        full_url: getFileUrl(item),
+        download_url: getFileDownloadUrl(item)
+      },
       message: "Preview: " + item.name
     });
   };
@@ -20711,6 +20741,7 @@ var useStyles26 = (0, import_antd_style35.createStyles)(({ token, css }) => ({
 var FileRenderer = ({
   fileUrl,
   fileName = "",
+  downloadUrl,
   showHeader = true
 }) => {
   const { styles } = useStyles26();
@@ -20745,7 +20776,14 @@ var FileRenderer = ({
   switch (category) {
     case "markdown":
     case "text":
-      return /* @__PURE__ */ (0, import_jsx_runtime77.jsx)("div", { className: styles.container, children: /* @__PURE__ */ (0, import_jsx_runtime77.jsx)(MarkdownViewer, { content, fileName }) });
+      return /* @__PURE__ */ (0, import_jsx_runtime77.jsx)("div", { className: styles.container, children: /* @__PURE__ */ (0, import_jsx_runtime77.jsx)(
+        MarkdownViewer,
+        {
+          content,
+          fileName,
+          downloadUrl
+        }
+      ) });
     case "image":
       return /* @__PURE__ */ (0, import_jsx_runtime77.jsx)("div", { className: styles.container, children: /* @__PURE__ */ (0, import_jsx_runtime77.jsx)("div", { className: styles.contentWrapper, children: /* @__PURE__ */ (0, import_jsx_runtime77.jsx)("div", { className: styles.imageContainer, children: /* @__PURE__ */ (0, import_jsx_runtime77.jsx)("img", { src: fileUrl, alt: fileName, className: styles.image }) }) }) });
     case "pdf":
@@ -20772,19 +20810,24 @@ var FileRenderer = ({
       return /* @__PURE__ */ (0, import_jsx_runtime77.jsx)("div", { className: styles.container, children: /* @__PURE__ */ (0, import_jsx_runtime77.jsxs)("div", { className: styles.downloadContainer, children: [
         /* @__PURE__ */ (0, import_jsx_runtime77.jsx)(import_icons17.FileUnknownOutlined, { className: styles.fileIcon }),
         /* @__PURE__ */ (0, import_jsx_runtime77.jsx)("span", { className: styles.fileName, children: fileName || "Unknown file" }),
-        /* @__PURE__ */ (0, import_jsx_runtime77.jsx)(import_antd57.Button, { type: "primary", href: fileUrl, download: fileName, children: "Download File" })
+        /* @__PURE__ */ (0, import_jsx_runtime77.jsx)(import_antd57.Button, { type: "primary", href: downloadUrl || fileUrl, download: fileName, children: "Download File" })
       ] }) });
   }
 };
 
 // src/components/GenUI/RenderMDFromURL.tsx
 var import_jsx_runtime78 = require("react/jsx-runtime");
-var RenderMDFromURL = ({ url, file_id }) => {
+var RenderMDFromURL = ({
+  url,
+  file_id,
+  downloadUrl
+}) => {
   return /* @__PURE__ */ (0, import_jsx_runtime78.jsx)(
     FileRenderer,
     {
       fileUrl: url,
       fileName: file_id,
+      downloadUrl,
       showHeader: true
     }
   );
@@ -21029,7 +21072,7 @@ function AttachmentsViewerSideApp({
 }) {
   const [loading, setLoading] = (0, import_react80.useState)(false);
   const [shareOpen, setShareOpen] = (0, import_react80.useState)(false);
-  const { file_id, url, full_url } = data ?? {};
+  const { file_id, url, full_url, download_url } = data ?? {};
   const [fileUri, setFileUri] = (0, import_react80.useState)({
     url: full_url || "",
     fileName: extractFileName(file_id || "")
@@ -21044,13 +21087,21 @@ function AttachmentsViewerSideApp({
     return /* @__PURE__ */ (0, import_jsx_runtime80.jsx)(import_antd59.Skeleton, { active: true });
   }
   const fileUrl = fileUri?.url || url;
+  const downloadUrl = download_url || fileUrl || "";
   const fileName = fileUri?.fileName || "";
   const isText = isTextFile(fileName);
   const isPreviewable = canPreviewInline(fileName);
   const isDir = !!fileName && !fileName.includes(".");
   let content;
   if (isText && fileUrl) {
-    content = /* @__PURE__ */ (0, import_jsx_runtime80.jsx)(RenderMDFromURL, { url: fileUrl, file_id });
+    content = /* @__PURE__ */ (0, import_jsx_runtime80.jsx)(
+      RenderMDFromURL,
+      {
+        url: fileUrl,
+        file_id,
+        downloadUrl
+      }
+    );
   } else if (isPreviewable || url || isDir) {
     content = /* @__PURE__ */ (0, import_jsx_runtime80.jsx)(
       "iframe",
@@ -21065,10 +21116,10 @@ function AttachmentsViewerSideApp({
       {
         description: /* @__PURE__ */ (0, import_jsx_runtime80.jsxs)(import_jsx_runtime80.Fragment, { children: [
           /* @__PURE__ */ (0, import_jsx_runtime80.jsx)("div", { children: "Preview not supported. Please download to view." }),
-          /* @__PURE__ */ (0, import_jsx_runtime80.jsxs)(import_antd59.Button, { type: "link", href: fileUri?.url, download: fileUri?.fileName, children: [
+          downloadUrl ? /* @__PURE__ */ (0, import_jsx_runtime80.jsxs)(import_antd59.Button, { type: "link", href: downloadUrl, download: fileName, children: [
             "Download ",
-            fileUri?.fileName
-          ] })
+            fileName
+          ] }) : null
         ] }),
         image: import_antd59.Empty.PRESENTED_IMAGE_DEFAULT
       }
@@ -21076,17 +21127,17 @@ function AttachmentsViewerSideApp({
   }
   return /* @__PURE__ */ (0, import_jsx_runtime80.jsxs)("div", { style: { display: "flex", flexDirection: "column", height: "100%" }, children: [
     /* @__PURE__ */ (0, import_jsx_runtime80.jsxs)(import_antd59.Space, { style: { padding: "8px 12px", borderBottom: "1px solid #f0f0f0", width: "100%" }, children: [
-      /* @__PURE__ */ (0, import_jsx_runtime80.jsx)(
+      downloadUrl ? /* @__PURE__ */ (0, import_jsx_runtime80.jsx)(
         import_antd59.Button,
         {
           type: "text",
           size: "small",
           icon: /* @__PURE__ */ (0, import_jsx_runtime80.jsx)(import_icons19.DownloadOutlined, {}),
-          href: fileUrl,
+          href: downloadUrl,
           download: fileName,
           children: "Download"
         }
-      ),
+      ) : null,
       /* @__PURE__ */ (0, import_jsx_runtime80.jsx)(
         import_antd59.Button,
         {
@@ -31914,25 +31965,28 @@ var FileLink = ({ data }) => {
   const openSideApp = useSideAppOpener();
   const { threadId, assistantId } = useConversationContext();
   const { config } = useLatticeChatShellContext();
-  const fileBaseURL = (0, import_react112.useMemo)(() => {
+  const fileEndpoint = (0, import_react112.useMemo)(() => {
     if (!config?.baseURL || !assistantId || !threadId) {
       return null;
     }
-    return `${config.baseURL}/api/assistants/${assistantId}/threads/${threadId}/sandbox/downloadfile?path=`;
+    return `${config.baseURL}/api/assistants/${assistantId}/threads/${threadId}/sandbox/downloadfile`;
   }, [config?.baseURL, assistantId, threadId]);
   const handleItemClick = (0, import_react112.useCallback)((path) => {
-    if (!fileBaseURL) return;
+    if (!fileEndpoint) return;
+    const params = new URLSearchParams({ path });
+    const fileUrl = `${fileEndpoint}?${params.toString()}`;
     openSideApp({
       component_key: "attachments",
       data: {
         file_id: path,
         message: "\u9884\u89C8\uFF1A" + path.split("/").pop(),
-        full_url: `${fileBaseURL}${path}`
+        full_url: fileUrl,
+        download_url: fileUrl
       },
       message: "\u9884\u89C8\uFF1A" + path.split("/").pop()
     });
-  }, [fileBaseURL, openSideApp]);
-  if (!fileBaseURL) {
+  }, [fileEndpoint, openSideApp]);
+  if (!fileEndpoint) {
     return /* @__PURE__ */ (0, import_jsx_runtime137.jsx)("span", { children: data });
   }
   return /* @__PURE__ */ (0, import_jsx_runtime137.jsx)(import_antd104.Button, { type: "link", size: "small", onClick: () => handleItemClick(data), children: data });
@@ -46363,7 +46417,7 @@ var FileDirectoryPanel = ({
 }) => {
   const { styles } = useStyles48();
   const { currentAssistant } = useAssistantContext();
-  const { getFileViewUrl, getDirectFileUrl } = useWorkspaceContext();
+  const { getFileDownloadUrl, getDirectFileUrl } = useWorkspaceContext();
   const { openSideApp } = useChatUIContext();
   const { currentTenant } = useAuth();
   const tenantId = currentTenant?.id || "";
@@ -46635,7 +46689,7 @@ var FileDirectoryPanel = ({
                 icon: /* @__PURE__ */ (0, import_jsx_runtime194.jsx)(import_icons50.DownloadOutlined, {}),
                 label: "Download",
                 onClick: () => {
-                  const url = getFileViewUrl(
+                  const url = getFileDownloadUrl(
                     entry.path,
                     currentAssistant?.id
                   );
@@ -46661,7 +46715,7 @@ var FileDirectoryPanel = ({
         }
       )
     ] }, entry.path);
-  }, [directoryChildren, directoryExpanded, directoryLoading, onAssetClick, onToggleDirectory, styles, getFileViewUrl, currentAssistant?.id, enableCustomMenu, pinStates, handleDropdownOpen, handleUnpin]);
+  }, [directoryChildren, directoryExpanded, directoryLoading, onAssetClick, onToggleDirectory, styles, getFileDownloadUrl, currentAssistant?.id, enableCustomMenu, pinStates, handleDropdownOpen, handleUnpin]);
   if (!projectId) {
     return /* @__PURE__ */ (0, import_jsx_runtime194.jsx)("div", { className: styles.emptyState, children: "Please select a project to browse files" });
   }
@@ -46764,6 +46818,7 @@ var ToolPanelFiles = () => {
     listPath,
     listPathByFolder,
     getFileViewUrl,
+    getFileDownloadUrl,
     uploadFileToFolder
   } = useWorkspaceContext();
   const { currentAssistant } = useAssistantContext();
@@ -46841,7 +46896,8 @@ var ToolPanelFiles = () => {
   }, [resourceFolders, loadAssetsForFolder]);
   const handleAssetClick = (0, import_react171.useCallback)((asset) => {
     const fileUrl = getFileViewUrl(asset.path, currentAssistant?.id);
-    if (!fileUrl) {
+    const downloadUrl = getFileDownloadUrl(asset.path, currentAssistant?.id);
+    if (!fileUrl || !downloadUrl) {
       import_antd147.message.warning("Please select a workspace and project first.");
       return;
     }
@@ -46850,11 +46906,12 @@ var ToolPanelFiles = () => {
       data: {
         file_id: asset.path,
         message: `Preview: ${asset.name || asset.path}`,
-        full_url: fileUrl
+        full_url: fileUrl,
+        download_url: downloadUrl
       },
       message: `Preview: ${asset.name || asset.path}`
     });
-  }, [getFileViewUrl, openSideApp, currentAssistant?.id]);
+  }, [getFileViewUrl, getFileDownloadUrl, openSideApp, currentAssistant?.id]);
   const handleUploadFolder = (0, import_react171.useCallback)(async (folderName) => {
     if (!workspaceId || !projectId) {
       import_antd147.message.warning("Please select a workspace and project before uploading.");