@arbidocs/react 0.4.3 → 0.4.5

package/dist/index.js

@@ -1,13 +1,19 @@
-import { formatRelativeTime } from './chunk-EJ2BMHYK.js';
-export { DEFAULT_NOTIFY_TONE, NotificationBell, Toaster, formatRelativeTime, notify, seedNotifications, selectUnread, toast, useNotify } from './chunk-EJ2BMHYK.js';
-import { normalizeCitationMarkup, MarkMenu, AiMarkdown, Card, Button, Badge, TagEntity, MarkdownRenderer, Input, FILE_TYPE_COLORS, getEntityIcon, UserEntity, getEntityColorStyles, AgentBadge, Textarea, Slider, Dialog, DialogContent, DialogHeader, DialogTitle, DialogFooter, DocumentEntity, DocumentCard, useAudioRecorder, MicWaveformIndicator, IconToggle, classifyMediaError, formatElapsed, PassageCard } from './chunk-MIJRORQL.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, MarkdownRenderer, MarkdownStreamingContext, 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, badgeVariants, buttonVariants, cardVariants, configureEntityRegistry, configureRagLabels, configureStatusRegistry, formatElapsed, getDocIcon, getEntityColor, getEntityColorStyles, getEntityIcon, getEntityLabel, getStatusVariant, getTagTypeIcon, getTagTypeLabel, getWorkspaceColor, markdownBaseComponents, normalizeCitationMarkup, splitMarkdownBlocks, toggleVariants, useAudioRecorder } from './chunk-MIJRORQL.js';
-import { cn, ScrollArea, DropdownMenu, DropdownMenuTrigger, DropdownMenuContent, DropdownMenuItem, DropdownMenuSeparator } from './chunk-6ALMBER7.js';
-export { DropdownMenu, DropdownMenuCheckboxItem, DropdownMenuContent, DropdownMenuGroup, DropdownMenuItem, DropdownMenuLabel, DropdownMenuPortal, DropdownMenuRadioGroup, DropdownMenuRadioItem, DropdownMenuSeparator, DropdownMenuShortcut, DropdownMenuSub, DropdownMenuSubContent, DropdownMenuSubTrigger, DropdownMenuTrigger, ScrollArea, ScrollBar, cn } from './chunk-6ALMBER7.js';
-import { useArbi, AudioViewer, ImageViewer, VideoViewer } from './chunk-JKO3NR5L.js';
-export { ArbiProvider, useArbi } from './chunk-JKO3NR5L.js';
+import { useDocuments, useAssistantQuery, arbiQueryKeys, useWorkspaces, useParsedDocument, useTags, useActivity, useThumbnails } from './chunk-ZY53R6I4.js';
+export { DEFAULT_USD_PER_CREDIT, arbiQueryKeys, creditsToUsd, formatCredits, isUsageMetered, useActivity, useAgents, useAssistantQuery, useConfigs, useConversations, useDeleteAgents, useDocuments, useFacts, useHealth, useParsedDocument, useProjectInvoices, useProjectUsage, useProjects, useProjectsUsage, useSessions, useTags, useTasks, useThumbnails, useUserUsageToday, useWorkspaces } from './chunk-ZY53R6I4.js';
+import { formatRelativeTime } from './chunk-FSMIN5VX.js';
+export { DEFAULT_NOTIFY_TONE, NotificationBell, Toaster, formatRelativeTime, notify, seedNotifications, selectUnread, toast, useNotify } from './chunk-FSMIN5VX.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-DVRJFLE7.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, badgeVariants, buttonVariants, cardVariants, configureEntityRegistry, configureRagLabels, configureStatusRegistry, formatElapsed, getDocIcon, getEntityColor, getEntityColorStyles, getEntityIcon, getEntityLabel, getStatusVariant, getTagTypeIcon, getTagTypeLabel, getWorkspaceColor, toggleVariants, useAudioRecorder } from './chunk-DVRJFLE7.js';
+import { normalizeCitationMarkup, MarkdownRenderer } from './chunk-H7ZZQ4ZV.js';
+export { MarkdownRenderer, MarkdownStreamingContext, markdownBaseComponents, normalizeCitationMarkup, splitMarkdownBlocks } from './chunk-H7ZZQ4ZV.js';
+import { ScrollArea, DropdownMenu, DropdownMenuTrigger, DropdownMenuContent, DropdownMenuItem, DropdownMenuSeparator } from './chunk-N5SVBDLC.js';
+export { DropdownMenu, DropdownMenuCheckboxItem, DropdownMenuContent, DropdownMenuGroup, DropdownMenuItem, DropdownMenuLabel, DropdownMenuPortal, DropdownMenuRadioGroup, DropdownMenuRadioItem, DropdownMenuSeparator, DropdownMenuShortcut, DropdownMenuSub, DropdownMenuSubContent, DropdownMenuSubTrigger, DropdownMenuTrigger, ScrollArea, ScrollBar } from './chunk-N5SVBDLC.js';
+import { cn } from './chunk-SN5YV6FF.js';
+export { cn } from './chunk-SN5YV6FF.js';
+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 { useQuery, useMutation, useQueries, useQueryClient } from '@tanstack/react-query';
 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, MessageCircleMore, FileX2, FileCheck2, Pencil, FolderPlus, SquareArrowOutUpRight, MessagesSquare, ArrowRight, MoreHorizontal, SquarePen, Mic, Pause, Play, Square, RotateCcw, Camera, Circle } from 'lucide-react';
 import { jsx, jsxs, Fragment } from 'react/jsx-runtime';
 import { connectWithReconnect } from '@arbidocs/sdk/browser';
