@aerol-ai/aerolvm-sdk 0.1.3 → 0.1.5

package/dist/MicroVM.js

@@ -1,6 +1,6 @@
 import { APIClient } from "./internal/client.js";
 import { Sandbox } from "./Sandbox.js";
-const defaultAPIURL = "http://127.0.0.1:8080";
+const defaultAPIURL = "http://127.0.0.1:21212";
 const authRequiredErrorMessage = "PAT token is required. Set patToken or SB_PAT_TOKEN.";
 export class MicroVM {
     apiUrl;

package/dist/internal/client.d.ts

@@ -32,6 +32,19 @@ export declare class APIClient {
     uploadFile(id: string, targetPath: string, data: BinaryLike): Promise<void>;
     downloadFile(id: string, targetPath: string): Promise<Uint8Array>;
     exposePort(id: string, port: number): Promise<string>;
+    /**
+     * Publish a raw TCP port through caddy-l4. The returned URL is in
+     * `tcp://<host>:<port>` form and works directly with `psql`, `redis-cli`,
+     * `mysql`, `mongosh`, and any other native-protocol client.
+     */
+    exposeTCPPort(id: string, port: number): Promise<string>;
+    /**
+     * Publish a TCP port behind the shared TLS-SNI multiplexer. Returns
+     * `tls://<id>-<port>.<domain>:<l4-port>`. Requires the daemon to have a
+     * domain configured AND `SB_L4_TLS_LISTEN` set.
+     */
+    exposeTLSPort(id: string, port: number): Promise<string>;
+    private exposePortWithProtocol;
     unexposePort(id: string, port: number): Promise<void>;
     reconcile(): Promise<void>;
     health(): Promise<HealthStatus>;
@@ -82,6 +95,8 @@ export declare class SandboxResource implements Sandbox {
     uploadFile(targetPath: string, data: BinaryLike): Promise<void>;
     downloadFile(targetPath: string): Promise<Uint8Array>;
     exposePort(port: number): Promise<string>;
+    exposeTCPPort(port: number): Promise<string>;
+    exposeTLSPort(port: number): Promise<string>;
     unexposePort(port: number): Promise<void>;
     start(): Promise<this>;
     stop(): Promise<this>;

package/dist/internal/client.js

@@ -93,7 +93,27 @@ export class APIClient {
         return new Uint8Array(await response.arrayBuffer());
     }
     async exposePort(id, port) {
-        const response = await this.doJSON("POST", `/v1/sandboxes/${id}/ports/${port}`);
+        return this.exposePortWithProtocol(id, port);
+    }
+    /**
+     * Publish a raw TCP port through caddy-l4. The returned URL is in
+     * `tcp://<host>:<port>` form and works directly with `psql`, `redis-cli`,
+     * `mysql`, `mongosh`, and any other native-protocol client.
+     */
+    async exposeTCPPort(id, port) {
+        return this.exposePortWithProtocol(id, port, "tcp");
+    }
+    /**
+     * Publish a TCP port behind the shared TLS-SNI multiplexer. Returns
+     * `tls://<id>-<port>.<domain>:<l4-port>`. Requires the daemon to have a
+     * domain configured AND `SB_L4_TLS_LISTEN` set.
+     */
+    async exposeTLSPort(id, port) {
+        return this.exposePortWithProtocol(id, port, "tls");
+    }
+    async exposePortWithProtocol(id, port, protocol) {
+        const body = protocol ? { protocol } : undefined;
+        const response = await this.doJSON("POST", `/v1/sandboxes/${id}/ports/${port}`, body);
         return response.public_url;
     }
     async unexposePort(id, port) {
@@ -202,6 +222,12 @@ export class SandboxResource {
     async exposePort(port) {
         return this.client.exposePort(this.id, port);
     }
+    async exposeTCPPort(port) {
+        return this.client.exposeTCPPort(this.id, port);
+    }
+    async exposeTLSPort(port) {
+        return this.client.exposeTLSPort(this.id, port);
+    }
     async unexposePort(port) {
         await this.client.unexposePort(this.id, port);
     }

package/dist/types.d.ts

@@ -32,6 +32,34 @@ export interface Lifecycle {
     destroyAtAge?: number;
 }
 export type UpdateLifecycleOptions = Lifecycle;
+export type GPUVendor = "nvidia" | "amd" | "apple";
+export interface GPUOptions {
+    /**
+     * GPU hardware vendor. Required when gpus is set.
+     * - "nvidia": NVIDIA GPUs via nvidia-container-runtime. Requires
+     *   nvidia-container-toolkit on the host.
+     * - "amd": AMD GPUs via ROCm (/dev/kfd + /dev/dri). Requires ROCm
+     *   drivers on the host.
+     * - "apple": Apple Silicon GPU via Docker Desktop's experimental Metal
+     *   support. Only functional on macOS with Docker Desktop.
+     *
+     * GPU access is not supported with runtime="gvisor" — the API returns
+     * an error if both gpus and runtime="gvisor" are set.
+     */
+    vendor: GPUVendor;
+    /**
+     * Number of GPUs to allocate. Use -1 to request all GPUs on the host.
+     * Defaults to 1 when omitted. Ignored for AMD (all AMD GPUs are exposed
+     * via /dev/kfd and /dev/dri).
+     */
+    count?: number;
+    /**
+     * Pin the sandbox to specific GPU device indices or UUIDs.
+     * For NVIDIA: indices ("0", "1") or UUIDs ("GPU-abc123...").
+     * For AMD and Apple: ignored.
+     */
+    deviceIDs?: string[];
+}
 export interface CreateOptions {
     image: string;
     /** Number of CPU cores to allocate. Fractional values are supported (e.g. 0.5 = half a core). */
@@ -52,8 +80,15 @@ export interface CreateOptions {
      * privileged containers and ignores per-sandbox disk quotas. `"kata"` is
      * reserved for future Kata Containers support and is rejected by the API
      * today.
+     *
+     * GPU access is not supported with `"gvisor"`.
      */
     runtime?: "docker" | "gvisor" | "kata";
+    /**
+     * Attach GPU resources to the sandbox. Omit for CPU-only workloads.
+     * Not compatible with runtime="gvisor".
+     */
+    gpus?: GPUOptions;
 }
 export interface ResizeOptions {
     /** Number of CPU cores. Fractional values supported (e.g. 0.5). */
@@ -122,6 +157,8 @@ export interface Sandbox {
      * a pre-migration row that resolves to the host default at start time.
      */
     runtime: "" | "docker" | "gvisor" | "kata";
+    /** GPU configuration this sandbox was created with. Absent means no GPU. */
+    gpus?: GPUOptions;
 }
 export interface ExecRequest {
     command: string;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aerol-ai/aerolvm-sdk",
-  "version": "0.1.3",
+  "version": "0.1.5",
   "private": false,
   "type": "module",
   "main": "./dist/index.js",