@athenaintel/react 0.9.10 → 0.9.11

package/dist/index.cjs

@@ -23855,6 +23855,10 @@ const nullableToOptionalString = preprocessType(
   stringType().optional()
 );
 const metadataRecordSchema = recordType(stringType(), unknownType());
+const nullableToOptionalMetadataRecord = preprocessType(
+  (val) => val === null ? void 0 : val,
+  metadataRecordSchema.optional()
+);
 const systemMessageSchema = objectType({
   id: nullableToOptionalString,
   type: literalType("system"),
@@ -23893,8 +23897,8 @@ const aiMessageSchema = objectType({
   tool_calls: arrayType(anyType()).optional(),
   status: anyType().optional(),
   additional_kwargs: metadataRecordSchema.optional(),
-  usage_metadata: metadataRecordSchema.optional(),
-  response_metadata: metadataRecordSchema.optional()
+  usage_metadata: nullableToOptionalMetadataRecord,
+  response_metadata: nullableToOptionalMetadataRecord
 });
 const formatValidationError = (error2, msg) => {
   const msgObj = msg && typeof msg === "object" ? msg : {};
@@ -64362,16 +64366,21 @@ const AthenaReasoningPart = ({
 }) => {
   const inheritedTextComponent = React.useContext(AthenaReasoningTextComponentContext);
   const EffectiveTextComponent = TextComponent ?? inheritedTextComponent ?? TiptapText;
+  const isMessageRunning = useAuiState((state) => {
+    var _a2;
+    return ((_a2 = state.message.status) == null ? void 0 : _a2.type) === "running";
+  });
   const reasoningTokens = useAuiState(
     (state) => getReasoningTokensFromMetadata(state.message.metadata)
   );
-  const isRunning = (status == null ? void 0 : status.type) === "running";
+  const isRunning = isMessageRunning || (status == null ? void 0 : status.type) === "running";
   const [isCollapsed, setIsCollapsed] = React.useState(!isRunning);
   const userOverrodeOpenStateRef = React.useRef(false);
+  const hasText = text2.trim().length > 0;
   const reasoningTextProps = {
     type: "text",
     text: text2,
-    status
+    status: isRunning ? { type: "running" } : status
   };
   React.useEffect(() => {
     if (userOverrodeOpenStateRef.current) {
@@ -64379,7 +64388,7 @@ const AthenaReasoningPart = ({
     }
     setIsCollapsed(!isRunning);
   }, [isRunning]);
-  if (!text2 && !isRunning) {
+  if (!hasText && !isRunning) {
     return null;
   }
   return /* @__PURE__ */ jsxRuntime.jsxs(
@@ -64420,10 +64429,10 @@ const AthenaReasoningPart = ({
             ]
           }
         ) }),
-        /* @__PURE__ */ jsxRuntime.jsx(CollapsibleContent, { className: "pt-3", children: isRunning && !text2 ? /* @__PURE__ */ jsxRuntime.jsx("span", { className: "shimmer text-sm text-muted-foreground", children: "Analyzing..." }) : /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "aui-assistant-reasoning-body text-sm", children: [
-          /* @__PURE__ */ jsxRuntime.jsx(EffectiveTextComponent, { ...reasoningTextProps }),
-          isRunning && /* @__PURE__ */ jsxRuntime.jsx("span", { className: "mt-2 inline-block shimmer text-sm text-muted-foreground", children: "Thinking..." })
-        ] }) })
+        /* @__PURE__ */ jsxRuntime.jsx(CollapsibleContent, { className: "pt-3", children: isRunning && !hasText ? /* @__PURE__ */ jsxRuntime.jsx("span", { className: "shimmer text-sm text-muted-foreground", children: "Analyzing..." }) : isRunning ? /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "whitespace-pre-wrap break-words text-sm leading-relaxed text-foreground", children: [
+          text2,
+          /* @__PURE__ */ jsxRuntime.jsx("span", { className: "mt-2 inline-block shimmer text-sm text-muted-foreground", children: "Thinking..." })
+        ] }) : /* @__PURE__ */ jsxRuntime.jsx("div", { className: "aui-assistant-reasoning-body text-sm", children: /* @__PURE__ */ jsxRuntime.jsx(EffectiveTextComponent, { ...reasoningTextProps }) }) })
       ]
     }
   );