@arbidocs/react 0.4.27 → 0.4.28

package/dist/index.js

@@ -1,7 +1,7 @@
-import { useDocuments, useConversations, useAssistantQuery, useConversationThreads, arbiQueryKeys, useWorkspaces, useParsedDocument, useTags, useActivity, useThumbnails } from './chunk-DLT354KA.js';
-export { DEFAULT_USD_PER_CREDIT, arbiQueryKeys, cleanDraftMarkdown, cleanDraftWithReport, creditsToUsd, formatCredits, isUsageMetered, promoteBoldTitle, repairParsedStructure, repairParsedTitle, stripAgentPreamble, unescapeMarkdown, unwrapCitationLinks, useActivity, useAgents, useAssistantQuery, useConfigs, useConversationThreads, useConversations, useDeleteAgents, useDocuments, useFacts, useHealth, useParsedDocument, useProjectInvoices, useProjectUsage, useProjects, useProjectsUsage, useSessions, useTags, useThumbnails, useUserUsageToday, useWorkspaces } from './chunk-DLT354KA.js';
-import { formatRelativeTime } from './chunk-FSMIN5VX.js';
-export { DEFAULT_NOTIFY_TONE, NotificationBell, Toaster, formatRelativeTime, notify, seedNotifications, selectUnread, toast, useNotify } from './chunk-FSMIN5VX.js';
+import { arbiQueryKeys, useDocuments, useConversations, useAssistantQuery, useConversationThreads, useWorkspaces, useParsedDocument, useTags, useActivity, useThumbnails } from './chunk-NDG44JWI.js';
+export { DEFAULT_USD_PER_CREDIT, arbiQueryKeys, cleanDraftMarkdown, cleanDraftWithReport, creditsToUsd, formatCredits, isUsageMetered, promoteBoldTitle, repairParsedStructure, repairParsedTitle, stripAgentPreamble, unescapeMarkdown, unwrapCitationLinks, useActivity, useAgents, useAssistantQuery, useConfigs, useConversationThreads, useConversations, useDeleteAgents, useDocuments, useFacts, useHealth, useParsedDocument, useProjectInvoices, useProjectUsage, useProjects, useProjectsUsage, useSessions, useTags, useThumbnails, useUserUsageToday, useWorkspaceUsers, useWorkspaces } from './chunk-NDG44JWI.js';
+import { formatRelativeTime } from './chunk-OBTCAI4E.js';
+export { DEFAULT_NOTIFY_TONE, NotificationBell, Toaster, formatRelativeTime, notify, seedNotifications, selectUnread, toast, useNotify } from './chunk-OBTCAI4E.js';
 import { MarkMenu, AiMarkdown, Card, Button, Badge, TagEntity, Input, FILE_TYPE_COLORS, getEntityIcon, UserEntity, getEntityColorStyles, AgentBadge, Textarea, Label, Slider, Dialog, DialogContent, DialogHeader, DialogTitle, DialogFooter, DocumentEntity, DocumentCard, useAudioRecorder, MicWaveformIndicator, IconToggle, classifyMediaError, formatElapsed, PassageCard } from './chunk-SDUISFTV.js';
 export { AgentBadge, AiMarkdown, ArbiBotHeadIcon, AudIcon, Avatar, AvatarFallback, AvatarImage, BASE_COLORS, Badge, Button, Card, CardContent, CardDescription, CardFooter, CardHeader, CardTitle, Checkbox, Collapsible, CollapsibleContent, CollapsibleTrigger, Command, CommandDialog, CommandEmpty, CommandGroup, CommandInput, CommandItem, CommandList, CommandSeparator, CommandShortcut, Dialog, DialogClose, DialogContent, DialogDescription, DialogFooter, DialogHeader, DialogOverlay, DialogPortal, DialogTitle, DialogTrigger, DictateProvider, DocumentCard, DocumentEntity, ENTITY_MAPPINGS, EmlIcon, EntityIcon, EntityPill, FILE_TYPE_COLORS, HoverCard, HoverCardContent, HoverCardTrigger, IconToggle, ImgIcon, InlineMicButton, Input, Label, MarkMenu, MicRecordingOverlay, MicWaveformIndicator, NEUTRAL_GRAY, PLACEHOLDER_COLOR, PassageCard, PdfIcon, Popover, PopoverAnchor, PopoverContent, PopoverTrigger, PresenceLabel, Progress, ProjectEntity, RadioGroup, RadioGroupItem, RagDot, RelevanceDot, Select, SelectContent, SelectGroup, SelectItem, SelectLabel, SelectScrollDownButton, SelectScrollUpButton, SelectSeparator, SelectTrigger, SelectValue, Separator, Slider, StatusPill, Switch, TAG_TYPE_ICONS, TAG_TYPE_LABELS, Table, TableBody, TableCaption, TableCell, TableFooter, TableHead, TableHeader, TableRow, Tabs, TabsContent, TabsList, TabsTrigger, TagEntity, Textarea, Toggle, ToggleGroup, ToggleGroupItem, Tooltip, TooltipContent, TooltipProvider, TooltipTrigger, UserEntity, UserMenu, VISIBILITY_ICONS, VidIcon, VisibilityIcon, VisibilityToggle, badgeVariants, buttonVariants, cardVariants, configureEntityRegistry, configureRagLabels, configureStatusRegistry, formatElapsed, getDocIcon, getEntityColor, getEntityColorStyles, getEntityIcon, getEntityLabel, getStatusVariant, getTagTypeIcon, getTagTypeLabel, getWorkspaceColor, toggleVariants, useAudioRecorder } from './chunk-SDUISFTV.js';
 import { normalizeCitationMarkup, MarkdownRenderer } from './chunk-6YSHODMG.js';
@@ -14,6 +14,7 @@ import { AudioViewer, ImageViewer, VideoViewer } from './chunk-HDCRPPBW.js';
 import { useArbi } from './chunk-J4MX2W4H.js';
 export { ArbiProvider, useArbi } from './chunk-J4MX2W4H.js';
 import React, { memo, useState, createContext, useMemo, useCallback, useEffect, useRef, useLayoutEffect, useContext, useId, Fragment as Fragment$1 } from 'react';
+import { useQueryClient, useMutation, useQuery } from '@tanstack/react-query';
 import { resolveCitations, summarizeCitations, connectWithReconnect } from '@arbidocs/sdk/browser';
 export { Arbi, ArbiError } from '@arbidocs/sdk/browser';
 import { Link, Trash2, CheckCheck, Check, Loader2, AlertTriangle, ShieldCheck, Copy, Sparkles, FileText, GitCompareArrows, Download, Image as Image$1, Search, CornerDownLeft, X, ArrowLeft, Send, Tags, Crown, Info, Activity, Gauge, Gift, PartyPopper, UserCheck, AtSign, MessageCircleMore, FileX2, FileCheck2, Pencil, FolderPlus, SquareArrowOutUpRight, MessagesSquare, ArrowRight, MoreHorizontal, SquarePen, Mic, Pause, Play, Square, RotateCcw, Camera, Circle } from 'lucide-react';
