@augmentcode/auggie-sdk 0.1.6 → 0.1.8

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

@@ -0,0 +1,84 @@
+import type { FileSystemContextOptions } from "./types";
+/**
+ * FileSystem Context - Local directory retrieval via MCP protocol
+ *
+ * ⚠️ **EXPERIMENTAL API** - This API is experimental and may change without notice.
+ *
+ * @experimental
+ *
+ * This class spawns `auggie --mcp` and communicates with it via the MCP protocol
+ * to provide codebase retrieval from a local directory.
+ */
+export declare class FileSystemContext {
+    private readonly auggiePath;
+    private readonly directory;
+    private readonly debug;
+    private readonly apiKey;
+    private readonly apiUrl;
+    private mcpClient;
+    /**
+     * Create a new FileSystemContext instance
+     * @param options Configuration options
+     */
+    constructor(options: FileSystemContextOptions);
+    /**
+     * Create and initialize a new FileSystemContext instance
+     */
+    static create(options: FileSystemContextOptions): Promise<FileSystemContext>;
+    /**
+     * Log a debug message if debug mode is enabled
+     */
+    private log;
+    /**
+     * Connect to Auggie MCP server
+     */
+    private connect;
+    /**
+     * Search the codebase using natural language and return formatted results.
+     *
+     * The results are returned as a formatted string designed for use in LLM prompts.
+     * The format includes file paths, line numbers, and code content in a structured,
+     * readable format that can be passed directly to LLM APIs like `generate()`.
+     *
+     * @param query - The search query describing what code you're looking for
+     * @returns A formatted string containing the search results, ready for LLM consumption
+     *
+     * @example
+     * ```typescript
+     * const context = await FileSystemContext.create({ directory: "./my-project" });
+     * const results = await context.search("authentication logic");
+     * console.log(results); // Formatted string with file paths, line numbers, and code
+     *
+     * // Use directly with generate API
+     * const response = await generate(
+     *   `Based on this code:\n\n${results}\n\nHow does authentication work?`,
+     *   { model: "sonnet4.5" }
+     * );
+     * ```
+     */
+    search(query: string): Promise<string>;
+    /**
+     * Search the indexed codebase and ask an LLM a question about the results.
+     *
+     * This is a convenience method that combines search() with an LLM call to answer
+     * questions about your codebase.
+     *
+     * @param searchQuery - The semantic search query to find relevant code (also used as the prompt if no separate prompt is provided)
+     * @param prompt - Optional prompt to ask the LLM about the search results. If not provided, searchQuery is used as the prompt.
+     * @returns The LLM's answer to your question
+     *
+     * @example
+     * ```typescript
+     * const answer = await context.searchAndAsk(
+     *   "How does the authentication flow work?"
+     * );
+     * console.log(answer);
+     * ```
+     */
+    searchAndAsk(searchQuery: string, prompt?: string): Promise<string>;
+    /**
+     * Close the MCP connection and cleanup resources
+     */
+    close(): Promise<void>;
+}
+//# sourceMappingURL=filesystem-context.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"filesystem-context.d.ts","sourceRoot":"","sources":["../../src/context/filesystem-context.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,SAAS,CAAC;AAcxD;;;;;;;;;GASG;AACH,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAU;IAChC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAqB;IAC5C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAqB;IAE5C,OAAO,CAAC,SAAS,CAA0B;IAE3C;;;OAGG;gBACS,OAAO,EAAE,wBAAwB;IAQ7C;;OAEG;WACU,MAAM,CACjB,OAAO,EAAE,wBAAwB,GAChC,OAAO,CAAC,iBAAiB,CAAC;IAM7B;;OAEG;IACH,OAAO,CAAC,GAAG;IAMX;;OAEG;IACH,OAAO,CAAC,OAAO;IA6Cf;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACG,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAyD5C;;;;;;;;;;;;;;;;;OAiBG;IACG,YAAY,CAAC,WAAW,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAsBzE;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAc7B"}

package/dist/context/filesystem-context.js

