@arbidocs/sdk 0.3.109 → 0.3.111

package/dist/index.d.cts

@@ -1,5 +1,5 @@
-import { n as ConfigStore, l as CliConfig, m as CliCredentials, C as ChatSession, a2 as SkippedFile, aa as UploadResult, i as AuthHeaders, a6 as UploadBatchResult, a9 as UploadOptions, a7 as UploadDirectOptions, a8 as UploadDirectResult, a0 as SUPPORTED_EXTENSIONS, r as DmCryptoContext } from './browser-C4gqnXaX.cjs';
-export { a as AgentStepEvent, b as Arbi, c as ArbiApiError, d as ArbiError, f as ArbiOptions, g as ArtifactEvent, h as AuthContext, j as AuthenticatedClient, k as CitationSummary, o as ConnectOptions, D as DOC_TERMINAL_STATUSES, s as DocNameMap, t as DocumentListFields, u as DocumentListOrder, v as DocumentWaiter, w as DocumentWaiterOptions, F as FormattedWsMessage, L as ListAllOptions, x as ListPaginatedOptions, B as MessageLevel, E as MessageMetadataPayload, G as MessageQueuedEvent, O as OutputTokensDetails, P as ParsedSlashCommand, Q as QueryOptions, R as ReconnectOptions, H as ReconnectableWsConnection, I as ResolvedCitation, J as ResponseCompletedEvent, K as ResponseContentPartAddedEvent, N as ResponseCreatedEvent, S as ResponseFailedEvent, T as ResponseOutputItemAddedEvent, U as ResponseOutputItemDoneEvent, V as ResponseOutputTextDeltaEvent, W as ResponseOutputTextDoneEvent, X as ResponseUsage, Y as SSEEvent, Z as SSEStreamCallbacks, _ as SSEStreamResult, $ as SSEStreamStartData, a1 as SkillSummary, ab as UserInfo, ac as UserInputRequestEvent, ad as UserMessageEvent, ae as WebSocketAuthError, af as WorkspaceContext, ag as WsConnection, ah as agentconfig, ai as assistant, aj as authenticatedFetch, ak as buildRetrievalChunkTool, al as buildRetrievalFullContextTool, am as buildRetrievalTocTool, an as connectWebSocket, ao as connectWithReconnect, ap as consumeSSEStream, aq as contacts, ar as conversations, as as countCitations, at as createAuthenticatedClient, au as createDocumentWaiter, av as dm, aw as doctags, ax as documents, ay as files, az as filterSkills, aA as formatAgentStepLabel, aB as formatFileSize, aC as formatStreamSummary, aD as formatUserName, aE as formatWorkspaceChoices, aF as formatWsMessage, aG as generateEncryptedWorkspaceKey, aH as generateNewWorkspaceKey, aI as getErrorCode, aJ as getErrorMessage, aK as getRawWorkspaceKey, aL as health, aN as parseSSEEvents, aO as parseSlashCommand, aP as parseSlashTokenInProgress, aQ as performPasswordLogin, aR as performSigningKeyLogin, aS as performSsoDeviceFlowLogin, aT as requireData, aU as requireOk, aV as resolveAuth, aW as resolveCitations, aX as resolveWorkspace, aY as responses, aZ as selectWorkspace, a_ as selectWorkspaceById, a$ as settings, b0 as streamSSE, b1 as stripCitationMarkdown, b2 as summarizeCitations, b3 as tags, b5 as workspaces } from './browser-C4gqnXaX.cjs';
+import { o as ConfigStore, l as CliConfig, m as CliCredentials, C as ChatSession, a3 as SkippedFile, ab as UploadResult, i as AuthHeaders, a7 as UploadBatchResult, aa as UploadOptions, a8 as UploadDirectOptions, a9 as UploadDirectResult, a1 as SUPPORTED_EXTENSIONS, s as DmCryptoContext } from './browser-B5b3SDW6.cjs';
+export { a as AgentStepEvent, b as Arbi, c as ArbiApiError, d as ArbiError, f as ArbiOptions, g as ArtifactEvent, h as AuthContext, j as AuthenticatedClient, k as CitationSummary, n as CommandDescriptor, p as ConnectOptions, D as DOC_TERMINAL_STATUSES, t as DocNameMap, u as DocumentListFields, v as DocumentListOrder, w as DocumentWaiter, x as DocumentWaiterOptions, F as FormattedWsMessage, L as ListAllOptions, y as ListPaginatedOptions, E as MessageLevel, G as MessageMetadataPayload, H as MessageQueuedEvent, O as OutputTokensDetails, P as ParsedSlashCommand, Q as QueryOptions, R as ReconnectOptions, I as ReconnectableWsConnection, J as ResolvedCitation, K as ResponseCompletedEvent, N as ResponseContentPartAddedEvent, S as ResponseCreatedEvent, T as ResponseFailedEvent, U as ResponseOutputItemAddedEvent, V as ResponseOutputItemDoneEvent, W as ResponseOutputTextDeltaEvent, X as ResponseOutputTextDoneEvent, Y as ResponseUsage, Z as SSEEvent, _ as SSEStreamCallbacks, $ as SSEStreamResult, a0 as SSEStreamStartData, a2 as SkillSummary, ac as UserInfo, ad as UserInputRequestEvent, ae as UserMessageEvent, af as WebSocketAuthError, ag as WorkspaceContext, ah as WsConnection, ai as agentconfig, aj as assistant, ak as authenticatedFetch, al as buildRetrievalChunkTool, am as buildRetrievalFullContextTool, an as buildRetrievalTocTool, ao as connectWebSocket, ap as connectWithReconnect, aq as consumeSSEStream, ar as contacts, as as conversations, at as countCitations, au as createAuthenticatedClient, av as createDocumentWaiter, aw as dm, ax as doctags, ay as documents, az as files, aA as filterSkills, aB as formatAgentStepLabel, aC as formatFileSize, aD as formatStreamSummary, aE as formatUserName, aF as formatWorkspaceChoices, aG as formatWsMessage, aH as generateEncryptedWorkspaceKey, aI as generateNewWorkspaceKey, aJ as getErrorCode, aK as getErrorMessage, aL as getRawWorkspaceKey, aM as health, aO as parseSSEEvents, aP as parseSlashCommand, aQ as parseSlashTokenInProgress, aR as performPasswordLogin, aS as performSigningKeyLogin, aT as performSsoDeviceFlowLogin, aU as requireData, aV as requireOk, aW as resolveAuth, aX as resolveCitations, aY as resolveWorkspace, aZ as responses, a_ as selectWorkspace, a$ as selectWorkspaceById, b0 as settings, b1 as streamSSE, b2 as stripCitationMarkdown, b3 as summarizeCitations, b4 as tags, b6 as workspaces } from './browser-B5b3SDW6.cjs';
 import { SessionStorageProvider, ArbiClient, components } from '@arbidocs/client';
 
 /**
@@ -450,6 +450,9 @@ declare function getAgentConfig(arbi: ArbiClient, agentExtId: string): Promise<{
         TEMPERATURE: number;
         MAX_TOKENS: number;
         MAX_TOKEN_SIZE_TO_ANSWER: number;
+        MAX_CHUNKS_PER_EVAL_CALL: number;
+        MAX_CONCURRENT_EVAL_BATCHES: number;
+        EVAL_BATCH_TIMEOUT_S: number;
     };
     TitleLLM: {
         API_TYPE: "local" | "remote";

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { n as ConfigStore, l as CliConfig, m as CliCredentials, C as ChatSession, a2 as SkippedFile, aa as UploadResult, i as AuthHeaders, a6 as UploadBatchResult, a9 as UploadOptions, a7 as UploadDirectOptions, a8 as UploadDirectResult, a0 as SUPPORTED_EXTENSIONS, r as DmCryptoContext } from './browser-C4gqnXaX.js';
-export { a as AgentStepEvent, b as Arbi, c as ArbiApiError, d as ArbiError, f as ArbiOptions, g as ArtifactEvent, h as AuthContext, j as AuthenticatedClient, k as CitationSummary, o as ConnectOptions, D as DOC_TERMINAL_STATUSES, s as DocNameMap, t as DocumentListFields, u as DocumentListOrder, v as DocumentWaiter, w as DocumentWaiterOptions, F as FormattedWsMessage, L as ListAllOptions, x as ListPaginatedOptions, B as MessageLevel, E as MessageMetadataPayload, G as MessageQueuedEvent, O as OutputTokensDetails, P as ParsedSlashCommand, Q as QueryOptions, R as ReconnectOptions, H as ReconnectableWsConnection, I as ResolvedCitation, J as ResponseCompletedEvent, K as ResponseContentPartAddedEvent, N as ResponseCreatedEvent, S as ResponseFailedEvent, T as ResponseOutputItemAddedEvent, U as ResponseOutputItemDoneEvent, V as ResponseOutputTextDeltaEvent, W as ResponseOutputTextDoneEvent, X as ResponseUsage, Y as SSEEvent, Z as SSEStreamCallbacks, _ as SSEStreamResult, $ as SSEStreamStartData, a1 as SkillSummary, ab as UserInfo, ac as UserInputRequestEvent, ad as UserMessageEvent, ae as WebSocketAuthError, af as WorkspaceContext, ag as WsConnection, ah as agentconfig, ai as assistant, aj as authenticatedFetch, ak as buildRetrievalChunkTool, al as buildRetrievalFullContextTool, am as buildRetrievalTocTool, an as connectWebSocket, ao as connectWithReconnect, ap as consumeSSEStream, aq as contacts, ar as conversations, as as countCitations, at as createAuthenticatedClient, au as createDocumentWaiter, av as dm, aw as doctags, ax as documents, ay as files, az as filterSkills, aA as formatAgentStepLabel, aB as formatFileSize, aC as formatStreamSummary, aD as formatUserName, aE as formatWorkspaceChoices, aF as formatWsMessage, aG as generateEncryptedWorkspaceKey, aH as generateNewWorkspaceKey, aI as getErrorCode, aJ as getErrorMessage, aK as getRawWorkspaceKey, aL as health, aN as parseSSEEvents, aO as parseSlashCommand, aP as parseSlashTokenInProgress, aQ as performPasswordLogin, aR as performSigningKeyLogin, aS as performSsoDeviceFlowLogin, aT as requireData, aU as requireOk, aV as resolveAuth, aW as resolveCitations, aX as resolveWorkspace, aY as responses, aZ as selectWorkspace, a_ as selectWorkspaceById, a$ as settings, b0 as streamSSE, b1 as stripCitationMarkdown, b2 as summarizeCitations, b3 as tags, b5 as workspaces } from './browser-C4gqnXaX.js';
+import { o as ConfigStore, l as CliConfig, m as CliCredentials, C as ChatSession, a3 as SkippedFile, ab as UploadResult, i as AuthHeaders, a7 as UploadBatchResult, aa as UploadOptions, a8 as UploadDirectOptions, a9 as UploadDirectResult, a1 as SUPPORTED_EXTENSIONS, s as DmCryptoContext } from './browser-B5b3SDW6.js';
+export { a as AgentStepEvent, b as Arbi, c as ArbiApiError, d as ArbiError, f as ArbiOptions, g as ArtifactEvent, h as AuthContext, j as AuthenticatedClient, k as CitationSummary, n as CommandDescriptor, p as ConnectOptions, D as DOC_TERMINAL_STATUSES, t as DocNameMap, u as DocumentListFields, v as DocumentListOrder, w as DocumentWaiter, x as DocumentWaiterOptions, F as FormattedWsMessage, L as ListAllOptions, y as ListPaginatedOptions, E as MessageLevel, G as MessageMetadataPayload, H as MessageQueuedEvent, O as OutputTokensDetails, P as ParsedSlashCommand, Q as QueryOptions, R as ReconnectOptions, I as ReconnectableWsConnection, J as ResolvedCitation, K as ResponseCompletedEvent, N as ResponseContentPartAddedEvent, S as ResponseCreatedEvent, T as ResponseFailedEvent, U as ResponseOutputItemAddedEvent, V as ResponseOutputItemDoneEvent, W as ResponseOutputTextDeltaEvent, X as ResponseOutputTextDoneEvent, Y as ResponseUsage, Z as SSEEvent, _ as SSEStreamCallbacks, $ as SSEStreamResult, a0 as SSEStreamStartData, a2 as SkillSummary, ac as UserInfo, ad as UserInputRequestEvent, ae as UserMessageEvent, af as WebSocketAuthError, ag as WorkspaceContext, ah as WsConnection, ai as agentconfig, aj as assistant, ak as authenticatedFetch, al as buildRetrievalChunkTool, am as buildRetrievalFullContextTool, an as buildRetrievalTocTool, ao as connectWebSocket, ap as connectWithReconnect, aq as consumeSSEStream, ar as contacts, as as conversations, at as countCitations, au as createAuthenticatedClient, av as createDocumentWaiter, aw as dm, ax as doctags, ay as documents, az as files, aA as filterSkills, aB as formatAgentStepLabel, aC as formatFileSize, aD as formatStreamSummary, aE as formatUserName, aF as formatWorkspaceChoices, aG as formatWsMessage, aH as generateEncryptedWorkspaceKey, aI as generateNewWorkspaceKey, aJ as getErrorCode, aK as getErrorMessage, aL as getRawWorkspaceKey, aM as health, aO as parseSSEEvents, aP as parseSlashCommand, aQ as parseSlashTokenInProgress, aR as performPasswordLogin, aS as performSigningKeyLogin, aT as performSsoDeviceFlowLogin, aU as requireData, aV as requireOk, aW as resolveAuth, aX as resolveCitations, aY as resolveWorkspace, aZ as responses, a_ as selectWorkspace, a$ as selectWorkspaceById, b0 as settings, b1 as streamSSE, b2 as stripCitationMarkdown, b3 as summarizeCitations, b4 as tags, b6 as workspaces } from './browser-B5b3SDW6.js';
 import { SessionStorageProvider, ArbiClient, components } from '@arbidocs/client';
 
 /**
@@ -450,6 +450,9 @@ declare function getAgentConfig(arbi: ArbiClient, agentExtId: string): Promise<{
         TEMPERATURE: number;
         MAX_TOKENS: number;
         MAX_TOKEN_SIZE_TO_ANSWER: number;
+        MAX_CHUNKS_PER_EVAL_CALL: number;
+        MAX_CONCURRENT_EVAL_BATCHES: number;
+        EVAL_BATCH_TIMEOUT_S: number;
     };
     TitleLLM: {
         API_TYPE: "local" | "remote";

package/dist/index.js

@@ -4456,7 +4456,7 @@ var WebSocketAuthError = class extends Error {
   }
 };
 function connectWebSocket(options) {
-  const { baseUrl, accessToken, onMessage, onClose } = options;
+  const { baseUrl, accessToken, onMessage, onClose, onAuthenticated } = options;
   const url = buildWebSocketUrl(baseUrl);
   return new Promise((resolve2, reject) => {
     const ws = new WebSocket(url);
@@ -4479,6 +4479,7 @@ function connectWebSocket(options) {
           clearTimeout(timeout);
           if (msg.success) {
             authenticated = true;
+            onAuthenticated?.(msg);
             resolve2({ close: () => ws.close() });
           } else {
             ws.close();
@@ -4512,6 +4513,8 @@ async function connectWithReconnect(options) {
     onReconnectFailed,
     onClose,
     onMessage,
+    onAuthenticated,
+    shouldReconnect,
     baseUrl,
     accessToken
   } = options;
@@ -4526,10 +4529,11 @@ async function connectWithReconnect(options) {
       baseUrl,
       accessToken: token,
       onMessage,
+      onAuthenticated,
       onClose: (code, reason) => {
         currentConnection = null;
         onClose?.(code, reason);
-        if (!closed) void scheduleReconnect();
+        if (!closed && (shouldReconnect?.(code, reason) ?? true)) void scheduleReconnect();
       }
     });
   };
@@ -4560,7 +4564,14 @@ async function connectWithReconnect(options) {
     }
     if (!closed) onReconnectFailed?.();
   };
-  currentConnection = await connectOnce();
+  try {
+    currentConnection = await connectOnce();
+  } catch (err) {
+    if (!(err instanceof WebSocketAuthError) || !refreshAuth) throw err;
+    await refreshAuth().catch(() => {
+    });
+    currentConnection = await connectOnce();
+  }
   return {
     close: () => {
       closed = true;
@@ -5327,6 +5338,7 @@ __export(assistant_exports, {
   buildRetrievalFullContextTool: () => buildRetrievalFullContextTool,
   buildRetrievalTocTool: () => buildRetrievalTocTool,
   filterSkills: () => filterSkills,
+  listCommands: () => listCommands,
   listSkills: () => listSkills,
   parseSlashCommand: () => parseSlashCommand,
   parseSlashTokenInProgress: () => parseSlashTokenInProgress,
@@ -5434,6 +5446,13 @@ async function listSkills(arbi, options) {
   );
   return result.skills;
 }
+async function listCommands(arbi) {
+  const result = requireData(
+    await arbi.fetch.GET("/v1/assistant/commands", {}),
+    "Failed to list commands"
+  );
+  return result.commands;
+}
 function parseSlashCommand(input) {
   const m2 = /^\s*\/([a-zA-Z0-9_-]+)(?:\s+([\s\S]*))?$/.exec(input);
   if (!m2) return null;
@@ -6093,6 +6112,19 @@ var Arbi = class {
   get hasWorkspace() {
     return this.currentWorkspaceId !== null;
   }
+  /**
+   * Live connection info for opening a raw transport (e.g. a WebSocket).
+   *
+   * Reads the access token from the session on each call, so a token refreshed
+   * elsewhere is reflected on the next (re)connect. Returns `null` when there is
+   * no client/token yet (not logged in) rather than throwing, so callers can
+   * poll for readiness instead of guarding against exceptions.
+   */
+  getConnectionInfo() {
+    const accessToken = this.client?.session.getState().accessToken;
+    if (!accessToken) return null;
+    return { baseUrl: this.options.url, accessToken };
+  }
   // ── Auth headers (for raw fetch operations) ───────────────────────────
   getAuthHeaders() {
     const client = this.requireClient();