@@ -21,7 +22,6 @@ import { jsx, jsxs, Fragment } from 'react/jsx-runtime';
 import { isMessageType, passwordChangeErrorMessage } from '@arbidocs/client';
 export { isMessageType } from '@arbidocs/client';
 import { create } from 'zustand';
-import { useQueryClient, useMutation, useQuery } from '@tanstack/react-query';
 import { diffWords } from 'diff';
 import { AnimatePresence, motion } from 'framer-motion';
 import { createPortal } from 'react-dom';
@@ -30,6 +30,319 @@ import Cropper from 'react-easy-crop';
 import { AgGridReact } from 'ag-grid-react';
 import { ModuleRegistry, AllCommunityModule } from 'ag-grid-community';
 
+function useUpdateConversation() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: ({
+      conversationId,
+      patch
+    }) => arbi.conversations.update(conversationId, patch),
+    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.allConversations() })
+  });
+}
+function useCreateWork() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: (body) => arbi.conversations.createWork(body),
+    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.allConversations() })
+  });
+}
+function useRunConversation() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: (conversationId) => arbi.conversations.runWithAuth(conversationId),
+    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.allConversations() })
+  });
+}
+function useReviewConversation() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: ({
+      conversationId,
+      review
+    }) => arbi.conversations.review(conversationId, review),
+    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.allConversations() })
+  });
+}
+function useRequestConversationApproval() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: ({
+      conversationId,
+      request
+    }) => arbi.conversations.requestApproval(conversationId, request),
+    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.allConversations() })
+  });
+}
+function usePlanWork() {
+  const arbi = useArbi();
+  return useMutation({
+    mutationFn: (body) => arbi.conversations.plan(body)
+  });
+}
+function useConversationPendingAuth(conversationId, options) {
+  const arbi = useArbi();
+  return useQuery({
+    queryKey: [...arbiQueryKeys.allConversations(), conversationId ?? "", "pending-auth"],
+    queryFn: () => arbi.conversations.pendingAuth(conversationId),
+    enabled: (options?.enabled ?? true) && !!conversationId && arbi.isLoggedIn,
+    retry: false
+  });
+}
+function useConversationActivity(conversationId, options) {
+  const arbi = useArbi();
+  return useQuery({
+    queryKey: [...arbiQueryKeys.allConversations(), conversationId ?? "", "activity"],
+    queryFn: () => arbi.conversations.activity(conversationId),
+    enabled: (options?.enabled ?? true) && !!conversationId && arbi.isLoggedIn,
+    refetchInterval: options?.live ? 2500 : false
+  });
+}
+function useWorkflows(options) {
+  const arbi = useArbi();
+  return useQuery({
+    queryKey: arbiQueryKeys.workflows(),
+    queryFn: () => arbi.workflows.list(),
+    refetchInterval: options?.live ? 15e3 : false,
+    enabled: options?.enabled ?? arbi.isLoggedIn
+  });
+}
+function useCreateWorkflow() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: (body) => arbi.workflows.create(body),
+    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.workflows() })
+  });
+}
+function useUpdateWorkflow() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: ({
+      workflowId,
+      patch
+    }) => arbi.workflows.update(workflowId, patch),
+    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.workflows() })
+  });
+}
+function useDeleteWorkflow() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: (workflowId) => arbi.workflows.delete(workflowId),
+    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.workflows() })
+  });
+}
+function useRunWorkflow() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: (workflowId) => arbi.workflows.run(workflowId),
+    onSuccess: () => Promise.all([
+      qc.invalidateQueries({ queryKey: arbiQueryKeys.workflows() }),
+      qc.invalidateQueries({ queryKey: arbiQueryKeys.allConversations() })
+    ])
+  });
+}
+function useCreateAgent() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: (name) => arbi.agents.create(name),
+    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.agents() })
+  });
+}
+function useAgentConfig(agentExtId, options) {
+  const arbi = useArbi();
+  return useQuery({
+    queryKey: arbiQueryKeys.agentConfig(agentExtId),
+    queryFn: async () => agentExtId ? await arbi.agents.getConfig(agentExtId) : await arbi.agentConfig.get("default"),
+    enabled: (options?.enabled ?? true) && arbi.isLoggedIn
+  });
+}
+function useAgentModels(options) {
+  const arbi = useArbi();
+  return useQuery({
+    queryKey: arbiQueryKeys.models(),
+    queryFn: async () => (await arbi.agentConfig.getModels()).models.map((m) => m.model_name),
+    enabled: (options?.enabled ?? true) && arbi.isLoggedIn
+  });
+}
+function useSaveAgentConfig() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: async ({ agentExtId, title, sections }) => {
+      const pinned = await arbi.agents.getConfig(agentExtId);
+      const base = pinned ?? await arbi.agentConfig.get("default");
+      const merged = structuredClone(base);
+      for (const [section, params] of Object.entries(sections)) {
+        merged[section] = { ...merged[section] ?? {}, ...params };
+      }
+      return arbi.agentConfig.save({
+        ...merged,
+        agent_ext_id: agentExtId,
+        title: title ?? `agent-${agentExtId}`
+      });
+    },
+    onSuccess: (_saved, { agentExtId }) => Promise.all([
+      qc.invalidateQueries({ queryKey: arbiQueryKeys.configs() }),
+      qc.invalidateQueries({ queryKey: arbiQueryKeys.agentConfig(agentExtId) }),
+      // The agent list shows the pinned persona/model, so it refreshes too.
+      qc.invalidateQueries({ queryKey: arbiQueryKeys.agents() })
+    ])
+  });
+}
+function useAgentWorkspaces(agentExtId, options) {
+  const arbi = useArbi();
+  return useQuery({
+    queryKey: arbiQueryKeys.agentWorkspaces(agentExtId ?? ""),
+    queryFn: () => arbi.agents.listWorkspaces(agentExtId),
+    enabled: (options?.enabled ?? true) && !!agentExtId && arbi.isLoggedIn
+  });
+}
+function useGrantAgentAccess() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: async ({
+      agentExtId,
+      workspaceId,
+      mode = { kind: "standing" }
+    }) => {
+      if (mode.kind === "temporary") {
+        return arbi.agents.grantTemporaryAccess(agentExtId, workspaceId, mode.ttlSeconds);
+      }
+      const agent = (await arbi.agents.list()).find((a) => a.external_id === agentExtId);
+      if (!agent?.email) throw new Error("The agent's address is unknown.");
+      return arbi.workspaces.grantAccess({ workspaceId, emails: [agent.email] });
+    },
+    onSuccess: (_res, { agentExtId }) => qc.invalidateQueries({ queryKey: arbiQueryKeys.agentWorkspaces(agentExtId) })
+  });
+}
+function useRevokeAgentAccess() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: async ({
+      agentExtId,
+      workspaceId
+    }) => {
+      const previous = arbi.hasWorkspace ? arbi.getWorkspaceId() : null;
+      await arbi.selectWorkspace(workspaceId);
+      try {
+        await arbi.workspaces.removeUsers([agentExtId]);
+      } finally {
+        if (previous && previous !== workspaceId) await arbi.selectWorkspace(previous);
+      }
+    },
+    onSuccess: (_res, { agentExtId }) => qc.invalidateQueries({ queryKey: arbiQueryKeys.agentWorkspaces(agentExtId) })
+  });
+}
+function useSkills(options) {
+  const arbi = useArbi();
+  return useQuery({
+    queryKey: [...arbiQueryKeys.skills(), options?.includeHidden ?? false],
+    queryFn: () => arbi.assistant.listSkills({ includeHidden: options?.includeHidden }),
+    refetchInterval: options?.live ? 5e3 : false,
+    enabled: (options?.enabled ?? true) && arbi.isLoggedIn,
+    staleTime: 3e4
+  });
+}
+function useUploadSkillFiles() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: async (entries) => {
+      if (!entries.length) throw new Error("Choose at least one skill file.");
+      for (const entry of entries) {
+        if (!/^skills\/[a-z0-9][a-z0-9_-]*(?:\/[^/]+)*$/.test(entry.folder) || entry.folder.split("/").some((segment) => segment === ".." || segment === "."))
+          throw new Error("Skill files need a folder under skills/<name>.");
+      }
+      const results = [];
+      for (const entry of entries) {
+        const result = await arbi.documents.uploadFile(entry.content, entry.fileName, {
+          folder: entry.folder,
+          wpType: "skill"
+        });
+        if (result.skipped?.length)
+          throw new Error(
+            result.skipped.map((file) => `${file.file_name}: ${file.reason}`).join("; ")
+          );
+        results.push(result);
+      }
+      return results;
+    },
+    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.skills() })
+  });
+}
+function useStartWorkQuery() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: async ({
+      question,
+      ...options
+    }) => {
+      const response = await arbi.assistant.submit(question, options);
+      for (let attempt = 0; attempt < 30; attempt++) {
+        if (attempt) await new Promise((resolve) => setTimeout(resolve, 500));
+        try {
+          const message = await arbi.conversations.getMessage(response.id);
+          if (message.conversation_ext_id)
+            return { conversationId: message.conversation_ext_id, responseId: response.id };
+        } catch (error) {
+          if (!(error && typeof error === "object" && "status" in error && error.status === 404))
+            throw error;
+        }
+      }
+      throw new Error(
+        `The run ${response.id} started, but its thread is not available yet. Do not submit it again.`
+      );
+    },
+    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.allConversations() })
+  });
+}
+function useAgentSession(agentId, options) {
+  const arbi = useArbi();
+  return useQuery({
+    queryKey: ["arbi", "agent-session", agentId],
+    queryFn: () => arbi.agents.getSession(agentId),
+    enabled: !!agentId && arbi.isLoggedIn && (options?.enabled ?? true),
+    refetchInterval: 5e3
+  });
+}
+function useAgentChats(agentId) {
+  const arbi = useArbi();
+  return useQuery({
+    queryKey: ["arbi", "agent-chats", agentId],
+    queryFn: () => arbi.agents.listChats(agentId),
+    enabled: !!agentId && arbi.isLoggedIn,
+    refetchInterval: 5e3
+  });
+}
+function useStartAgentSession() {
+  const arbi = useArbi();
+  const cache = useQueryClient();
+  return useMutation({
+    mutationFn: ({ agentId, workspaceId }) => arbi.agents.startSession(agentId, workspaceId),
+    onSuccess: () => cache.invalidateQueries({ queryKey: ["arbi", "agent-session"] })
+  });
+}
+function useCloseAgentSession() {
+  const arbi = useArbi();
+  const cache = useQueryClient();
+  return useMutation({
+    mutationFn: (agentId) => arbi.agents.closeSession(agentId),
+    onSuccess: () => cache.invalidateQueries({ queryKey: ["arbi", "agent-session"] })
+  });
+}
 function previewText(content) {
   return content.replace(/\s+/g, " ").trim().replace(/^(?:(?:[-*•>]|#{1,6})\s+)+/, "");
 }
@@ -79,26 +392,50 @@ var UNFINISHED_STATUSES = /* @__PURE__ */ new Set(["failed", "in_progress", "que
 var FAILED_MESSAGE = "The assistant request failed.";
 var UNFINISHED_MESSAGE = "The answer did not finish. Try again.";
 var isUnfinished = (m) => UNFINISHED_STATUSES.has(m.status);
-function messagesFromHistory(history) {
+function messagesFromHistory(history, live = false) {
   return history.filter(
-    (m) => m.role !== "system" && (m.content.trim() !== "" || m.role === "assistant" && isUnfinished(m))
+    (m) => m.role !== "system" && (m.content.trim() !== "" || m.role === "assistant" && (isUnfinished(m) || m.status === "cancelled"))
   ).map((m) => {
     if (m.role === "user") {
-      return { id: m.external_id, role: "user", content: m.content, status: "complete" };
+      return {
+        id: m.external_id,
+        role: "user",
+        content: m.content,
+        status: "complete",
+        ...m.created_by_ext_id ? { authorId: m.created_by_ext_id } : {},
+        ...m.mentions?.length ? { mentions: m.mentions } : {}
+      };
     }
     const citations = toChatCitations({ tools: m.tools });
     const unfinished = isUnfinished(m);
+    const runningElsewhere = live && (m.status === "in_progress" || m.status === "queued");
     return {
       id: m.external_id,
       role: "assistant",
       content: m.content,
-      status: unfinished ? "error" : "complete",
-      ...unfinished ? { error: m.status === "failed" ? FAILED_MESSAGE : UNFINISHED_MESSAGE } : {},
+      status: runningElsewhere ? "streaming" : unfinished ? "error" : "complete",
+      ...m.status === "cancelled" ? { stopped: true } : {},
+      ...unfinished && !runningElsewhere ? { error: m.status === "failed" ? FAILED_MESSAGE : UNFINISHED_MESSAGE } : {},
       ...unfinished ? {} : { responseId: m.external_id },
       ...citations.length > 0 ? { citations } : {}
     };
   });
 }
+function mergeHistory(history, local) {
+  const merged = [...history];
+  for (const message of local) {
+    if (!message.pendingHistory) continue;
+    const question = local.find((candidate) => candidate.id === message.questionId);
+    const pending = question?.storedId && history.some((stored) => stored.id === question.storedId) ? { ...message, questionId: question.storedId } : message;
+    const storedId = message.storedId ?? message.responseId ?? message.id;
+    const index = merged.findIndex((stored) => stored.id === storedId);
+    if (index < 0) merged.push(pending);
+    else if (message.role === "assistant" && merged[index].status === "streaming") {
+      merged[index] = pending;
+    }
+  }
+  return merged;
+}
 function chainId(messages) {
   for (let i = messages.length - 1; i >= 0; i--) {
     const m = messages[i];
@@ -119,43 +456,61 @@ var messageOf = (e, fallback) => e instanceof Error ? e.message : fallback;
 var isAbort = (e) => e instanceof DOMException && e.name === "AbortError";
 var TITLE_REFRESH_DELAYS_MS = [2500, 6e3, 12e3];
 function useArbiChat(options = {}) {
-  const { workspaceId, model, titleLength = 48, streamRecovery } = options;
+  const { workspaceId, agentId, model, titleLength = 48, streamRecovery } = options;
   const recoveryIntervalMs = streamRecovery?.intervalMs ?? 3e3;
   const recoveryTimeoutMs = streamRecovery?.timeoutMs ?? 12e4;
   const arbi = useArbi();
   const { data: docs = [], isLoading: isLoadingDocuments, refetch } = useDocuments(workspaceId);
-  const {
-    data: stored = [],
-    isLoading: isLoadingThreads,
-    refetch: refetchConversations
-  } = useConversations(workspaceId);
+  const ordinaryHistory = useConversations(workspaceId, { enabled: !agentId, live: true });
+  const agentHistory = useAgentChats(agentId);
+  const stored = useMemo(
+    () => agentId ? (agentHistory.data ?? []).filter((chat) => chat.workspace_ext_id === workspaceId).map((chat) => ({
+      external_id: chat.conversation_ext_id,
+      title: chat.title,
+      updated_at: chat.updated_at,
+      message_count: 1,
+      status: null,
+      assistant_mode: "always"
+    })) : ordinaryHistory.data ?? [],
+    [agentId, agentHistory.data, ordinaryHistory.data, workspaceId]
+  );
+  const isLoadingThreads = agentId ? agentHistory.isLoading : ordinaryHistory.isLoading;
+  const refetchConversations = agentId ? agentHistory.refetch : ordinaryHistory.refetch;
   const query = useAssistantQuery();
   const { mutateAsync } = query;
+  const [stagedFiles, setStagedFiles] = useState([]);
   const [threads, setThreads] = useState(() => [emptyThread()]);
   const [activeKey, setActiveKey] = useState(() => threads[0].key);
   const [deselected, setDeselected] = useState(/* @__PURE__ */ new Set());
   const [isRunning, setIsRunning] = useState(false);
+  const [isPostingComment, setIsPostingComment] = useState(false);
+  const [isUpdatingAssistantMode, setIsUpdatingAssistantMode] = useState(false);
   const [isUploading, setIsUploading] = useState(false);
   const [uploadError, setUploadError] = useState(null);
   const [historyError, setHistoryError] = useState(null);
   const abortRef = useRef(null);
+  const runIdRef = useRef(null);
+  const stopRequestedRef = useRef(false);
   const timersRef = useRef([]);
   useEffect(() => {
     const timers = timersRef.current;
     return () => timers.forEach(clearTimeout);
   }, []);
-  const scopeRef = useRef(workspaceId);
+  const chatScope = `${workspaceId ?? ""}:${agentId ?? ""}`;
+  const scopeRef = useRef(chatScope);
   useEffect(() => {
-    if (scopeRef.current === workspaceId) return;
-    scopeRef.current = workspaceId;
+    if (scopeRef.current === chatScope) return;
+    scopeRef.current = chatScope;
+    setStagedFiles([]);
     const fresh = emptyThread();
     setThreads([fresh]);
     setActiveKey(fresh.key);
     setDeselected(/* @__PURE__ */ new Set());
     setUploadError(null);
     setHistoryError(null);
-  }, [workspaceId]);
+  }, [chatScope]);
   const active = threads.find((t) => t.key === activeKey) ?? threads[0];
+  const assistantMode = active.assistantMode ?? stored.find((c) => c.external_id === active.conversationId)?.assistant_mode ?? "always";
   const threadEntries = useMemo(() => {
     const storedIds = new Set(stored.map((c) => c.external_id));
     const byConversation = new Map(
@@ -163,7 +518,7 @@ function useArbiChat(options = {}) {
     );
     const fresh = threads.filter((t) => !t.conversationId);
     const justSaved = threads.filter((t) => t.conversationId && !storedIds.has(t.conversationId));
-    const persisted = [...stored].filter((c) => c.message_count > 0 || byConversation.has(c.external_id)).sort((a, b) => b.updated_at.localeCompare(a.updated_at)).map((c) => {
+    const persisted = [...stored].filter((c) => c.status != null || c.message_count > 0 || byConversation.has(c.external_id)).sort((a, b) => b.updated_at.localeCompare(a.updated_at)).map((c) => {
       const local = byConversation.get(c.external_id);
       return {
         key: local?.key ?? c.external_id,
@@ -177,41 +532,71 @@ function useArbiChat(options = {}) {
       ...persisted
     ];
   }, [threads, stored]);
+  const liveHistory = assistantMode === "mentioned" || stored.some(
+    (conversation) => conversation.external_id === active.conversationId && conversation.status === "generating"
+  ) || active.messages.some((message) => message.status === "streaming");
   const needsLoad = !!active.conversationId && !active.isLoaded;
   const loaded = useConversationThreads(workspaceId, active.conversationId ?? void 0, {
-    enabled: needsLoad
+    enabled: needsLoad || liveHistory && !isRunning && !isPostingComment,
+    live: liveHistory
   });
+  const refetchThread = loaded.refetch;
+  const selectedStored = stored.find(
+    (conversation) => conversation.external_id === active.conversationId
+  );
+  const historyRevision = selectedStored ? `${selectedStored.external_id}:${selectedStored.updated_at}:${selectedStored.message_count}:${selectedStored.status}` : void 0;
+  const observedHistoryRevision = useRef(void 0);
+  useEffect(() => {
+    const previous = observedHistoryRevision.current;
+    observedHistoryRevision.current = historyRevision;
+    if (previous && previous !== historyRevision && active.isLoaded && !isRunning && !isPostingComment)
+      void refetchThread();
+  }, [historyRevision, active.isLoaded, isRunning, isPostingComment, refetchThread]);
   useEffect(() => {
     const data = loaded.data;
-    if (!needsLoad || !data || data.conversation_ext_id !== active.conversationId) return;
+    if (isRunning || isPostingComment || !data || data.conversation_ext_id !== active.conversationId)
+      return;
     const primary = [...data.threads].sort((a, b) => b.history.length - a.history.length)[0];
-    const messages = messagesFromHistory(primary?.history ?? []);
+    const messages = messagesFromHistory(primary?.history ?? [], liveHistory);
     setThreads(
       (all) => all.map(
-        (t) => t.key === active.key ? {
-          ...t,
-          isLoaded: true,
-          messages,
-          responseId: chainId(messages)
-        } : t
+        (t) => t.key === active.key ? (() => {
+          const merged = mergeHistory(messages, t.messages);
+          return {
+            ...t,
+            isLoaded: true,
+            assistantMode: data.assistant_mode,
+            messages: merged,
+            responseId: chainId(merged)
+          };
+        })() : t
       )
     );
-  }, [loaded.data, needsLoad, active.conversationId, active.key]);
+  }, [
+    loaded.data,
+    liveHistory,
+    needsLoad,
+    active.conversationId,
+    active.key,
+    assistantMode,
+    isRunning,
+    isPostingComment
+  ]);
   useEffect(() => {
     if (loaded.error)
       setHistoryError(messageOf(loaded.error, "The conversation could not be loaded."));
   }, [loaded.error]);
   const selectedIds = useMemo(
-    () => docs.map((d) => d.external_id).filter((id) => !!id && !deselected.has(id)),
-    [docs, deselected]
+    () => (agentId ? [] : docs).map((d) => d.external_id).filter((id) => !!id && !deselected.has(id)),
+    [docs, deselected, agentId]
   );
   const documents = useMemo(
-    () => docs.filter((d) => !!d.external_id).map((d) => ({
+    () => agentId ? stagedFiles.map((file) => ({ ...file, isSelected: !deselected.has(file.id) })) : docs.filter((d) => !!d.external_id).map((d) => ({
       id: d.external_id,
       name: d.file_name ?? d.external_id,
       isSelected: !deselected.has(d.external_id)
     })),
-    [docs, deselected]
+    [docs, deselected, agentId, stagedFiles]
   );
   const patchThread = useCallback(
     (key, patch) => setThreads((all) => all.map((t) => t.key === key ? patch(t) : t)),
@@ -254,15 +639,31 @@ function useArbiChat(options = {}) {
     },
     [arbi, recoveryIntervalMs, recoveryTimeoutMs]
   );
+  const cancelNativeRun = useCallback(
+    async (responseId, controller) => {
+      try {
+        await arbi.assistant.cancelRun(responseId);
+        if (abortRef.current === controller) controller.abort();
+      } catch (error2) {
+        stopRequestedRef.current = false;
+        setHistoryError(messageOf(error2, "The agent run could not be stopped."));
+      }
+    },
+    [arbi]
+  );
   const ask = useCallback(
-    async (thread, text) => {
+    async (thread, text, mentions) => {
       const userMessage = {
+        pendingHistory: true,
+        ...mentions?.length ? { mentions } : {},
         id: nextId("msg"),
         role: "user",
         content: text,
         status: "complete"
       };
       const answer = {
+        pendingHistory: true,
+        questionId: userMessage.id,
         id: nextId("msg"),
         role: "assistant",
         content: "",
@@ -275,6 +676,8 @@ function useArbiChat(options = {}) {
       }));
       const controller = new AbortController();
       abortRef.current = controller;
+      runIdRef.current = null;
+      stopRequestedRef.current = false;
       setIsRunning(true);
       let buffer = "";
       let steps = 0;
@@ -291,15 +694,25 @@ function useArbiChat(options = {}) {
         revertChain();
       };
       try {
+        const lastMessage2 = thread.messages[thread.messages.length - 1];
+        const parentId = lastMessage2?.role === "user" && (lastMessage2.storedId || !lastMessage2.pendingHistory) ? lastMessage2.storedId ?? lastMessage2.id : thread.responseId;
         const result = await mutateAsync({
-          question: text,
+          question: agentId && stagedFiles.some((file) => !deselected.has(file.id)) ? text + "\n\nSource files staged in your sandbox:\n" + stagedFiles.filter((file) => !deselected.has(file.id)).map((file) => `${file.name}: ${file.id}`).join("\n") : text,
+          ...agentId ? { agentId, workspaceId } : {},
           docIds: selectedIds,
-          previousResponseId: thread.responseId,
+          previousResponseId: parentId,
+          ...thread.conversationId ? { conversationId: thread.conversationId } : {},
+          ...!thread.conversationId && thread.assistantMode === "mentioned" ? { assistantMode: "mentioned" } : {},
+          ...mentions?.length ? { mentions } : {},
           model,
           signal: controller.signal,
           onStreamStart: (data) => {
             if (data.assistant_message_ext_id) {
               backendMessageId = data.assistant_message_ext_id;
+              runIdRef.current = backendMessageId;
+              if (agentId && stopRequestedRef.current)
+                void cancelNativeRun(backendMessageId, controller);
+              patchMessage(thread.key, answer.id, { storedId: backendMessageId });
               patchThread(thread.key, (t) => ({
                 ...t,
                 responseId: data.assistant_message_ext_id ?? t.responseId
@@ -308,6 +721,7 @@ function useArbiChat(options = {}) {
           },
           onUserMessage: (ev) => {
             conversationId = conversationId ?? ev.conversation_ext_id;
+            patchMessage(thread.key, userMessage.id, { storedId: ev.external_id });
             patchThread(thread.key, (t) => ({
               ...t,
               conversationId: t.conversationId ?? ev.conversation_ext_id
@@ -389,37 +803,139 @@ function useArbiChat(options = {}) {
       }
     },
     [
-      mutateAsync,
-      patchMessage,
+      mutateAsync,
+      patchMessage,
+      patchThread,
+      refreshTitles,
+      recoverStoredAnswer,
+      selectedIds,
+      model,
+      titleLength,
+      agentId,
+      cancelNativeRun,
+      workspaceId,
+      stagedFiles,
+      deselected
+    ]
+  );
+  const send = useCallback(
+    async (text, options2) => {
+      const trimmed = text.trim();
+      if (!trimmed || isRunning || isPostingComment || isUpdatingAssistantMode || !active.isLoaded)
+        return;
+      setHistoryError(null);
+      if (assistantMode === "mentioned" && !options2?.mentions?.some((m) => m.type === "assistant")) {
+        setIsPostingComment(true);
+        setHistoryError(null);
+        try {
+          const saved = await arbi.conversations.postMessage({
+            content: trimmed,
+            assistant_mode: "mentioned",
+            mentions: options2?.mentions ?? [],
+            ...active.conversationId ? { conversation_ext_id: active.conversationId } : {}
+          });
+          patchThread(active.key, (t) => ({
+            ...t,
+            conversationId: saved.conversation_ext_id,
+            title: t.title || titleFromQuestion(trimmed, titleLength),
+            assistantMode: "mentioned",
+            responseId: saved.external_id,
+            messages: [
+              ...t.messages,
+              {
+                pendingHistory: true,
+                id: saved.external_id,
+                role: "user",
+                content: saved.content,
+                status: "complete",
+                ...saved.created_by_ext_id ? { authorId: saved.created_by_ext_id } : {}
+              }
+            ]
+          }));
+          await refetchConversations();
+          if (active.conversationId) await refetchThread();
+        } catch (e) {
+          setHistoryError(messageOf(e, "The comment could not be posted."));
+          throw e;
+        } finally {
+          setIsPostingComment(false);
+        }
+        return;
+      }
+      if (active.messages.some((message) => message.status === "streaming")) {
+        const message = "An assistant reply is already running in this thread. You can still post a comment.";
+        setHistoryError(message);
+        throw new Error(message);
+      }
+      await ask(active, trimmed, options2?.mentions);
+    },
+    [
+      active,
+      ask,
+      isRunning,
+      isPostingComment,
+      isUpdatingAssistantMode,
+      assistantMode,
+      arbi,
       patchThread,
-      refreshTitles,
-      recoverStoredAnswer,
-      selectedIds,
-      model,
-      titleLength
+      titleLength,
+      refetchConversations,
+      refetchThread
     ]
   );
-  const send = useCallback(
-    async (text) => {
-      const trimmed = text.trim();
-      if (!trimmed || isRunning || !active.isLoaded) return;
-      await ask(active, trimmed);
+  const setAssistantMode = useCallback(
+    async (mode) => {
+      if (isRunning || isPostingComment || isUpdatingAssistantMode) return;
+      setIsUpdatingAssistantMode(true);
+      setHistoryError(null);
+      try {
+        if (active.conversationId) {
+          await arbi.conversations.setAssistantMode(active.conversationId, mode);
+          await refetchThread();
+        }
+        patchThread(active.key, (t) => ({ ...t, assistantMode: mode }));
+        await refetchConversations();
+      } catch (e) {
+        setHistoryError(messageOf(e, "The assistant mode could not be changed."));
+      } finally {
+        setIsUpdatingAssistantMode(false);
+      }
     },
-    [active, ask, isRunning]
+    [
+      active,
+      arbi,
+      patchThread,
+      refetchConversations,
+      refetchThread,
+      isRunning,
+      isPostingComment,
+      isUpdatingAssistantMode
+    ]
   );
-  const stop = useCallback(() => abortRef.current?.abort(), []);
+  const stop = useCallback(() => {
+    const controller = abortRef.current;
+    if (!controller) return;
+    if (!agentId) {
+      controller.abort();
+      return;
+    }
+    stopRequestedRef.current = true;
+    if (runIdRef.current) void cancelNativeRun(runIdRef.current, controller);
+  }, [agentId, cancelNativeRun]);
   const lastMessage = active.messages[active.messages.length - 1];
   const error = lastMessage?.status === "error" ? lastMessage.error ?? null : null;
   const retry = useCallback(async () => {
     if (isRunning) return;
     const messages = active.messages;
     const failed = messages[messages.length - 1];
-    const question = messages[messages.length - 2];
+    const question = failed?.questionId ? messages.find((message) => message.id === failed.questionId) : messages[messages.length - 2];
     if (failed?.status !== "error" || question?.role !== "user") return;
-    const rest = messages.slice(0, -2);
+    const rest = messages.filter(
+      (message) => message.id !== failed.id && message.id !== question.id
+    );
     const trimmed = { ...active, messages: rest, responseId: chainId(rest) };
     patchThread(active.key, () => trimmed);
-    await ask(trimmed, question.content);
+    await ask(trimmed, question.content, question.mentions);
   }, [active, ask, isRunning, patchThread]);
   const newThread = useCallback(() => {
     setHistoryError(null);
@@ -482,17 +998,29 @@ function useArbiChat(options = {}) {
       setIsUploading(true);
       setUploadError(null);
       try {
-        for (const file of files) await arbi.documents.uploadFile(file, file.name);
-        await refetch();
+        for (const file of files) {
+          if (agentId) {
+            const staged = await arbi.agents.stageFile(agentId, file, file.name);
+            setStagedFiles((all) => [
+              ...all.filter((existing) => existing.id !== staged.path),
+              { id: staged.path, name: staged.name }
+            ]);
+          } else await arbi.documents.uploadFile(file, file.name);
+        }
+        if (!agentId) await refetch();
       } catch (e) {
         setUploadError(messageOf(e, "The upload failed."));
       } finally {
         setIsUploading(false);
       }
     },
-    [arbi, refetch]
+    [arbi, refetch, agentId]
   );
   return {
+    assistantMode,
+    isUpdatingAssistantMode,
+    isPostingComment,
+    setAssistantMode,
     threads: threadEntries.map(({ id, title }) => ({ id, title })),
     activeThreadId: publicId(active),
     messages: active.messages,
@@ -935,229 +1463,6 @@ function useArbiShellWorkspaces(options = {}) {
     select: select ?? defaultSelect
   });
 }
-function useUpdateConversation() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  return useMutation({
-    mutationFn: ({
-      conversationId,
-      patch
-    }) => arbi.conversations.update(conversationId, patch),
-    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.allConversations() })
-  });
-}
-function useCreateWork() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  return useMutation({
-    mutationFn: (body) => arbi.conversations.createWork(body),
-    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.allConversations() })
-  });
-}
-function useRunConversation() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  return useMutation({
-    mutationFn: (conversationId) => arbi.conversations.runWithAuth(conversationId),
-    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.allConversations() })
-  });
-}
-function useReviewConversation() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  return useMutation({
-    mutationFn: ({
-      conversationId,
-      review
-    }) => arbi.conversations.review(conversationId, review),
-    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.allConversations() })
-  });
-}
-function useRequestConversationApproval() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  return useMutation({
-    mutationFn: ({
-      conversationId,
-      request
-    }) => arbi.conversations.requestApproval(conversationId, request),
-    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.allConversations() })
-  });
-}
-function usePlanWork() {
-  const arbi = useArbi();
-  return useMutation({
-    mutationFn: (body) => arbi.conversations.plan(body)
-  });
-}
-function useConversationPendingAuth(conversationId, options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: [...arbiQueryKeys.allConversations(), conversationId ?? "", "pending-auth"],
-    queryFn: () => arbi.conversations.pendingAuth(conversationId),
-    enabled: (options?.enabled ?? true) && !!conversationId && arbi.isLoggedIn,
-    retry: false
-  });
-}
-function useConversationActivity(conversationId, options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: [...arbiQueryKeys.allConversations(), conversationId ?? "", "activity"],
-    queryFn: () => arbi.conversations.activity(conversationId),
-    enabled: (options?.enabled ?? true) && !!conversationId && arbi.isLoggedIn,
-    refetchInterval: options?.live ? 2500 : false
-  });
-}
-function useWorkflows(options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.workflows(),
-    queryFn: () => arbi.workflows.list(),
-    enabled: options?.enabled ?? arbi.isLoggedIn
-  });
-}
-function useCreateWorkflow() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  return useMutation({
-    mutationFn: (body) => arbi.workflows.create(body),
-    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.workflows() })
-  });
-}
-function useUpdateWorkflow() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  return useMutation({
-    mutationFn: ({
-      workflowId,
-      patch
-    }) => arbi.workflows.update(workflowId, patch),
-    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.workflows() })
-  });
-}
-function useDeleteWorkflow() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  return useMutation({
-    mutationFn: (workflowId) => arbi.workflows.delete(workflowId),
-    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.workflows() })
-  });
-}
-function useRunWorkflow() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  return useMutation({
-    mutationFn: (workflowId) => arbi.workflows.run(workflowId),
-    onSuccess: () => Promise.all([
-      qc.invalidateQueries({ queryKey: arbiQueryKeys.workflows() }),
-      qc.invalidateQueries({ queryKey: arbiQueryKeys.allConversations() })
-    ])
-  });
-}
-function useCreateAgent() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  return useMutation({
-    mutationFn: (name) => arbi.agents.create(name),
-    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.agents() })
-  });
-}
-function useAgentConfig(agentExtId, options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.agentConfig(agentExtId),
-    queryFn: async () => agentExtId ? await arbi.agents.getConfig(agentExtId) : await arbi.agentConfig.get("default"),
-    enabled: (options?.enabled ?? true) && arbi.isLoggedIn
-  });
-}
-function useAgentModels(options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.models(),
-    queryFn: async () => (await arbi.agentConfig.getModels()).models.map((m) => m.model_name),
-    enabled: (options?.enabled ?? true) && arbi.isLoggedIn
-  });
-}
-function useSaveAgentConfig() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  return useMutation({
-    mutationFn: async ({ agentExtId, title, sections }) => {
-      const pinned = await arbi.agents.getConfig(agentExtId);
-      const base = pinned ?? await arbi.agentConfig.get("default");
-      const merged = structuredClone(base);
-      for (const [section, params] of Object.entries(sections)) {
-        merged[section] = { ...merged[section] ?? {}, ...params };
-      }
-      return arbi.agentConfig.save({
-        ...merged,
-        agent_ext_id: agentExtId,
-        title: title ?? `agent-${agentExtId}`
-      });
-    },
-    onSuccess: (_saved, { agentExtId }) => Promise.all([
-      qc.invalidateQueries({ queryKey: arbiQueryKeys.configs() }),
-      qc.invalidateQueries({ queryKey: arbiQueryKeys.agentConfig(agentExtId) }),
-      // The agent list shows the pinned persona/model, so it refreshes too.
-      qc.invalidateQueries({ queryKey: arbiQueryKeys.agents() })
-    ])
-  });
-}
-function useAgentWorkspaces(agentExtId, options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.agentWorkspaces(agentExtId ?? ""),
-    queryFn: () => arbi.agents.listWorkspaces(agentExtId),
-    enabled: (options?.enabled ?? true) && !!agentExtId && arbi.isLoggedIn
-  });
-}
-function useGrantAgentAccess() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  return useMutation({
-    mutationFn: async ({
-      agentExtId,
-      workspaceId,
-      mode = { kind: "standing" }
-    }) => {
-      if (mode.kind === "temporary") {
-        return arbi.agents.grantTemporaryAccess(agentExtId, workspaceId, mode.ttlSeconds);
-      }
-      const agent = (await arbi.agents.list()).find((a) => a.external_id === agentExtId);
-      if (!agent?.email) throw new Error("The agent's address is unknown.");
-      return arbi.workspaces.grantAccess({ workspaceId, emails: [agent.email] });
-    },
-    onSuccess: (_res, { agentExtId }) => qc.invalidateQueries({ queryKey: arbiQueryKeys.agentWorkspaces(agentExtId) })
-  });
-}
-function useRevokeAgentAccess() {
-  const arbi = useArbi();
-  const qc = useQueryClient();
-  return useMutation({
-    mutationFn: async ({
-      agentExtId,
-      workspaceId
-    }) => {
-      const previous = arbi.hasWorkspace ? arbi.getWorkspaceId() : null;
-      await arbi.selectWorkspace(workspaceId);
-      try {
-        await arbi.workspaces.removeUsers([agentExtId]);
-      } finally {
-        if (previous && previous !== workspaceId) await arbi.selectWorkspace(previous);
-      }
-    },
-    onSuccess: (_res, { agentExtId }) => qc.invalidateQueries({ queryKey: arbiQueryKeys.agentWorkspaces(agentExtId) })
-  });
-}
-function useSkills(options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: [...arbiQueryKeys.skills(), options?.includeHidden ?? false],
-    queryFn: () => arbi.assistant.listSkills({ includeHidden: options?.includeHidden }),
-    enabled: (options?.enabled ?? true) && arbi.isLoggedIn,
-    staleTime: 3e4
-  });
-}
 function useContacts(options) {
   const arbi = useArbi();
   return useQuery({
@@ -3765,6 +4070,34 @@ function ActivityFeed({
     );
   }) });
 }