@@ -15,6 +21,7 @@ export { Arbi, ArbiError } from '@arbidocs/sdk/browser';
 import { isMessageType } 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';
@@ -23,241 +30,6 @@ import Cropper from 'react-easy-crop';
 import { AgGridReact } from 'ag-grid-react';
 import { ModuleRegistry, AllCommunityModule } from 'ag-grid-community';
 
-var arbiQueryKeys = {
-  workspaces: () => ["arbi", "workspaces"],
-  documents: (workspaceId) => ["arbi", "documents", workspaceId],
-  conversations: (workspaceId) => ["arbi", "conversations", workspaceId],
-  tags: (workspaceId) => ["arbi", "tags", workspaceId],
-  thumbnails: (docIds) => ["arbi", "thumbnails", [...docIds].sort().join(",")],
-  parsedDoc: (docId, stage) => ["arbi", "parsedDoc", docId, stage],
-  agents: () => ["arbi", "agents"],
-  sessions: () => ["arbi", "sessions"],
-  health: () => ["arbi", "health"],
-  tasks: () => ["arbi", "tasks"],
-  facts: (workspaceId) => ["arbi", "facts", workspaceId],
-  configs: () => ["arbi", "configs"],
-  activity: () => ["arbi", "activity"],
-  projects: () => ["arbi", "projects"],
-  projectUsage: (projectExtId, monthsBack) => ["arbi", "projectUsage", projectExtId, monthsBack],
-  projectInvoices: (projectExtId) => ["arbi", "projectInvoices", projectExtId],
-  userUsageToday: () => ["arbi", "userUsageToday"]
-};
-function useWorkspaces(options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.workspaces(),
-    queryFn: () => arbi.workspaces.list(),
-    enabled: options?.enabled ?? arbi.isLoggedIn
-  });
-}
-function useDocuments(workspaceId, options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.documents(workspaceId ?? ""),
-    queryFn: () => arbi.documents.list(),
-    enabled: (options?.enabled ?? true) && !!workspaceId && arbi.hasWorkspace
-  });
-}
-function useConversations(workspaceId, options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.conversations(workspaceId ?? ""),
-    queryFn: () => arbi.conversations.list(),
-    enabled: (options?.enabled ?? true) && !!workspaceId && arbi.hasWorkspace
-  });
-}
-function useTags(workspaceId, options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.tags(workspaceId ?? ""),
-    queryFn: () => arbi.tags.list(),
-    enabled: (options?.enabled ?? true) && !!workspaceId && arbi.hasWorkspace
-  });
-}
-function useThumbnails(docIds, options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.thumbnails(docIds),
-    queryFn: () => arbi.documents.getThumbnails(docIds),
-    enabled: (options?.enabled ?? true) && docIds.length > 0 && arbi.hasWorkspace,
-    staleTime: 5 * 6e4
-  });
-}
-function useParsedDocument(docId, stage = "final", options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.parsedDoc(docId ?? "", stage),
-    queryFn: () => arbi.documents.getParsedContent(docId, stage),
-    enabled: (options?.enabled ?? true) && !!docId && arbi.hasWorkspace
-  });
-}
-function useAssistantQuery() {
-  const arbi = useArbi();
-  return useMutation({
-    mutationFn: ({ question, ...options }) => arbi.assistant.query(question, options)
-  });
-}
-function useAgents(options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.agents(),
-    queryFn: () => arbi.agents.list(),
-    enabled: options?.enabled ?? arbi.isLoggedIn
-  });
-}
-function useDeleteAgents() {
-  const arbi = useArbi();
-  return useMutation({
-    mutationFn: ({ agentExtIds }) => arbi.agents.delete(agentExtIds)
-  });
-}
-function useSessions(options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.sessions(),
-    queryFn: () => arbi.sessions.list(),
-    enabled: options?.enabled ?? arbi.isLoggedIn
-  });
-}
-function useHealth(options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.health(),
-    queryFn: () => arbi.health.check(),
-    enabled: options?.enabled
-  });
-}
-function useTasks(options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.tasks(),
-    queryFn: () => arbi.tasks.list(),
-    enabled: options?.enabled ?? arbi.isLoggedIn
-  });
-}
-function useFacts(workspaceId, options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.facts(workspaceId ?? ""),
-    queryFn: () => arbi.facts.list({ pinnedOnly: options?.pinnedOnly }),
-    enabled: (options?.enabled ?? true) && !!workspaceId && arbi.hasWorkspace
-  });
-}
-function useConfigs(options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.configs(),
-    queryFn: () => arbi.agentConfig.list(),
-    enabled: options?.enabled ?? arbi.isLoggedIn
-  });
-}
-function useActivity(options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.activity(),
-    queryFn: () => arbi.dm.listActivity({ limit: options?.limit, offset: options?.offset }),
-    enabled: options?.enabled ?? arbi.isLoggedIn
-  });
-}
-function useProjects(options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.projects(),
-    queryFn: () => arbi.projects.list(),
-    enabled: options?.enabled ?? arbi.isLoggedIn
-  });
-}
-function useProjectUsage(projectExtId, monthsBack = 0, options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.projectUsage(projectExtId ?? "", monthsBack),
-    queryFn: () => arbi.projects.usage(projectExtId, monthsBack),
-    enabled: (options?.enabled ?? true) && !!projectExtId && arbi.isLoggedIn
-  });
-}
-function useProjectsUsage(projectExtIds, monthsBack = 0, options) {
-  const arbi = useArbi();
-  const enabled = (options?.enabled ?? true) && arbi.isLoggedIn;
-  const results = useQueries({
-    queries: projectExtIds.map((id) => ({
-      queryKey: arbiQueryKeys.projectUsage(id, monthsBack),
-      queryFn: () => arbi.projects.usage(id, monthsBack),
-      enabled: enabled && !!id
-    }))
-  });
-  const signature = results.map((r) => `${r.status}:${r.dataUpdatedAt ?? 0}`).join("|");
-  return useMemo(() => {
-    const dailyMap = /* @__PURE__ */ new Map();
-    const modelMap = /* @__PURE__ */ new Map();
-    let totalSpend = 0;
-    let totalTokens = 0;
-    let totalRequests = 0;
-    for (const r of results) {
-      const d = r.data;
-      if (!d) continue;
-      totalSpend += d.total_spend ?? 0;
-      totalTokens += d.total_tokens ?? 0;
-      totalRequests += d.total_requests ?? 0;
-      for (const day of d.daily ?? []) {
-        const cur = dailyMap.get(day.date) ?? {
-          date: day.date,
-          spend: 0,
-          totalTokens: 0,
-          apiRequests: 0
-        };
-        cur.spend += day.spend ?? 0;
-        cur.totalTokens += day.total_tokens ?? 0;
-        cur.apiRequests += day.api_requests ?? 0;
-        dailyMap.set(day.date, cur);
-      }
-      for (const [model, mg] of Object.entries(d.model_groups ?? {})) {
-        const cur = modelMap.get(model) ?? {
-          model,
-          spend: 0,
-          totalTokens: 0,
-          promptTokens: 0,
-          completionTokens: 0,
-          successfulRequests: 0
-        };
-        cur.spend += mg.spend ?? 0;
-        cur.totalTokens += mg.total_tokens ?? 0;
-        cur.promptTokens += mg.prompt_tokens ?? 0;
-        cur.completionTokens += mg.completion_tokens ?? 0;
-        cur.successfulRequests += mg.successful_requests ?? 0;
-        modelMap.set(model, cur);
-      }
-    }
-    return {
-      totalSpend,
-      totalTokens,
-      totalRequests,
-      daily: [...dailyMap.values()].sort((a, b) => a.date.localeCompare(b.date)),
-      modelGroups: [...modelMap.values()].sort(
-        (a, b) => b.spend - a.spend || b.totalTokens - a.totalTokens
-      ),
-      isLoading: results.some((r) => r.isLoading),
-      isError: results.some((r) => r.isError)
-    };
-  }, [signature]);
-}
-function useProjectInvoices(projectExtId, options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.projectInvoices(projectExtId ?? ""),
-    queryFn: () => arbi.projects.invoices(projectExtId),
-    enabled: (options?.enabled ?? true) && !!projectExtId && arbi.isLoggedIn
-  });
-}
-function useUserUsageToday(options) {
-  const arbi = useArbi();
-  return useQuery({
-    queryKey: arbiQueryKeys.userUsageToday(),
-    queryFn: () => arbi.user.usageToday(),
-    enabled: options?.enabled ?? arbi.isLoggedIn
-  });
-}
-
-// src/chat/useArbiChat.ts
 var counter = 0;
 var nextId = (prefix) => `${prefix}-${Date.now().toString(36)}-${(counter++).toString(36)}`;
 var emptyThread = () => ({
@@ -890,6 +662,186 @@ function useArbiShellWorkspaces(options = {}) {
     select: select ?? defaultSelect
   });
 }
