@augmentcode/auggie-sdk 0.1.14 → 0.1.15

package/dist/auggie/ai-sdk-provider.d.ts

@@ -25,6 +25,12 @@ interface AugmentLanguageModelConfig {
     apiUrl: string;
     /** Enable debug logging */
     debug?: boolean;
+    /**
+     * Custom User-Agent string to use instead of the default SDK User-Agent.
+     * If not provided, defaults to "augment.sdk.ai/{version} (typescript)".
+     * Example: "context-connectors/0.1.3 via:cli-agent"
+     */
+    clientUserAgent?: string;
 }
 /** Resolved credentials for Augment API (re-export for convenience) */
 type AugmentCredentials = ResolvedCredentials;
@@ -86,6 +92,7 @@ declare class AugmentLanguageModel implements LanguageModelV2 {
     private readonly apiUrl;
     private readonly sessionId;
     private readonly debug;
+    private readonly userAgent;
     constructor(modelId: string, config: AugmentLanguageModelConfig);
     private log;
     private getHeaders;

package/dist/auggie/ai-sdk-provider.js

@@ -1,4 +1,5 @@
 import { v4 as uuidv4 } from "uuid";
+import { getSDKVersion } from "../version.js";
 import { resolveCredentials } from "../context/internal/credentials.js";
 async function resolveAugmentCredentials() {
   return resolveCredentials();
@@ -233,12 +234,14 @@ class AugmentLanguageModel {
   apiUrl;
   sessionId;
   debug;
+  userAgent;
   constructor(modelId, config) {
     this.modelId = modelId;
     this.apiKey = config.apiKey;
     this.apiUrl = config.apiUrl.endsWith("/") ? config.apiUrl.slice(0, -1) : config.apiUrl;
     this.sessionId = uuidv4();
     this.debug = config.debug ?? false;
+    this.userAgent = config.clientUserAgent?.trim() || `augment.sdk.ai/${getSDKVersion()} (typescript)`;
   }
   log(message) {
     if (this.debug) {
@@ -253,7 +256,7 @@ class AugmentLanguageModel {
       "X-Request-Id": requestId,
       "conversation-id": this.sessionId,
       "X-Mode": "sdk",
-      "User-Agent": "auggie-sdk/1.0.0 (typescript)"
+      "User-Agent": this.userAgent
     };
   }
   buildPayload(options) {

package/dist/context/direct-context.js

@@ -68,7 +68,7 @@ class DirectContext {
    */
   static async create(options = {}) {
     const { apiKey, apiUrl } = await resolveCredentials(options);
-    return new DirectContext(apiKey, apiUrl, options.debug ?? false);
+    return new DirectContext(apiKey, apiUrl, options.debug ?? false, options.clientUserAgent);
   }
   /**
    * Import a DirectContext instance from a saved state object
@@ -79,7 +79,7 @@ class DirectContext {
    */
   static async import(state, options = {}) {
     const { apiKey, apiUrl } = await resolveCredentials(options);
-    const instance = new DirectContext(apiKey, apiUrl, options.debug ?? false);
+    const instance = new DirectContext(apiKey, apiUrl, options.debug ?? false, options.clientUserAgent);
     await instance.doImport(state);
     return instance;
   }
@@ -101,12 +101,13 @@ class DirectContext {
    * @param apiUrl API URL for the tenant
    * @param debug Enable debug logging
    */
-  constructor(apiKey, apiUrl, debug) {
+  constructor(apiKey, apiUrl, debug, clientUserAgent) {
     this.debug = debug;
     this.apiClient = new ContextAPIClient({
       apiKey,
       apiUrl,
-      debug
+      debug,
+      clientUserAgent
     });
     this.blobCalculator = new BlobNameCalculator(
       DirectContext.MAX_FILE_SIZE_BYTES

package/dist/context/filesystem-context.d.ts

@@ -16,6 +16,7 @@ declare class FileSystemContext {
     private readonly debug;
     private readonly apiKey;
     private readonly apiUrl;
+    private readonly clientUserAgent;
     private mcpClient;
     private apiClient;
     /**

package/dist/context/filesystem-context.js

@@ -9,6 +9,7 @@ class FileSystemContext {
   debug;
   apiKey;
   apiUrl;
+  clientUserAgent;
   mcpClient = null;
   apiClient = null;
   /**
@@ -23,6 +24,7 @@ class FileSystemContext {
     this.debug = options.debug ?? false;
     this.apiKey = apiKey;
     this.apiUrl = apiUrl;
+    this.clientUserAgent = options.clientUserAgent;
   }
   /**
    * Create and initialize a new FileSystemContext instance
@@ -172,7 +174,8 @@ class FileSystemContext {
       this.apiClient = new ContextAPIClient({
         apiKey: this.apiKey,
         apiUrl: this.apiUrl,
-        debug: this.debug
+        debug: this.debug,
+        clientUserAgent: this.clientUserAgent
       });
     }
     return await chatWithRetry(this.apiClient, llmPrompt, this.debug);

package/dist/context/internal/api-client.d.ts

@@ -37,11 +37,13 @@ type ContextAPIClientOptions = {
     apiKey: string;
     apiUrl: string;
     debug?: boolean;
+    /**
+     * Custom User-Agent string to use instead of the default SDK User-Agent.
+     * If not provided, defaults to "augment.sdk.context/{version} (typescript)".
+     * Example: "context-connectors/0.1.3 via:cli-search"
+     */
+    clientUserAgent?: string;
 };
-/**
- * Get user agent string for Context API requests
- */
-declare function getUserAgent(): string;
 /**
  * API client for Context operations
  */
@@ -50,6 +52,7 @@ declare class ContextAPIClient {
     private readonly apiUrl;
     private readonly sessionId;
     private readonly debug;
+    private readonly userAgent;
     constructor(options: ContextAPIClientOptions);
     private log;
     createRequestId(): string;
@@ -76,4 +79,4 @@ declare class ContextAPIClient {
     chat(prompt: string): Promise<string>;
 }
 
-export { APIError, type AgentCodebaseRetrievalResult, type BatchUploadResult, type CheckpointBlobsResult, ContextAPIClient, type ContextAPIClientOptions, type FindMissingResult, type UploadBlob, getUserAgent };
+export { APIError, type AgentCodebaseRetrievalResult, type BatchUploadResult, type CheckpointBlobsResult, ContextAPIClient, type ContextAPIClientOptions, type FindMissingResult, type UploadBlob };

package/dist/context/internal/api-client.js

@@ -11,19 +11,18 @@ class APIError extends Error {
     this.statusText = statusText;
   }
 }
-function getUserAgent() {
-  return `augment.sdk.context/${getSDKVersion()} (typescript)`;
-}
 class ContextAPIClient {
   apiKey;
   apiUrl;
   sessionId;
   debug;
+  userAgent;
   constructor(options) {
     this.apiKey = options.apiKey;
     this.apiUrl = options.apiUrl;
     this.sessionId = uuidv4();
     this.debug = options.debug ?? false;
+    this.userAgent = options.clientUserAgent?.trim() || `augment.sdk.context/${getSDKVersion()} (typescript)`;
   }
   log(message) {
     if (this.debug) {
@@ -46,7 +45,7 @@ class ContextAPIClient {
         Authorization: `Bearer ${this.apiKey}`,
         "X-Request-Session-Id": this.sessionId,
         "X-Request-Id": requestId,
-        "User-Agent": getUserAgent()
+        "User-Agent": this.userAgent
       },
       body: JSON.stringify(payload)
     });
@@ -219,7 +218,7 @@ class ContextAPIClient {
           "X-Request-Id": requestId,
           "conversation-id": this.sessionId,
           "X-Mode": "sdk",
-          "User-Agent": getUserAgent()
+          "User-Agent": this.userAgent
         },
         body: JSON.stringify(payload)
       });
@@ -242,6 +241,5 @@ class ContextAPIClient {
 }
 export {
   APIError,
-  ContextAPIClient,
-  getUserAgent
+  ContextAPIClient
 };

package/dist/context/types.d.ts

@@ -162,6 +162,12 @@ type DirectContextOptions = {
      * Default: false
      */
     debug?: boolean;
+    /**
+     * Custom User-Agent string to use instead of the default SDK User-Agent.
+     * If not provided, defaults to "augment.sdk.context/{version} (typescript)".
+     * Example: "context-connectors/0.1.3 via:cli-search"
+     */
+    clientUserAgent?: string;
 };
 /**
  * Full state for Direct Context with all blob information
@@ -227,6 +233,12 @@ type FileSystemContextOptions = {
     auggiePath?: string;
     /** Enable debug logging */
     debug?: boolean;
+    /**
+     * Custom User-Agent string to use instead of the default SDK User-Agent.
+     * If not provided, defaults to "augment.sdk.context/{version} (typescript)".
+     * Example: "context-connectors/0.1.3 via:mcp"
+     */
+    clientUserAgent?: string;
 };
 
 export type { AddToIndexOptions, BlobEntry, BlobInfo, Blobs, Chunk, DirectContextOptions, DirectContextState, ExportOptions, File, FileSystemContextOptions, FullContextState, IndexingProgress, IndexingResult, SearchOnlyContextState, SearchResult };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@augmentcode/auggie-sdk",
-  "version": "0.1.14",
+  "version": "0.1.15",
   "description": "TypeScript SDK for Auggie",
   "type": "module",
   "main": "./dist/index.js",