+var workTypes = [
+  "mention",
+  "workspace_access_request",
+  "conversation_ready_for_review",
+  "conversation_approval_request",
+  "conversation_done",
+  "workflow_failed"
+];
+function useWorkNotifications(options) {
+  const arbi = useArbi();
+  const queries = useQueryClient();
+  const queryKey = ["work-notifications", options.apiUrl, options.userId];
+  const query = useQuery({
+    queryKey,
+    queryFn: async () => (await Promise.all(workTypes.map((type) => arbi.notifications.list(type)))).flat(),
+    enabled: options.enabled && Boolean(options.userId),
+    refetchInterval: 3e4
+  });
+  useArbiWebSocket((message) => {
+    if (workTypes.some((type) => type === message.type))
+      void queries.invalidateQueries({ queryKey });
+  });
+  const receipt = useMutation({
+    mutationFn: (ids) => arbi.notifications.markRead(ids),
+    onSuccess: () => queries.invalidateQueries({ queryKey })
+  });
+  return { ...query, markRead: receipt.mutateAsync };
+}
 
 // src/directmessages/utils.ts
 function userInitials(user) {
@@ -5833,6 +6166,6 @@ function hasVideoTrack(blob) {
   });
 }
 
-export { ActivityFeed, AiActionCard, AiAnswer, AppleIcon, ArbiBotIcon, ArbiIcon, ArbiWebSocketProvider, ArtifactCanvas, ChatBubble, CitationRenderer, ClaudeIcon, CommandPalette, CommandPaletteProvider, ConversationList, DEFAULT_ASPECT_OPTIONS, DEFAULT_PASSWORD_CHANGE_FORM_MESSAGES, DEFAULT_START_BY_EMAIL_LABELS, DateSeparator, DirectMessages, DmAvatar, DmComposer, DocIcon, DocQaPanel, DocSearch, DocumentCardMenu, DocumentGallery, DocumentReview, DocumentTable, DraftTitleControls, FileIcon, FireworksIcon, GeminiIcon, GitHubIcon, GoogleIcon, HtmlIcon, ImageCropModal, ImageCropper, LinkedInIcon, MAX_AVATAR_FILE_BYTES, MessageArtifactCard, MessageBubble, MessageThread, OpenAIIcon, PhotoCapture, ProviderIcon, RUN_WINDOW_MS, SemanticSearch, StartByEmail, TxtIcon, TypingIndicator, VideoCapture, VoiceNoteRecorder, XlsIcon, annotateThread, captureFrameAsBlob, cropImage, dayKey, describeActivity, getNotificationTypeMeta, getWorkspaceActionText, getWorkspaceKind, getWorkspaceSentPhrase, hasVideoTrack, isWorkspaceNotificationType, parseActivityContent, parseDeleteAccountBlockers, pickRecentWorkspaces, redlineStats, resolveActiveWorkspace, textToArtifact, titleFromQuestion, toChatCitations, toNotificationPreferences, toProfileView, toRedlineMarkdown, toShellWorkspace, useAddContacts, useAgentConfig, useAgentModels, useAgentWorkspaces, useAiTask, useArbiChat, useArbiConnected, useArbiDocumentSync, useArbiShellWorkspaces, useArbiUiStore, useArbiWebSocket, useArbiWsContext, useArbiWsMessage, useArbiWsStatus, useArbiWsStore, useArbiWsTask, useArbiWsTasks, useAvatarCrop, useCameraStream, useCommandPalette, useConfirmEmailChange, useContacts, useConversationActivity, useConversationPendingAuth, useCreateAgent, useCreateWork, useCreateWorkflow, useCurrentSessionId, useDeleteAccount, useDeleteAccountFlow, useDeleteWorkflow, useDirectMessages, useEmailChange, useGrantAgentAccess, useLogoutAllSessions, useNameEditor, useNotificationPreferences, usePasswordChange, usePasswordChangeForm, usePendingContacts, usePlanWork, useProfile, useRemoveContacts, useRequestConversationApproval, useRequestEmailChange, useReviewConversation, useRevokeAgentAccess, useRunConversation, useRunWorkflow, useSaveAgentConfig, useSemanticSearch, useShellWorkspaces, useSkills, useTerminateSession, useUpdateConversation, useUpdateProfile, useUpdateWorkflow, useUserSettings, useVideoRecorder, useWorkflows, useWorkspaceDocs, userDisplayName, userInitials };
+export { ActivityFeed, AiActionCard, AiAnswer, AppleIcon, ArbiBotIcon, ArbiIcon, ArbiWebSocketProvider, ArtifactCanvas, ChatBubble, CitationRenderer, ClaudeIcon, CommandPalette, CommandPaletteProvider, ConversationList, DEFAULT_ASPECT_OPTIONS, DEFAULT_PASSWORD_CHANGE_FORM_MESSAGES, DEFAULT_START_BY_EMAIL_LABELS, DateSeparator, DirectMessages, DmAvatar, DmComposer, DocIcon, DocQaPanel, DocSearch, DocumentCardMenu, DocumentGallery, DocumentReview, DocumentTable, DraftTitleControls, FileIcon, FireworksIcon, GeminiIcon, GitHubIcon, GoogleIcon, HtmlIcon, ImageCropModal, ImageCropper, LinkedInIcon, MAX_AVATAR_FILE_BYTES, MessageArtifactCard, MessageBubble, MessageThread, OpenAIIcon, PhotoCapture, ProviderIcon, RUN_WINDOW_MS, SemanticSearch, StartByEmail, TxtIcon, TypingIndicator, VideoCapture, VoiceNoteRecorder, XlsIcon, annotateThread, captureFrameAsBlob, cropImage, dayKey, describeActivity, getNotificationTypeMeta, getWorkspaceActionText, getWorkspaceKind, getWorkspaceSentPhrase, hasVideoTrack, isWorkspaceNotificationType, parseActivityContent, parseDeleteAccountBlockers, pickRecentWorkspaces, redlineStats, resolveActiveWorkspace, textToArtifact, titleFromQuestion, toChatCitations, toNotificationPreferences, toProfileView, toRedlineMarkdown, toShellWorkspace, useAddContacts, useAgentChats, useAgentConfig, useAgentModels, useAgentSession, useAgentWorkspaces, useAiTask, useArbiChat, useArbiConnected, useArbiDocumentSync, useArbiShellWorkspaces, useArbiUiStore, useArbiWebSocket, useArbiWsContext, useArbiWsMessage, useArbiWsStatus, useArbiWsStore, useArbiWsTask, useArbiWsTasks, useAvatarCrop, useCameraStream, useCloseAgentSession, useCommandPalette, useConfirmEmailChange, useContacts, useConversationActivity, useConversationPendingAuth, useCreateAgent, useCreateWork, useCreateWorkflow, useCurrentSessionId, useDeleteAccount, useDeleteAccountFlow, useDeleteWorkflow, useDirectMessages, useEmailChange, useGrantAgentAccess, useLogoutAllSessions, useNameEditor, useNotificationPreferences, usePasswordChange, usePasswordChangeForm, usePendingContacts, usePlanWork, useProfile, useRemoveContacts, useRequestConversationApproval, useRequestEmailChange, useReviewConversation, useRevokeAgentAccess, useRunConversation, useRunWorkflow, useSaveAgentConfig, useSemanticSearch, useShellWorkspaces, useSkills, useStartAgentSession, useStartWorkQuery, useTerminateSession, useUpdateConversation, useUpdateProfile, useUpdateWorkflow, useUploadSkillFiles, useUserSettings, useVideoRecorder, useWorkNotifications, useWorkflows, useWorkspaceDocs, userDisplayName, userInitials };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map