@augmentcode/auggie-sdk 0.1.9 → 0.1.11

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

@@ -1,98 +1,111 @@
-/**
- * Mock implementation of ContextAPIClient for testing
- */
-import { ContextAPIClient, } from "../api-client";
-export class MockContextAPIClient extends ContextAPIClient {
-    constructor(config = {}) {
-        // Call parent constructor with dummy credentials
-        super({
-            apiKey: "mock-api-key",
-            apiUrl: "https://mock.api.augmentcode.com",
-            debug: false,
-        });
-        // Track method calls for assertions
-        this.findMissingCalls = [];
-        this.batchUploadCalls = [];
-        this.checkpointBlobsCalls = [];
-        this.agentCodebaseRetrievalCalls = [];
-        this.chatCalls = [];
-        this.knownBlobs = config.knownBlobs || new Set();
-        this.indexedBlobs = config.indexedBlobs || new Set();
-        this.delay = config.delay || 0;
-        this.shouldFailFindMissing = config.shouldFailFindMissing ?? false;
-        this.shouldFailBatchUpload = config.shouldFailBatchUpload ?? false;
-        this.shouldFailCheckpoint = config.shouldFailCheckpoint ?? false;
+import {
+  ContextAPIClient
+} from "../api-client.js";
+class MockContextAPIClient extends ContextAPIClient {
+  knownBlobs;
+  indexedBlobs;
+  delay;
+  shouldFailFindMissing;
+  shouldFailBatchUpload;
+  shouldFailCheckpoint;
+  blobIdTransform;
+  // Track method calls for assertions
+  findMissingCalls = [];
+  batchUploadCalls = [];
+  checkpointBlobsCalls = [];
+  agentCodebaseRetrievalCalls = [];
+  chatCalls = [];
+  constructor(config = {}) {
+    super({
+      apiKey: "mock-api-key",
+      apiUrl: "https://mock.api.augmentcode.com",
+      debug: false
+    });
+    this.knownBlobs = config.knownBlobs || /* @__PURE__ */ new Set();
+    this.indexedBlobs = config.indexedBlobs || /* @__PURE__ */ new Set();
+    this.delay = config.delay || 0;
+    this.shouldFailFindMissing = config.shouldFailFindMissing ?? false;
+    this.shouldFailBatchUpload = config.shouldFailBatchUpload ?? false;
+    this.shouldFailCheckpoint = config.shouldFailCheckpoint ?? false;
+    this.blobIdTransform = config.blobIdTransform;
+  }
+  createRequestId() {
+    return `mock-req-${Date.now()}`;
+  }
+  async simulateDelay() {
+    if (this.delay > 0) {
+      await new Promise((resolve) => setTimeout(resolve, this.delay));
     }
-    createRequestId() {
-        return `mock-req-${Date.now()}`;
+  }
+  async findMissing(blobNames) {
+    this.findMissingCalls.push([...blobNames]);
+    if (this.shouldFailFindMissing) {
+      throw new Error("Mock findMissing error");
     }
-    async simulateDelay() {
-        if (this.delay > 0) {
-            await new Promise((resolve) => setTimeout(resolve, this.delay));
-        }
+    await this.simulateDelay();
+    const unknownBlobNames = blobNames.filter(
+      (name) => !this.knownBlobs.has(name)
+    );
+    const nonindexedBlobNames = blobNames.filter(
+      (name) => this.knownBlobs.has(name) && !this.indexedBlobs.has(name)
+    );
+    return {
+      unknownBlobNames,
+      nonindexedBlobNames
+    };
+  }
+  async batchUpload(blobs) {
+    this.batchUploadCalls.push([...blobs]);
+    if (this.shouldFailBatchUpload) {
+      throw new Error("Mock batchUpload error");
     }
-    async findMissing(blobNames) {
-        this.findMissingCalls.push([...blobNames]);
-        if (this.shouldFailFindMissing) {
-            throw new Error("Mock findMissing error");
-        }
-        await this.simulateDelay();
-        const unknownBlobNames = blobNames.filter((name) => !this.knownBlobs.has(name));
-        const nonindexedBlobNames = blobNames.filter((name) => this.knownBlobs.has(name) && !this.indexedBlobs.has(name));
-        return {
-            unknownBlobNames,
-            nonindexedBlobNames,
-        };
+    await this.simulateDelay();
+    const blobNames = blobs.map(
+      (b) => this.blobIdTransform ? this.blobIdTransform(b.blobName) : b.blobName
+    );
+    for (const blobName of blobNames) {
+      this.knownBlobs.add(blobName);
     }
-    async batchUpload(blobs) {
-        this.batchUploadCalls.push([...blobs]);
-        if (this.shouldFailBatchUpload) {
-            throw new Error("Mock batchUpload error");
-        }
-        await this.simulateDelay();
-        // Add uploaded blobs to known blobs (but not indexed yet)
-        const blobNames = blobs.map((b) => b.blobName);
-        for (const blobName of blobNames) {
-            this.knownBlobs.add(blobName);
-        }
-        return { blobNames };
+    return { blobNames };
+  }
+  async checkpointBlobs(blobs) {
+    this.checkpointBlobsCalls.push({ ...blobs });
+    if (this.shouldFailCheckpoint) {
+      throw new Error("Mock checkpointBlobs error");
     }
-    async checkpointBlobs(blobs) {
-        this.checkpointBlobsCalls.push({ ...blobs });
-        if (this.shouldFailCheckpoint) {
-            throw new Error("Mock checkpointBlobs error");
-        }
-        await this.simulateDelay();
-        return {
-            newCheckpointId: `checkpoint-${Date.now()}`,
-        };
-    }
-    async agentCodebaseRetrieval(query, blobs, maxOutputLength) {
-        this.agentCodebaseRetrievalCalls.push({ query, blobs, maxOutputLength });
-        await this.simulateDelay();
-        return {
-            formattedRetrieval: `Mock search results for: ${query}`,
-        };
-    }
-    async chat(prompt) {
-        this.chatCalls.push(prompt);
-        await this.simulateDelay();
-        return `Mock LLM response to: ${prompt}`;
-    }
-    // Helper methods for test setup
-    /** Mark blobs as indexed (simulates backend indexing completion) */
-    markAsIndexed(blobNames) {
-        for (const blobName of blobNames) {
-            this.indexedBlobs.add(blobName);
-        }
-    }
-    /** Reset all call tracking */
-    resetCalls() {
-        this.findMissingCalls = [];
-        this.batchUploadCalls = [];
-        this.checkpointBlobsCalls = [];
-        this.agentCodebaseRetrievalCalls = [];
-        this.chatCalls = [];
+    await this.simulateDelay();
+    return {
+      newCheckpointId: `checkpoint-${Date.now()}`
+    };
+  }
+  async agentCodebaseRetrieval(query, blobs, maxOutputLength) {
+    this.agentCodebaseRetrievalCalls.push({ query, blobs, maxOutputLength });
+    await this.simulateDelay();
+    return {
+      formattedRetrieval: `Mock search results for: ${query}`
+    };
+  }
+  async chat(prompt) {
+    this.chatCalls.push(prompt);
+    await this.simulateDelay();
+    return `Mock LLM response to: ${prompt}`;
+  }
+  // Helper methods for test setup
+  /** Mark blobs as indexed (simulates backend indexing completion) */
+  markAsIndexed(blobNames) {
+    for (const blobName of blobNames) {
+      this.indexedBlobs.add(blobName);
     }
+  }
+  /** Reset all call tracking */
+  resetCalls() {
+    this.findMissingCalls = [];
+    this.batchUploadCalls = [];
+    this.checkpointBlobsCalls = [];
+    this.agentCodebaseRetrievalCalls = [];
+    this.chatCalls = [];
+  }
 }
-//# sourceMappingURL=api-client.js.map
+export {
+  MockContextAPIClient
+};

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

@@ -1,37 +1,39 @@
+import { Blobs } from '../types.js';
+
 /**
  * API client for Context operations
  * Handles both indexing endpoints and LLM chat endpoint
  */
-import type { Blobs } from "../types";
+
 /**
  * Error thrown when an API request fails
  * Includes the HTTP status code to enable retry logic for transient failures
  */
-export declare class APIError extends Error {
+declare class APIError extends Error {
     readonly status: number;
     readonly statusText: string;
     constructor(status: number, statusText: string, message: string);
 }
-export type UploadBlob = {
+type UploadBlob = {
     blobName: string;
     pathName: string;
     text: string;
     metadata: string[];
 };
-export type FindMissingResult = {
+type FindMissingResult = {
     unknownBlobNames: string[];
     nonindexedBlobNames: string[];
 };
-export type CheckpointBlobsResult = {
+type CheckpointBlobsResult = {
     newCheckpointId: string;
 };
-export type BatchUploadResult = {
+type BatchUploadResult = {
     blobNames: string[];
 };
-export type AgentCodebaseRetrievalResult = {
+type AgentCodebaseRetrievalResult = {
     formattedRetrieval: string;
 };
-export type ContextAPIClientOptions = {
+type ContextAPIClientOptions = {
     apiKey: string;
     apiUrl: string;
     debug?: boolean;
@@ -39,11 +41,11 @@ export type ContextAPIClientOptions = {
 /**
  * Get user agent string for Context API requests
  */
-export declare function getUserAgent(): string;
+declare function getUserAgent(): string;
 /**
  * API client for Context operations
  */
-export declare class ContextAPIClient {
+declare class ContextAPIClient {
     private readonly apiKey;
     private readonly apiUrl;
     private readonly sessionId;
@@ -60,12 +62,9 @@ export declare class ContextAPIClient {
     batchUpload(blobs: UploadBlob[]): Promise<BatchUploadResult>;
     checkpointBlobs(blobs: Blobs): Promise<CheckpointBlobsResult>;
     agentCodebaseRetrieval(query: string, blobs: Blobs, maxOutputLength?: number): Promise<AgentCodebaseRetrievalResult>;
-    /**
-     * Parse a single JSON line from the stream and extract text if present
-     */
-    private parseStreamLine;
     /**
      * Parse streaming response and accumulate text chunks
+     * Uses a buffer to handle JSON objects that span multiple lines or chunks
      */
     private parseSSEStream;
     /**
@@ -76,4 +75,5 @@ export declare class ContextAPIClient {
      */
     chat(prompt: string): Promise<string>;
 }
-//# sourceMappingURL=api-client.d.ts.map
+
+export { APIError, type AgentCodebaseRetrievalResult, type BatchUploadResult, type CheckpointBlobsResult, ContextAPIClient, type ContextAPIClientOptions, type FindMissingResult, type UploadBlob, getUserAgent };