@@ -0,0 +1,198 @@
+import { MCPClient } from "@mastra/mcp";
+import { ContextAPIClient } from "./internal/api-client";
+import { formatSearchPrompt } from "./internal/search-utils";
+/**
+ * FileSystem Context - Local directory retrieval via MCP protocol
+ *
+ * ⚠️ **EXPERIMENTAL API** - This API is experimental and may change without notice.
+ *
+ * @experimental
+ *
+ * This class spawns `auggie --mcp` and communicates with it via the MCP protocol
+ * to provide codebase retrieval from a local directory.
+ */
+export class FileSystemContext {
+    /**
+     * Create a new FileSystemContext instance
+     * @param options Configuration options
+     */
+    constructor(options) {
+        this.mcpClient = null;
+        this.auggiePath = options.auggiePath || "auggie";
+        this.directory = options.directory;
+        this.debug = options.debug ?? false;
+        this.apiKey = process.env.AUGMENT_API_TOKEN;
+        this.apiUrl = process.env.AUGMENT_API_URL;
+    }
+    /**
+     * Create and initialize a new FileSystemContext instance
+     */
+    static async create(options) {
+        const instance = new FileSystemContext(options);
+        await Promise.resolve(instance.connect());
+        return instance;
+    }
+    /**
+     * Log a debug message if debug mode is enabled
+     */
+    log(message) {
+        if (this.debug) {
+            console.log(`[FileSystemContext] ${message}`);
+        }
+    }
+    /**
+     * Connect to Auggie MCP server
+     */
+    connect() {
+        this.log(`Starting MCP server for directory: ${this.directory}`);
+        const command = this.auggiePath.trim();
+        if (!command) {
+            throw new Error("Invalid auggiePath: cannot be empty");
+        }
+        const args = ["--mcp", "--workspace-root", this.directory];
+        // Build environment variables
+        const env = {};
+        for (const [key, value] of Object.entries(process.env)) {
+            if (value !== undefined) {
+                env[key] = value;
+            }
+        }
+        if (this.apiKey) {
+            env.AUGMENT_API_TOKEN = this.apiKey;
+        }
+        if (this.apiUrl) {
+            env.AUGMENT_API_URL = this.apiUrl;
+        }
+        // Create MCP client using MCPClient
+        this.mcpClient = new MCPClient({
+            servers: {
+                auggie: {
+                    command,
+                    args,
+                    env,
+                    logger: this.debug
+                        ? (logMessage) => {
+                            console.log(`[MCP ${logMessage.level}] ${logMessage.message}`, logMessage.details);
+                        }
+                        : undefined,
+                },
+            },
+        });
+        this.log("Connected to Auggie MCP server");
+    }
+    /**
+     * Search the codebase using natural language and return formatted results.
+     *
+     * The results are returned as a formatted string designed for use in LLM prompts.
+     * The format includes file paths, line numbers, and code content in a structured,
+     * readable format that can be passed directly to LLM APIs like `generate()`.
+     *
+     * @param query - The search query describing what code you're looking for
+     * @returns A formatted string containing the search results, ready for LLM consumption
+     *
+     * @example
+     * ```typescript
+     * const context = await FileSystemContext.create({ directory: "./my-project" });
+     * const results = await context.search("authentication logic");
+     * console.log(results); // Formatted string with file paths, line numbers, and code
+     *
+     * // Use directly with generate API
+     * const response = await generate(
+     *   `Based on this code:\n\n${results}\n\nHow does authentication work?`,
+     *   { model: "sonnet4.5" }
+     * );
+     * ```
+     */
+    async search(query) {
+        if (!this.mcpClient) {
+            throw new Error("MCP client not initialized. Call create() first.");
+        }
+        this.log(`Searching for: "${query}"`);
+        try {
+            // Get the tools from the MCP client
+            // getTools() returns tools with server name prefix (auggie_codebase-retrieval)
+            const tools = await this.mcpClient.getTools();
+            const codebaseRetrievalTool = tools["auggie_codebase-retrieval"];
+            if (!codebaseRetrievalTool) {
+                throw new Error(`codebase-retrieval tool not found. Available tools: ${Object.keys(tools).join(", ")}`);
+            }
+            // Execute the MCP tool
+            //
+            // MCP Tool Input Format:
+            // Mastra MCP wraps tool parameters in a 'context' property:
+            //   { context: { param1: value1, param2: value2, ... } }
+            //
+            // MCP Tool Output Format:
+            // MCP tools return results in the Model Context Protocol format:
+            //   { content: [{ type: "text", text: "..." }, ...] }
+            // This allows tools to return multiple content blocks of different types.
+            const result = await codebaseRetrievalTool.execute({
+                context: {
+                    information_request: query,
+                },
+            });
+            // Extract text from MCP result (standard MCP format)
+            if (!(result?.content && Array.isArray(result.content))) {
+                throw new Error(`Unexpected MCP response format: expected { content: [...] }, got ${JSON.stringify(result)}`);
+            }
+            const text = result.content
+                .filter((item) => item.type === "text")
+                .map((item) => item.text || "")
+                .join("\n");
+            this.log(`Search completed (${text.length} characters)`);
+            return text;
+        }
+        catch (error) {
+            throw new Error(`Codebase retrieval failed: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+    /**
+     * Search the indexed codebase and ask an LLM a question about the results.
+     *
+     * This is a convenience method that combines search() with an LLM call to answer
+     * questions about your codebase.
+     *
+     * @param searchQuery - The semantic search query to find relevant code (also used as the prompt if no separate prompt is provided)
+     * @param prompt - Optional prompt to ask the LLM about the search results. If not provided, searchQuery is used as the prompt.
+     * @returns The LLM's answer to your question
+     *
+     * @example
+     * ```typescript
+     * const answer = await context.searchAndAsk(
+     *   "How does the authentication flow work?"
+     * );
+     * console.log(answer);
+     * ```
+     */
+    async searchAndAsk(searchQuery, prompt) {
+        const results = await this.search(searchQuery);
+        const llmPrompt = formatSearchPrompt(prompt ?? searchQuery, results);
+        const apiKey = this.apiKey || process.env.AUGMENT_API_TOKEN;
+        if (!apiKey) {
+            throw new Error("API key is required for searchAndAsk(). Provide it via AUGMENT_API_TOKEN environment variable.");
+        }
+        const apiUrl = this.apiUrl || process.env.AUGMENT_API_URL;
+        if (!apiUrl) {
+            throw new Error("API URL is required for searchAndAsk(). Set AUGMENT_API_URL environment variable to your tenant-specific URL (e.g., 'https://your-tenant.api.augmentcode.com').");
+        }
+        const apiClient = new ContextAPIClient({ apiKey, apiUrl });
+        return await apiClient.chat(llmPrompt);
+    }
+    /**
+     * Close the MCP connection and cleanup resources
+     */
+    async close() {
+        this.log("Closing MCP connection");
+        if (this.mcpClient) {
+            try {
+                await this.mcpClient.disconnect();
+            }
+            catch (error) {
+                console.error("Error closing MCP client:", error);
+            }
+            this.mcpClient = null;
+        }
+        this.log("MCP connection closed");
+    }
+}
+//# sourceMappingURL=filesystem-context.js.map

package/dist/context/filesystem-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"filesystem-context.js","sourceRoot":"","sources":["../../src/context/filesystem-context.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAe7D;;;;;;;;;GASG;AACH,MAAM,OAAO,iBAAiB;IAS5B;;;OAGG;IACH,YAAY,OAAiC;QANrC,cAAS,GAAqB,IAAI,CAAC;QAOzC,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,QAAQ,CAAC;QACjD,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,KAAK,CAAC;QACpC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;QAC5C,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;IAC5C,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,MAAM,CACjB,OAAiC;QAEjC,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAChD,MAAM,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC;QAC1C,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;OAEG;IACK,GAAG,CAAC,OAAe;QACzB,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,GAAG,CAAC,uBAAuB,OAAO,EAAE,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,OAAO;QACb,IAAI,CAAC,GAAG,CAAC,sCAAsC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;QAEjE,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;QACvC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;QACzD,CAAC;QACD,MAAM,IAAI,GAAG,CAAC,OAAO,EAAE,kBAAkB,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAE3D,8BAA8B;QAC9B,MAAM,GAAG,GAA2B,EAAE,CAAC;QACvC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;YACvD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,GAAG,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YACnB,CAAC;QACH,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,GAAG,CAAC,iBAAiB,GAAG,IAAI,CAAC,MAAM,CAAC;QACtC,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,GAAG,CAAC,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC;QACpC,CAAC;QAED,oCAAoC;QACpC,IAAI,CAAC,SAAS,GAAG,IAAI,SAAS,CAAC;YAC7B,OAAO,EAAE;gBACP,MAAM,EAAE;oBACN,OAAO;oBACP,IAAI;oBACJ,GAAG;oBACH,MAAM,EAAE,IAAI,CAAC,KAAK;wBAChB,CAAC,CAAC,CAAC,UAAyB,EAAE,EAAE;4BAC5B,OAAO,CAAC,GAAG,CACT,QAAQ,UAAU,CAAC,KAAK,KAAK,UAAU,CAAC,OAAO,EAAE,EACjD,UAAU,CAAC,OAAO,CACnB,CAAC;wBACJ,CAAC;wBACH,CAAC,CAAC,SAAS;iBACd;aACF;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;IAC7C,CAAC;IAED;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACH,KAAK,CAAC,MAAM,CAAC,KAAa;QACxB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;QACtE,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,mBAAmB,KAAK,GAAG,CAAC,CAAC;QAEtC,IAAI,CAAC;YACH,oCAAoC;YACpC,+EAA+E;YAC/E,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC;YAC9C,MAAM,qBAAqB,GAAG,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAEjE,IAAI,CAAC,qBAAqB,EAAE,CAAC;gBAC3B,MAAM,IAAI,KAAK,CACb,uDAAuD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACvF,CAAC;YACJ,CAAC;YAED,uBAAuB;YACvB,EAAE;YACF,yBAAyB;YACzB,4DAA4D;YAC5D,yDAAyD;YACzD,EAAE;YACF,0BAA0B;YAC1B,iEAAiE;YACjE,sDAAsD;YACtD,0EAA0E;YAC1E,MAAM,MAAM,GAAG,MAAM,qBAAqB,CAAC,OAAO,CAAC;gBACjD,OAAO,EAAE;oBACP,mBAAmB,EAAE,KAAK;iBAC3B;aACF,CAAC,CAAC;YAEH,qDAAqD;YACrD,IAAI,CAAC,CAAC,MAAM,EAAE,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;gBACxD,MAAM,IAAI,KAAK,CACb,oEAAoE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAC7F,CAAC;YACJ,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO;iBACxB,MAAM,CAAC,CAAC,IAAoB,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;iBACtD,GAAG,CAAC,CAAC,IAAoB,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;iBAC9C,IAAI,CAAC,IAAI,CAAC,CAAC;YAEd,IAAI,CAAC,GAAG,CAAC,qBAAqB,IAAI,CAAC,MAAM,cAAc,CAAC,CAAC;YAEzD,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CACb,8BAA8B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CACvF,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;;;;;;;;;;;;;;;OAiBG;IACH,KAAK,CAAC,YAAY,CAAC,WAAmB,EAAE,MAAe;QACrD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QAC/C,MAAM,SAAS,GAAG,kBAAkB,CAAC,MAAM,IAAI,WAAW,EAAE,OAAO,CAAC,CAAC;QAErE,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;QAC5D,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CACb,gGAAgG,CACjG,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;QAC1D,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CACb,iKAAiK,CAClK,CAAC;QACJ,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,gBAAgB,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QAC3D,OAAO,MAAM,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACzC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;QAEnC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC;YACpC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,KAAK,CAAC,CAAC;YACpD,CAAC;YACD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACxB,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IACpC,CAAC;CACF"}

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

@@ -0,0 +1,54 @@
+/**
+ * Mock implementation of ContextAPIClient for testing
+ */
+import type { AgentCodebaseRetrievalResult, BatchUploadResult, CheckpointBlobsResult, FindMissingResult, UploadBlob } from "../api-client";
+import type { Blobs } from "../../types";
+export type MockAPIClientConfig = {
+    /** Blobs that the server knows about (uploaded) */
+    knownBlobs?: Set<string>;
+    /** Blobs that are indexed (subset of knownBlobs) */
+    indexedBlobs?: Set<string>;
+    /** Delay in ms for async operations (simulates network latency) */
+    delay?: number;
+    /** Should findMissing throw an error? */
+    shouldFailFindMissing?: boolean;
+    /** Should batchUpload throw an error? */
+    shouldFailBatchUpload?: boolean;
+    /** Should checkpointBlobs throw an error? */
+    shouldFailCheckpoint?: boolean;
+};
+/**
+ * Mock implementation that provides the same public API as ContextAPIClient
+ * Note: Does not formally implement ContextAPIClient due to TypeScript limitations
+ * with private members, but provides all the same public methods for testing.
+ */
+export declare class MockContextAPIClient {
+    private knownBlobs;
+    private indexedBlobs;
+    private delay;
+    private shouldFailFindMissing;
+    private shouldFailBatchUpload;
+    private shouldFailCheckpoint;
+    findMissingCalls: string[][];
+    batchUploadCalls: UploadBlob[][];
+    checkpointBlobsCalls: Blobs[];
+    agentCodebaseRetrievalCalls: Array<{
+        query: string;
+        blobs: Blobs;
+        maxOutputLength?: number;
+    }>;
+    chatCalls: string[];
+    constructor(config?: MockAPIClientConfig);
+    createRequestId(): string;
+    private simulateDelay;
+    findMissing(blobNames: string[]): Promise<FindMissingResult>;
+    batchUpload(blobs: UploadBlob[]): Promise<BatchUploadResult>;
+    checkpointBlobs(blobs: Blobs): Promise<CheckpointBlobsResult>;
+    agentCodebaseRetrieval(_requestId: string, query: string, blobs: Blobs, maxOutputLength?: number): Promise<AgentCodebaseRetrievalResult>;
+    chat(prompt: string): Promise<string>;
+    /** Mark blobs as indexed (simulates backend indexing completion) */
+    markAsIndexed(blobNames: string[]): void;
+    /** Reset all call tracking */
+    resetCalls(): void;
+}
+//# sourceMappingURL=api-client.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"api-client.d.ts","sourceRoot":"","sources":["../../../../src/context/internal/__mocks__/api-client.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EACV,4BAA4B,EAC5B,iBAAiB,EACjB,qBAAqB,EACrB,iBAAiB,EACjB,UAAU,EACX,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAEzC,MAAM,MAAM,mBAAmB,GAAG;IAChC,mDAAmD;IACnD,UAAU,CAAC,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IACzB,oDAAoD;IACpD,YAAY,CAAC,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAC3B,mEAAmE;IACnE,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,yCAAyC;IACzC,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC,yCAAyC;IACzC,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC,6CAA6C;IAC7C,oBAAoB,CAAC,EAAE,OAAO,CAAC;CAChC,CAAC;AAEF;;;;GAIG;AACH,qBAAa,oBAAoB;IAE/B,OAAO,CAAC,UAAU,CAAc;IAChC,OAAO,CAAC,YAAY,CAAc;IAClC,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,qBAAqB,CAAU;IACvC,OAAO,CAAC,qBAAqB,CAAU;IACvC,OAAO,CAAC,oBAAoB,CAAU;IAG/B,gBAAgB,EAAE,MAAM,EAAE,EAAE,CAAM;IAClC,gBAAgB,EAAE,UAAU,EAAE,EAAE,CAAM;IACtC,oBAAoB,EAAE,KAAK,EAAE,CAAM;IACnC,2BAA2B,EAAE,KAAK,CAAC;QACxC,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,EAAE,KAAK,CAAC;QACb,eAAe,CAAC,EAAE,MAAM,CAAC;KAC1B,CAAC,CAAM;IACD,SAAS,EAAE,MAAM,EAAE,CAAM;gBAEpB,MAAM,GAAE,mBAAwB;IAS5C,eAAe,IAAI,MAAM;YAIX,aAAa;IAMrB,WAAW,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAsB5D,WAAW,CAAC,KAAK,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAkB5D,eAAe,CAAC,KAAK,EAAE,KAAK,GAAG,OAAO,CAAC,qBAAqB,CAAC;IAc7D,sBAAsB,CAC1B,UAAU,EAAE,MAAM,EAClB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,KAAK,EACZ,eAAe,CAAC,EAAE,MAAM,GACvB,OAAO,CAAC,4BAA4B,CAAC;IAUlC,IAAI,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAU3C,oEAAoE;IACpE,aAAa,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,IAAI;IAMxC,8BAA8B;IAC9B,UAAU,IAAI,IAAI;CAOnB"}

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

@@ -0,0 +1,96 @@
+/**
+ * Mock implementation of ContextAPIClient for testing
+ */
+/**
+ * Mock implementation that provides the same public API as ContextAPIClient
+ * Note: Does not formally implement ContextAPIClient due to TypeScript limitations
+ * with private members, but provides all the same public methods for testing.
+ */
+export class MockContextAPIClient {
+    constructor(config = {}) {
+        // 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;
+    }
+    createRequestId() {
+        return `mock-req-${Date.now()}`;
+    }
+    async simulateDelay() {
+        if (this.delay > 0) {
+            await new Promise((resolve) => setTimeout(resolve, this.delay));
+        }
+    }
+    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,
+        };
+    }
+    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 };
+    }
+    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(_requestId, 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

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

@@ -0,0 +1 @@
+{"version":3,"file":"api-client.js","sourceRoot":"","sources":["../../../../src/context/internal/__mocks__/api-client.ts"],"names":[],"mappings":"AAAA;;GAEG;AA0BH;;;;GAIG;AACH,MAAM,OAAO,oBAAoB;IAoB/B,YAAY,SAA8B,EAAE;QAX5C,oCAAoC;QAC7B,qBAAgB,GAAe,EAAE,CAAC;QAClC,qBAAgB,GAAmB,EAAE,CAAC;QACtC,yBAAoB,GAAY,EAAE,CAAC;QACnC,gCAA2B,GAI7B,EAAE,CAAC;QACD,cAAS,GAAa,EAAE,CAAC;QAG9B,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,IAAI,GAAG,EAAE,CAAC;QACjD,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,IAAI,GAAG,EAAE,CAAC;QACrD,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,CAAC,CAAC;QAC/B,IAAI,CAAC,qBAAqB,GAAG,MAAM,CAAC,qBAAqB,IAAI,KAAK,CAAC;QACnE,IAAI,CAAC,qBAAqB,GAAG,MAAM,CAAC,qBAAqB,IAAI,KAAK,CAAC;QACnE,IAAI,CAAC,oBAAoB,GAAG,MAAM,CAAC,oBAAoB,IAAI,KAAK,CAAC;IACnE,CAAC;IAED,eAAe;QACb,OAAO,YAAY,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;IAClC,CAAC;IAEO,KAAK,CAAC,aAAa;QACzB,IAAI,IAAI,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC;YACnB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QAClE,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,SAAmB;QACnC,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC,CAAC;QAE3C,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAC/B,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC5C,CAAC;QAED,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAE3B,MAAM,gBAAgB,GAAG,SAAS,CAAC,MAAM,CACvC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CACrC,CAAC;QACF,MAAM,mBAAmB,GAAG,SAAS,CAAC,MAAM,CAC1C,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CACpE,CAAC;QAEF,OAAO;YACL,gBAAgB;YAChB,mBAAmB;SACpB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAmB;QACnC,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC;QAEvC,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAC/B,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC5C,CAAC;QAED,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAE3B,0DAA0D;QAC1D,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;QAC/C,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YACjC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,EAAE,SAAS,EAAE,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,KAAY;QAChC,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC;QAE7C,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,CAAC;QAED,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAE3B,OAAO;YACL,eAAe,EAAE,cAAc,IAAI,CAAC,GAAG,EAAE,EAAE;SAC5C,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,sBAAsB,CAC1B,UAAkB,EAClB,KAAa,EACb,KAAY,EACZ,eAAwB;QAExB,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC,CAAC;QAEzE,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAE3B,OAAO;YACL,kBAAkB,EAAE,4BAA4B,KAAK,EAAE;SACxD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAAc;QACvB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAE5B,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAE3B,OAAO,yBAAyB,MAAM,EAAE,CAAC;IAC3C,CAAC;IAED,gCAAgC;IAEhC,oEAAoE;IACpE,aAAa,CAAC,SAAmB;QAC/B,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YACjC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAED,8BAA8B;IAC9B,UAAU;QACR,IAAI,CAAC,gBAAgB,GAAG,EAAE,CAAC;QAC3B,IAAI,CAAC,gBAAgB,GAAG,EAAE,CAAC;QAC3B,IAAI,CAAC,oBAAoB,GAAG,EAAE,CAAC;QAC/B,IAAI,CAAC,2BAA2B,GAAG,EAAE,CAAC;QACtC,IAAI,CAAC,SAAS,GAAG,EAAE,CAAC;IACtB,CAAC;CACF"}

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

@@ -0,0 +1,75 @@
+/**
+ * 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 {
+    readonly status: number;
+    readonly statusText: string;
+    constructor(status: number, statusText: string, message: string);
+}
+export type UploadBlob = {
+    blobName: string;
+    pathName: string;
+    text: string;
+    metadata: string[];
+};
+export type FindMissingResult = {
+    unknownBlobNames: string[];
+    nonindexedBlobNames: string[];
+};
+export type CheckpointBlobsResult = {
+    newCheckpointId: string;
+};
+export type BatchUploadResult = {
+    blobNames: string[];
+};
+export type AgentCodebaseRetrievalResult = {
+    formattedRetrieval: string;
+};
+export type ContextAPIClientOptions = {
+    apiKey: string;
+    apiUrl: string;
+    debug?: boolean;
+};
+/**
+ * Get user agent string for Context API requests
+ */
+export declare function getUserAgent(): string;
+/**
+ * API client for Context operations
+ */
+export declare class ContextAPIClient {
+    private readonly apiKey;
+    private readonly apiUrl;
+    private readonly sessionId;
+    private readonly debug;
+    constructor(options: ContextAPIClientOptions);
+    private log;
+    createRequestId(): string;
+    private callApi;
+    findMissing(blobNames: string[]): Promise<FindMissingResult>;
+    batchUpload(blobs: UploadBlob[]): Promise<BatchUploadResult>;
+    checkpointBlobs(blobs: Blobs): Promise<CheckpointBlobsResult>;
+    agentCodebaseRetrieval(_requestId: string, 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
+     */
+    private parseSSEStream;
+    /**
+     * Call the LLM chat streaming API with a formatted prompt
+     *
+     * @param prompt - The formatted prompt to send to the LLM
+     * @returns The LLM's response text
+     */
+    chat(prompt: string): Promise<string>;
+}
+//# sourceMappingURL=api-client.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"api-client.d.ts","sourceRoot":"","sources":["../../../src/context/internal/api-client.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAIH,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,UAAU,CAAC;AAEtC;;;GAGG;AACH,qBAAa,QAAS,SAAQ,KAAK;IACjC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;gBAEhB,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM;CAMhE;AAED,MAAM,MAAM,UAAU,GAAG;IACvB,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,EAAE,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,gBAAgB,EAAE,MAAM,EAAE,CAAC;IAC3B,mBAAmB,EAAE,MAAM,EAAE,CAAC;CAC/B,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAAG;IAClC,eAAe,EAAE,MAAM,CAAC;CACzB,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,SAAS,EAAE,MAAM,EAAE,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG;IACzC,kBAAkB,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEF,MAAM,MAAM,uBAAuB,GAAG;IACpC,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,OAAO,CAAC;CACjB,CAAC;AAEF;;GAEG;AACH,wBAAgB,YAAY,IAAI,MAAM,CAErC;AAED;;GAEG;AACH,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAChC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAChC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAU;gBAEpB,OAAO,EAAE,uBAAuB;IAO5C,OAAO,CAAC,GAAG;IAMX,eAAe,IAAI,MAAM;YAIX,OAAO;IAuCf,WAAW,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAc5D,WAAW,CAAC,KAAK,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAiB5D,eAAe,CAAC,KAAK,EAAE,KAAK,GAAG,OAAO,CAAC,qBAAqB,CAAC;IAiB7D,sBAAsB,CAC1B,UAAU,EAAE,MAAM,EAClB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,KAAK,EACZ,eAAe,CAAC,EAAE,MAAM,GACvB,OAAO,CAAC,4BAA4B,CAAC;IAkCxC;;OAEG;IACH,OAAO,CAAC,eAAe;IAqBvB;;OAEG;YACW,cAAc;IA2B5B;;;;;OAKG;IACG,IAAI,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;CAsD5C"}