@arbidocs/blocks 0.3.112 → 0.3.114

package/dist/index.js

@@ -2,10 +2,10 @@ import { create } from 'zustand';
 import { persist } from 'zustand/middleware';
 import { createContext, useState, useEffect, useCallback, useMemo, useContext, useRef } from 'react';
 import { MessageSquare, Cpu, Code, Cloud, Layers, Target, Award, MapPin, Phone, Mail, Lock, Globe, Heart, Star, Rocket, Shield, Zap, Sparkles, Gavel, Scale, Bell, Activity, BarChart3, DollarSign, TrendingUp, Info, AlertCircle, CheckCircle, Clock, Calendar, Briefcase, Users, Folder, FileText, RotateCcw, Palette, Search, ArrowUpRight, ArrowDownRight, ImageIcon, ArrowRight, Quote, ChevronDown, XCircle, AlertTriangle, CheckCircle2, Check, ArrowLeft, Loader2, ChevronUp, BrainCircuit, Bot, LayoutGrid, FileJson, Save, Download, ImagePlus, Upload, Link2 } from 'lucide-react';
-import { Input, Card, CardHeader, CardTitle, CardContent, useArbi, ArbiProvider, useAiTask, useWorkspaceDocs as useWorkspaceDocs$1, useSemanticSearch as useSemanticSearch$1, cn, Table, TableHeader, TableRow, TableHead, TableBody, TableCell, Button, Avatar, AvatarImage, AvatarFallback, AiMarkdown, Badge, Switch, useConfigs, useAgents, Checkbox, Tabs as Tabs$1, TabsList, TabsTrigger, TabsContent, Separator, ArbiWebSocketProvider, useThumbnails, useDocuments } from '@arbidocs/react';
+import { Input, Card, CardHeader, CardTitle, CardContent, useArbi, ArbiProvider, useAiTask, useWorkspaceDocs as useWorkspaceDocs$1, useSemanticSearch as useSemanticSearch$1, cn, Table, TableHeader, TableRow, TableHead, TableBody, TableCell, Button, Avatar, AvatarImage, AvatarFallback, AiMarkdown, Badge, Switch, useConfigs, useAgents, Checkbox, Tabs as Tabs$1, TabsList, TabsTrigger, TabsContent, Separator, ArbiWebSocketProvider, useDocuments, useThumbnails } from '@arbidocs/react';
 export { Table, TableBody, TableCell, TableHead, TableHeader, TableRow, redlineStats, textToArtifact, toRedlineMarkdown } from '@arbidocs/react';
 import { jsxs, jsx, Fragment } from 'react/jsx-runtime';
-import { QueryClient, QueryClientProvider, useQuery } from '@tanstack/react-query';
+import { QueryClient, QueryClientProvider, useQuery, useQueryClient } from '@tanstack/react-query';
 import { AgGridReact } from 'ag-grid-react';
 import { themeQuartz, AllCommunityModule } from 'ag-grid-community';
 import { Link, useParams } from 'react-router-dom';
@@ -1450,14 +1450,35 @@ function useAssetResolverActive() {
 function isAssetRef(src) {
   return !!src && src.startsWith(ASSET_REF_PREFIX);
 }
+function useAssetImage(docId) {
+  const arbi = useArbi();
+  const { data: thumbs } = useThumbnails([docId]);
+  const thumb = thumbs?.[docId];
+  const { data: objectUrl } = useQuery({
+    queryKey: ["arbi", "asset-image", docId],
+    queryFn: async () => {
+      const res = await arbi.documents.download(docId);
+      const blob = await res.blob();
+      return URL.createObjectURL(blob);
+    },
+    // Only download when the thumbnail is genuinely absent (not merely loading).
+    enabled: !!docId && thumbs != null && !thumb && arbi.hasWorkspace,
+    staleTime: 10 * 6e4
+  });
+  useEffect(() => {
+    return () => {
+      if (objectUrl) URL.revokeObjectURL(objectUrl);
+    };
+  }, [objectUrl]);
+  return thumb ?? objectUrl;
+}
 function AssetThumb({
   docId,
   alt,
   className,
   testId
 }) {
-  const { data } = useThumbnails([docId]);
-  const src = data?.[docId];
+  const src = useAssetImage(docId);
   return /* @__PURE__ */ jsx("img", { src: src ?? void 0, alt: alt ?? "", className, "data-testid": testId });
 }
 var ASPECT = {
@@ -2864,7 +2885,9 @@ function ImageFieldControl({
   const [busy, setBusy] = useState(false);
   const arbi = useArbi();
   const gen = useImageGen();
+  const queryClient = useQueryClient();
   const tp = (...q) => tid(`${testIdPrefix}-image`, name, ...q);
+  const refreshAssets = () => queryClient.invalidateQueries({ queryKey: ["arbi", "documents", workspaceId] });
   const { data: docs } = useDocuments(workspaceId);
   const imageDocs = useMemo(
     () => (docs ?? []).filter(
@@ -2885,7 +2908,10 @@ function ImageFieldControl({
       await arbi.selectWorkspace(workspaceId);
       const res = await arbi.documents.uploadFile(file, file.name, { folder });
       const id = res.doc_ext_ids?.[0];
-      if (id) onChange(asAssetRef(id));
+      if (id) {
+        onChange(asAssetRef(id));
+        void refreshAssets();
+      }
     } finally {
       setBusy(false);
     }
@@ -2893,7 +2919,10 @@ function ImageFieldControl({
   const onGenerate = async () => {
     if (!prompt.trim() || gen.isGenerating) return;
     const id = await gen.generate(prompt);
-    if (id) onChange(asAssetRef(id));
+    if (id) {
+      onChange(asAssetRef(id));
+      void refreshAssets();
+    }
   };
   const MODES = [
     { key: "asset", label: "Assets", icon: ImagePlus },