@athenaintel/react 0.10.2 → 0.10.3

package/dist/index.js

@@ -24957,6 +24957,7 @@ function AthenaWithThreadList({
   linkClicks,
   citationLinks
 }) {
+  var _a2;
   const adapter = useAthenaThreadListAdapter({
     backendUrl,
     apiKey,
@@ -24999,15 +25000,17 @@ function AthenaWithThreadList({
     adapter
   });
   const handleRefresh = useCallback(() => {
-    var _a2;
-    const core = (_a2 = runtime == null ? void 0 : runtime._core) == null ? void 0 : _a2.threads;
+    var _a3;
+    const core = (_a3 = runtime == null ? void 0 : runtime._core) == null ? void 0 : _a3.threads;
     if (core) {
       core._loadThreadsPromise = void 0;
       core.__internal_load();
     }
   }, [runtime]);
   const previousAuthRefreshKeyRef = useRef(null);
-  const authRefreshKey = `${backendUrl}::${apiKey ?? ""}::${token ?? ""}`;
+  const authContext = useContext(AthenaAuthContext);
+  const authUserId = ((_a2 = authContext == null ? void 0 : authContext.user) == null ? void 0 : _a2.userId) ?? "";
+  const authRefreshKey = `${backendUrl}::${apiKey ?? ""}::${authUserId}`;
   useEffect(() => {
     const previousAuthRefreshKey = previousAuthRefreshKeyRef.current;
     previousAuthRefreshKeyRef.current = authRefreshKey;
@@ -64826,11 +64829,24 @@ function highlightSql(code2) {
       i += str.length;
       continue;
     }
+    if (code2.slice(i, i + 2) === "/*") {
+      const end = code2.indexOf("*/", i + 2);
+      const str = end >= 0 ? code2.slice(i, end + 2) : code2.slice(i);
+      result += `<span style="color:var(--aui-syn-comment, #6a9955)">${escapeHtml(str)}</span>`;
+      i += str.length;
+      continue;
+    }
     if (code2[i] === "'" || code2[i] === '"') {
       const q = code2[i];
       let j = i + 1;
-      while (j < code2.length && code2[j] !== q) {
-        if (code2[j] === "\\") j++;
+      while (j < code2.length) {
+        if (code2[j] === q) {
+          if (j + 1 < code2.length && code2[j + 1] === q) {
+            j += 2;
+            continue;
+          }
+          break;
+        }
         j++;
       }
       const str = code2.slice(i, j + 1);
@@ -64979,9 +64995,11 @@ const RunSqlToolUIImpl = ({
               i
             )) })
           ] }) }),
-          parsed.rows.length > 50 && /* @__PURE__ */ jsxs("div", { className: "border-t border-border/30 px-4 py-1.5 text-[10px] text-muted-foreground", children: [
-            "Showing first 50 of ",
-            parsed.rows.length,
+          (parsed.rows.length > 50 || parsed.rowCount !== null && parsed.rowCount > parsed.rows.length) && /* @__PURE__ */ jsxs("div", { className: "border-t border-border/30 px-4 py-1.5 text-[10px] text-muted-foreground", children: [
+            "Showing first ",
+            Math.min(50, parsed.rows.length),
+            " of ",
+            parsed.rowCount ?? parsed.rows.length,
             " rows"
           ] })
         ] }),