@augmentcode/auggie-sdk 0.1.0

package/README.md

@@ -0,0 +1,395 @@
+# Auggie SDK
+
+A TypeScript SDK for connecting to Auggie as an ACP (Agent Client Protocol) server with support for custom tools via MCP (Model Context Protocol).
+
+## Overview
+
+This SDK provides a simple way to:
+1. Launch Auggie in ACP mode
+2. Establish a bidirectional communication channel
+3. Initialize sessions with custom tools
+4. Send prompts and receive responses
+5. Handle session updates and permission requests
+6. Integrate custom tools using the AI SDK tool format
+
+## Installation
+
+### From tarball (recommended for distribution)
+```bash
+npm install /path/to/augmentcode-auggie-sdk-0.1.0.tgz
+# or
+pnpm add /path/to/augmentcode-auggie-sdk-0.1.0.tgz
+```
+
+### From local directory (for development)
+```bash
+npm install /path/to/auggie-sdk
+# or
+pnpm add /path/to/auggie-sdk
+```
+
+## Usage
+
+### Basic Example
+
+```typescript
+import { Auggie } from "@augmentcode/auggie-sdk";
+
+async function main() {
+  // Create and initialize Auggie (automatically connects and creates a session)
+  const client = await Auggie.create({
+    model: "sonnet4.5", // Optional: haiku4.5, opus4.1, sonnet4.5, sonnet4
+    debug: false,       // Optional: enable debug logging
+  });
+
+  try {
+    // Listen for session updates
+    client.onSessionUpdate((update) => {
+      console.log("Received update:", update);
+    });
+
+    // Send a prompt
+    const response = await client.sendPrompt("What files are in the current directory?");
+    console.log("Response:", response);
+
+    // Close the connection
+    await client.close();
+  } catch (error) {
+    console.error("Error:", error);
+    await client.close();
+  }
+}
+
+main();
+```
+
+### Using Custom Tools
+
+You can provide custom tools to Auggie using the AI SDK tool format:
+
+```typescript
+import { Auggie } from "@augmentcode/auggie-sdk";
+import { tool } from "ai";
+import z from "zod";
+
+const weatherTool = tool({
+  name: "get_weather",
+  description: "Get the weather in a location",
+  inputSchema: z.object({
+    location: z.string().describe("The location to get the weather for"),
+  }),
+  execute: async ({ location }) => {
+    // Your tool implementation
+    return `The weather in ${location} is sunny.`;
+  },
+});
+
+async function main() {
+  // Pass tools when creating the instance
+  const client = await Auggie.create({
+    model: "sonnet4.5",
+    toolsMap: { weather_tool: weatherTool }
+  });
+
+  try {
+    const response = await client.sendPrompt("What's the weather in San Francisco?");
+    console.log(response);
+
+    await client.close();
+  } catch (error) {
+    console.error("Error:", error);
+    await client.close();
+  }
+}
+
+main();
+```
+
+### Getting Only the Final Response
+
+When the agent uses tools (like file operations, web searches, etc.), you can choose to get only the final answer after all tool executions complete:
+
+```typescript
+// Get all agent responses (default behavior)
+const fullResponse = await client.sendPrompt("List all TypeScript files in this project");
+
+// Get only the final response after tool calls
+const finalAnswer = await client.sendPrompt(
+  "List all TypeScript files in this project",
+  { isAnswerOnly: true }
+);
+```
+
+This is useful when you want to:
+- Display only the final answer to users, hiding intermediate thinking
+- Process only the conclusive response in automated workflows
+- Reduce token usage by ignoring intermediate agent messages
+
+### Streaming Session Updates
+
+You can listen to real-time updates from the agent:
+
+```typescript
+client.onSessionUpdate((event) => {
+  switch (event.update.sessionUpdate) {
+    case "agent_message_chunk":
+      if (event.update.content.type === "text") {
+        process.stdout.write(event.update.content.text);
+      }
+      break;
+    case "tool_call":
+      console.log("Tool called:", event.update.title);
+      console.log("Input:", event.update.rawInput);
+      break;
+    case "tool_call_update":
+      console.log("Tool output:", event.update.rawOutput);
+      break;
+  }
+});
+```
+
+### Running the Examples
+
+```bash
+# Run the dev example with streaming output
+npm run dev "What's the weather in Tokyo?"
+
+# Run the silent example (no streaming)
+npm run silent "What's the weather in Paris?"
+```
+
+### Custom Auggie Path and Workspace
+
+```typescript
+const client = await Auggie.create({
+  auggiePath: "/path/to/auggie",
+  workspaceRoot: "/path/to/workspace",
+  model: "sonnet4.5",
+  debug: true,
+  allowIndexing: true,
+});
+```
+
+## API Reference
+
+### `Auggie`
+
+#### Static Factory Method
+
+```typescript
+static async create(options?: AuggieOptions): Promise<Auggie>
+```
+
+Creates a new Auggie instance and automatically connects and initializes a session.
+
+**Options:**
+- `auggiePath?: string` - Path to the Auggie executable (default: "auggie")
+- `workspaceRoot?: string` - Working directory for the Auggie process (default: `process.cwd()`)
+- `debug?: boolean` - Enable debug logging (default: false)
+- `model?: "haiku4.5" | "opus4.1" | "sonnet4.5" | "sonnet4"` - Model to use (default: "haiku4.5")
+- `allowIndexing?: boolean` - Allow codebase indexing (default: true)
+- `toolsMap?: Record<string, Tool>` - Custom tools to provide to Auggie (optional)
+
+#### Methods
+
+##### `connect(): Promise<void>`
+
+Launches Auggie in ACP mode and establishes a connection. (Called automatically by `create()`)
+
+##### `createSession(toolsMap?: Record<string, Tool>): Promise<string>`
+
+Creates a new session. (Called automatically by `create()`)
+
+Creates a new conversation session with the agent.
+
+- `toolsMap`: Optional map of tool names to AI SDK Tool objects
+- Returns: Session ID
+
+When tools are provided, an MCP server is automatically started and connected to the session.
+
+##### `sendPrompt(prompt: string, options?: { isAnswerOnly?: boolean }): Promise<string>`
+
+Sends a prompt to the agent and waits for completion.
+
+- `prompt`: The text prompt to send
+- `options.isAnswerOnly`: Optional boolean (default: `false`). When set to `true`, returns only the final response after all tool calls complete. When `false` or omitted, returns all agent message chunks including any intermediate responses.
+- Returns: The accumulated text response from the agent
+
+##### `onSessionUpdate(callback: (update: SessionNotification) => void): void`
+
+Registers a callback to receive session updates from the agent.
+
+- `callback`: Function to handle session updates
+
+##### `cancel(): Promise<void>`
+
+Cancels the current ongoing request.
+
+##### `close(): Promise<void>`
+
+Closes the connection, stops the MCP server (if running), and terminates the Auggie process.
+
+## How It Works
+
+### ACP Protocol
+
+The Agent Client Protocol (ACP) is a standardized protocol for communication between code editors and AI coding agents. This SDK implements the client side of the protocol.
+
+### MCP Integration
+
+When you provide custom tools, the SDK automatically:
+1. Starts a local MCP (Model Context Protocol) server
+2. Wraps your AI SDK tools to work with Mastra's MCP implementation
+3. Registers the MCP server with the Auggie session
+4. Handles tool execution requests from Auggie
+5. Cleans up the MCP server when the session closes
+
+### Connection Flow
+
+1. **Launch**: Spawns Auggie with the `--acp` flag and optional parameters
+2. **Streams**: Creates bidirectional streams using stdio
+3. **Initialize**: Negotiates protocol version and capabilities
+4. **MCP Setup**: If tools are provided, starts an MCP server on a free port
+5. **Session**: Creates a conversation session with MCP server configuration
+6. **Prompt**: Sends prompts and receives responses
+7. **Updates**: Receives real-time updates via callbacks
+8. **Cleanup**: Closes connections and terminates processes
+
+### Client Implementation
+
+The SDK implements the `Client` interface required by ACP:
+
+- `sessionUpdate`: Handles real-time updates from the agent
+- `requestPermission`: Handles permission requests (auto-selects first option)
+
+## Examples
+
+### Complete Example with Tools and Streaming
+
+```typescript
+import { Auggie } from "@augmentcode/auggie-sdk";
+import { tool } from "ai";
+import z from "zod";
+
+const weatherTool = tool({
+  name: "get_weather",
+  description: "Get the weather in a location",
+  inputSchema: z.object({
+    location: z.string().describe("The location to get the weather for"),
+  }),
+  execute: async ({ location }) => {
+    console.log(`Weather tool called for: ${location}`);
+    return `The weather in ${location} is sunny.`;
+  },
+});
+
+async function main() {
+  const client = await Auggie.create({
+    model: "sonnet4.5",
+    debug: true,
+    toolsMap: { weather_tool: weatherTool }
+  });
+
+  try {
+    // Stream updates in real-time
+    client.onSessionUpdate((event) => {
+      switch (event.update.sessionUpdate) {
+        case "agent_message_chunk":
+          if (event.update.content.type === "text") {
+            process.stdout.write(event.update.content.text);
+          }
+          break;
+        case "tool_call":
+          console.log(`\nTool: ${event.update.title}`);
+          console.log("Input:", event.update.rawInput);
+          break;
+        case "tool_call_update":
+          console.log("Output:", event.update.rawOutput);
+          break;
+      }
+    });
+
+    const response = await client.sendPrompt(
+      "What's the weather in Tokyo?",
+      { isAnswerOnly: true }
+    );
+
+    console.log("\nFinal answer:", response);
+  } catch (error) {
+    console.error("Error:", error);
+  } finally {
+    await client.close();
+  }
+}
+
+main();
+```
+
+### Error Handling
+
+```typescript
+try {
+  const client = await Auggie.create();
+  await client.sendPrompt("Hello!");
+  await client.close();
+} catch (error) {
+  if (error instanceof Error) {
+    console.error("Error:", error.message);
+  }
+}
+```
+
+## Requirements
+
+- Node.js 18+
+- Auggie CLI installed and accessible in PATH (or provide custom path)
+- Auggie must be authenticated (`auggie login`)
+
+## Building the Package
+
+To build the TypeScript SDK:
+
+```bash
+npm run build
+```
+
+This compiles the TypeScript source to JavaScript in the `dist/` directory with:
+- ES2020 module format
+- Type declaration files (`.d.ts`)
+- Source maps for debugging
+
+To create a distributable tarball:
+
+```bash
+npm pack
+```
+
+This creates `augmentcode-auggie-sdk-0.1.0.tgz` which can be shared and installed.
+
+## Launching Auggie as ACP Server
+
+To launch Auggie in ACP mode manually:
+
+```bash
+auggie --acp
+```
+
+This starts Auggie in ACP server mode, communicating via stdio using newline-delimited JSON (ndjson).
+
+### Available CLI Options
+
+- `--acp`: Enable ACP mode
+- `--model <model>`: Specify model (haiku4.5, opus4.1, sonnet4.5, sonnet4)
+- `--workspace-root <path>`: Set workspace directory
+- `--allow-indexing`: Enable codebase indexing
+
+## References
+
+- [Agent Client Protocol Documentation](https://agentclientprotocol.com/)
+- [ACP TypeScript SDK](https://github.com/agentclientprotocol/typescript-sdk)
+- [AI SDK Documentation](https://sdk.vercel.ai/docs)
+- [Mastra MCP Documentation](https://mastra.ai/docs/mcp)
+
+## License
+
+See the main repository license.

package/dist/auggie/sdk-acp-client.d.ts

@@ -0,0 +1,76 @@
+import * as acp from "@agentclientprotocol/sdk";
+import { type Tool } from "ai";
+type models = "haiku4.5" | "opus4.1" | "sonnet4.5" | "sonnet4";
+type AuggieOptions = {
+    auggiePath?: string;
+    workspaceRoot?: string;
+    debug?: boolean;
+    model?: models;
+    allowIndexing?: boolean;
+};
+/**
+ * ACP Client for connecting to Auggie as an ACP server
+ *
+ * This client demonstrates how to:
+ * 1. Launch Auggie in ACP mode
+ * 2. Establish a bidirectional communication channel
+ * 3. Initialize a session
+ * 4. Send prompts and receive responses
+ */
+declare class Auggie {
+    private connection;
+    private sessionId;
+    private auggiePath;
+    private workspaceRoot;
+    private auggieProcess;
+    private sessionUpdateCallback;
+    private debug;
+    private mcpServer;
+    private model;
+    private allowIndexing;
+    constructor({ auggiePath, workspaceRoot, debug, model, allowIndexing }?: AuggieOptions);
+    /**
+     * Connect to Auggie ACP server
+     */
+    connect(): Promise<void>;
+    /**
+     * Create a Client implementation to handle agent requests
+     */
+    private createClient;
+    /**
+     * Initialize the ACP connection
+     */
+    private initialize;
+    /**
+     * Create a new session
+     */
+    createSession(toolsMap?: Record<string, Tool>): Promise<string>;
+    /**
+     * Send a prompt to the agent and accumulate the text response
+     * @param prompt - The text prompt to send to the agent
+     * @param isAnswerOnly - If true, returns only the final response after all tool calls complete.
+     *                        If false or undefined, returns all agent message chunks (default: false)
+     * @returns The accumulated text response from the agent
+     */
+    sendPrompt(prompt: string, { isAnswerOnly }?: {
+        isAnswerOnly?: boolean;
+    } | undefined): Promise<string>;
+    /**
+     * Listen for session update events from the agent
+     */
+    onSessionUpdate(callback: (update: acp.SessionNotification) => void): void;
+    /**
+     * Cancel the current request
+     */
+    cancel(): Promise<void>;
+    /**
+     * Close the connection and kill the Auggie process
+     */
+    close(): Promise<void>;
+    /**
+     * Kill the Auggie process if it's still running
+     */
+    private killProcess;
+}
+export { Auggie };
+//# sourceMappingURL=sdk-acp-client.d.ts.map

package/dist/auggie/sdk-acp-client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sdk-acp-client.d.ts","sourceRoot":"","sources":["../../src/auggie/sdk-acp-client.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,GAAG,MAAM,0BAA0B,CAAC;AAGhD,OAAO,EAAE,KAAK,IAAI,EAAE,MAAM,IAAI,CAAC;AAG/B,KAAK,MAAM,GACT,UAAU,GACV,SAAS,GACT,WAAW,GACX,SAAS,CACV;AAED,KAAK,aAAa,GAAG;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB,CAAC;AAEF;;;;;;;;GAQG;AACH,cAAM,MAAM;IACV,OAAO,CAAC,UAAU,CAAyC;IAC3D,OAAO,CAAC,SAAS,CAAuB;IACxC,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,aAAa,CAAiC;IACtD,OAAO,CAAC,aAAa,CAA6B;IAClD,OAAO,CAAC,qBAAqB,CAA4D;IACzF,OAAO,CAAC,KAAK,CAAU;IACvB,OAAO,CAAC,SAAS,CAAgC;IACjD,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,aAAa,CAAU;gBACnB,EAAE,UAAqB,EAAE,aAAa,EAAE,KAAa,EAAE,KAAkB,EAAE,aAAoB,EAAE,GAAE,aAAkB;IAQjI;;OAEG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAwD9B;;OAEG;IACH,OAAO,CAAC,YAAY;IAoCpB;;OAEG;YACW,UAAU;IAmBxB;;OAEG;IACG,aAAa,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IA2CrE;;;;;;OAMG;IACG,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,YAAoB,EAAE,GAAE;QAAE,YAAY,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,SAAmC,GACxH,OAAO,CAAC,MAAM,CAAC;IAgFlB;;OAEG;IACH,eAAe,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,GAAG,CAAC,mBAAmB,KAAK,IAAI,GAAG,IAAI;IAI1E;;OAEG;IACG,MAAM,IAAI,OAAO,CAAC,IAAI,CAAC;IAc7B;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IA0B5B;;OAEG;IACH,OAAO,CAAC,WAAW;CAOpB;AAED,OAAO,EAAE,MAAM,EAAE,CAAC"}

package/dist/auggie/sdk-acp-client.js

@@ -0,0 +1,306 @@
+import * as acp from "@agentclientprotocol/sdk";
+import { spawn, ChildProcess } from "node:child_process";
+import { Readable, Writable } from "node:stream";
+import {} from "ai";
+import { AuggieMCPServer } from "./sdk-mcp-server";
+/**
+ * ACP Client for connecting to Auggie as an ACP server
+ *
+ * This client demonstrates how to:
+ * 1. Launch Auggie in ACP mode
+ * 2. Establish a bidirectional communication channel
+ * 3. Initialize a session
+ * 4. Send prompts and receive responses
+ */
+class Auggie {
+    constructor({ auggiePath = "auggie", workspaceRoot, debug = false, model = "haiku4.5", allowIndexing = true } = {}) {
+        this.connection = null;
+        this.sessionId = null;
+        this.workspaceRoot = undefined;
+        this.auggieProcess = null;
+        this.sessionUpdateCallback = null;
+        this.mcpServer = null;
+        this.auggiePath = auggiePath;
+        this.workspaceRoot = workspaceRoot;
+        this.debug = debug;
+        this.model = model;
+        this.allowIndexing = allowIndexing;
+    }
+    /**
+     * Connect to Auggie ACP server
+     */
+    async connect() {
+        const args = ["--acp"];
+        if (this.model) {
+            args.push("--model", this.model);
+        }
+        if (this.allowIndexing) {
+            args.push("--allow-indexing");
+        }
+        if (this.workspaceRoot) {
+            args.push("--workspace-root", this.workspaceRoot);
+        }
+        // Parse auggiePath to handle cases where it includes arguments (e.g., "node /path/to/file.js")
+        const pathParts = this.auggiePath.trim().split(/\s+/);
+        if (pathParts.length === 0 || !pathParts[0]) {
+            throw new Error("Invalid auggiePath: cannot be empty");
+        }
+        const command = pathParts[0];
+        const commandArgs = [...pathParts.slice(1), ...args];
+        this.auggieProcess = spawn(command, commandArgs, {
+            stdio: ["pipe", "pipe", "inherit"],
+        });
+        // Handle process errors
+        this.auggieProcess.on("error", (error) => {
+            console.error("Failed to start Auggie process:", error);
+            throw error;
+        });
+        this.auggieProcess.on("exit", (code, signal) => {
+            if (this.debug)
+                console.log(`Auggie process exited with code ${code} and signal ${signal}`);
+        });
+        // Create streams for ACP communication
+        // Note: Auggie writes to stdout and reads from stdin
+        const input = Readable.toWeb(this.auggieProcess.stdout);
+        const output = Writable.toWeb(this.auggieProcess.stdin);
+        // Create ndjson stream for ACP protocol
+        const stream = acp.ndJsonStream(output, input);
+        // Create client-side connection with a Client implementation
+        this.connection = new acp.ClientSideConnection((agent) => this.createClient(agent), stream);
+        if (this.debug) {
+            console.log("Connected to Auggie ACP server");
+        }
+        // Initialize the connection
+        await this.initialize();
+    }
+    /**
+     * Create a Client implementation to handle agent requests
+     */
+    createClient(_agent) {
+        return {
+            // Handle session updates from the agent
+            sessionUpdate: async (params) => {
+                if (this.sessionUpdateCallback) {
+                    this.sessionUpdateCallback(params);
+                }
+            },
+            // Handle permission requests from the agent
+            requestPermission: async (params) => {
+                // For this example, auto-select the first option if available
+                // In a real implementation, you'd prompt the user
+                if (this.debug)
+                    console.log("Permission requested:", params);
+                if (params.options && params.options.length > 0) {
+                    return {
+                        outcome: {
+                            outcome: "selected",
+                            optionId: params.options?.[0]?.name || '',
+                        },
+                    };
+                }
+                // If no options, return cancelled
+                return {
+                    outcome: {
+                        outcome: "cancelled",
+                    },
+                };
+            },
+        };
+    }
+    /**
+     * Initialize the ACP connection
+     */
+    async initialize() {
+        if (!this.connection) {
+            throw new Error("Not connected. Call connect() first.");
+        }
+        if (this.debug)
+            console.log("Initializing ACP connection...");
+        const initResponse = await this.connection.initialize({
+            protocolVersion: acp.PROTOCOL_VERSION,
+            clientInfo: {
+                name: "auggie-sdk-client",
+                version: "0.1.0",
+            },
+            clientCapabilities: {},
+        });
+        if (this.debug)
+            console.log("Initialized:", JSON.stringify(initResponse, null, 2));
+    }
+    /**
+     * Create a new session
+     */
+    async createSession(toolsMap) {
+        if (!this.connection) {
+            throw new Error("Not connected. Call connect() first.");
+        }
+        if (this.debug)
+            console.log("Creating new session...");
+        // Build MCP servers array for the session
+        const mcpServers = [];
+        // If tools are provided, create and start an MCP server
+        if (toolsMap && Object.keys(toolsMap).length > 0) {
+            if (this.debug)
+                console.log("Starting MCP server with provided tools...");
+            this.mcpServer = new AuggieMCPServer(toolsMap, undefined);
+            const serverUrl = await this.mcpServer.start();
+            if (this.debug)
+                console.log(`MCP server started at: ${serverUrl}`);
+            // Add the MCP server to the session configuration
+            mcpServers.push({
+                type: "http",
+                name: "custom_tool",
+                url: serverUrl,
+                headers: [],
+            });
+        }
+        const sessionResponse = await this.connection.newSession({
+            cwd: this.workspaceRoot ? this.workspaceRoot : process.cwd(),
+            mcpServers,
+        });
+        this.sessionId = sessionResponse.sessionId;
+        if (this.debug) {
+            console.log(`Session created: ${this.sessionId}`);
+            console.log(`   Available modes:`, sessionResponse.modes?.availableModes);
+        }
+        return this.sessionId;
+    }
+    /**
+     * Send a prompt to the agent and accumulate the text response
+     * @param prompt - The text prompt to send to the agent
+     * @param isAnswerOnly - If true, returns only the final response after all tool calls complete.
+     *                        If false or undefined, returns all agent message chunks (default: false)
+     * @returns The accumulated text response from the agent
+     */
+    async sendPrompt(prompt, { isAnswerOnly = false } = { isAnswerOnly: false }) {
+        if (!this.connection || !this.sessionId) {
+            throw new Error("Not connected or no session. Call connect() and createSession() first.");
+        }
+        if (this.debug)
+            console.log(`\nSending prompt: "${prompt}"`);
+        // Accumulator for text responses
+        let accumulatedText = "";
+        // Track the final response separately (text after the last tool call)
+        let finalResponseText = "";
+        // Track if we've seen a tool call
+        let hasSeenToolCall = false;
+        // Temporary callback to accumulate text
+        const originalCallback = this.sessionUpdateCallback;
+        const accumulatorCallback = (update) => {
+            // Track tool calls to know when to start capturing final response
+            if (update.update.sessionUpdate === "tool_call") {
+                hasSeenToolCall = true;
+                // Reset final response text when a new tool call starts
+                finalResponseText = "";
+            }
+            // Accumulate text content
+            if (update.update.sessionUpdate === "agent_message_chunk" && update.update.content.type === "text") {
+                const textChunk = update.update.content.text;
+                accumulatedText += textChunk;
+                // If we've seen a tool call, accumulate text for final response
+                // This captures the agent's response after tool execution
+                if (hasSeenToolCall) {
+                    finalResponseText += textChunk;
+                }
+            }
+            // Also call the original callback if it exists
+            if (originalCallback) {
+                originalCallback(update);
+            }
+        };
+        // Temporarily replace the callback
+        this.sessionUpdateCallback = accumulatorCallback;
+        try {
+            // Build content blocks
+            const content = [
+                {
+                    type: "text",
+                    text: prompt,
+                },
+            ];
+            // Send the prompt and wait for response
+            const response = await this.connection.prompt({
+                sessionId: this.sessionId,
+                prompt: content,
+            });
+            if (this.debug)
+                console.log("Prompt completed with stop reason:", response.stopReason);
+            // Restore original callback
+            this.sessionUpdateCallback = originalCallback;
+            // Return final response if requested and we had tool calls, otherwise return all text
+            if (isAnswerOnly && hasSeenToolCall) {
+                return finalResponseText;
+            }
+            return accumulatedText;
+        }
+        catch (error) {
+            console.error("Error during prompt:", error);
+            // Restore original callback before closing
+            this.sessionUpdateCallback = originalCallback;
+            // Kill the process on error
+            await this.close();
+            throw error;
+        }
+    }
+    /**
+     * Listen for session update events from the agent
+     */
+    onSessionUpdate(callback) {
+        this.sessionUpdateCallback = callback;
+    }
+    /**
+     * Cancel the current request
+     */
+    async cancel() {
+        if (!this.connection || !this.sessionId) {
+            throw new Error("Not connected or no session.");
+        }
+        if (this.debug)
+            console.log(" Cancelling current request...");
+        await this.connection.cancel({
+            sessionId: this.sessionId,
+        });
+        if (this.debug)
+            console.log(" Request cancelled");
+    }
+    /**
+     * Close the connection and kill the Auggie process
+     */
+    async close() {
+        if (this.debug)
+            console.log(" Closing connection...");
+        // Kill the Auggie process first - this will naturally close all MCP connections
+        this.killProcess();
+        // Stop the MCP server (should be fast since Auggie is already killed)
+        if (this.mcpServer) {
+            if (this.debug)
+                console.log(" Stopping MCP server...");
+            // Use a short timeout since the connections are already closed
+            await Promise.race([
+                this.mcpServer.stop(),
+                new Promise((resolve) => setTimeout(resolve, 100))
+            ]);
+            this.mcpServer = null;
+        }
+        // Clean up connection state
+        if (this.connection) {
+            this.connection = null;
+            this.sessionId = null;
+        }
+        if (this.debug)
+            console.log(" Connection closed");
+    }
+    /**
+     * Kill the Auggie process if it's still running
+     */
+    killProcess() {
+        if (this.auggieProcess && !this.auggieProcess.killed) {
+            if (this.debug)
+                console.log(" Killing Auggie process...");
+            // Use SIGKILL for immediate termination
+            this.auggieProcess.kill("SIGKILL");
+        }
+    }
+}
+export { Auggie };
+//# sourceMappingURL=sdk-acp-client.js.map

package/dist/auggie/sdk-acp-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sdk-acp-client.js","sourceRoot":"","sources":["../../src/auggie/sdk-acp-client.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,GAAG,MAAM,0BAA0B,CAAC;AAChD,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,EAAa,MAAM,IAAI,CAAC;AAC/B,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAiBnD;;;;;;;;GAQG;AACH,MAAM,MAAM;IAWV,YAAY,EAAE,UAAU,GAAG,QAAQ,EAAE,aAAa,EAAE,KAAK,GAAG,KAAK,EAAE,KAAK,GAAG,UAAU,EAAE,aAAa,GAAG,IAAI,KAAoB,EAAE;QAVzH,eAAU,GAAoC,IAAI,CAAC;QACnD,cAAS,GAAkB,IAAI,CAAC;QAEhC,kBAAa,GAAuB,SAAS,CAAC;QAC9C,kBAAa,GAAwB,IAAI,CAAC;QAC1C,0BAAqB,GAAuD,IAAI,CAAC;QAEjF,cAAS,GAA2B,IAAI,CAAC;QAI/C,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO;QACX,MAAM,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC;QACvB,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,CAAC;QACD,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAChC,CAAC;QACD,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QACpD,CAAC;QAED,+FAA+F;QAC/F,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACtD,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5C,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;QACzD,CAAC;QACD,MAAM,OAAO,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;QAC7B,MAAM,WAAW,GAAG,CAAC,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QAErD,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE;YAC/C,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC;SACnC,CAAC,CAAC;QAEH,wBAAwB;QACxB,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YACvC,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;YACxD,MAAM,KAAK,CAAC;QACd,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;YAC7C,IAAI,IAAI,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,mCAAmC,IAAI,eAAe,MAAM,EAAE,CAAC,CAAC;QAC9F,CAAC,CAAC,CAAC;QAEH,uCAAuC;QACvC,qDAAqD;QACrD,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,MAAO,CAA0C,CAAC;QAClG,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,KAAM,CAAC,CAAC;QAEzD,wCAAwC;QACxC,MAAM,MAAM,GAAG,GAAG,CAAC,YAAY,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAE/C,6DAA6D;QAC7D,IAAI,CAAC,UAAU,GAAG,IAAI,GAAG,CAAC,oBAAoB,CAC5C,CAAC,KAAgB,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,EAC9C,MAAM,CACP,CAAC;QAEF,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;QAChD,CAAC;QAED,4BAA4B;QAC5B,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;IAC1B,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,MAAiB;QACpC,OAAO;YACL,wCAAwC;YACxC,aAAa,EAAE,KAAK,EAAE,MAA+B,EAAiB,EAAE;gBACtE,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;oBAC/B,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;gBACrC,CAAC;YACH,CAAC;YAED,4CAA4C;YAC5C,iBAAiB,EAAE,KAAK,EACtB,MAAoC,EACI,EAAE;gBAC1C,8DAA8D;gBAC9D,kDAAkD;gBAClD,IAAI,IAAI,CAAC,KAAK;oBAAE,OAAO,CAAC,GAAG,CAAC,uBAAuB,EAAE,MAAM,CAAC,CAAC;gBAE7D,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAChD,OAAO;wBACL,OAAO,EAAE;4BACP,OAAO,EAAE,UAAU;4BACnB,QAAQ,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE;yBAC1C;qBACF,CAAC;gBACJ,CAAC;gBAED,kCAAkC;gBAClC,OAAO;oBACL,OAAO,EAAE;wBACP,OAAO,EAAE,WAAW;qBACrB;iBACF,CAAC;YACJ,CAAC;SACF,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,UAAU;QACtB,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC1D,CAAC;QAED,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;QAE9D,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;YACpD,eAAe,EAAE,GAAG,CAAC,gBAAgB;YACrC,UAAU,EAAE;gBACV,IAAI,EAAE,mBAAmB;gBACzB,OAAO,EAAE,OAAO;aACjB;YACD,kBAAkB,EAAE,EAAE;SACvB,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACrF,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,aAAa,CAAC,QAA+B;QACjD,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC1D,CAAC;QAED,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;QAEvD,0CAA0C;QAC1C,MAAM,UAAU,GAAoB,EAAE,CAAC;QAEvC,wDAAwD;QACxD,IAAI,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjD,IAAI,IAAI,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;YAE1E,IAAI,CAAC,SAAS,GAAG,IAAI,eAAe,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;YAE1D,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;YAE/C,IAAI,IAAI,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,0BAA0B,SAAS,EAAE,CAAC,CAAC;YAEnE,kDAAkD;YAClD,UAAU,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,aAAa;gBACnB,GAAG,EAAE,SAAS;gBACd,OAAO,EAAE,EAAE;aACZ,CAAC,CAAC;QACL,CAAC;QAED,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;YACvD,GAAG,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;YAC5D,UAAU;SACX,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,GAAG,eAAe,CAAC,SAAS,CAAC;QAC3C,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YAClD,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,eAAe,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;QAC5E,CAAC;QAED,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,UAAU,CAAC,MAAc,EAAE,EAAE,YAAY,GAAG,KAAK,KAA6C,EAAE,YAAY,EAAE,KAAK,EAAE;QAEzH,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACxC,MAAM,IAAI,KAAK,CAAC,wEAAwE,CAAC,CAAC;QAC5F,CAAC;QAED,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,sBAAsB,MAAM,GAAG,CAAC,CAAC;QAE7D,iCAAiC;QACjC,IAAI,eAAe,GAAG,EAAE,CAAC;QACzB,sEAAsE;QACtE,IAAI,iBAAiB,GAAG,EAAE,CAAC;QAC3B,kCAAkC;QAClC,IAAI,eAAe,GAAG,KAAK,CAAC;QAE5B,wCAAwC;QACxC,MAAM,gBAAgB,GAAG,IAAI,CAAC,qBAAqB,CAAC;QACpD,MAAM,mBAAmB,GAAG,CAAC,MAA+B,EAAE,EAAE;YAC9D,kEAAkE;YAClE,IAAI,MAAM,CAAC,MAAM,CAAC,aAAa,KAAK,WAAW,EAAE,CAAC;gBAChD,eAAe,GAAG,IAAI,CAAC;gBACvB,wDAAwD;gBACxD,iBAAiB,GAAG,EAAE,CAAC;YACzB,CAAC;YAED,0BAA0B;YAC1B,IAAI,MAAM,CAAC,MAAM,CAAC,aAAa,KAAK,qBAAqB,IAAI,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACnG,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC;gBAC7C,eAAe,IAAI,SAAS,CAAC;gBAE7B,gEAAgE;gBAChE,0DAA0D;gBAC1D,IAAI,eAAe,EAAE,CAAC;oBACpB,iBAAiB,IAAI,SAAS,CAAC;gBACjC,CAAC;YACH,CAAC;YAED,+CAA+C;YAC/C,IAAI,gBAAgB,EAAE,CAAC;gBACrB,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC,CAAC;QAEF,mCAAmC;QACnC,IAAI,CAAC,qBAAqB,GAAG,mBAAmB,CAAC;QAEjD,IAAI,CAAC;YACH,uBAAuB;YACvB,MAAM,OAAO,GAAuB;gBAClC;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,MAAM;iBACb;aACF,CAAC;YAEF,wCAAwC;YACxC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;gBAC5C,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,MAAM,EAAE,OAAO;aAChB,CAAC,CAAC;YAEH,IAAI,IAAI,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,oCAAoC,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;YAEvF,4BAA4B;YAC5B,IAAI,CAAC,qBAAqB,GAAG,gBAAgB,CAAC;YAE9C,sFAAsF;YACtF,IAAI,YAAY,IAAI,eAAe,EAAE,CAAC;gBACpC,OAAO,iBAAiB,CAAC;YAC3B,CAAC;YACD,OAAO,eAAe,CAAC;QACzB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAC;YAC7C,2CAA2C;YAC3C,IAAI,CAAC,qBAAqB,GAAG,gBAAgB,CAAC;YAC9C,4BAA4B;YAC5B,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;YACnB,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,QAAmD;QACjE,IAAI,CAAC,qBAAqB,GAAG,QAAQ,CAAC;IACxC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM;QACV,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACxC,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAClD,CAAC;QAED,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;QAE9D,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;YAC3B,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;IACpD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;QAEtD,gFAAgF;QAChF,IAAI,CAAC,WAAW,EAAE,CAAC;QAEnB,sEAAsE;QACtE,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,IAAI,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;YACvD,+DAA+D;YAC/D,MAAM,OAAO,CAAC,IAAI,CAAC;gBACjB,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE;gBACrB,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;aACzD,CAAC,CAAC;YACH,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACxB,CAAC;QAED,4BAA4B;QAC5B,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;YACvB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACxB,CAAC;QAED,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;IACpD,CAAC;IAED;;OAEG;IACK,WAAW;QACjB,IAAI,IAAI,CAAC,aAAa,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;YACrD,IAAI,IAAI,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;YAC1D,wCAAwC;YACxC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;CACF;AAED,OAAO,EAAE,MAAM,EAAE,CAAC"}

package/dist/auggie/sdk-mcp-server.d.ts

@@ -0,0 +1,12 @@
+import { type Tool } from "ai";
+export declare class AuggieMCPServer {
+    private server;
+    private httpServer;
+    private PORT;
+    private serverPATH;
+    private activeSockets;
+    constructor(tools: Record<string, Tool>, PORT: number | undefined);
+    start(): Promise<string>;
+    stop(): Promise<void>;
+}
+//# sourceMappingURL=sdk-mcp-server.d.ts.map

package/dist/auggie/sdk-mcp-server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sdk-mcp-server.d.ts","sourceRoot":"","sources":["../../src/auggie/sdk-mcp-server.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,IAAI,EAAE,MAAM,IAAI,CAAC;AA2B/B,qBAAa,eAAe;IAC1B,OAAO,CAAC,MAAM,CAAY;IAC1B,OAAO,CAAC,UAAU,CAA4B;IAC9C,OAAO,CAAC,IAAI,CAAqB;IACjC,OAAO,CAAC,UAAU,CAAU;IAC5B,OAAO,CAAC,aAAa,CAAwC;gBAEjD,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,GAAG,SAAS;IA+B3D,KAAK,IAAI,OAAO,CAAC,MAAM,CAAC;IAyCxB,IAAI;CA6BX"}

package/dist/auggie/sdk-mcp-server.js

@@ -0,0 +1,124 @@
+import { MCPServer } from "@mastra/mcp";
+import {} from "ai";
+import * as http from "http";
+import * as net from "net";
+/**
+ * Find a free port by creating a temporary server on port 0
+ * and getting the assigned port number
+ */
+async function findFreePort() {
+    return new Promise((resolve, reject) => {
+        const server = net.createServer();
+        server.unref();
+        server.on('error', reject);
+        server.listen(0, () => {
+            const address = server.address();
+            if (address && typeof address !== 'string') {
+                const port = address.port;
+                server.close(() => {
+                    resolve(port);
+                });
+            }
+            else {
+                reject(new Error('Failed to get port from server address'));
+            }
+        });
+    });
+}
+export class AuggieMCPServer {
+    constructor(tools, PORT) {
+        this.httpServer = null;
+        this.serverPATH = "/mcp";
+        this.activeSockets = new Set();
+        // Wrap AI SDK tools to match Mastra's expected signature
+        // Mastra wraps the actual tool parameters in a 'context' property
+        // AI SDK tools expect: execute({ param1, param2 })
+        // Mastra passes: execute({ context: { param1, param2 }, runtimeContext, ... }, mcpContext)
+        const wrappedTools = {};
+        for (const [name, tool] of Object.entries(tools)) {
+            wrappedTools[name] = {
+                ...tool,
+                execute: async (args, mcpContext) => {
+                    // Extract the actual tool parameters from the 'context' property
+                    const toolParams = args?.context || args;
+                    // Call the original AI SDK tool's execute with the unwrapped parameters
+                    // AI SDK tools may expect (params, options) signature
+                    if (tool.execute) {
+                        return await tool.execute(toolParams, mcpContext);
+                    }
+                    throw new Error(`Tool ${name} does not have an execute function`);
+                },
+            };
+        }
+        this.server = new MCPServer({
+            name: "auggie-mcp",
+            version: "1.0.0",
+            tools: wrappedTools,
+        });
+        this.PORT = PORT;
+    }
+    async start() {
+        // If PORT is undefined, find a free port
+        if (this.PORT === undefined) {
+            this.PORT = await findFreePort();
+        }
+        const httpServer = http.createServer(async (req, res) => {
+            await this.server.startHTTP({
+                url: new URL(req.url || "", `http://localhost:${this.PORT}`),
+                httpPath: this.serverPATH,
+                req,
+                res,
+                options: {
+                    sessionIdGenerator: undefined,
+                },
+            });
+        });
+        // Store the server instance
+        this.httpServer = httpServer;
+        // Track active connections for faster shutdown
+        httpServer.on('connection', (socket) => {
+            this.activeSockets.add(socket);
+            socket.on('close', () => {
+                this.activeSockets.delete(socket);
+            });
+        });
+        // Wait for the server to start listening before returning
+        await new Promise((resolve, reject) => {
+            httpServer.listen(this.PORT, () => {
+                resolve();
+            });
+            httpServer.on('error', reject);
+        });
+        const serverURL = new URL(this.serverPATH, `http://localhost:${this.PORT}`);
+        return serverURL.toString();
+    }
+    async stop() {
+        // Close the HTTP server first with a timeout
+        if (this.httpServer) {
+            await Promise.race([
+                new Promise((resolve, reject) => {
+                    this.httpServer.close((err) => {
+                        if (err)
+                            reject(err);
+                        else
+                            resolve();
+                    });
+                }),
+                // Timeout after 1 second
+                new Promise((resolve) => setTimeout(resolve, 1000))
+            ]);
+            // Force close by destroying all connections if still open
+            if (this.httpServer.listening) {
+                this.httpServer.closeAllConnections?.();
+            }
+            this.httpServer = null;
+        }
+        // Then close the MCP server with a timeout
+        await Promise.race([
+            this.server.close(),
+            // Timeout after 1 second
+            new Promise((resolve) => setTimeout(resolve, 1000))
+        ]);
+    }
+}
+//# sourceMappingURL=sdk-mcp-server.js.map

package/dist/auggie/sdk-mcp-server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sdk-mcp-server.js","sourceRoot":"","sources":["../../src/auggie/sdk-mcp-server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAa,MAAM,IAAI,CAAC;AAC/B,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,GAAG,MAAM,KAAK,CAAC;AAE3B;;;GAGG;AACH,KAAK,UAAU,YAAY;IACzB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,MAAM,GAAG,GAAG,CAAC,YAAY,EAAE,CAAC;QAClC,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,EAAE;YACpB,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;YACjC,IAAI,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;gBAC3C,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;gBAC1B,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE;oBAChB,OAAO,CAAC,IAAI,CAAC,CAAC;gBAChB,CAAC,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,OAAO,eAAe;IAO1B,YAAY,KAA2B,EAAE,IAAwB;QALzD,eAAU,GAAuB,IAAI,CAAC;QAEtC,eAAU,GAAG,MAAM,CAAC;QACpB,kBAAa,GAA8B,IAAI,GAAG,EAAE,CAAC;QAG3D,yDAAyD;QACzD,kEAAkE;QAClE,mDAAmD;QACnD,2FAA2F;QAC3F,MAAM,YAAY,GAAwB,EAAE,CAAC;QAC7C,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACjD,YAAY,CAAC,IAAI,CAAC,GAAG;gBACnB,GAAG,IAAI;gBACP,OAAO,EAAE,KAAK,EAAE,IAAS,EAAE,UAAgB,EAAE,EAAE;oBAC7C,iEAAiE;oBACjE,MAAM,UAAU,GAAG,IAAI,EAAE,OAAO,IAAI,IAAI,CAAC;oBAEzC,wEAAwE;oBACxE,sDAAsD;oBACtD,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;wBACjB,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;oBACpD,CAAC;oBACD,MAAM,IAAI,KAAK,CAAC,QAAQ,IAAI,oCAAoC,CAAC,CAAC;gBACpE,CAAC;aACF,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC;YAC1B,IAAI,EAAE,YAAY;YAClB,OAAO,EAAE,OAAO;YAChB,KAAK,EAAE,YAAY;SACpB,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,KAAK;QACT,yCAAyC;QACzC,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAC5B,IAAI,CAAC,IAAI,GAAG,MAAM,YAAY,EAAE,CAAC;QACnC,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,GAAyB,EAAE,GAAwB,EAAE,EAAE;YACjG,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;gBAC1B,GAAG,EAAE,IAAI,GAAG,CAAC,GAAG,CAAC,GAAG,IAAI,EAAE,EAAE,oBAAoB,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC5D,QAAQ,EAAE,IAAI,CAAC,UAAU;gBACzB,GAAG;gBACH,GAAG;gBACH,OAAO,EAAE;oBACP,kBAAkB,EAAE,SAAS;iBAC9B;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,4BAA4B;QAC5B,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,+CAA+C;QAC/C,UAAU,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,MAAkB,EAAE,EAAE;YACjD,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC/B,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBACtB,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACpC,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,0DAA0D;QAC1D,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC1C,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE;gBAChC,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;YACH,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,UAAU,EAAE,oBAAoB,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAE5E,OAAO,SAAS,CAAC,QAAQ,EAAE,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,IAAI;QACR,6CAA6C;QAC7C,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,MAAM,OAAO,CAAC,IAAI,CAAC;gBACjB,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;oBACpC,IAAI,CAAC,UAAW,CAAC,KAAK,CAAC,CAAC,GAAW,EAAE,EAAE;wBACrC,IAAI,GAAG;4BAAE,MAAM,CAAC,GAAG,CAAC,CAAC;;4BAChB,OAAO,EAAE,CAAC;oBACjB,CAAC,CAAC,CAAC;gBACL,CAAC,CAAC;gBACF,yBAAyB;gBACzB,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;aAC1D,CAAC,CAAC;YAEH,0DAA0D;YAC1D,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC;gBAC9B,IAAI,CAAC,UAAU,CAAC,mBAAmB,EAAE,EAAE,CAAC;YAC1C,CAAC;YAED,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACzB,CAAC;QAED,2CAA2C;QAC3C,MAAM,OAAO,CAAC,IAAI,CAAC;YACjB,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE;YACnB,yBAAyB;YACzB,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;SAC1D,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/index.d.ts

@@ -0,0 +1,2 @@
+export { Auggie } from "./auggie/sdk-acp-client";
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC"}

package/dist/index.js

@@ -0,0 +1,2 @@
+export { Auggie } from "./auggie/sdk-acp-client";
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC"}

package/package.json

@@ -0,0 +1,39 @@
+{
+  "name": "@augmentcode/auggie-sdk",
+  "version": "0.1.0",
+  "description": "TypeScript SDK for Auggie",
+  "type": "module",
+  "main": "./dist/index.js",
+  "module": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "build": "tsc --project tsconfig.build.json",
+    "dev": "bun ./examples/dev.ts",
+    "silent": "bun ./examples/silent.ts",
+    "prepare": "husky"
+  },
+  "devDependencies": {
+    "@biomejs/biome": "2.3.4",
+    "@types/bun": "latest",
+    "husky": "^9.1.7",
+    "ultracite": "6.3.0"
+  },
+  "peerDependencies": {
+    "typescript": "^5"
+  },
+  "dependencies": {
+    "@agentclientprotocol/sdk": "^0.5.1",
+    "@mastra/mcp": "^0.14.1",
+    "ai": "^5.0.86",
+    "zod": "^4.1.12"
+  }
+}