+function useCreateTask() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: (body) => arbi.tasks.create(body),
+    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.tasks() })
+  });
+}
+function useUpdateTask() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: ({ taskId, patch }) => arbi.tasks.update(taskId, patch),
+    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.tasks() })
+  });
+}
+function useRunTask() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: (taskId) => arbi.tasks.runWithAuth(taskId),
+    onSuccess: (_task, taskId) => Promise.all([
+      qc.invalidateQueries({ queryKey: arbiQueryKeys.tasks() }),
+      qc.invalidateQueries({ queryKey: arbiQueryKeys.taskRuns(taskId) })
+    ])
+  });
+}
+function useDeleteTask() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: (taskId) => arbi.tasks.delete(taskId),
+    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.tasks() })
+  });
+}
+function useTaskRuns(taskId, options) {
+  const arbi = useArbi();
+  return useQuery({
+    queryKey: arbiQueryKeys.taskRuns(taskId ?? ""),
+    queryFn: () => arbi.tasks.listRuns(taskId),
+    enabled: (options?.enabled ?? true) && !!taskId && arbi.isLoggedIn,
+    refetchInterval: (q) => q.state.data?.some((r) => r.status === "running") ? 3e3 : false
+  });
+}
+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) })
+    ])
+  });
+}
+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({
+    queryKey: arbiQueryKeys.contacts(),
+    queryFn: () => arbi.contacts.list(),
+    enabled: (options?.enabled ?? true) && arbi.isLoggedIn
+  });
+}
+function usePendingContacts(options) {
+  const contacts = useContacts(options);
+  const pending = useMemo(
+    () => (contacts.data ?? []).filter(
+      (c) => c.status === "invitation_pending" || c.status === "invitation_expired"
+    ),
+    [contacts.data]
+  );
+  return { ...contacts, pending };
+}
+function useAddContacts() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: (emails) => arbi.contacts.add(emails),
+    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.contacts() })
+  });
+}
+function useRemoveContacts() {
+  const arbi = useArbi();
+  const qc = useQueryClient();
+  return useMutation({
+    mutationFn: (contactIds) => arbi.contacts.remove(contactIds),
+    onSuccess: () => qc.invalidateQueries({ queryKey: arbiQueryKeys.contacts() })
+  });
+}
 function useWorkspaceDocs(workspaceId) {
   const { data: docs = [], isLoading, isError } = useDocuments(workspaceId);
   const docIds = useMemo(
@@ -3065,7 +3017,6 @@ function isAgentUser(user) {
   return user.external_id.startsWith("agt-");
 }
 var DM_QUERY_KEY = ["arbi", "directMessages"];
-var CONTACTS_QUERY_KEY = ["arbi", "contacts"];
 function mergeContacts(threads, contacts) {
   const byPeer = /* @__PURE__ */ new Map();
   for (const t of threads) byPeer.set(t.peer.external_id, t);
@@ -3131,7 +3082,7 @@ function useDirectMessages(options) {
     enabled
   });
   const contactsQuery = useQuery({
-    queryKey: CONTACTS_QUERY_KEY,
+    queryKey: arbiQueryKeys.contacts(),
     queryFn: () => arbi.contacts.list(),
     enabled
   });
@@ -3216,6 +3167,7 @@ function ConversationList({
   classNames,
   emptyState,
   header,
+  lead,
   className
 }) {
   return /* @__PURE__ */ jsxs(
@@ -3235,6 +3187,7 @@ function ConversationList({
             conversations.length === 1 ? "colleague" : "colleagues"
           ] })
         ] }) : header,
