@bytespell/amux-client 0.0.19

package/dist/types.d.ts

@@ -0,0 +1,166 @@
+import type { SessionUpdate } from '@bytespell/amux-types';
+export type { SessionUpdate };
+/**
+ * Content block types for accumulated messages
+ */
+export type ContentBlock = {
+    type: 'text';
+    text: string;
+} | {
+    type: 'thinking';
+    text: string;
+} | {
+    type: 'tool_use';
+    id: string;
+    name: string;
+    input: unknown;
+} | {
+    type: 'tool_result';
+    toolUseId: string;
+    content: unknown;
+    isError?: boolean;
+} | {
+    type: 'diff';
+    path: string;
+    oldText: string;
+    newText: string;
+} | {
+    type: 'image';
+    source: {
+        type: 'base64';
+        mediaType: string;
+        data: string;
+    };
+};
+/**
+ * Accumulated message
+ */
+export interface Message {
+    id: string;
+    role: 'user' | 'assistant';
+    content: ContentBlock[];
+    status: 'streaming' | 'complete';
+    timestamp: number;
+}
+/**
+ * Tool call state tracking
+ */
+export interface ToolCallState {
+    id: string;
+    name: string;
+    input: unknown;
+    status: 'pending' | 'running' | 'complete' | 'error';
+    result?: unknown;
+    error?: string;
+}
+/**
+ * Turn state - groups user message and assistant response
+ */
+export interface Turn {
+    id: string;
+    userMessage: Message | null;
+    assistantMessage: Message | null;
+    toolCalls: ToolCallState[];
+    status: 'idle' | 'streaming' | 'awaiting_permission' | 'complete';
+}
+/**
+ * Connection status
+ */
+export type ConnectionStatus = 'disconnected' | 'connecting' | 'connected' | 'ready';
+/**
+ * Permission option (from server)
+ */
+export interface PermissionOptionData {
+    id?: string;
+    optionId?: string;
+    label?: string;
+    title?: string;
+    description?: string;
+}
+/**
+ * Client event types
+ */
+export interface AmuxClientEvents {
+    ready: {
+        cwd: string;
+        sessionId: string | null;
+        capabilities: unknown;
+        agent: {
+            type: string;
+            name: string;
+        };
+        availableAgents: Array<{
+            id: string;
+            name: string;
+        }>;
+        availableModels?: Array<{
+            id: string;
+            name: string;
+            provider?: string;
+            supportsThinking?: boolean;
+        }>;
+        currentModelId?: string;
+    };
+    connecting: Record<string, never>;
+    update: SessionUpdate;
+    turn_start: Record<string, never>;
+    turn_end: Record<string, never>;
+    permission_request: {
+        requestId: string;
+        toolCall?: unknown;
+        options: PermissionOptionData[];
+    };
+    prompt_complete: {
+        stopReason?: string;
+    };
+    session_created: {
+        sessionId: string | null;
+    };
+    session_switched: {
+        sessionId: string;
+    };
+    history_replay: {
+        previousSessionId: string;
+        events: unknown[];
+        eventCount: number;
+    };
+    history: {
+        events: unknown[];
+        sessionId: string | null;
+    };
+    sessions: {
+        sessions: unknown[];
+    };
+    error: {
+        message: string;
+    };
+    agent_exit: {
+        code: number | null;
+        signal: string | null;
+    };
+    connection_status: {
+        status: ConnectionStatus;
+    };
+    messages_updated: {
+        messages: Message[];
+    };
+    turns_updated: {
+        turns: Turn[];
+    };
+}
+/**
+ * Options for creating an AmuxClient
+ */
+export interface AmuxClientOptions {
+    /** WebSocket URL to connect to */
+    url: string;
+    /** Auto-connect on creation (default: true) */
+    autoConnect?: boolean;
+    /** Auto-reconnect on disconnect (default: true) */
+    reconnect?: boolean;
+    /** Reconnect interval in ms (default: 1000) */
+    reconnectInterval?: number;
+    /** Max reconnect attempts (default: Infinity) */
+    maxReconnectAttempts?: number;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAG3D,YAAY,EAAE,aAAa,EAAE,CAAC;AAE9B;;GAEG;AACH,MAAM,MAAM,YAAY,GACpB;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GAC9B;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GAClC;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,EAAE,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,OAAO,CAAA;CAAE,GAC9D;IAAE,IAAI,EAAE,aAAa,CAAC;IAAC,SAAS,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,OAAO,CAAC;IAAC,OAAO,CAAC,EAAE,OAAO,CAAA;CAAE,GAC/E;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GAChE;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,MAAM,EAAE;QAAE,IAAI,EAAE,QAAQ,CAAC;QAAC,SAAS,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,CAAC;AAEnF;;GAEG;AACH,MAAM,WAAW,OAAO;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC;IAC3B,OAAO,EAAE,YAAY,EAAE,CAAC;IACxB,MAAM,EAAE,WAAW,GAAG,UAAU,CAAC;IACjC,SAAS,EAAE,MAAM,CAAC;CACnB;AAED;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,OAAO,CAAC;IACf,MAAM,EAAE,SAAS,GAAG,SAAS,GAAG,UAAU,GAAG,OAAO,CAAC;IACrD,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED;;GAEG;AACH,MAAM,WAAW,IAAI;IACnB,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,OAAO,GAAG,IAAI,CAAC;IAC5B,gBAAgB,EAAE,OAAO,GAAG,IAAI,CAAC;IACjC,SAAS,EAAE,aAAa,EAAE,CAAC;IAC3B,MAAM,EAAE,MAAM,GAAG,WAAW,GAAG,qBAAqB,GAAG,UAAU,CAAC;CACnE;AAED;;GAEG;AACH,MAAM,MAAM,gBAAgB,GAAG,cAAc,GAAG,YAAY,GAAG,WAAW,GAAG,OAAO,CAAC;AAErF;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,KAAK,EAAE;QACL,GAAG,EAAE,MAAM,CAAC;QACZ,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;QACzB,YAAY,EAAE,OAAO,CAAC;QACtB,KAAK,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,IAAI,EAAE,MAAM,CAAA;SAAE,CAAC;QACtC,eAAe,EAAE,KAAK,CAAC;YAAE,EAAE,EAAE,MAAM,CAAC;YAAC,IAAI,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QACrD,eAAe,CAAC,EAAE,KAAK,CAAC;YAAE,EAAE,EAAE,MAAM,CAAC;YAAC,IAAI,EAAE,MAAM,CAAC;YAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;YAAC,gBAAgB,CAAC,EAAE,OAAO,CAAA;SAAE,CAAC,CAAC;QACrG,cAAc,CAAC,EAAE,MAAM,CAAC;KACzB,CAAC;IACF,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAClC,MAAM,EAAE,aAAa,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAClC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAChC,kBAAkB,EAAE;QAClB,SAAS,EAAE,MAAM,CAAC;QAClB,QAAQ,CAAC,EAAE,OAAO,CAAC;QACnB,OAAO,EAAE,oBAAoB,EAAE,CAAC;KACjC,CAAC;IACF,eAAe,EAAE;QAAE,UAAU,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IACzC,eAAe,EAAE;QAAE,SAAS,EAAE,MAAM,GAAG,IAAI,CAAA;KAAE,CAAC;IAC9C,gBAAgB,EAAE;QAAE,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC;IACxC,cAAc,EAAE;QACd,iBAAiB,EAAE,MAAM,CAAC;QAC1B,MAAM,EAAE,OAAO,EAAE,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;KACpB,CAAC;IACF,OAAO,EAAE;QAAE,MAAM,EAAE,OAAO,EAAE,CAAC;QAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAA;KAAE,CAAC;IACzD,QAAQ,EAAE;QAAE,QAAQ,EAAE,OAAO,EAAE,CAAA;KAAE,CAAC;IAClC,KAAK,EAAE;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC;IAC3B,UAAU,EAAE;QAAE,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;QAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAA;KAAE,CAAC;IAC3D,iBAAiB,EAAE;QAAE,MAAM,EAAE,gBAAgB,CAAA;KAAE,CAAC;IAChD,gBAAgB,EAAE;QAAE,QAAQ,EAAE,OAAO,EAAE,CAAA;KAAE,CAAC;IAC1C,aAAa,EAAE;QAAE,KAAK,EAAE,IAAI,EAAE,CAAA;KAAE,CAAC;CAClC;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,kCAAkC;IAClC,GAAG,EAAE,MAAM,CAAC;IACZ,+CAA+C;IAC/C,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,mDAAmD;IACnD,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,+CAA+C;IAC/C,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,iDAAiD;IACjD,oBAAoB,CAAC,EAAE,MAAM,CAAC;CAC/B"}

package/dist/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":""}

package/package.json

@@ -0,0 +1,48 @@
+{
+  "name": "@bytespell/amux-client",
+  "version": "0.0.19",
+  "description": "Client library for consuming amux over WebSocket",
+  "type": "module",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    },
+    "./react": {
+      "types": "./dist/react/index.d.ts",
+      "import": "./dist/react/index.js"
+    }
+  },
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "typecheck": "tsc --noEmit"
+  },
+  "keywords": [
+    "amux",
+    "client",
+    "websocket",
+    "acp",
+    "react"
+  ],
+  "author": "",
+  "license": "MIT",
+  "dependencies": {
+    "@bytespell/amux-types": "*"
+  },
+  "peerDependencies": {
+    "react": "^18.0.0 || ^19.0.0"
+  },
+  "peerDependenciesMeta": {
+    "react": {
+      "optional": true
+    }
+  },
+  "devDependencies": {
+    "@types/react": "^18.0.0",
+    "react": "^18.0.0",
+    "typescript": "~5.7.0"
+  }
+}

