@firstlovecenter/ai-chat 0.9.2 → 0.9.4

package/dist/ui/index.cjs

@@ -7,6 +7,8 @@ var Link = require('next/link');
 var lucideReact = require('lucide-react');
 var radixUi = require('radix-ui');
 var jsxRuntime = require('react/jsx-runtime');
+var ReactMarkdown = require('react-markdown');
+var remarkGfm = require('remark-gfm');
 var RechartsPrimitive = require('recharts');
 var react = require('@ai-sdk/react');
 
@@ -32,6 +34,8 @@ function _interopNamespace(e) {
 
 var React__namespace = /*#__PURE__*/_interopNamespace(React);
 var Link__default = /*#__PURE__*/_interopDefault(Link);
+var ReactMarkdown__default = /*#__PURE__*/_interopDefault(ReactMarkdown);
+var remarkGfm__default = /*#__PURE__*/_interopDefault(remarkGfm);
 var RechartsPrimitive__namespace = /*#__PURE__*/_interopNamespace(RechartsPrimitive);
 
 // src/ui/_shared/cn.ts
@@ -512,22 +516,54 @@ function isBlockEmpty(b) {
   if (b.kind === "table") return !Array.isArray(b.rows) || b.rows.length === 0;
   return false;
 }
+var PROSE_MD_COMPONENTS = {
+  p: ({ children }) => /* @__PURE__ */ jsxRuntime.jsx("p", { className: "mb-2 last:mb-0", children }),
+  ul: ({ children }) => /* @__PURE__ */ jsxRuntime.jsx("ul", { className: "my-1 list-disc pl-6", children }),
+  ol: ({ children }) => /* @__PURE__ */ jsxRuntime.jsx("ol", { className: "my-1 list-decimal pl-6", children }),
+  li: ({ children }) => /* @__PURE__ */ jsxRuntime.jsx("li", { className: "leading-6", children }),
+  strong: ({ children }) => /* @__PURE__ */ jsxRuntime.jsx("strong", { className: "font-semibold", children }),
+  em: ({ children }) => /* @__PURE__ */ jsxRuntime.jsx("em", { className: "italic", children }),
+  code: ({ children }) => /* @__PURE__ */ jsxRuntime.jsx("code", { className: "rounded bg-muted px-1 py-0.5 font-mono text-[0.85em]", children }),
+  a: ({ children, href }) => /* @__PURE__ */ jsxRuntime.jsx(
+    "a",
+    {
+      href,
+      target: "_blank",
+      rel: "noreferrer noopener",
+      className: "underline underline-offset-2",
+      children
+    }
+  )
+};
+var INLINE_MD_COMPONENTS = {
+  ...PROSE_MD_COMPONENTS,
+  p: ({ children }) => /* @__PURE__ */ jsxRuntime.jsx(jsxRuntime.Fragment, { children })
+};
+function ProseMarkdown({ children }) {
+  return /* @__PURE__ */ jsxRuntime.jsx(ReactMarkdown__default.default, { remarkPlugins: [remarkGfm__default.default], components: PROSE_MD_COMPONENTS, children });
+}
+function InlineMarkdown({ children }) {
+  return /* @__PURE__ */ jsxRuntime.jsx(ReactMarkdown__default.default, { remarkPlugins: [remarkGfm__default.default], components: INLINE_MD_COMPONENTS, children });
+}
 function AnswerBlocks({ blocks }) {
   const visible = blocks.filter((b) => !isBlockEmpty(b));
   return /* @__PURE__ */ jsxRuntime.jsx("div", { className: "flex flex-col gap-3", children: visible.map((b, i) => /* @__PURE__ */ jsxRuntime.jsx(BlockView, { block: b }, i)) });
 }
 function BlockView({ block }) {
   if (block.kind === "paragraph_brief") {
-    return /* @__PURE__ */ jsxRuntime.jsx("p", { className: "whitespace-pre-wrap text-sm leading-6 text-foreground", children: block.prose || /* @__PURE__ */ jsxRuntime.jsxs("span", { className: "text-muted-foreground", children: [
-      block.key_facts.join(". "),
-      "\u2026"
-    ] }) });
+    if (!block.prose) {
+      return /* @__PURE__ */ jsxRuntime.jsxs("p", { className: "text-sm leading-6 text-muted-foreground", children: [
+        block.key_facts.join(". "),
+        "\u2026"
+      ] });
+    }
+    return /* @__PURE__ */ jsxRuntime.jsx("div", { className: "text-sm leading-6 text-foreground", children: /* @__PURE__ */ jsxRuntime.jsx(ProseMarkdown, { children: block.prose }) });
   }
   if (block.kind === "list") {
     const Tag = block.style === "numbered" ? "ol" : "ul";
     return /* @__PURE__ */ jsxRuntime.jsxs("div", { children: [
       block.title && /* @__PURE__ */ jsxRuntime.jsx("p", { className: "mb-1 text-sm font-medium", children: block.title }),
-      /* @__PURE__ */ jsxRuntime.jsx(Tag, { className: block.style === "numbered" ? "list-decimal pl-6" : "list-disc pl-6", children: block.items.map((it, i) => /* @__PURE__ */ jsxRuntime.jsx("li", { className: "text-sm leading-6", children: it }, i)) })
+      /* @__PURE__ */ jsxRuntime.jsx(Tag, { className: block.style === "numbered" ? "list-decimal pl-6" : "list-disc pl-6", children: block.items.map((it, i) => /* @__PURE__ */ jsxRuntime.jsx("li", { className: "text-sm leading-6", children: /* @__PURE__ */ jsxRuntime.jsx(InlineMarkdown, { children: it }) }, i)) })
     ] });
   }
   if (block.kind === "chart") {
@@ -544,7 +580,7 @@ function BlockView({ block }) {
   }
   if (block.kind === "callout") {
     const cls = block.tone === "warn" ? "border-amber-300 dark:border-amber-800 bg-amber-50 dark:bg-amber-950/30 text-amber-900 dark:text-amber-100" : block.tone === "success" ? "border-emerald-300 dark:border-emerald-800 bg-emerald-50 dark:bg-emerald-950/30 text-emerald-900 dark:text-emerald-100" : "border-sky-300 bg-sky-50 text-sky-900";
-    return /* @__PURE__ */ jsxRuntime.jsx("div", { className: `rounded border p-3 text-sm ${cls}`, children: block.text });
+    return /* @__PURE__ */ jsxRuntime.jsx("div", { className: `rounded border p-3 text-sm leading-6 ${cls}`, children: /* @__PURE__ */ jsxRuntime.jsx(ProseMarkdown, { children: block.text }) });
   }
   return null;
 }
@@ -723,6 +759,11 @@ function AiChat({
       cancelled = true;
     };
   }, []);
+  React.useEffect(() => {
+    if (initialSessionId != null && activeSessionId !== initialSessionId) {
+      setActiveSessionId(initialSessionId);
+    }
+  }, [initialSessionId]);
   React.useEffect(() => {
     if (initialSessionId == null) {
       setAnswers([]);
@@ -735,13 +776,18 @@ function AiChat({
         const res = await fetch(`/api/chat/sessions/${id}`, { cache: "no-store" });
         if (cancelled) return;
         if (!res.ok) {
+          console.warn(
+            "[AiChat] hydration: GET /api/chat/sessions failed",
+            { id, status: res.status }
+          );
           setAnswers([]);
           return;
         }
         const data = await res.json();
         if (cancelled) return;
         setAnswers(messagesToAnswers(data.messages ?? []));
-      } catch {
+      } catch (err) {
+        console.error("[AiChat] hydration threw \u2014 setting empty answers", err);
         if (!cancelled) setAnswers([]);
       } finally {
         if (!cancelled) setLoadingSession(false);
@@ -1483,6 +1529,15 @@ function updateLast(prev, updater) {
   next[next.length - 1] = updater(next[next.length - 1]);
   return next;
 }
+function coerceJsonField(v) {
+  if (v == null) return null;
+  if (typeof v !== "string") return v;
+  try {
+    return JSON.parse(v);
+  } catch {
+    return null;
+  }
+}
 function messagesToAnswers(messages) {
   const out = [];
   let pendingUser = null;
@@ -1493,19 +1548,24 @@ function messagesToAnswers(messages) {
       const question = pendingUser ?? "";
       pendingUser = null;
       const blocks = [];
-      const stored = m.blocks ?? [];
-      stored.forEach((b, i) => {
-        const sanitised = sanitiseBlock({ ...b});
-        if (sanitised.kind === "paragraph_brief") {
-          sanitised.prose = m.prose?.[String(i)] ?? "";
-        }
-        blocks[i] = sanitised;
-      });
+      const storedBlocks = coerceJsonField(m.blocks) ?? [];
+      const storedProse = coerceJsonField(m.prose) ?? {};
+      const storedError = coerceJsonField(m.errorJson);
+      if (Array.isArray(storedBlocks)) {
+        storedBlocks.forEach((b, i) => {
+          const sanitised = sanitiseBlock({
+            ...b});
+          if (sanitised.kind === "paragraph_brief") {
+            sanitised.prose = storedProse[String(i)] ?? "";
+          }
+          blocks[i] = sanitised;
+        });
+      }
       out.push({
         question,
         blocks,
         done: true,
-        error: m.errorJson ?? void 0
+        error: storedError ?? void 0
       });
     }
   }
@@ -2401,6 +2461,15 @@ function UserChip2({ text }) {
     )
   ] });
 }
+function coerceJsonField2(v) {
+  if (v == null) return null;
+  if (typeof v !== "string") return v;
+  try {
+    return JSON.parse(v);
+  } catch {
+    return null;
+  }
+}
 function storedToUseChat(stored) {
   const uiMessages = [];
   const blocksMap = {};
@@ -2416,15 +2485,20 @@ function storedToUseChat(stored) {
     if (m.role === "assistant") {
       const id = `assistant-${m.id}`;
       const blocks = [];
-      const stored2 = m.blocks ?? [];
-      stored2.forEach((b, i) => {
-        const sanitised = sanitiseBlock({ ...b});
-        if (sanitised.kind === "paragraph_brief") {
-          sanitised.prose = m.prose?.[String(i)] ?? "";
-        }
-        blocks[i] = sanitised;
-      });
-      const joinedProse = Object.values(m.prose ?? {}).filter(Boolean).join("\n\n");
+      const storedBlocks = coerceJsonField2(m.blocks) ?? [];
+      const storedProse = coerceJsonField2(m.prose) ?? {};
+      const storedError = coerceJsonField2(m.errorJson);
+      if (Array.isArray(storedBlocks)) {
+        storedBlocks.forEach((b, i) => {
+          const sanitised = sanitiseBlock({
+            ...b});
+          if (sanitised.kind === "paragraph_brief") {
+            sanitised.prose = storedProse[String(i)] ?? "";
+          }
+          blocks[i] = sanitised;
+        });
+      }
+      const joinedProse = Object.values(storedProse).filter(Boolean).join("\n\n");
       uiMessages.push({
         id,
         role: "assistant",
@@ -2432,7 +2506,7 @@ function storedToUseChat(stored) {
       });
       blocksMap[id] = blocks;
       if (joinedProse) proseMap[id] = joinedProse;
-      if (m.errorJson) errorsMap[id] = m.errorJson;
+      if (storedError) errorsMap[id] = storedError;
     }
   }
   return { uiMessages, blocksMap, proseMap, errorsMap };