+        lead,
         /* @__PURE__ */ jsx("div", { className: "min-h-0 flex-1 overflow-y-auto", children: conversations.length === 0 ? /* @__PURE__ */ jsx("div", { className: "px-4 py-6 text-xs text-muted-foreground", children: emptyState ?? "No conversations yet." }) : /* @__PURE__ */ jsx("ul", { className: "w-full", children: conversations.map((c, i) => {
           const { peer, lastMessage, unreadCount } = c;
           const newSection = c.sectionLabel && c.sectionLabel !== conversations[i - 1]?.sectionLabel;
@@ -3521,14 +3474,117 @@ function DmComposer({
     error && /* @__PURE__ */ jsx("p", { className: "mt-2 text-xs text-destructive", "data-testid": "arbi-dm-error", children: error })
   ] });
 }
+var DEFAULT_START_BY_EMAIL_LABELS = {
+  email: "Email address",
+  emailPlaceholder: "name@example.com",
+  message: "Message",
+  messagePlaceholder: "Write the first message\u2026",
+  send: "Send",
+  sending: "Sending\u2026",
+  invalidEmail: "Enter a valid email address."
+};
+var EMAIL = /^[^\s@]+@[^\s@]+\.[^\s@]+$/;
+function StartByEmail({
+  onStart,
+  onStarted,
+  classNames,
+  labels,
+  className
+}) {
+  const l = { ...DEFAULT_START_BY_EMAIL_LABELS, ...labels };
+  const uid = useId();
+  const [email, setEmail] = useState("");
+  const [text, setText] = useState("");
+  const [busy, setBusy] = useState(false);
+  const [error, setError] = useState(null);
+  const address = email.trim().toLowerCase();
+  const valid = EMAIL.test(address);
+  const submit = async () => {
+    if (!valid || !text.trim() || busy) return;
+    setBusy(true);
+    setError(null);
+    try {
+      const result = await onStart(address, text.trim());
+      setEmail("");
+      setText("");
+      onStarted?.(result);
+    } catch (e) {
+      setError(e instanceof Error ? e.message : String(e));
+    } finally {
+      setBusy(false);
+    }
+  };
+  return /* @__PURE__ */ jsxs(
+    "form",
+    {
+      "data-testid": "arbi-dm-start",
+      className: cn("flex flex-col gap-2 border-b border-border px-4 py-3", className),
+      onSubmit: (e) => {
+        e.preventDefault();
+        void submit();
+      },
+      children: [
+        /* @__PURE__ */ jsx(Label, { htmlFor: `${uid}-email`, children: l.email }),
+        /* @__PURE__ */ jsx(
+          Input,
+          {
+            id: `${uid}-email`,
+            type: "email",
+            value: email,
+            onChange: (e) => setEmail(e.target.value),
+            placeholder: l.emailPlaceholder,
+            "aria-invalid": email !== "" && !valid,
+            "aria-describedby": email !== "" && !valid ? `${uid}-bad` : void 0,
+            disabled: busy,
+            "data-testid": "arbi-dm-start-email"
+          }
+        ),
+        email !== "" && !valid && /* @__PURE__ */ jsx("p", { id: `${uid}-bad`, className: "text-xs text-destructive", children: l.invalidEmail }),
+        /* @__PURE__ */ jsx(Label, { htmlFor: `${uid}-text`, children: l.message }),
+        /* @__PURE__ */ jsx(
+          Textarea,
+          {
+            id: `${uid}-text`,
+            value: text,
+            onChange: (e) => setText(e.target.value),
+            placeholder: l.messagePlaceholder,
+            rows: 2,
+            disabled: busy,
+            "data-testid": "arbi-dm-start-message"
+          }
+        ),
+        error && /* @__PURE__ */ jsx("p", { role: "alert", className: "text-xs text-destructive", "data-testid": "arbi-dm-start-error", children: error }),
+        /* @__PURE__ */ jsxs(
+          Button,
+          {
+            type: "submit",
+            size: "sm",
+            disabled: !valid || !text.trim() || busy,
+            className: cn("self-end", classNames?.sendButton),
+            "data-testid": "arbi-dm-start-send",
+            children: [
+              busy ? /* @__PURE__ */ jsx(Loader2, { className: "size-4 animate-spin", "aria-hidden": true }) : /* @__PURE__ */ jsx(Send, { className: "size-4", "aria-hidden": true }),
+              busy ? l.sending : l.send
+            ]
+          }
+        )
+      ]
+    }
+  );
+}
 function DirectMessages({
   className,
   classNames,
   emptyThread: emptyThread2,
-  emptyList
+  emptyList,
+  listHeader,
+  listLead,
+  startByEmail,
+  hideOwnContact
 }) {
   const {
-    conversations,
+    conversations: allConversations,
+    myExtId,
     isLoading,
     isError,
     error,
@@ -3538,6 +3594,14 @@ function DirectMessages({
     markRead
   } = useDirectMessages();
   const [selectedPeerId, setSelectedPeerId] = useState(null);
+  const conversations = useMemo(
+    () => hideOwnContact && myExtId ? allConversations.filter((c) => c.peer.external_id !== myExtId) : allConversations,
+    [allConversations, hideOwnContact, myExtId]
+  );
+  const openStarted = (created) => {
+    const peer = Array.isArray(created) ? created[0]?.recipient?.external_id : void 0;
+    if (peer) setSelectedPeerId(peer);
+  };
   const loadError = isError ? error instanceof Error ? error.message : "Couldn\u2019t load messages." : null;
   const selected = useMemo(
     () => conversations.find((c) => c.peer.external_id === selectedPeerId) ?? null,
@@ -3581,7 +3645,20 @@ function DirectMessages({
             selectedPeerId: selected?.peer.external_id ?? null,
             onSelect: setSelectedPeerId,
             classNames,
-            emptyState: isLoading ? "Loading messages\u2026" : emptyList
+            emptyState: isLoading ? "Loading messages\u2026" : emptyList,
+            header: listHeader,
+            lead: /* @__PURE__ */ jsxs(Fragment, { children: [
+              listLead,
+              startByEmail && /* @__PURE__ */ jsx(
+                StartByEmail,
+                {
+                  onStart: sendMessageTo,
+                  onStarted: openStarted,
+                  classNames,
+                  labels: typeof startByEmail === "object" ? startByEmail.labels : void 0
+                }
+              )
+            ] })
           }
         ),
         /* @__PURE__ */ jsxs("div", { className: "flex min-h-0 min-w-0 flex-1 flex-col", children: [
@@ -4990,6 +5067,6 @@ function hasVideoTrack(blob) {
   });
 }
 
-export { ActivityFeed, AiActionCard, AiAnswer, AppleIcon, ArbiBotIcon, ArbiIcon, ArbiWebSocketProvider, ArtifactCanvas, ChatBubble, CitationRenderer, ClaudeIcon, CommandPalette, CommandPaletteProvider, ConversationList, DEFAULT_ASPECT_OPTIONS, DateSeparator, DirectMessages, DmAvatar, DmComposer, DocIcon, DocQaPanel, DocSearch, DocumentCardMenu, DocumentGallery, DocumentReview, DocumentTable, DraftTitleControls, FileIcon, FireworksIcon, GeminiIcon, GitHubIcon, GoogleIcon, HtmlIcon, ImageCropModal, ImageCropper, LinkedInIcon, MessageArtifactCard, MessageBubble, MessageThread, OpenAIIcon, PhotoCapture, ProviderIcon, RUN_WINDOW_MS, SemanticSearch, TxtIcon, TypingIndicator, VideoCapture, VoiceNoteRecorder, XlsIcon, annotateThread, arbiQueryKeys, captureFrameAsBlob, cropImage, dayKey, describeActivity, getNotificationTypeMeta, getWorkspaceActionText, getWorkspaceKind, getWorkspaceSentPhrase, hasVideoTrack, isWorkspaceNotificationType, parseActivityContent, pickRecentWorkspaces, redlineStats, resolveActiveWorkspace, textToArtifact, titleFromQuestion, toRedlineMarkdown, toShellWorkspace, useActivity, useAgents, useAiTask, useArbiChat, useArbiConnected, useArbiDocumentSync, useArbiShellWorkspaces, useArbiUiStore, useArbiWebSocket, useArbiWsContext, useArbiWsMessage, useArbiWsStatus, useArbiWsStore, useArbiWsTask, useArbiWsTasks, useAssistantQuery, useCameraStream, useCommandPalette, useConfigs, useConversations, useDeleteAgents, useDirectMessages, useDocuments, useFacts, useHealth, useParsedDocument, useProjectInvoices, useProjectUsage, useProjects, useProjectsUsage, useSemanticSearch, useSessions, useShellWorkspaces, useTags, useTasks, useThumbnails, useUserUsageToday, useVideoRecorder, useWorkspaceDocs, useWorkspaces, userDisplayName, userInitials };
+export { ActivityFeed, AiActionCard, AiAnswer, AppleIcon, ArbiBotIcon, ArbiIcon, ArbiWebSocketProvider, ArtifactCanvas, ChatBubble, CitationRenderer, ClaudeIcon, CommandPalette, CommandPaletteProvider, ConversationList, DEFAULT_ASPECT_OPTIONS, 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, 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, pickRecentWorkspaces, redlineStats, resolveActiveWorkspace, textToArtifact, titleFromQuestion, toRedlineMarkdown, toShellWorkspace, useAddContacts, useAgentConfig, useAgentModels, useAgentWorkspaces, useAiTask, useArbiChat, useArbiConnected, useArbiDocumentSync, useArbiShellWorkspaces, useArbiUiStore, useArbiWebSocket, useArbiWsContext, useArbiWsMessage, useArbiWsStatus, useArbiWsStore, useArbiWsTask, useArbiWsTasks, useCameraStream, useCommandPalette, useContacts, useCreateAgent, useCreateTask, useDeleteTask, useDirectMessages, useGrantAgentAccess, usePendingContacts, useRemoveContacts, useRevokeAgentAccess, useRunTask, useSaveAgentConfig, useSemanticSearch, useShellWorkspaces, useSkills, useTaskRuns, useUpdateTask, useVideoRecorder, useWorkspaceDocs, userDisplayName, userInitials };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map