@augmentcode/auggie-sdk 0.1.10 → 0.1.12

package/dist/context/internal/credentials.js

@@ -1,40 +1,26 @@
-import { readSessionFile } from "./session-reader";
-/**
- * Resolve API credentials from options, environment, or session file
- *
- * Priority order:
- * 1. options.apiKey and options.apiUrl (if provided)
- * 2. AUGMENT_API_TOKEN and AUGMENT_API_URL environment variables
- * 3. ~/.augment/session.json (created by `auggie login`)
- *
- * @param options - Optional credential options
- * @returns Resolved credentials
- * @throws Error if credentials cannot be resolved
- */
-export async function resolveCredentials(options = {}) {
-    let apiKey = options.apiKey || process.env.AUGMENT_API_TOKEN;
-    let apiUrl = options.apiUrl || process.env.AUGMENT_API_URL;
-    // If credentials not provided, try session.json
-    if (!(apiKey && apiUrl)) {
-        const session = await readSessionFile();
-        if (session) {
-            apiKey = apiKey || session.accessToken;
-            apiUrl = apiUrl || session.tenantURL;
-        }
+import { readSessionFile } from "./session-reader.js";
+async function resolveCredentials(options = {}) {
+  let apiKey = options.apiKey || process.env.AUGMENT_API_TOKEN;
+  let apiUrl = options.apiUrl || process.env.AUGMENT_API_URL;
+  if (!(apiKey && apiUrl)) {
+    const session = await readSessionFile();
+    if (session) {
+      apiKey = apiKey || session.accessToken;
+      apiUrl = apiUrl || session.tenantURL;
     }
-    // Validate we have credentials
-    if (!apiKey) {
-        throw new Error("API key is required. Provide it via:\n" +
-            "1. options.apiKey parameter\n" +
-            "2. AUGMENT_API_TOKEN environment variable\n" +
-            "3. Run 'auggie login' to create ~/.augment/session.json");
-    }
-    if (!apiUrl) {
-        throw new Error("API URL is required. Provide it via:\n" +
-            "1. options.apiUrl parameter\n" +
-            "2. AUGMENT_API_URL environment variable\n" +
-            "3. Run 'auggie login' to create ~/.augment/session.json");
-    }
-    return { apiKey, apiUrl };
+  }
+  if (!apiKey) {
+    throw new Error(
+      "API key is required. Provide it via:\n1. options.apiKey parameter\n2. AUGMENT_API_TOKEN environment variable\n3. Run 'auggie login' to create ~/.augment/session.json"
+    );
+  }
+  if (!apiUrl) {
+    throw new Error(
+      "API URL is required. Provide it via:\n1. options.apiUrl parameter\n2. AUGMENT_API_URL environment variable\n3. Run 'auggie login' to create ~/.augment/session.json"
+    );
+  }
+  return { apiKey, apiUrl };
 }
-//# sourceMappingURL=credentials.js.map
+export {
+  resolveCredentials
+};

package/dist/context/internal/retry-utils.d.ts

@@ -2,7 +2,7 @@
  * Retry utilities with exponential backoff
  * Based on clients/sidecar/libs/src/utils/promise-utils.ts
  */
-export type BackoffParams = {
+type BackoffParams = {
     initialMS: number;
     mult: number;
     maxMS: number;
@@ -18,7 +18,7 @@ export type BackoffParams = {
  * @param backoffParams Backoff parameters
  * @returns Promise that resolves to the function's return value
  */
-export declare function retryWithBackoff<T>(fn: () => Promise<T>, debug?: boolean, backoffParams?: BackoffParams): Promise<T>;
+declare function retryWithBackoff<T>(fn: () => Promise<T>, debug?: boolean, backoffParams?: BackoffParams): Promise<T>;
 /**
  * Retry a chat function with exponential backoff
  * Uses chat-specific retry logic that includes 429 and 529 status codes
@@ -28,5 +28,6 @@ export declare function retryWithBackoff<T>(fn: () => Promise<T>, debug?: boolea
  * @param backoffParams Backoff parameters (defaults to chat-optimized settings)
  * @returns Promise that resolves to the function's return value
  */
-export declare function retryChat<T>(fn: () => Promise<T>, debug?: boolean, backoffParams?: BackoffParams): Promise<T>;
-//# sourceMappingURL=retry-utils.d.ts.map
+declare function retryChat<T>(fn: () => Promise<T>, debug?: boolean, backoffParams?: BackoffParams): Promise<T>;
+
+export { type BackoffParams, retryChat, retryWithBackoff };

package/dist/context/internal/retry-utils.js

@@ -1,125 +1,71 @@
-/**
- * Retry utilities with exponential backoff
- * Based on clients/sidecar/libs/src/utils/promise-utils.ts
- */
 const defaultBackoffParams = {
-    initialMS: 100,
-    mult: 2,
-    maxMS: 30000,
+  initialMS: 100,
+  mult: 2,
+  maxMS: 3e4
 };
-/**
- * Delay for the specified number of milliseconds
- */
 function delayMs(ms) {
-    if (ms === 0) {
-        return Promise.resolve();
-    }
-    return new Promise((resolve) => {
-        setTimeout(resolve, ms);
-    });
+  if (ms === 0) {
+    return Promise.resolve();
+  }
+  return new Promise((resolve) => {
+    setTimeout(resolve, ms);
+  });
 }
-/**
- * Check if an error is retriable based on its properties
- * Retriable errors include network errors and server errors (5xx)
- */
 function isRetriableError(e) {
-    // Check for HTTP status codes
-    if (e && typeof e === "object" && "status" in e) {
-        const status = e.status;
-        // Match the default VSCode client retry logic:
-        // - 499 (cancelled)
-        // - 503 (unavailable)
-        // - 5xx (unavailable)
-        //
-        // Note: We do NOT retry 429 or 504 by default
-        // (only chat streams retry those)
-        return status === 499 || status === 503 || (status >= 500 && status < 600);
-    }
-    return false;
+  if (e && typeof e === "object" && "status" in e) {
+    const status = e.status;
+    return status === 499 || status === 503 || status >= 500 && status < 600;
+  }
+  return false;
 }
-/**
- * Check if an error is retriable for chat streaming endpoints
- * Chat streams retry additional status codes beyond the default retry logic
- */
 function isChatRetriableError(e) {
-    // Check for HTTP status codes
-    if (e && typeof e === "object" && "status" in e) {
-        const status = e.status;
-        // Chat streams retry:
-        // - 429 (rate limit)
-        // - 499 (cancelled)
-        // - 503 (unavailable)
-        // - 529 (overloaded)
-        // - 5xx (server errors)
-        return (status === 429 ||
-            status === 499 ||
-            status === 503 ||
-            status === 529 ||
-            (status >= 500 && status < 600));
-    }
-    return false;
+  if (e && typeof e === "object" && "status" in e) {
+    const status = e.status;
+    return status === 429 || status === 499 || status === 503 || status === 529 || status >= 500 && status < 600;
+  }
+  return false;
 }
-/**
- * Retry a function with exponential backoff
- *
- * @param fn The function to retry
- * @param debug Whether to log debug messages
- * @param backoffParams Backoff parameters
- * @returns Promise that resolves to the function's return value
- */
-export async function retryWithBackoff(fn, debug = false, backoffParams = defaultBackoffParams) {
-    const canRetryFn = backoffParams.canRetry ?? isRetriableError;
-    let backoffMs = 0;
-    const startTime = Date.now();
-    for (let tries = 0;; tries++) {
-        try {
-            const ret = await fn();
-            if (tries > 0 && debug) {
-                console.log(`[RetryUtils] Operation succeeded after ${tries} transient failures`);
-            }
-            return ret;
-        }
-        catch (e) {
-            const currTryCount = tries + 1;
-            // Check if we have exceeded the max number of retries
-            if (backoffParams.maxTries !== undefined &&
-                currTryCount >= backoffParams.maxTries) {
-                throw e;
-            }
-            // Check if we should retry this error
-            if (!canRetryFn(e)) {
-                throw e;
-            }
-            // Calculate backoff delay
-            backoffMs =
-                backoffMs === 0
-                    ? backoffParams.initialMS
-                    : Math.min(backoffMs * backoffParams.mult, backoffParams.maxMS);
-            if (debug) {
-                console.log(`[RetryUtils] Operation failed with error ${e}, retrying in ${backoffMs} ms; retries = ${tries}`);
-            }
-            // Check if the backoff delay will exceed total time
-            if (backoffParams.maxTotalMs !== undefined &&
-                Date.now() - startTime + backoffMs > backoffParams.maxTotalMs) {
-                throw e;
-            }
-            await delayMs(backoffMs);
-        }
+async function retryWithBackoff(fn, debug = false, backoffParams = defaultBackoffParams) {
+  const canRetryFn = backoffParams.canRetry ?? isRetriableError;
+  let backoffMs = 0;
+  const startTime = Date.now();
+  for (let tries = 0; ; tries++) {
+    try {
+      const ret = await fn();
+      if (tries > 0 && debug) {
+        console.log(
+          `[RetryUtils] Operation succeeded after ${tries} transient failures`
+        );
+      }
+      return ret;
+    } catch (e) {
+      const currTryCount = tries + 1;
+      if (backoffParams.maxTries !== void 0 && currTryCount >= backoffParams.maxTries) {
+        throw e;
+      }
+      if (!canRetryFn(e)) {
+        throw e;
+      }
+      backoffMs = backoffMs === 0 ? backoffParams.initialMS : Math.min(backoffMs * backoffParams.mult, backoffParams.maxMS);
+      if (debug) {
+        console.log(
+          `[RetryUtils] Operation failed with error ${e}, retrying in ${backoffMs} ms; retries = ${tries}`
+        );
+      }
+      if (backoffParams.maxTotalMs !== void 0 && Date.now() - startTime + backoffMs > backoffParams.maxTotalMs) {
+        throw e;
+      }
+      await delayMs(backoffMs);
     }
+  }
 }
-/**
- * Retry a chat function with exponential backoff
- * Uses chat-specific retry logic that includes 429 and 529 status codes
- *
- * @param fn The function to retry
- * @param debug Whether to log debug messages
- * @param backoffParams Backoff parameters (defaults to chat-optimized settings)
- * @returns Promise that resolves to the function's return value
- */
-export function retryChat(fn, debug = false, backoffParams = defaultBackoffParams) {
-    return retryWithBackoff(fn, debug, {
-        ...backoffParams,
-        canRetry: isChatRetriableError,
-    });
+function retryChat(fn, debug = false, backoffParams = defaultBackoffParams) {
+  return retryWithBackoff(fn, debug, {
+    ...backoffParams,
+    canRetry: isChatRetriableError
+  });
 }
-//# sourceMappingURL=retry-utils.js.map
+export {
+  retryChat,
+  retryWithBackoff
+};

package/dist/context/internal/search-utils.d.ts

@@ -4,5 +4,6 @@
 /**
  * Format a question and search results into a prompt for the LLM
  */
-export declare function formatSearchPrompt(question: string, results: string): string;
-//# sourceMappingURL=search-utils.d.ts.map
+declare function formatSearchPrompt(question: string, results: string): string;
+
+export { formatSearchPrompt };

package/dist/context/internal/search-utils.js

@@ -1,10 +1,9 @@
-/**
- * Shared utilities for search functionality across context modes
- */
-/**
- * Format a question and search results into a prompt for the LLM
- */
-export function formatSearchPrompt(question, results) {
-    return `Relevant context:\n${results}\n\n${question}`;
+function formatSearchPrompt(question, results) {
+  return `Relevant context:
+${results}
+
+${question}`;
 }
-//# sourceMappingURL=search-utils.js.map
+export {
+  formatSearchPrompt
+};

package/dist/context/internal/session-reader.d.ts

@@ -4,7 +4,7 @@
 /**
  * Structure of the session.json file created by `auggie login`
  */
-export type SessionData = {
+type SessionData = {
     accessToken: string;
     tenantURL: string;
     scopes?: string[];
@@ -13,5 +13,6 @@ export type SessionData = {
  * Read session data from ~/.augment/session.json
  * Returns null if the file doesn't exist or can't be read
  */
-export declare function readSessionFile(): Promise<SessionData | null>;
-//# sourceMappingURL=session-reader.d.ts.map
+declare function readSessionFile(): Promise<SessionData | null>;
+
+export { type SessionData, readSessionFile };

package/dist/context/internal/session-reader.js

@@ -1,27 +1,19 @@
-/**
- * Utility for reading Augment session file
- */
 import { readFile } from "node:fs/promises";
 import { homedir } from "node:os";
 import { join } from "node:path";
-/**
- * Read session data from ~/.augment/session.json
- * Returns null if the file doesn't exist or can't be read
- */
-export async function readSessionFile() {
-    try {
-        const sessionPath = join(homedir(), ".augment", "session.json");
-        const content = await readFile(sessionPath, "utf-8");
-        const data = JSON.parse(content);
-        // Validate required fields
-        if (!(data.accessToken && data.tenantURL)) {
-            return null;
-        }
-        return data;
-    }
-    catch {
-        // File doesn't exist or can't be read
-        return null;
+async function readSessionFile() {
+  try {
+    const sessionPath = join(homedir(), ".augment", "session.json");
+    const content = await readFile(sessionPath, "utf-8");
+    const data = JSON.parse(content);
+    if (!(data.accessToken && data.tenantURL)) {
+      return null;
     }
+    return data;
+  } catch {
+    return null;
+  }
 }
-//# sourceMappingURL=session-reader.js.map
+export {
+  readSessionFile
+};

package/dist/context/types.d.ts

@@ -4,7 +4,7 @@
 /**
  * Represents a file with path and contents (source-agnostic)
  */
-export type File = {
+type File = {
     /** Relative path (e.g., "src/main.py") */
     path: string;
     /** File contents as string */
@@ -13,7 +13,7 @@ export type File = {
 /**
  * A single code chunk from the retrieval results
  */
-export type Chunk = {
+type Chunk = {
     /** File path relative to workspace root */
     path: string;
     /** Starting line number (1-based) */
@@ -26,7 +26,7 @@ export type Chunk = {
 /**
  * Result from a codebase search query
  */
-export type SearchResult = {
+type SearchResult = {
     /** Formatted retrieval results as markdown text */
     formattedRetrieval: string;
     /** Structured list of code chunks */
@@ -35,20 +35,28 @@ export type SearchResult = {
 /**
  * Blob information for a file
  */
-export type BlobInfo = {
+type BlobInfo = {
     /** SHA-256 hash of the file path and contents */
     blobName: string;
     /** Relative path from workspace root */
     relativePath: string;
 };
 /**
- * Blob entry in persistent state - tuple of [blobName, path]
+ * Blob entry in persistent state - tuple of [serverBlobId, path, clientBlobId?]
+ *
+ * The third element (clientBlobId) is optional for backwards compatibility.
+ * When present, it indicates the server computed a different blob ID than the client.
+ * When absent, assume clientBlobId === serverBlobId.
  */
-export type BlobEntry = [blobName: string, path: string];
+type BlobEntry = [
+    serverBlobId: string,
+    path: string,
+    clientBlobId?: string
+];
 /**
  * Blobs payload for API requests
  */
-export type Blobs = {
+type Blobs = {
     /** Optional checkpoint ID from previous indexing */
     checkpointId?: string;
     /** List of added blob names */
@@ -59,16 +67,85 @@ export type Blobs = {
 /**
  * Result from addToIndex operation
  */
-export type IndexingResult = {
+type IndexingResult = {
     /** Paths that were newly uploaded to the backend (queued for indexing) */
     newlyUploaded: string[];
     /** Paths that were already uploaded (content unchanged or already on server) */
     alreadyUploaded: string[];
 };
+/**
+ * Progress information for indexing operations
+ */
+type IndexingProgress = {
+    /**
+     * Current stage of the indexing operation
+     * - 'uploading': Files are being uploaded to the backend
+     * - 'indexing': Backend is processing and indexing the uploaded files
+     */
+    stage: "uploading" | "indexing";
+    /**
+     * Number of files uploaded so far
+     */
+    uploaded: number;
+    /**
+     * Number of files indexed so far
+     */
+    indexed: number;
+    /**
+     * Total number of files to upload and index
+     */
+    total: number;
+    /**
+     * Path of the file currently being processed (only available during upload stage)
+     */
+    currentFile?: string;
+    /**
+     * Total bytes uploaded so far (only available during upload stage)
+     */
+    bytesUploaded?: number;
+    /**
+     * When the operation started
+     */
+    startedAt: Date;
+};
+/**
+ * Options for addToIndex operation
+ */
+type AddToIndexOptions = {
+    /**
+     * If true (default), waits for the newly added files to be indexed before returning.
+     * If false, returns immediately after upload completes (indexing continues asynchronously).
+     * Default: true
+     */
+    waitForIndexing?: boolean;
+    /**
+     * Timeout in milliseconds for the indexing operation when waitForIndexing is true.
+     * If the backend does not finish indexing within this time, an error is thrown.
+     * Default: undefined (no timeout - waits indefinitely)
+     *
+     * Set this to a specific value (e.g., 600000 for 10 minutes) if you want to fail fast
+     * in case of backend issues or unexpectedly slow indexing.
+     */
+    timeout?: number;
+    /**
+     * Optional callback to receive progress updates during the indexing operation.
+     * Called periodically with information about the current stage, files uploaded/indexed, etc.
+     *
+     * @example
+     * ```typescript
+     * await context.addToIndex(files, {
+     *   onProgress: (progress) => {
+     *     console.log(`[${progress.stage}] Uploaded: ${progress.uploaded}/${progress.total}, Indexed: ${progress.indexed}/${progress.total}`);
+     *   }
+     * });
+     * ```
+     */
+    onProgress?: (progress: IndexingProgress) => void;
+};
 /**
  * Options for configuring the Direct Context
  */
-export type DirectContextOptions = {
+type DirectContextOptions = {
     /**
      * API key for authentication
      * Optional - falls back to AUGMENT_API_TOKEN env var, then ~/.augment/session.json
@@ -87,9 +164,11 @@ export type DirectContextOptions = {
     debug?: boolean;
 };
 /**
- * State for Direct Context that can be exported/imported
+ * Full state for Direct Context with all blob information
  */
-export type DirectContextState = {
+type FullContextState = {
+    /** Discriminator for state type */
+    mode: "full";
     /** Current checkpoint ID */
     checkpointId?: string;
     /** Array of blob names that have been added (pending checkpoint) */
@@ -99,10 +178,49 @@ export type DirectContextState = {
     /** List of blobs as [blobName, path] tuples */
     blobs: BlobEntry[];
 };
+/**
+ * Lightweight search-only state without blob information
+ *
+ * This state format is optimized for search-only use cases and reduces
+ * storage size for large codebases by excluding the blobs array.
+ *
+ * Use this when you only need to search the index and don't need to:
+ * - Add new files to the index
+ * - Remove files from the index
+ * - Get the list of indexed paths
+ */
+type SearchOnlyContextState = {
+    /** Discriminator for state type */
+    mode: "search-only";
+    /** Current checkpoint ID */
+    checkpointId?: string;
+    /** Array of blob names that have been added (pending checkpoint) */
+    addedBlobs: string[];
+    /** Array of blob names that have been deleted (pending checkpoint) */
+    deletedBlobs: string[];
+};
+/**
+ * State for Direct Context that can be exported/imported
+ *
+ * Can be either full state (with blobs array) or search-only state (without blobs array).
+ * For backwards compatibility, states without a mode field are treated as full state.
+ */
+type DirectContextState = FullContextState | SearchOnlyContextState;
+/**
+ * Options for exporting Direct Context state
+ */
+type ExportOptions = {
+    /**
+     * Export mode:
+     * - 'full': Include all blob information (default) - required for addToIndex/removeFromIndex
+     * - 'search-only': Exclude blob array for minimal storage - only supports search operations
+     */
+    mode?: "full" | "search-only";
+};
 /**
  * Options for FileSystem Context (MCP mode)
  */
-export type FileSystemContextOptions = {
+type FileSystemContextOptions = {
     /** Path to the workspace directory to index */
     directory: string;
     /** Path to auggie executable (default: "auggie") */
@@ -110,4 +228,5 @@ export type FileSystemContextOptions = {
     /** Enable debug logging */
     debug?: boolean;
 };
-//# sourceMappingURL=types.d.ts.map
+
+export type { AddToIndexOptions, BlobEntry, BlobInfo, Blobs, Chunk, DirectContextOptions, DirectContextState, ExportOptions, File, FileSystemContextOptions, FullContextState, IndexingProgress, IndexingResult, SearchOnlyContextState, SearchResult };

package/dist/context/types.js

@@ -1,5 +0,0 @@
-/**
- * Types for the Context SDK
- */
-export {};
-//# sourceMappingURL=types.js.map

package/dist/index.d.ts

@@ -1,7 +1,8 @@
-export { Auggie, type PredefinedToolType, type ToolIdentifier, } from "./auggie/sdk-acp-client";
-export { DirectContext } from "./context/direct-context";
-export { FileSystemContext } from "./context/filesystem-context";
-export { APIError } from "./context/internal/api-client";
-export { BlobTooLargeError } from "./context/internal/blob-name-calculator";
-export type { BlobEntry, BlobInfo, Blobs, DirectContextOptions, DirectContextState, File, FileSystemContextOptions, IndexingResult, } from "./context/types";
-//# sourceMappingURL=index.d.ts.map
+export { Auggie, PredefinedToolType, ToolIdentifier } from './auggie/sdk-acp-client.js';
+export { DirectContext } from './context/direct-context.js';
+export { FileSystemContext } from './context/filesystem-context.js';
+export { APIError } from './context/internal/api-client.js';
+export { BlobTooLargeError } from './context/internal/blob-name-calculator.js';
+export { AddToIndexOptions, BlobEntry, BlobInfo, Blobs, DirectContextOptions, DirectContextState, ExportOptions, File, FileSystemContextOptions, FullContextState, IndexingProgress, IndexingResult, SearchOnlyContextState } from './context/types.js';
+import '@agentclientprotocol/sdk';
+import 'ai';

package/dist/index.js

@@ -1,9 +1,14 @@
-// ACP Client (existing)
-export { Auggie, } from "./auggie/sdk-acp-client";
-// Context Modes
-export { DirectContext } from "./context/direct-context";
-export { FileSystemContext } from "./context/filesystem-context";
-// Context Errors
-export { APIError } from "./context/internal/api-client";
-export { BlobTooLargeError } from "./context/internal/blob-name-calculator";
-//# sourceMappingURL=index.js.map
+import {
+  Auggie
+} from "./auggie/sdk-acp-client.js";
+import { DirectContext } from "./context/direct-context.js";
+import { FileSystemContext } from "./context/filesystem-context.js";
+import { APIError } from "./context/internal/api-client.js";
+import { BlobTooLargeError } from "./context/internal/blob-name-calculator.js";
+export {
+  APIError,
+  Auggie,
+  BlobTooLargeError,
+  DirectContext,
+  FileSystemContext
+};

package/dist/version.d.ts

@@ -12,5 +12,6 @@
  *
  * @returns The version string from package.json, or "unknown" if it cannot be read
  */
-export declare function getSDKVersion(): string;
-//# sourceMappingURL=version.d.ts.map
+declare function getSDKVersion(): string;
+
+export { getSDKVersion };