@amaster.ai/pi-computer-use 0.1.2-beta.1 → 0.1.2-beta.3

package/README.md

@@ -0,0 +1,113 @@
+# @amaster.ai/pi-computer-use
+
+pi-coding-agent extension that wraps [cua-driver-rs](https://github.com/trycua/cua/), exposing desktop automation tools with a `computer_use_` prefix.
+
+## Features
+
+- **Zero external dependencies** — pre-compiled cua-driver-rs binaries bundled for all platforms
+- **MCP stdio communication** — spawns `cua-driver mcp` via `StdioClientTransport`, JSON-RPC over stdio
+- **Dynamic tool discovery** — auto-discovers upstream MCP tools and registers with `computer_use_` prefix
+- **Smart tool filtering** — excludes non-essential tools (agent cursor, recording, config), exposes 18 core tools
+- **Optional visual analysis** — `computer_use_analyze_screenshot` via configurable vision model
+- **Cross-platform** — darwin-arm64/x64, linux-x64, win32-x64/arm64
+
+## Install
+
+```bash
+bun add @amaster.ai/pi-computer-use
+```
+
+Requires Node.js >= 20 and `@earendil-works/pi-coding-agent >= 0.74.0`.
+
+## Usage
+
+Install the package and pi-coding-agent will automatically discover and load the extension. All tools are registered on `session_start`.
+
+Configure via `.pi/settings.json` (project-level) or `~/.pi/agent/settings.json` (user-level) under the `"pi-computer-use"` key:
+
+```json
+{
+  "pi-computer-use": {
+    "mode": "bundled"
+  }
+}
+```
+
+## Configuration
+
+| Option | Type | Default | Description |
+|--------|------|---------|-------------|
+| `mode` | `'bundled' \| 'path'` | `'bundled'` | Binary resolution strategy |
+| `binaryPath` | `string` | — | Custom cua-driver binary path (requires `mode: 'path'`) |
+| `extraArgs` | `string[]` | — | Extra CLI arguments passed to cua-driver |
+| `visionModel` | `VisionModelConfig` | — | Enable visual screenshot analysis |
+
+### Vision Model (Optional)
+
+Enable `computer_use_analyze_screenshot` by referencing a model already configured in Pi's model registry (`models.json`):
+
+```json
+{
+  "pi-computer-use": {
+    "visionModel": {
+      "provider": "openai",
+      "model": "gpt-4o"
+    }
+  }
+}
+```
+
+The extension resolves API key, base URL, and headers from the model registry automatically — no need to duplicate credentials here.
+
+## Exposed Tools (18)
+
+### Core
+
+| Tool | Description |
+|------|-------------|
+| `computer_use_screenshot` | Capture screen |
+| `computer_use_click` | Click at coordinates |
+| `computer_use_type_text` | Type text |
+| `computer_use_press_key` | Press a keyboard key |
+| `computer_use_scroll` | Scroll in a direction |
+| `computer_use_hotkey` | Press key combination |
+
+### Common
+
+| Tool | Description |
+|------|-------------|
+| `computer_use_double_click` | Double-click |
+| `computer_use_right_click` | Right-click (context menu) |
+| `computer_use_drag` | Drag from one point to another |
+| `computer_use_get_screen_size` | Get screen dimensions |
+| `computer_use_get_accessibility_tree` | Get accessibility tree for element discovery |
+| `computer_use_set_value` | Set form field value directly |
+| `computer_use_get_cursor_position` | Get current cursor position |
+
+### Situational
+
+| Tool | Description |
+|------|-------------|
+| `computer_use_list_apps` | List running applications |
+| `computer_use_list_windows` | List open windows |
+| `computer_use_get_window_state` | Get window state/position |
+| `computer_use_launch_app` | Launch an application |
+| `computer_use_kill_app` | Kill an application |
+
+## Excluded Tools (15)
+
+Agent cursor styling, recording/replay, config management, and redundant operations (covered by other tools) are filtered out.
+
+## Supported Platforms
+
+| Platform | Binary |
+|----------|--------|
+| macOS ARM64 | `bin/darwin-arm64/cua-driver` |
+| macOS x64 | `bin/darwin-x64/cua-driver` |
+| Linux x64 | `bin/linux-x64/cua-driver` |
+| Windows x64 | `bin/win32-x64/cua-driver.exe` |
+| Windows ARM64 | `bin/win32-arm64/cua-driver.exe` |
+
+## License
+
+Apache-2.0

package/dist/config.d.ts

@@ -1,31 +1,18 @@
+import { type PiSettingsOptions } from '@amaster.ai/pi-shared/settings';
 export interface VisionModelConfig {
     provider: string;
     model: string;
-    apiKey?: string;
-    baseUrl?: string;
 }
 export interface ComputerUseConfig {
-    /** 'managed' spawns cua-computer-server, 'external' connects to existing */
-    mode?: 'managed' | 'external';
-    /** Command to start cua-computer-server (default: 'uvx') */
-    command?: string;
-    /** Package/module to run (default: 'cua-computer-server') */
-    package?: string;
-    /** Extra CLI args for cua-computer-server */
+    /** 'bundled' uses the packaged binary, 'path' uses a custom binary path */
+    mode?: 'bundled' | 'path';
+    /** Custom cua-driver binary path (used when mode is 'path') */
+    binaryPath?: string;
+    /** Extra CLI args passed to cua-driver mcp */
     extraArgs?: string[];
-    /** Server host (default: '127.0.0.1') */
-    host?: string;
-    /** Server port (default: 8000) */
-    port?: number;
-    /** API key — enables wss + auth for cloud/remote */
-    apiKey?: string;
-    /** VM name for cloud routing */
-    vmName?: string;
-    /** Auto-screenshot after each action (default: true) */
-    autoScreenshot?: boolean;
     /** Vision model for screenshot analysis */
     visionModel?: VisionModelConfig;
 }
 export declare function resolveConfig(config?: ComputerUseConfig): ComputerUseConfig;
-export declare function loadConfigFromFile(): ComputerUseConfig;
+export declare function loadConfigFromFile(options?: PiSettingsOptions): ComputerUseConfig;
 //# sourceMappingURL=config.d.ts.map

package/dist/config.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,iBAAiB;IAChC,4EAA4E;IAC5E,IAAI,CAAC,EAAE,SAAS,GAAG,UAAU,CAAC;IAE9B,4DAA4D;IAC5D,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,6DAA6D;IAC7D,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,6CAA6C;IAC7C,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IAErB,yCAAyC;IACzC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,kCAAkC;IAClC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,oDAAoD;IACpD,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,gCAAgC;IAChC,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB,wDAAwD;IACxD,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,2CAA2C;IAC3C,WAAW,CAAC,EAAE,iBAAiB,CAAC;CACjC;AAWD,wBAAgB,aAAa,CAAC,MAAM,CAAC,EAAE,iBAAiB,GAAG,iBAAiB,CAE3E;AAED,wBAAgB,kBAAkB,IAAI,iBAAiB,CAEtD"}
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAkB,KAAK,iBAAiB,EAAE,MAAM,gCAAgC,CAAC;AAGxF,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,iBAAiB;IAChC,2EAA2E;IAC3E,IAAI,CAAC,EAAE,SAAS,GAAG,MAAM,CAAC;IAC1B,+DAA+D;IAC/D,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,8CAA8C;IAC9C,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,2CAA2C;IAC3C,WAAW,CAAC,EAAE,iBAAiB,CAAC;CACjC;AAMD,wBAAgB,aAAa,CAAC,MAAM,CAAC,EAAE,iBAAiB,GAAG,iBAAiB,CAE3E;AAED,wBAAgB,kBAAkB,CAAC,OAAO,CAAC,EAAE,iBAAiB,GAAG,iBAAiB,CAKjF"}

package/dist/config.js

@@ -1,16 +1,15 @@
 import { loadPiSettings } from '@amaster.ai/pi-shared/settings';
+import { getAgentDir } from '@earendil-works/pi-coding-agent';
 const DEFAULTS = {
-    mode: 'managed',
-    command: 'uvx',
-    package: 'cua-computer-server',
-    host: '127.0.0.1',
-    port: 8000,
-    autoScreenshot: true,
+    mode: 'bundled',
 };
 export function resolveConfig(config) {
     return { ...DEFAULTS, ...config };
 }
-export function loadConfigFromFile() {
-    return loadPiSettings('pi-computer-use');
+export function loadConfigFromFile(options) {
+    return loadPiSettings('pi-computer-use', {
+        agentDir: getAgentDir(),
+        ...options,
+    });
 }
 //# sourceMappingURL=config.js.map

package/dist/config.js.map

@@ -1 +1 @@
-{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,gCAAgC,CAAC;AAmChE,MAAM,QAAQ,GAA+B;IAC3C,IAAI,EAAE,SAAS;IACf,OAAO,EAAE,KAAK;IACd,OAAO,EAAE,qBAAqB;IAC9B,IAAI,EAAE,WAAW;IACjB,IAAI,EAAE,IAAI;IACV,cAAc,EAAE,IAAI;CACrB,CAAC;AAEF,MAAM,UAAU,aAAa,CAAC,MAA0B;IACtD,OAAO,EAAE,GAAG,QAAQ,EAAE,GAAG,MAAM,EAAE,CAAC;AACpC,CAAC;AAED,MAAM,UAAU,kBAAkB;IAChC,OAAO,cAAc,CAAoB,iBAAiB,CAAC,CAAC;AAC9D,CAAC"}
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAA0B,MAAM,gCAAgC,CAAC;AACxF,OAAO,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;AAkB9D,MAAM,QAAQ,GAA+B;IAC3C,IAAI,EAAE,SAAS;CAChB,CAAC;AAEF,MAAM,UAAU,aAAa,CAAC,MAA0B;IACtD,OAAO,EAAE,GAAG,QAAQ,EAAE,GAAG,MAAM,EAAE,CAAC;AACpC,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,OAA2B;IAC5D,OAAO,cAAc,CAAoB,iBAAiB,EAAE;QAC1D,QAAQ,EAAE,WAAW,EAAE;QACvB,GAAG,OAAO;KACX,CAAC,CAAC;AACL,CAAC"}

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAoB,MAAM,iCAAiC,CAAC;AAItF,OAAO,EAAE,KAAK,iBAAiB,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAIxF,YAAY,EAAE,iBAAiB,EAAE,CAAC;AAClC,OAAO,EAAE,kBAAkB,EAAE,aAAa,EAAE,CAAC;AAE7C,MAAM,CAAC,OAAO,UAAU,oBAAoB,CAAC,EAAE,EAAE,YAAY,GAAG,IAAI,CA2HnE"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAoB,MAAM,iCAAiC,CAAC;AAEtF,OAAO,EAAE,KAAK,iBAAiB,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAIxF,YAAY,EAAE,iBAAiB,EAAE,CAAC;AAClC,OAAO,EAAE,kBAAkB,EAAE,aAAa,EAAE,CAAC;AAsB7C,MAAM,CAAC,OAAO,UAAU,oBAAoB,CAAC,EAAE,EAAE,YAAY,GAAG,IAAI,CA+InE"}

package/dist/index.js

@@ -1,101 +1,129 @@
 import { Type } from 'typebox';
-import { dispatchAction } from './actions.js';
-import { ComputerClient } from './computer-client.js';
 import { loadConfigFromFile, resolveConfig } from './config.js';
-import { ComputerServerProcess } from './server-process.js';
+import { CuaDriverClient } from './mcp-client.js';
 import { createPiVisionCaller } from './vision.js';
 export { loadConfigFromFile, resolveConfig };
+const TOOL_PREFIX = 'computer_use_';
+const EXCLUDED_TOOLS = new Set([
+    'set_agent_cursor_enabled',
+    'set_agent_cursor_motion',
+    'set_agent_cursor_style',
+    'get_agent_cursor_state',
+    'set_recording',
+    'get_recording_state',
+    'replay_trajectory',
+    'check_permissions',
+    'get_config',
+    'set_config',
+    'move_cursor',
+    'zoom',
+    'type_text_chars',
+    'page',
+    'browser_eval',
+]);
 export default function computerUseExtension(pi) {
-    const config = resolveConfig(loadConfigFromFile());
-    const serverProcess = new ComputerServerProcess();
-    const client = new ComputerClient(config);
-    let started = false;
-    async function ensureRunning() {
-        if (started)
-            return;
-        if (config.mode !== 'external') {
-            await serverProcess.start(config);
+    let config;
+    let client;
+    let connected = false;
+    async function ensureConnected() {
+        if (!client)
+            throw new Error('pi-computer-use: session not started');
+        if (!connected) {
+            await client.connect();
+            connected = true;
+        }
+    }
+    async function registerUpstreamTools() {
+        await ensureConnected();
+        const upstreamTools = await client.listAllTools();
+        for (const tool of upstreamTools) {
+            if (EXCLUDED_TOOLS.has(tool.name))
+                continue;
+            const prefixedName = `${TOOL_PREFIX}${tool.name}`;
+            const originalName = tool.name;
+            pi.registerTool({
+                name: prefixedName,
+                label: prefixedName,
+                description: tool.description ?? '',
+                parameters: Type.Unsafe(tool.inputSchema),
+                async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
+                    await ensureConnected();
+                    const result = await client.callTool(originalName, params);
+                    if (originalName === 'screenshot' && config?.visionModel) {
+                        const imageContent = result.content?.find((c) => c.type === 'image' && c.data);
+                        if (imageContent?.data) {
+                            const callVision = createPiVisionCaller(config.visionModel, ctx);
+                            const analysis = await callVision('Describe the full screen: identify all visible windows, UI elements, buttons, text fields, and their positions.', imageContent.data, imageContent.mimeType ?? 'image/png');
+                            return {
+                                content: [{ type: 'text', text: analysis }],
+                                details: undefined,
+                            };
+                        }
+                    }
+                    const content = [];
+                    if (result.content) {
+                        for (const item of result.content) {
+                            if (item.type === 'text' && item.text) {
+                                content.push({ type: 'text', text: item.text });
+                            }
+                        }
+                    }
+                    if (content.length === 0) {
+                        content.push({ type: 'text', text: 'Action executed.' });
+                    }
+                    return result.isError
+                        ? { content, details: undefined, isError: true }
+                        : { content, details: undefined };
+                },
+            });
         }
-        await client.connect();
-        started = true;
     }
-    pi.registerTool({
-        name: 'computer_use',
-        label: 'computer_use',
-        description: 'Control a computer desktop. Actions: screenshot, click, double_click, type, keypress, scroll, move, drag, wait, run_command. Each call executes one action and returns the resulting screen state.',
-        promptSnippet: 'computer_use — control a desktop: screenshot, click, type, scroll, keypress, drag, run_command',
-        parameters: Type.Object({
-            action: Type.Object({
-                type: Type.String({
-                    description: 'Action type: screenshot | click | double_click | type | keypress | scroll | move | drag | wait | run_command',
-                }),
-                x: Type.Optional(Type.Number({ description: 'X coordinate (click, double_click, move)' })),
-                y: Type.Optional(Type.Number({ description: 'Y coordinate (click, double_click, move)' })),
-                button: Type.Optional(Type.String({ description: 'Mouse button: left | right (default: left)' })),
-                text: Type.Optional(Type.String({ description: 'Text to type (type action)' })),
-                keys: Type.Optional(Type.Array(Type.String(), { description: 'Keys to press (keypress action)' })),
-                scroll_x: Type.Optional(Type.Number({ description: 'Horizontal scroll amount' })),
-                scroll_y: Type.Optional(Type.Number({ description: 'Vertical scroll amount' })),
-                path: Type.Optional(Type.Array(Type.Tuple([Type.Number(), Type.Number()]), {
-                    description: 'Drag path as [[x,y], ...] coordinates',
+    async function registerVisionTool() {
+        if (!config?.visionModel)
+            return;
+        const visionConfig = config.visionModel;
+        pi.registerTool({
+            name: `${TOOL_PREFIX}analyze_screenshot`,
+            label: `${TOOL_PREFIX}analyze_screenshot`,
+            description: 'Take a screenshot and analyze it visually using a vision model. Use when you need to identify elements by visual attributes (color, layout, position) or need precise pixel coordinates.',
+            parameters: Type.Object({
+                instruction: Type.Optional(Type.String({
+                    description: 'What to identify or analyze visually (e.g., "Find the coordinates of the blue submit button").',
                 })),
-                command: Type.Optional(Type.String({ description: 'Shell command (run_command action)' })),
             }),
-        }),
-        async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
-            await ensureRunning();
-            if (params.action.type === 'screenshot') {
-                const screenshotBase64 = await client.screenshot();
-                if (config.visionModel) {
-                    const callVision = createPiVisionCaller(config.visionModel, ctx);
-                    const analysis = await callVision('Describe the full screen: identify all visible windows, UI elements, buttons, text fields, and their positions.', screenshotBase64, 'image/png');
-                    return { content: [{ type: 'text', text: analysis }], details: undefined };
-                }
-                return {
-                    content: [
-                        {
-                            type: 'text',
-                            text: 'Screenshot captured (no vision model configured to analyze it).',
-                        },
-                    ],
-                    details: undefined,
-                };
-            }
-            const actionResult = await dispatchAction(client, params.action);
-            if (config.autoScreenshot !== false) {
-                const screenshotBase64 = await client.screenshot();
-                if (config.visionModel) {
-                    const callVision = createPiVisionCaller(config.visionModel, ctx);
-                    const analysis = await callVision('Describe the current screen state after the action. Focus on what changed and what is now visible.', screenshotBase64, 'image/png');
+            async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
+                await ensureConnected();
+                const screenshotResult = await client.callTool('screenshot', {});
+                const imageContent = screenshotResult.content?.find((c) => c.type === 'image' && c.data);
+                if (!imageContent?.data) {
                     return {
-                        content: [
-                            {
-                                type: 'text',
-                                text: `${actionResult}\n\nScreen state:\n${analysis}`,
-                            },
-                        ],
+                        content: [{ type: 'text', text: 'Failed to capture screenshot.' }],
                         details: undefined,
+                        isError: true,
                     };
                 }
+                const callVision = createPiVisionCaller(visionConfig, ctx);
+                const instruction = params.instruction ??
+                    'Describe the full screen: identify all visible windows, UI elements, buttons, text fields, and their positions.';
+                const analysis = await callVision(instruction, imageContent.data, imageContent.mimeType ?? 'image/png');
                 return {
-                    content: [{ type: 'text', text: actionResult ?? 'Action executed.' }],
+                    content: [{ type: 'text', text: analysis }],
                     details: undefined,
                 };
-            }
-            return {
-                content: [{ type: 'text', text: actionResult ?? 'Action executed.' }],
-                details: undefined,
-            };
-        },
-    });
-    pi.on('session_start', async () => {
-        // Lazy — actual startup happens on first tool call
+            },
+        });
+    }
+    pi.on('session_start', async (_event, ctx) => {
+        config = resolveConfig(loadConfigFromFile({ cwd: ctx.cwd }));
+        client = new CuaDriverClient(config);
+        connected = false;
+        await registerUpstreamTools();
+        await registerVisionTool();
     });
     pi.on('session_shutdown', async () => {
-        if (started) {
+        if (connected && client) {
             await client.close();
-            await serverProcess.stop();
-            started = false;
+            connected = false;
         }
     });
 }

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAC/B,OAAO,EAAuB,cAAc,EAAE,MAAM,cAAc,CAAC;AACnE,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAA0B,kBAAkB,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AACxF,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAGnD,OAAO,EAAE,kBAAkB,EAAE,aAAa,EAAE,CAAC;AAE7C,MAAM,CAAC,OAAO,UAAU,oBAAoB,CAAC,EAAgB;IAC3D,MAAM,MAAM,GAAG,aAAa,CAAC,kBAAkB,EAAE,CAAC,CAAC;IACnD,MAAM,aAAa,GAAG,IAAI,qBAAqB,EAAE,CAAC;IAClD,MAAM,MAAM,GAAG,IAAI,cAAc,CAAC,MAAM,CAAC,CAAC;IAC1C,IAAI,OAAO,GAAG,KAAK,CAAC;IAEpB,KAAK,UAAU,aAAa;QAC1B,IAAI,OAAO;YAAE,OAAO;QACpB,IAAI,MAAM,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC/B,MAAM,aAAa,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACpC,CAAC;QACD,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;QACvB,OAAO,GAAG,IAAI,CAAC;IACjB,CAAC;IAED,EAAE,CAAC,YAAY,CAAC;QACd,IAAI,EAAE,cAAc;QACpB,KAAK,EAAE,cAAc;QACrB,WAAW,EACT,oMAAoM;QACtM,aAAa,EACX,gGAAgG;QAClG,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC;YACtB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC;gBAClB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC;oBAChB,WAAW,EACT,8GAA8G;iBACjH,CAAC;gBACF,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,0CAA0C,EAAE,CAAC,CAAC;gBAC1F,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,0CAA0C,EAAE,CAAC,CAAC;gBAC1F,MAAM,EAAE,IAAI,CAAC,QAAQ,CACnB,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,4CAA4C,EAAE,CAAC,CAC3E;gBACD,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,4BAA4B,EAAE,CAAC,CAAC;gBAC/E,IAAI,EAAE,IAAI,CAAC,QAAQ,CACjB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,EAAE,WAAW,EAAE,iCAAiC,EAAE,CAAC,CAC9E;gBACD,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,0BAA0B,EAAE,CAAC,CAAC;gBACjF,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,wBAAwB,EAAE,CAAC,CAAC;gBAC/E,IAAI,EAAE,IAAI,CAAC,QAAQ,CACjB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE;oBACrD,WAAW,EAAE,uCAAuC;iBACrD,CAAC,CACH;gBACD,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,oCAAoC,EAAE,CAAC,CAAC;aAC3F,CAAC;SACH,CAAC;QACF,KAAK,CAAC,OAAO,CACX,WAAmB,EACnB,MAAkC,EAClC,OAAgC,EAChC,SAAkB,EAClB,GAAqB;YAErB,MAAM,aAAa,EAAE,CAAC;YAEtB,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBACxC,MAAM,gBAAgB,GAAG,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;gBACnD,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;oBACvB,MAAM,UAAU,GAAG,oBAAoB,CAAC,MAAM,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;oBACjE,MAAM,QAAQ,GAAG,MAAM,UAAU,CAC/B,iHAAiH,EACjH,gBAAgB,EAChB,WAAW,CACZ,CAAC;oBACF,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;gBACtF,CAAC;gBACD,OAAO;oBACL,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,MAAe;4BACrB,IAAI,EAAE,iEAAiE;yBACxE;qBACF;oBACD,OAAO,EAAE,SAAS;iBACnB,CAAC;YACJ,CAAC;YAED,MAAM,YAAY,GAAG,MAAM,cAAc,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;YAEjE,IAAI,MAAM,CAAC,cAAc,KAAK,KAAK,EAAE,CAAC;gBACpC,MAAM,gBAAgB,GAAG,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;gBACnD,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;oBACvB,MAAM,UAAU,GAAG,oBAAoB,CAAC,MAAM,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;oBACjE,MAAM,QAAQ,GAAG,MAAM,UAAU,CAC/B,oGAAoG,EACpG,gBAAgB,EAChB,WAAW,CACZ,CAAC;oBACF,OAAO;wBACL,OAAO,EAAE;4BACP;gCACE,IAAI,EAAE,MAAe;gCACrB,IAAI,EAAE,GAAG,YAAY,sBAAsB,QAAQ,EAAE;6BACtD;yBACF;wBACD,OAAO,EAAE,SAAS;qBACnB,CAAC;gBACJ,CAAC;gBACD,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,YAAY,IAAI,kBAAkB,EAAE,CAAC;oBAC9E,OAAO,EAAE,SAAS;iBACnB,CAAC;YACJ,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,YAAY,IAAI,kBAAkB,EAAE,CAAC;gBAC9E,OAAO,EAAE,SAAS;aACnB,CAAC;QACJ,CAAC;KACF,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,eAAe,EAAE,KAAK,IAAI,EAAE;QAChC,mDAAmD;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,kBAAkB,EAAE,KAAK,IAAI,EAAE;QACnC,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACrB,MAAM,aAAa,CAAC,IAAI,EAAE,CAAC;YAC3B,OAAO,GAAG,KAAK,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAC/B,OAAO,EAA0B,kBAAkB,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AACxF,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAClD,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAGnD,OAAO,EAAE,kBAAkB,EAAE,aAAa,EAAE,CAAC;AAE7C,MAAM,WAAW,GAAG,eAAe,CAAC;AAEpC,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC;IAC7B,0BAA0B;IAC1B,yBAAyB;IACzB,wBAAwB;IACxB,wBAAwB;IACxB,eAAe;IACf,qBAAqB;IACrB,mBAAmB;IACnB,mBAAmB;IACnB,YAAY;IACZ,YAAY;IACZ,aAAa;IACb,MAAM;IACN,iBAAiB;IACjB,MAAM;IACN,cAAc;CACf,CAAC,CAAC;AAEH,MAAM,CAAC,OAAO,UAAU,oBAAoB,CAAC,EAAgB;IAC3D,IAAI,MAAqC,CAAC;IAC1C,IAAI,MAAmC,CAAC;IACxC,IAAI,SAAS,GAAG,KAAK,CAAC;IAEtB,KAAK,UAAU,eAAe;QAC5B,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QACrE,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;YACvB,SAAS,GAAG,IAAI,CAAC;QACnB,CAAC;IACH,CAAC;IAED,KAAK,UAAU,qBAAqB;QAClC,MAAM,eAAe,EAAE,CAAC;QACxB,MAAM,aAAa,GAAG,MAAM,MAAO,CAAC,YAAY,EAAE,CAAC;QAEnD,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;YACjC,IAAI,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;gBAAE,SAAS;YAE5C,MAAM,YAAY,GAAG,GAAG,WAAW,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;YAClD,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC;YAE/B,EAAE,CAAC,YAAY,CAAC;gBACd,IAAI,EAAE,YAAY;gBAClB,KAAK,EAAE,YAAY;gBACnB,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;gBACnC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC;gBACzC,KAAK,CAAC,OAAO,CACX,WAAmB,EACnB,MAA+B,EAC/B,OAAgC,EAChC,SAAkB,EAClB,GAAqB;oBAErB,MAAM,eAAe,EAAE,CAAC;oBACxB,MAAM,MAAM,GAAG,MAAM,MAAO,CAAC,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;oBAE5D,IAAI,YAAY,KAAK,YAAY,IAAI,MAAM,EAAE,WAAW,EAAE,CAAC;wBACzD,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC;wBAC/E,IAAI,YAAY,EAAE,IAAI,EAAE,CAAC;4BACvB,MAAM,UAAU,GAAG,oBAAoB,CAAC,MAAM,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;4BACjE,MAAM,QAAQ,GAAG,MAAM,UAAU,CAC/B,iHAAiH,EACjH,YAAY,CAAC,IAAI,EACjB,YAAY,CAAC,QAAQ,IAAI,WAAW,CACrC,CAAC;4BACF,OAAO;gCACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;gCACpD,OAAO,EAAE,SAAS;6BACnB,CAAC;wBACJ,CAAC;oBACH,CAAC;oBAED,MAAM,OAAO,GAA0C,EAAE,CAAC;oBAC1D,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;wBACnB,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;4BAClC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gCACtC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;4BAClD,CAAC;wBACH,CAAC;oBACH,CAAC;oBACD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;wBACzB,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,CAAC;oBAC3D,CAAC;oBAED,OAAO,MAAM,CAAC,OAAO;wBACnB,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,EAAE;wBAChD,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;gBACtC,CAAC;aACF,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,KAAK,UAAU,kBAAkB;QAC/B,IAAI,CAAC,MAAM,EAAE,WAAW;YAAE,OAAO;QACjC,MAAM,YAAY,GAAG,MAAM,CAAC,WAAW,CAAC;QAExC,EAAE,CAAC,YAAY,CAAC;YACd,IAAI,EAAE,GAAG,WAAW,oBAAoB;YACxC,KAAK,EAAE,GAAG,WAAW,oBAAoB;YACzC,WAAW,EACT,0LAA0L;YAC5L,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC;gBACtB,WAAW,EAAE,IAAI,CAAC,QAAQ,CACxB,IAAI,CAAC,MAAM,CAAC;oBACV,WAAW,EACT,gGAAgG;iBACnG,CAAC,CACH;aACF,CAAC;YACF,KAAK,CAAC,OAAO,CACX,WAAmB,EACnB,MAA+B,EAC/B,OAAgC,EAChC,SAAkB,EAClB,GAAqB;gBAErB,MAAM,eAAe,EAAE,CAAC;gBAExB,MAAM,gBAAgB,GAAG,MAAM,MAAO,CAAC,QAAQ,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;gBAClE,MAAM,YAAY,GAAG,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC;gBAEzF,IAAI,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC;oBACxB,OAAO;wBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,+BAA+B,EAAE,CAAC;wBAC3E,OAAO,EAAE,SAAS;wBAClB,OAAO,EAAE,IAAI;qBACd,CAAC;gBACJ,CAAC;gBAED,MAAM,UAAU,GAAG,oBAAoB,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC;gBAC3D,MAAM,WAAW,GACd,MAAM,CAAC,WAAsB;oBAC9B,iHAAiH,CAAC;gBACpH,MAAM,QAAQ,GAAG,MAAM,UAAU,CAC/B,WAAW,EACX,YAAY,CAAC,IAAI,EACjB,YAAY,CAAC,QAAQ,IAAI,WAAW,CACrC,CAAC;gBAEF,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;oBACpD,OAAO,EAAE,SAAS;iBACnB,CAAC;YACJ,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IAED,EAAE,CAAC,EAAE,CAAC,eAAe,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE;QAC3C,MAAM,GAAG,aAAa,CAAC,kBAAkB,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QAC7D,MAAM,GAAG,IAAI,eAAe,CAAC,MAAM,CAAC,CAAC;QACrC,SAAS,GAAG,KAAK,CAAC;QAClB,MAAM,qBAAqB,EAAE,CAAC;QAC9B,MAAM,kBAAkB,EAAE,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,kBAAkB,EAAE,KAAK,IAAI,EAAE;QACnC,IAAI,SAAS,IAAI,MAAM,EAAE,CAAC;YACxB,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACrB,SAAS,GAAG,KAAK,CAAC;QACpB,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/mcp-client.d.ts

@@ -0,0 +1,22 @@
+import type { Tool } from '@modelcontextprotocol/sdk/types.js';
+import type { ComputerUseConfig } from './config.js';
+export declare class CuaDriverClient {
+    private client;
+    private transport;
+    private config;
+    constructor(config: ComputerUseConfig);
+    connect(): Promise<void>;
+    listAllTools(): Promise<Tool[]>;
+    callTool(name: string, args: Record<string, unknown>): Promise<{
+        content?: Array<{
+            type: string;
+            text?: string;
+            data?: string;
+            mimeType?: string;
+        }>;
+        isError?: boolean;
+    }>;
+    close(): Promise<void>;
+    private resolveBinaryPath;
+}
+//# sourceMappingURL=mcp-client.d.ts.map

package/dist/mcp-client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-client.d.ts","sourceRoot":"","sources":["../src/mcp-client.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,oCAAoC,CAAC;AAC/D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAKrD,qBAAa,eAAe;IAC1B,OAAO,CAAC,MAAM,CAAuB;IACrC,OAAO,CAAC,SAAS,CAAqC;IACtD,OAAO,CAAC,MAAM,CAAoB;gBAEtB,MAAM,EAAE,iBAAiB;IAI/B,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAmBxB,YAAY,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;IAgB/B,QAAQ,CACZ,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC5B,OAAO,CAAC;QACT,OAAO,CAAC,EAAE,KAAK,CAAC;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,IAAI,CAAC,EAAE,MAAM,CAAC;YAAC,IAAI,CAAC,EAAE,MAAM,CAAC;YAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QACnF,OAAO,CAAC,EAAE,OAAO,CAAC;KACnB,CAAC;IAWI,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;YAQd,iBAAiB;CAmBhC"}

package/dist/mcp-client.js

@@ -0,0 +1,71 @@
+import { existsSync } from 'node:fs';
+import path from 'node:path';
+import { fileURLToPath } from 'node:url';
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js';
+const MCP_TIMEOUT_MS = 60_000;
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
+export class CuaDriverClient {
+    client = null;
+    transport = null;
+    config;
+    constructor(config) {
+        this.config = config;
+    }
+    async connect() {
+        const binaryPath = await this.resolveBinaryPath();
+        const args = ['mcp', ...(this.config.extraArgs ?? [])];
+        this.transport = new StdioClientTransport({
+            command: binaryPath,
+            args,
+            stderr: 'pipe',
+        });
+        this.transport.onerror = (error) => {
+            console.error(`[pi-computer-use] cua-driver transport error: ${error.message}`);
+        };
+        this.client = new Client({ name: 'pi-computer-use', version: '0.1.0' }, { capabilities: {} });
+        await this.client.connect(this.transport);
+    }
+    async listAllTools() {
+        if (!this.client)
+            throw new Error('CuaDriverClient not connected');
+        const allTools = [];
+        let cursor;
+        do {
+            const result = await this.client.listTools(cursor ? { cursor } : undefined, {
+                timeout: MCP_TIMEOUT_MS,
+            });
+            allTools.push(...result.tools);
+            cursor = result.nextCursor;
+        } while (cursor);
+        return allTools;
+    }
+    async callTool(name, args) {
+        if (!this.client)
+            throw new Error('CuaDriverClient not connected');
+        return (await this.client.callTool({ name, arguments: args }, undefined, {
+            timeout: MCP_TIMEOUT_MS,
+        }));
+    }
+    async close() {
+        if (this.client) {
+            await this.client.close();
+            this.client = null;
+        }
+        this.transport = null;
+    }
+    async resolveBinaryPath() {
+        if (this.config.mode === 'path' && this.config.binaryPath) {
+            return this.config.binaryPath;
+        }
+        const platform = process.platform === 'win32' ? 'win32-x64' : `${process.platform}-${process.arch}`;
+        const ext = process.platform === 'win32' ? '.exe' : '';
+        const binPath = path.resolve(__dirname, '..', 'bin', platform, `cua-driver${ext}`);
+        if (!existsSync(binPath)) {
+            throw new Error(`[pi-computer-use] cua-driver binary not found at ${binPath}. ` +
+                `Platform "${platform}" may not be supported, or set mode: "path" with binaryPath in config.`);
+        }
+        return binPath;
+    }
+}
+//# sourceMappingURL=mcp-client.js.map

package/dist/mcp-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-client.js","sourceRoot":"","sources":["../src/mcp-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AAIjF,MAAM,cAAc,GAAG,MAAM,CAAC;AAC9B,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;AAE/D,MAAM,OAAO,eAAe;IAClB,MAAM,GAAkB,IAAI,CAAC;IAC7B,SAAS,GAAgC,IAAI,CAAC;IAC9C,MAAM,CAAoB;IAElC,YAAY,MAAyB;QACnC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,OAAO;QACX,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAClD,MAAM,IAAI,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,CAAC;QAEvD,IAAI,CAAC,SAAS,GAAG,IAAI,oBAAoB,CAAC;YACxC,OAAO,EAAE,UAAU;YACnB,IAAI;YACJ,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,CAAC,OAAO,GAAG,CAAC,KAAY,EAAE,EAAE;YACxC,OAAO,CAAC,KAAK,CAAC,iDAAiD,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QAClF,CAAC,CAAC;QAEF,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;QAE9F,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;QAEnE,MAAM,QAAQ,GAAW,EAAE,CAAC;QAC5B,IAAI,MAA0B,CAAC;QAC/B,GAAG,CAAC;YACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,EAAE;gBAC1E,OAAO,EAAE,cAAc;aACxB,CAAC,CAAC;YACH,QAAQ,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;YAC/B,MAAM,GAAG,MAAM,CAAC,UAAU,CAAC;QAC7B,CAAC,QAAQ,MAAM,EAAE;QAEjB,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,IAAY,EACZ,IAA6B;QAK7B,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;QAEnE,OAAO,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,SAAS,EAAE;YACvE,OAAO,EAAE,cAAc;SACxB,CAAC,CAGD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YAC1B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACrB,CAAC;QACD,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;IACxB,CAAC;IAEO,KAAK,CAAC,iBAAiB;QAC7B,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;YAC1D,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;QAChC,CAAC;QAED,MAAM,QAAQ,GACZ,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;QACrF,MAAM,GAAG,GAAG,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;QACvD,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,aAAa,GAAG,EAAE,CAAC,CAAC;QAEnF,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YACzB,MAAM,IAAI,KAAK,CACb,oDAAoD,OAAO,IAAI;gBAC7D,aAAa,QAAQ,wEAAwE,CAChG,CAAC;QACJ,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;CACF"}

package/package.json

@@ -1,7 +1,16 @@
 {
   "name": "@amaster.ai/pi-computer-use",
-  "version": "0.1.2-beta.1",
-  "description": "Pi extension for CUA computer-use: spawns and controls cua-computer-server",
+  "version": "0.1.2-beta.3",
+  "description": "Pi extension for desktop automation via cua-driver-rs with computer_use_ prefixed tools",
+  "keywords": [
+    "pi",
+    "extension",
+    "computer-use",
+    "desktop",
+    "automation",
+    "cua-driver",
+    "mcp"
+  ],
   "license": "Apache-2.0",
   "type": "module",
   "sideEffects": false,
@@ -18,6 +27,7 @@
   },
   "files": [
     "dist",
+    "bin",
     "README.md"
   ],
   "pi": {
@@ -34,8 +44,8 @@
     "directory": "packages/pi-computer-use"
   },
   "dependencies": {
-    "ws": "^8.18.0",
-    "@amaster.ai/pi-shared": "0.1.2-beta.1"
+    "@modelcontextprotocol/sdk": "^1.29.0",
+    "@amaster.ai/pi-shared": "0.1.2-beta.3"
   },
   "peerDependencies": {
     "@earendil-works/pi-ai": ">=0.74.0",
@@ -45,7 +55,6 @@
   "devDependencies": {
     "@earendil-works/pi-ai": "0.74.0",
     "@earendil-works/pi-coding-agent": "0.74.0",
-    "@types/ws": "^8.5.0",
     "tsup": "^8.4.0",
     "typebox": "*",
     "vitest": "^4.0.0"

package/dist/__tests__/computer-client.test.d.ts

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=computer-client.test.d.ts.map

package/dist/__tests__/computer-client.test.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"computer-client.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/computer-client.test.ts"],"names":[],"mappings":""}