@distri/core 0.2.7 → 0.3.0

package/dist/index.js

@@ -170,7 +170,7 @@ function isDistriEvent(event) {
   return "type" in event && "data" in event;
 }
 
-// ../../../node_modules/.pnpm/@a2a-js+sdk@https+++codeload.github.com+v3g42+a2a-js+tar.gz+51444c9/node_modules/@a2a-js/sdk/dist/chunk-CUGIRVQB.js
+// ../../node_modules/.pnpm/@a2a-js+sdk@https+++codeload.github.com+v3g42+a2a-js+tar.gz+51444c9/node_modules/@a2a-js/sdk/dist/chunk-CUGIRVQB.js
 var A2AClient = class {
   /**
    * Constructs an A2AClient instance.
@@ -730,6 +730,17 @@ function convertA2AStatusUpdateToDistri(statusUpdate) {
       };
       return hookRequested;
     }
+    case "browser_session_started": {
+      const browserSessionStarted = {
+        type: "browser_session_started",
+        data: {
+          session_id: metadata.session_id || "",
+          viewer_url: metadata.viewer_url,
+          stream_url: metadata.stream_url
+        }
+      };
+      return browserSessionStarted;
+    }
     default: {
       console.warn(`Unhandled status update metadata type: ${metadata.type}`, metadata);
       const defaultResult = {
@@ -909,21 +920,29 @@ var _DistriClient = class _DistriClient {
     this.tokenRefreshSkewMs = config.tokenRefreshSkewMs ?? 6e4;
     this.onTokenRefresh = config.onTokenRefresh;
     this.config = {
-      baseUrl: config.baseUrl.replace(/\/$/, ""),
+      baseUrl: config.baseUrl?.replace(/\/$/, "") || DEFAULT_BASE_URL,
       apiVersion: config.apiVersion || "v1",
-      timeout: config.timeout || 3e4,
-      retryAttempts: config.retryAttempts || 3,
-      retryDelay: config.retryDelay || 1e3,
-      debug: config.debug || false,
+      timeout: config.timeout ?? 3e4,
+      retryAttempts: config.retryAttempts ?? 3,
+      retryDelay: config.retryDelay ?? 1e3,
+      debug: config.debug ?? false,
       headers,
-      interceptor: config.interceptor || ((init) => Promise.resolve(init))
+      interceptor: config.interceptor ?? (async (init) => Promise.resolve(init)),
+      onTokenRefresh: config.onTokenRefresh,
+      clientId: config.clientId
     };
-    this.debug("DistriClient initialized with config:", {
-      baseUrl: this.config.baseUrl,
-      hasAccessToken: !!this.accessToken,
-      hasRefreshToken: !!this.refreshToken,
-      timeout: this.config.timeout
-    });
+  }
+  /**
+   * Get the configured client ID.
+   */
+  get clientId() {
+    return this.config.clientId;
+  }
+  /**
+   * Set the client ID for embed token issuance.
+   */
+  set clientId(value) {
+    this.config.clientId = value;
   }
   /**
    * Create a client with default cloud configuration.
@@ -956,7 +975,7 @@ var _DistriClient = class _DistriClient {
     if (expiry) {
       body.expiry = typeof expiry === "string" ? expiry : expiry.toISOString();
     }
-    const resp = await this.fetch(`/session/${encodeURIComponent(sessionId)}/values`, {
+    const resp = await this.fetch(`/sessions/${encodeURIComponent(sessionId)}/values`, {
       method: "POST",
       headers: {
         "Content-Type": "application/json",
@@ -973,7 +992,7 @@ var _DistriClient = class _DistriClient {
    * Session store: get a single value
    */
   async getSessionValue(sessionId, key) {
-    const resp = await this.fetch(`/session/${encodeURIComponent(sessionId)}/values/${encodeURIComponent(key)}`, {
+    const resp = await this.fetch(`/sessions/${encodeURIComponent(sessionId)}/values/${encodeURIComponent(key)}`, {
       method: "GET",
       headers: {
         ...this.config.headers
@@ -990,7 +1009,7 @@ var _DistriClient = class _DistriClient {
    * Session store: get all values in a session
    */
   async getSessionValues(sessionId) {
-    const resp = await this.fetch(`/session/${encodeURIComponent(sessionId)}/values`, {
+    const resp = await this.fetch(`/sessions/${encodeURIComponent(sessionId)}/values`, {
       method: "GET",
       headers: {
         ...this.config.headers
@@ -1007,7 +1026,7 @@ var _DistriClient = class _DistriClient {
    * Session store: delete a single key
    */
   async deleteSessionValue(sessionId, key) {
-    const resp = await this.fetch(`/session/${encodeURIComponent(sessionId)}/values/${encodeURIComponent(key)}`, {
+    const resp = await this.fetch(`/sessions/${encodeURIComponent(sessionId)}/values/${encodeURIComponent(key)}`, {
       method: "DELETE",
       headers: {
         ...this.config.headers
@@ -1022,7 +1041,7 @@ var _DistriClient = class _DistriClient {
    * Session store: clear all keys in a session
    */
   async clearSession(sessionId) {
-    const resp = await this.fetch(`/session/${encodeURIComponent(sessionId)}`, {
+    const resp = await this.fetch(`/sessions/${encodeURIComponent(sessionId)}`, {
       method: "DELETE",
       headers: {
         ...this.config.headers
@@ -1033,30 +1052,6 @@ var _DistriClient = class _DistriClient {
       throw new ApiError(errorData.error || "Failed to clear session", resp.status);
     }
   }
-  /**
-   * Set additional user message parts for the next agent iteration.
-   * These parts will be appended to the user message in the prompt.
-   * @param sessionId - The thread/session ID
-   * @param parts - Array of DistriPart objects to append to user message
-   */
-  async setAdditionalUserParts(sessionId, parts) {
-    await this.setSessionValue(sessionId, _DistriClient.ADDITIONAL_PARTS_KEY, parts);
-  }
-  /**
-   * Get the current additional user message parts.
-   * @param sessionId - The thread/session ID
-   * @returns Array of DistriPart objects or null if not set
-   */
-  async getAdditionalUserParts(sessionId) {
-    return this.getSessionValue(sessionId, _DistriClient.ADDITIONAL_PARTS_KEY);
-  }
-  /**
-   * Clear/delete the additional user message parts.
-   * @param sessionId - The thread/session ID
-   */
-  async clearAdditionalUserParts(sessionId) {
-    await this.deleteSessionValue(sessionId, _DistriClient.ADDITIONAL_PARTS_KEY);
-  }
   /**
    * Issue an access token + refresh token for temporary authentication.
    * Requires an existing authenticated session (bearer token).
@@ -1086,7 +1081,7 @@ var _DistriClient = class _DistriClient {
     if (!tokens?.access_token || !tokens?.refresh_token || typeof tokens?.expires_at !== "number") {
       throw new ApiError("Invalid token response", response.status);
     }
-    this.applyTokens(tokens.access_token, tokens.refresh_token, false);
+    this.applyTokens(tokens.access_token, tokens.refresh_token);
     return tokens;
   }
   /**
@@ -1099,13 +1094,18 @@ var _DistriClient = class _DistriClient {
    * Update the access/refresh tokens in memory.
    */
   setTokens(tokens) {
-    if (tokens.accessToken !== void 0) {
-      this.accessToken = tokens.accessToken;
-    }
-    if (tokens.refreshToken !== void 0) {
+    this.accessToken = tokens.accessToken;
+    if (tokens.refreshToken) {
       this.refreshToken = tokens.refreshToken;
     }
   }
+  /**
+   * Reset all authentication tokens.
+   */
+  resetTokens() {
+    this.accessToken = void 0;
+    this.refreshToken = void 0;
+  }
   /**
    * Start streaming speech-to-text transcription via WebSocket
    */
@@ -1323,6 +1323,31 @@ var _DistriClient = class _DistriClient {
       throw new DistriError(`Failed to fetch agent ${agentId}`, "FETCH_ERROR", error);
     }
   }
+  /**
+   * Update an agent's definition (markdown only)
+   */
+  async updateAgent(agentId, update) {
+    try {
+      const response = await this.fetch(`/agents/${agentId}`, {
+        method: "PUT",
+        headers: {
+          "Content-Type": "application/json",
+          ...this.config.headers
+        },
+        body: JSON.stringify(update)
+      });
+      if (!response.ok) {
+        if (response.status === 404) {
+          throw new ApiError(`Agent not found: ${agentId}`, 404);
+        }
+        throw new ApiError(`Failed to update agent: ${response.statusText}`, response.status);
+      }
+      return await response.json();
+    } catch (error) {
+      if (error instanceof ApiError) throw error;
+      throw new DistriError(`Failed to update agent ${agentId}`, "UPDATE_ERROR", error);
+    }
+  }
   /**
    * Get or create A2AClient for an agent
    */
@@ -1408,11 +1433,22 @@ var _DistriClient = class _DistriClient {
     }
   }
   /**
-   * Get threads from Distri server
+   * Get threads from Distri server with filtering and pagination
    */
-  async getThreads() {
+  async getThreads(params = {}) {
     try {
-      const response = await this.fetch(`/threads`);
+      const searchParams = new URLSearchParams();
+      if (params.agent_id) searchParams.set("agent_id", params.agent_id);
+      if (params.external_id) searchParams.set("external_id", params.external_id);
+      if (params.search) searchParams.set("search", params.search);
+      if (params.from_date) searchParams.set("from_date", params.from_date);
+      if (params.to_date) searchParams.set("to_date", params.to_date);
+      if (params.tags?.length) searchParams.set("tags", params.tags.join(","));
+      if (params.limit !== void 0) searchParams.set("limit", params.limit.toString());
+      if (params.offset !== void 0) searchParams.set("offset", params.offset.toString());
+      const queryString = searchParams.toString();
+      const url = queryString ? `/threads?${queryString}` : "/threads";
+      const response = await this.fetch(url);
       if (!response.ok) {
         throw new ApiError(`Failed to fetch threads: ${response.statusText}`, response.status);
       }
@@ -1422,6 +1458,39 @@ var _DistriClient = class _DistriClient {
       throw new DistriError("Failed to fetch threads", "FETCH_ERROR", error);
     }
   }
+  /**
+   * Get agents sorted by thread count (most active first)
+   */
+  async getAgentsByUsage() {
+    try {
+      const response = await this.fetch("/threads/agents");
+      if (!response.ok) {
+        throw new ApiError(`Failed to fetch agents by usage: ${response.statusText}`, response.status);
+      }
+      return await response.json();
+    } catch (error) {
+      if (error instanceof ApiError) throw error;
+      throw new DistriError("Failed to fetch agents by usage", "FETCH_ERROR", error);
+    }
+  }
+  /**
+   * Create a new browser session
+   * Returns session info including viewer_url and stream_url from browsr
+   */
+  async createBrowserSession() {
+    try {
+      const response = await this.fetch("/browser/session", {
+        method: "POST"
+      });
+      if (!response.ok) {
+        throw new ApiError(`Failed to create browser session: ${response.statusText}`, response.status);
+      }
+      return await response.json();
+    } catch (error) {
+      if (error instanceof ApiError) throw error;
+      throw new DistriError("Failed to create browser session", "FETCH_ERROR", error);
+    }
+  }
   async getThread(threadId) {
     try {
       const response = await this.fetch(`/threads/${threadId}`);
@@ -1521,15 +1590,17 @@ var _DistriClient = class _DistriClient {
   get baseUrl() {
     return this.config.baseUrl;
   }
-  applyTokens(accessToken, refreshToken, notify) {
+  applyTokens(accessToken, refreshToken) {
     this.accessToken = accessToken;
-    this.refreshToken = refreshToken;
-    if (notify && this.onTokenRefresh) {
-      this.onTokenRefresh({ accessToken, refreshToken });
+    if (refreshToken) {
+      this.refreshToken = refreshToken;
     }
   }
+  /**
+   * Ensure access token is valid, refreshing if necessary
+   */
   async ensureAccessToken() {
-    if (!this.refreshToken) {
+    if (!this.refreshToken && !this.onTokenRefresh) {
       return;
     }
     if (!this.accessToken || this.isTokenExpiring(this.accessToken)) {
@@ -1541,7 +1612,7 @@ var _DistriClient = class _DistriClient {
     }
   }
   async refreshTokens() {
-    if (!this.refreshToken) {
+    if (!this.refreshToken && !this.onTokenRefresh) {
       return;
     }
     if (!this.refreshPromise) {
@@ -1552,6 +1623,17 @@ var _DistriClient = class _DistriClient {
     return this.refreshPromise;
   }
   async performTokenRefresh() {
+    if (this.onTokenRefresh) {
+      this.accessToken = void 0;
+      const newToken = await this.onTokenRefresh();
+      if (newToken) {
+        this.applyTokens(newToken);
+        return;
+      }
+    }
+    if (!this.refreshToken) {
+      return;
+    }
     const response = await this.fetchAbsolute(
       `${this.config.baseUrl}/token`,
       {
@@ -1575,7 +1657,7 @@ var _DistriClient = class _DistriClient {
     if (!tokens?.access_token || !tokens?.refresh_token) {
       throw new ApiError("Invalid token response", response.status);
     }
-    this.applyTokens(tokens.access_token, tokens.refresh_token, true);
+    this.applyTokens(tokens.access_token, tokens.refresh_token);
   }
   isTokenExpiring(token) {
     const expiresAt = this.getTokenExpiry(token);
@@ -1670,7 +1752,7 @@ var _DistriClient = class _DistriClient {
           headers
         });
         clearTimeout(timeoutId);
-        if (!skipAuth && retryOnAuth && response.status === 401 && this.refreshToken) {
+        if (!skipAuth && retryOnAuth && response.status === 401 && (this.refreshToken || this.onTokenRefresh)) {
           const refreshed = await this.refreshTokens().then(() => true).catch(() => false);
           if (refreshed) {
             return this.fetchAbsolute(url, initialInit, { skipAuth, retryOnAuth: false });
@@ -1762,13 +1844,6 @@ var _DistriClient = class _DistriClient {
   }
 };
 // ============================================================
-// Additional User Message Parts API
-// ============================================================
-// These methods allow external tools to append parts (text, images)
-// to the user message in the next agent iteration.
-// The parts are stored under the key "__additional_user_parts".
-_DistriClient.ADDITIONAL_PARTS_KEY = "__additional_user_parts";
-// ============================================================
 // Token API
 // ============================================================
 // Issue access + refresh tokens for temporary authentication (e.g., frontend use)