@puku-ai/sdk 4.0.3 → 4.0.4

package/dist/vendor/resources/beta/environments/work.cjs

@@ -0,0 +1,256 @@
+"use strict";
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.EnvironmentWorker = exports.WorkPoller = exports.Work = void 0;
+const resource_1 = require("../../../core/resource");
+const pagination_1 = require("../../../core/pagination");
+const headers_1 = require("../../../internal/headers");
+const path_1 = require("../../../internal/utils/path");
+const poller_1 = require("../../../lib/environments/poller");
+const worker_1 = require("../../../lib/environments/worker");
+class Work extends resource_1.APIResource {
+    /**
+     * Note: these endpoints are called automatically by the pre-built environment
+     * worker provided in the SDKs and CLI, for orchestrating sessions with self-hosted
+     * sandbox environments. They are included here as a reference; you do not need to
+     * invoke them directly.
+     *
+     * Retrieve detailed information about a specific work item.
+     *
+     * @example
+     * ```ts
+     * const betaSelfHostedWork =
+     *   await client.beta.environments.work.retrieve('work_id', {
+     *     environment_id: 'env_011CZkZ9X2dpNyB7HsEFoRfW',
+     *   });
+     * ```
+     */
+    retrieve(workID, params, options) {
+        const { environment_id, betas } = params;
+        return this._client.get((0, path_1.path) `/v1/environments/${environment_id}/work/${workID}?beta=true`, {
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'managed-agents-2026-04-01'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Note: these endpoints are called automatically by the pre-built environment
+     * worker provided in the SDKs and CLI, for orchestrating sessions with self-hosted
+     * sandbox environments. They are included here as a reference; you do not need to
+     * invoke them directly.
+     *
+     * Update work item metadata with merge semantics.
+     *
+     * @example
+     * ```ts
+     * const betaSelfHostedWork =
+     *   await client.beta.environments.work.update('work_id', {
+     *     environment_id: 'env_011CZkZ9X2dpNyB7HsEFoRfW',
+     *     metadata: { foo: 'string' },
+     *   });
+     * ```
+     */
+    update(workID, params, options) {
+        const { environment_id, betas, ...body } = params;
+        return this._client.post((0, path_1.path) `/v1/environments/${environment_id}/work/${workID}?beta=true`, {
+            body,
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'managed-agents-2026-04-01'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Note: these endpoints are called automatically by the pre-built environment
+     * worker provided in the SDKs and CLI, for orchestrating sessions with self-hosted
+     * sandbox environments. They are included here as a reference; you do not need to
+     * invoke them directly.
+     *
+     * List work items in an environment.
+     *
+     * @example
+     * ```ts
+     * // Automatically fetches more pages as needed.
+     * for await (const betaSelfHostedWork of client.beta.environments.work.list(
+     *   'env_011CZkZ9X2dpNyB7HsEFoRfW',
+     * )) {
+     *   // ...
+     * }
+     * ```
+     */
+    list(environmentID, params = {}, options) {
+        const { betas, ...query } = params ?? {};
+        return this._client.getAPIList((0, path_1.path) `/v1/environments/${environmentID}/work?beta=true`, (pagination_1.PageCursor), {
+            query,
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'managed-agents-2026-04-01'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Note: these endpoints are called automatically by the pre-built environment
+     * worker provided in the SDKs and CLI, for orchestrating sessions with self-hosted
+     * sandbox environments. They are included here as a reference; you do not need to
+     * invoke them directly.
+     *
+     * Acknowledge receipt of a work item, transitioning it from 'queued' to 'starting'
+     * and removing it from the queue.
+     *
+     * @example
+     * ```ts
+     * const betaSelfHostedWork =
+     *   await client.beta.environments.work.ack('work_id', {
+     *     environment_id: 'env_011CZkZ9X2dpNyB7HsEFoRfW',
+     *   });
+     * ```
+     */
+    ack(workID, params, options) {
+        const { environment_id, betas } = params;
+        return this._client.post((0, path_1.path) `/v1/environments/${environment_id}/work/${workID}/ack?beta=true`, {
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'managed-agents-2026-04-01'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Note: these endpoints are called automatically by the pre-built environment
+     * worker provided in the SDKs and CLI, for orchestrating sessions with self-hosted
+     * sandbox environments. They are included here as a reference; you do not need to
+     * invoke them directly.
+     *
+     * Record a heartbeat for a work item to maintain the lease.
+     *
+     * @example
+     * ```ts
+     * const betaSelfHostedWorkHeartbeatResponse =
+     *   await client.beta.environments.work.heartbeat('work_id', {
+     *     environment_id: 'env_011CZkZ9X2dpNyB7HsEFoRfW',
+     *   });
+     * ```
+     */
+    heartbeat(workID, params, options) {
+        const { environment_id, desired_ttl_seconds, expected_last_heartbeat, betas } = params;
+        return this._client.post((0, path_1.path) `/v1/environments/${environment_id}/work/${workID}/heartbeat?beta=true`, {
+            query: { desired_ttl_seconds, expected_last_heartbeat },
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'managed-agents-2026-04-01'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Note: these endpoints are called automatically by the pre-built environment
+     * worker provided in the SDKs and CLI, for orchestrating sessions with self-hosted
+     * sandbox environments. They are included here as a reference; you do not need to
+     * invoke them directly.
+     *
+     * Long poll for work items in the queue.
+     *
+     * @example
+     * ```ts
+     * const betaSelfHostedWork =
+     *   await client.beta.environments.work.poll(
+     *     'env_011CZkZ9X2dpNyB7HsEFoRfW',
+     *   );
+     * ```
+     */
+    poll(environmentID, params = {}, options) {
+        const { betas, 'PukuAI-Worker-ID': pukuWorkerID, ...query } = params ?? {};
+        return this._client.get((0, path_1.path) `/v1/environments/${environmentID}/work/poll?beta=true`, {
+            query,
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                {
+                    'puku-beta': [...(betas ?? []), 'managed-agents-2026-04-01'].toString(),
+                    ...(pukuWorkerID != null ? { 'PukuAI-Worker-ID': pukuWorkerID } : undefined),
+                },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Get statistics about the work queue for an environment.
+     *
+     * @example
+     * ```ts
+     * const betaSelfHostedWorkQueueStats =
+     *   await client.beta.environments.work.stats(
+     *     'env_011CZkZ9X2dpNyB7HsEFoRfW',
+     *   );
+     * ```
+     */
+    stats(environmentID, params = {}, options) {
+        const { betas } = params ?? {};
+        return this._client.get((0, path_1.path) `/v1/environments/${environmentID}/work/stats?beta=true`, {
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'managed-agents-2026-04-01'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Note: these endpoints are called automatically by the pre-built environment
+     * worker provided in the SDKs and CLI, for orchestrating sessions with self-hosted
+     * sandbox environments. They are included here as a reference; you do not need to
+     * invoke them directly.
+     *
+     * Stop a work item, initiating graceful or forced shutdown.
+     *
+     * @example
+     * ```ts
+     * const betaSelfHostedWork =
+     *   await client.beta.environments.work.stop('work_id', {
+     *     environment_id: 'env_011CZkZ9X2dpNyB7HsEFoRfW',
+     *   });
+     * ```
+     */
+    stop(workID, params, options) {
+        const { environment_id, betas, ...body } = params;
+        return this._client.post((0, path_1.path) `/v1/environments/${environment_id}/work/${workID}/stop?beta=true`, {
+            body,
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'managed-agents-2026-04-01'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Continuously claim work from a self-hosted environment, ack each item,
+     * and yield it. Posts `stop` automatically when the consumer's loop body
+     * returns or when iteration ends.
+     *
+     * @example
+     * ```ts
+     * for await (const work of client.beta.environments.work.poller({
+     *   environmentId,
+     *   environmentKey,
+     * })) {
+     *   if (work.data.type !== 'session') continue;
+     *   // ...service the work...
+     * }
+     * ```
+     */
+    poller(opts) {
+        return new poller_1.WorkPoller({ ...opts, client: this._client });
+    }
+    worker(opts) {
+        return new worker_1.EnvironmentWorker({ ...opts, client: this._client });
+    }
+}
+exports.Work = Work;
+var poller_2 = require("../../../lib/environments/poller");
+Object.defineProperty(exports, "WorkPoller", { enumerable: true, get: function () { return poller_2.WorkPoller; } });
+var worker_2 = require("../../../lib/environments/worker");
+Object.defineProperty(exports, "EnvironmentWorker", { enumerable: true, get: function () { return worker_2.EnvironmentWorker; } });
+Work.WorkPoller = poller_1.WorkPoller;
+Work.EnvironmentWorker = worker_1.EnvironmentWorker;

package/dist/vendor/resources/beta/environments.cjs

@@ -0,0 +1,18 @@
+"use strict";
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./environments/index"), exports);

package/dist/vendor/resources/beta/files.cjs

@@ -0,0 +1,123 @@
+"use strict";
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Files = void 0;
+const resource_1 = require("../../core/resource");
+const pagination_1 = require("../../core/pagination");
+const headers_1 = require("../../internal/headers");
+const stainless_helper_header_1 = require("../../internal/stainless-helper-header");
+const uploads_1 = require("../../internal/uploads");
+const path_1 = require("../../internal/utils/path");
+class Files extends resource_1.APIResource {
+    /**
+     * List Files
+     *
+     * @example
+     * ```ts
+     * // Automatically fetches more pages as needed.
+     * for await (const betaFileMetadata of client.beta.files.list()) {
+     *   // ...
+     * }
+     * ```
+     */
+    list(params = {}, options) {
+        const { betas, ...query } = params ?? {};
+        return this._client.getAPIList('/v1/files?beta=true', (pagination_1.PageCursor), {
+            query,
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { ...(betas?.toString() != null ? { 'puku-beta': betas?.toString() } : undefined) },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Delete File
+     *
+     * @example
+     * ```ts
+     * const betaDeletedFile = await client.beta.files.delete(
+     *   'file_id',
+     * );
+     * ```
+     */
+    delete(fileID, params = {}, options) {
+        const { betas } = params ?? {};
+        return this._client.delete((0, path_1.path) `/v1/files/${fileID}?beta=true`, {
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { ...(betas?.toString() != null ? { 'puku-beta': betas?.toString() } : undefined) },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Download File
+     *
+     * @example
+     * ```ts
+     * const response = await client.beta.files.download(
+     *   'file_id',
+     * );
+     *
+     * const content = await response.blob();
+     * console.log(content);
+     * ```
+     */
+    download(fileID, params = {}, options) {
+        const { betas } = params ?? {};
+        return this._client.get((0, path_1.path) `/v1/files/${fileID}/content?beta=true`, {
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                {
+                    Accept: 'application/binary',
+                    ...(betas?.toString() != null ? { 'puku-beta': betas?.toString() } : undefined),
+                },
+                options?.headers,
+            ]),
+            __binaryResponse: true,
+        });
+    }
+    /**
+     * Get File Metadata
+     *
+     * @example
+     * ```ts
+     * const betaFileMetadata =
+     *   await client.beta.files.retrieveMetadata('file_id');
+     * ```
+     */
+    retrieveMetadata(fileID, params = {}, options) {
+        const { betas } = params ?? {};
+        return this._client.get((0, path_1.path) `/v1/files/${fileID}?beta=true`, {
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { ...(betas?.toString() != null ? { 'puku-beta': betas?.toString() } : undefined) },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Upload File
+     *
+     * @example
+     * ```ts
+     * const betaFileMetadata = await client.beta.files.upload({
+     *   file: fs.createReadStream('path/to/file'),
+     * });
+     * ```
+     */
+    upload(params, options) {
+        const { betas, ...body } = params;
+        return this._client.post('/v1/files?beta=true', (0, uploads_1.multipartFormRequestOptions)({
+            body,
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { ...(betas?.toString() != null ? { 'puku-beta': betas?.toString() } : undefined) },
+                (0, stainless_helper_header_1.stainlessHelperHeaderFromFile)(body.file),
+                options?.headers,
+            ]),
+        }, this._client));
+    }
+}
+exports.Files = Files;

package/dist/vendor/resources/beta/index.cjs

@@ -0,0 +1,38 @@
+"use strict";
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Webhooks = exports.Vaults = exports.UserProfiles = exports.Tunnels = exports.Skills = exports.Sessions = exports.Organization = exports.Models = exports.Messages = exports.MemoryStores = exports.Files = exports.Environments = exports.Dreams = exports.Deployments = exports.DeploymentRuns = exports.Beta = exports.Agents = void 0;
+var index_1 = require("./agents/index");
+Object.defineProperty(exports, "Agents", { enumerable: true, get: function () { return index_1.Agents; } });
+var beta_1 = require("./beta");
+Object.defineProperty(exports, "Beta", { enumerable: true, get: function () { return beta_1.Beta; } });
+var deployment_runs_1 = require("./deployment-runs");
+Object.defineProperty(exports, "DeploymentRuns", { enumerable: true, get: function () { return deployment_runs_1.DeploymentRuns; } });
+var deployments_1 = require("./deployments");
+Object.defineProperty(exports, "Deployments", { enumerable: true, get: function () { return deployments_1.Deployments; } });
+var dreams_1 = require("./dreams");
+Object.defineProperty(exports, "Dreams", { enumerable: true, get: function () { return dreams_1.Dreams; } });
+var index_2 = require("./environments/index");
+Object.defineProperty(exports, "Environments", { enumerable: true, get: function () { return index_2.Environments; } });
+var files_1 = require("./files");
+Object.defineProperty(exports, "Files", { enumerable: true, get: function () { return files_1.Files; } });
+var index_3 = require("./memory-stores/index");
+Object.defineProperty(exports, "MemoryStores", { enumerable: true, get: function () { return index_3.MemoryStores; } });
+var index_4 = require("./messages/index");
+Object.defineProperty(exports, "Messages", { enumerable: true, get: function () { return index_4.Messages; } });
+var models_1 = require("./models");
+Object.defineProperty(exports, "Models", { enumerable: true, get: function () { return models_1.Models; } });
+var index_5 = require("./organization/index");
+Object.defineProperty(exports, "Organization", { enumerable: true, get: function () { return index_5.Organization; } });
+var index_6 = require("./sessions/index");
+Object.defineProperty(exports, "Sessions", { enumerable: true, get: function () { return index_6.Sessions; } });
+var index_7 = require("./skills/index");
+Object.defineProperty(exports, "Skills", { enumerable: true, get: function () { return index_7.Skills; } });
+var index_8 = require("./tunnels/index");
+Object.defineProperty(exports, "Tunnels", { enumerable: true, get: function () { return index_8.Tunnels; } });
+var user_profiles_1 = require("./user-profiles");
+Object.defineProperty(exports, "UserProfiles", { enumerable: true, get: function () { return user_profiles_1.UserProfiles; } });
+var index_9 = require("./vaults/index");
+Object.defineProperty(exports, "Vaults", { enumerable: true, get: function () { return index_9.Vaults; } });
+var webhooks_1 = require("./webhooks");
+Object.defineProperty(exports, "Webhooks", { enumerable: true, get: function () { return webhooks_1.Webhooks; } });

package/dist/vendor/resources/beta/memory-stores/index.cjs

@@ -0,0 +1,10 @@
+"use strict";
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MemoryVersions = exports.MemoryStores = exports.Memories = void 0;
+var memories_1 = require("./memories");
+Object.defineProperty(exports, "Memories", { enumerable: true, get: function () { return memories_1.Memories; } });
+var memory_stores_1 = require("./memory-stores");
+Object.defineProperty(exports, "MemoryStores", { enumerable: true, get: function () { return memory_stores_1.MemoryStores; } });
+var memory_versions_1 = require("./memory-versions");
+Object.defineProperty(exports, "MemoryVersions", { enumerable: true, get: function () { return memory_versions_1.MemoryVersions; } });

package/dist/vendor/resources/beta/memory-stores/memories.cjs

@@ -0,0 +1,129 @@
+"use strict";
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Memories = void 0;
+const resource_1 = require("../../../core/resource");
+const pagination_1 = require("../../../core/pagination");
+const headers_1 = require("../../../internal/headers");
+const path_1 = require("../../../internal/utils/path");
+class Memories extends resource_1.APIResource {
+    /**
+     * Create a memory
+     *
+     * @example
+     * ```ts
+     * const betaManagedAgentsMemory =
+     *   await client.beta.memoryStores.memories.create(
+     *     'memory_store_id',
+     *     { content: 'content', path: 'xx' },
+     *   );
+     * ```
+     */
+    create(memoryStoreID, params, options) {
+        const { view, betas, ...body } = params;
+        return this._client.post((0, path_1.path) `/v1/memory_stores/${memoryStoreID}/memories?beta=true`, {
+            query: { view },
+            body,
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'agent-memory-2026-07-22'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Retrieve a memory
+     *
+     * @example
+     * ```ts
+     * const betaManagedAgentsMemory =
+     *   await client.beta.memoryStores.memories.retrieve(
+     *     'memory_id',
+     *     { memory_store_id: 'memory_store_id' },
+     *   );
+     * ```
+     */
+    retrieve(memoryID, params, options) {
+        const { memory_store_id, betas, ...query } = params;
+        return this._client.get((0, path_1.path) `/v1/memory_stores/${memory_store_id}/memories/${memoryID}?beta=true`, {
+            query,
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'agent-memory-2026-07-22'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Update a memory
+     *
+     * @example
+     * ```ts
+     * const betaManagedAgentsMemory =
+     *   await client.beta.memoryStores.memories.update(
+     *     'memory_id',
+     *     { memory_store_id: 'memory_store_id' },
+     *   );
+     * ```
+     */
+    update(memoryID, params, options) {
+        const { memory_store_id, view, betas, ...body } = params;
+        return this._client.post((0, path_1.path) `/v1/memory_stores/${memory_store_id}/memories/${memoryID}?beta=true`, {
+            query: { view },
+            body,
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'agent-memory-2026-07-22'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * List memories
+     *
+     * @example
+     * ```ts
+     * // Automatically fetches more pages as needed.
+     * for await (const betaManagedAgentsMemoryListItem of client.beta.memoryStores.memories.list(
+     *   'memory_store_id',
+     * )) {
+     *   // ...
+     * }
+     * ```
+     */
+    list(memoryStoreID, params = {}, options) {
+        const { betas, ...query } = params ?? {};
+        return this._client.getAPIList((0, path_1.path) `/v1/memory_stores/${memoryStoreID}/memories?beta=true`, (pagination_1.PageCursor), {
+            query,
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'agent-memory-2026-07-22'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Delete a memory
+     *
+     * @example
+     * ```ts
+     * const betaManagedAgentsDeletedMemory =
+     *   await client.beta.memoryStores.memories.delete(
+     *     'memory_id',
+     *     { memory_store_id: 'memory_store_id' },
+     *   );
+     * ```
+     */
+    delete(memoryID, params, options) {
+        const { memory_store_id, expected_content_sha256, betas } = params;
+        return this._client.delete((0, path_1.path) `/v1/memory_stores/${memory_store_id}/memories/${memoryID}?beta=true`, {
+            query: { expected_content_sha256 },
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'agent-memory-2026-07-22'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+}
+exports.Memories = Memories;