@lastbrain/ai-ui-react 1.0.10 → 1.0.12

package/src/components/AiSelect.tsx

@@ -5,6 +5,7 @@ import type { BaseAiProps } from "../types";
 import { useAiCallText } from "../hooks/useAiCallText";
 import { useAiModels } from "../hooks/useAiModels";
 import { AiPromptPanel } from "./AiPromptPanel";
+import { UsageToast, useUsageToast } from "./UsageToast";
 import { aiStyles } from "../styles/inline";
 
 export interface AiSelectProps
@@ -31,6 +32,7 @@ export function AiSelect({
 }: AiSelectProps) {
   const [isOpen, setIsOpen] = useState(false);
   const [isFocused, setIsFocused] = useState(false);
+  const { showUsageToast, toastData, toastKey, clearToast } = useUsageToast();
 
   const { models } = useAiModels({ baseUrl, apiKeyId });
   const { generateText, loading } = useAiCallText({ baseUrl, apiKeyId });
@@ -58,6 +60,7 @@ export function AiSelect({
       if (result.text) {
         onValue?.(result.text);
         onToast?.({ type: "success", message: "AI suggestion ready" });
+        showUsageToast(result);
       }
     } catch (error) {
       onToast?.({ type: "error", message: "Failed to generate suggestion" });
@@ -95,6 +98,14 @@ export function AiSelect({
           models={models || []}
         />
       )}
+      {Boolean(toastData) && (
+        <UsageToast
+          key={toastKey}
+          result={toastData}
+          position="bottom-right"
+          onComplete={clearToast}
+        />
+      )}
     </div>
   );
 }

package/src/components/AiSettingsButton.tsx

@@ -66,10 +66,12 @@ export function AiSettingsButton({
                         <span>Tokens:</span>
                         <span>{status.balance?.total || 0}</span>
                       </div>
-                      {status.storage?.total_mb !== undefined && (
+                      {status.storage?.allocated_mb !== undefined && (
                         <div data-ai-status-item>
                           <span>Storage:</span>
-                          <span>{status.storage.total_mb.toFixed(2)} MB</span>
+                          <span>
+                            {status.storage.allocated_mb.toFixed(2)} MB
+                          </span>
                         </div>
                       )}
                     </div>