package/src/accumulator.ts

@@ -0,0 +1,307 @@
+import type { SessionUpdate } from '@bytespell/amux-types';
+import type { Message, Turn, ToolCallState } from './types.js';
+
+/**
+ * Generate a unique ID
+ */
+function generateId(): string {
+  return Math.random().toString(36).substring(2, 15);
+}
+
+/**
+ * Message and Turn accumulator
+ *
+ * Processes streaming session updates and builds structured conversation data.
+ */
+export class Accumulator {
+  private _messages: Message[] = [];
+  private _turns: Turn[] = [];
+  private _currentTurn: Turn | null = null;
+
+  /**
+   * All accumulated messages
+   */
+  get messages(): Message[] {
+    return this._messages;
+  }
+
+  /**
+   * All turns (grouped user/assistant pairs)
+   */
+  get turns(): Turn[] {
+    return this._turns;
+  }
+
+  /**
+   * Current active turn (if streaming)
+   */
+  get currentTurn(): Turn | null {
+    return this._currentTurn;
+  }
+
+  /**
+   * Last user message
+   */
+  get lastUserMessage(): Message | null {
+    for (let i = this._messages.length - 1; i >= 0; i--) {
+      if (this._messages[i]?.role === 'user') {
+        return this._messages[i] ?? null;
+      }
+    }
+    return null;
+  }
+
+  /**
+   * Last assistant message
+   */
+  get lastAssistantMessage(): Message | null {
+    for (let i = this._messages.length - 1; i >= 0; i--) {
+      if (this._messages[i]?.role === 'assistant') {
+        return this._messages[i] ?? null;
+      }
+    }
+    return null;
+  }
+
+  /**
+   * Start a new turn
+   */
+  startTurn(): void {
+    this._currentTurn = {
+      id: generateId(),
+      userMessage: null,
+      assistantMessage: null,
+      toolCalls: [],
+      status: 'streaming',
+    };
+    this._turns.push(this._currentTurn);
+  }
+
+  /**
+   * End the current turn
+   */
+  endTurn(): void {
+    if (this._currentTurn) {
+      this._currentTurn.status = 'complete';
+      if (this._currentTurn.userMessage) {
+        this._currentTurn.userMessage.status = 'complete';
+      }
+      if (this._currentTurn.assistantMessage) {
+        this._currentTurn.assistantMessage.status = 'complete';
+      }
+      this._currentTurn = null;
+    }
+  }
+
+  /**
+   * Set turn status to awaiting permission
+   */
+  setAwaitingPermission(): void {
+    if (this._currentTurn) {
+      this._currentTurn.status = 'awaiting_permission';
+    }
+  }
+
+  /**
+   * Resume streaming after permission
+   */
+  resumeStreaming(): void {
+    if (this._currentTurn && this._currentTurn.status === 'awaiting_permission') {
+      this._currentTurn.status = 'streaming';
+    }
+  }
+
+  /**
+   * Process a session update
+   */
+  processUpdate(update: SessionUpdate): void {
+    const updateType = update.sessionUpdate;
+
+    switch (updateType) {
+      case 'user_message_chunk':
+        this.processUserChunk(update);
+        break;
+      case 'agent_message_chunk':
+        this.processAssistantChunk(update);
+        break;
+      case 'tool_call':
+        this.processToolCall(update);
+        break;
+      case 'tool_call_update':
+        this.processToolCallUpdate(update);
+        break;
+    }
+  }
+
+  /**
+   * Process user message chunk
+   */
+  private processUserChunk(update: SessionUpdate): void {
+    if (update.sessionUpdate !== 'user_message_chunk') return;
+
+    // Create or get current turn
+    if (!this._currentTurn) {
+      this.startTurn();
+    }
+
+    // Create user message if needed
+    if (!this._currentTurn!.userMessage) {
+      const userMessage: Message = {
+        id: generateId(),
+        role: 'user',
+        content: [],
+        status: 'streaming',
+        timestamp: Date.now(),
+      };
+      this._currentTurn!.userMessage = userMessage;
+      this._messages.push(userMessage);
+    }
+
+    // Add content
+    const content = update.content as { type: string; text?: string };
+    if (content?.type === 'text' && content.text) {
+      const lastBlock = this._currentTurn!.userMessage!.content[
+        this._currentTurn!.userMessage!.content.length - 1
+      ];
+      if (lastBlock?.type === 'text') {
+        lastBlock.text += content.text;
+      } else {
+        this._currentTurn!.userMessage!.content.push({
+          type: 'text',
+          text: content.text,
+        });
+      }
+    }
+  }
+
+  /**
+   * Process assistant message chunk
+   */
+  private processAssistantChunk(update: SessionUpdate): void {
+    if (update.sessionUpdate !== 'agent_message_chunk') return;
+
+    // Create turn if needed
+    if (!this._currentTurn) {
+      this.startTurn();
+    }
+
+    // Create assistant message if needed
+    if (!this._currentTurn!.assistantMessage) {
+      const assistantMessage: Message = {
+        id: generateId(),
+        role: 'assistant',
+        content: [],
+        status: 'streaming',
+        timestamp: Date.now(),
+      };
+      this._currentTurn!.assistantMessage = assistantMessage;
+      this._messages.push(assistantMessage);
+    }
+
+    // Add content based on type
+    const content = update.content as { type: string; text?: string; [key: string]: unknown };
+    if (!content) return;
+
+    const assistantContent = this._currentTurn!.assistantMessage!.content;
+
+    if (content.type === 'text' && content.text) {
+      const lastBlock = assistantContent[assistantContent.length - 1];
+      if (lastBlock?.type === 'text') {
+        lastBlock.text += content.text;
+      } else {
+        assistantContent.push({ type: 'text', text: content.text });
+      }
+    } else if (content.type === 'thinking' && content.text) {
+      const lastBlock = assistantContent[assistantContent.length - 1];
+      if (lastBlock?.type === 'thinking') {
+        lastBlock.text += content.text;
+      } else {
+        assistantContent.push({ type: 'thinking', text: content.text });
+      }
+    }
+  }
+
+  /**
+   * Process tool call
+   */
+  private processToolCall(update: SessionUpdate): void {
+    if (update.sessionUpdate !== 'tool_call') return;
+
+    if (!this._currentTurn) return;
+
+    const toolCall: ToolCallState = {
+      id: (update as { toolCallId?: string }).toolCallId ?? generateId(),
+      name: (update as { title?: string }).title ?? 'unknown',
+      input: (update as { input?: unknown }).input,
+      status: 'pending',
+    };
+
+    this._currentTurn.toolCalls.push(toolCall);
+
+    // Also add to assistant message content
+    if (this._currentTurn.assistantMessage) {
+      this._currentTurn.assistantMessage.content.push({
+        type: 'tool_use',
+        id: toolCall.id,
+        name: toolCall.name,
+        input: toolCall.input,
+      });
+    }
+  }
+
+  /**
+   * Process tool call update
+   */
+  private processToolCallUpdate(update: SessionUpdate): void {
+    if (update.sessionUpdate !== 'tool_call_update') return;
+
+    if (!this._currentTurn) return;
+
+    const toolCallId = (update as { toolCallId?: string }).toolCallId;
+    if (!toolCallId) return;
+
+    const toolCall = this._currentTurn.toolCalls.find((tc) => tc.id === toolCallId);
+    if (!toolCall) return;
+
+    const status = (update as { status?: string }).status;
+    if (status === 'running') {
+      toolCall.status = 'running';
+    } else if (status === 'completed') {
+      toolCall.status = 'complete';
+      toolCall.result = (update as { output?: unknown }).output;
+    } else if (status === 'error') {
+      toolCall.status = 'error';
+      toolCall.error = (update as { error?: string }).error;
+    }
+  }
+
+  /**
+   * Replay history events
+   */
+  replayHistory(events: unknown[]): void {
+    // Clear current state
+    this._messages = [];
+    this._turns = [];
+    this._currentTurn = null;
+
+    for (const event of events) {
+      const e = event as { type: string; update?: SessionUpdate };
+      if (e.type === 'turn_start') {
+        this.startTurn();
+      } else if (e.type === 'turn_end') {
+        this.endTurn();
+      } else if (e.type === 'session_update' && e.update) {
+        this.processUpdate(e.update);
+      }
+    }
+  }
+
+  /**
+   * Clear all accumulated data
+   */
+  clear(): void {
+    this._messages = [];
+    this._turns = [];
+    this._currentTurn = null;
+  }
+}