@puku-ai/sdk 4.0.3 → 4.0.4

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

@@ -0,0 +1,173 @@
+"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 __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MemoryStores = void 0;
+const resource_1 = require("../../../core/resource");
+const MemoriesAPI = __importStar(require("./memories"));
+const memories_1 = require("./memories");
+const MemoryVersionsAPI = __importStar(require("./memory-versions"));
+const memory_versions_1 = require("./memory-versions");
+const pagination_1 = require("../../../core/pagination");
+const headers_1 = require("../../../internal/headers");
+const path_1 = require("../../../internal/utils/path");
+class MemoryStores extends resource_1.APIResource {
+    memories = new MemoriesAPI.Memories(this._client);
+    memoryVersions = new MemoryVersionsAPI.MemoryVersions(this._client);
+    /**
+     * Create a memory store
+     *
+     * @example
+     * ```ts
+     * const betaManagedAgentsMemoryStore =
+     *   await client.beta.memoryStores.create({ name: 'x' });
+     * ```
+     */
+    create(params, options) {
+        const { betas, ...body } = params;
+        return this._client.post('/v1/memory_stores?beta=true', {
+            body,
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'agent-memory-2026-07-22'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Retrieve a memory store
+     *
+     * @example
+     * ```ts
+     * const betaManagedAgentsMemoryStore =
+     *   await client.beta.memoryStores.retrieve(
+     *     'memory_store_id',
+     *   );
+     * ```
+     */
+    retrieve(memoryStoreID, params = {}, options) {
+        const { betas } = params ?? {};
+        return this._client.get((0, path_1.path) `/v1/memory_stores/${memoryStoreID}?beta=true`, {
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'agent-memory-2026-07-22'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Update a memory store
+     *
+     * @example
+     * ```ts
+     * const betaManagedAgentsMemoryStore =
+     *   await client.beta.memoryStores.update('memory_store_id');
+     * ```
+     */
+    update(memoryStoreID, params, options) {
+        const { betas, ...body } = params;
+        return this._client.post((0, path_1.path) `/v1/memory_stores/${memoryStoreID}?beta=true`, {
+            body,
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'agent-memory-2026-07-22'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * List memory stores
+     *
+     * @example
+     * ```ts
+     * // Automatically fetches more pages as needed.
+     * for await (const betaManagedAgentsMemoryStore of client.beta.memoryStores.list()) {
+     *   // ...
+     * }
+     * ```
+     */
+    list(params = {}, options) {
+        const { betas, ...query } = params ?? {};
+        return this._client.getAPIList('/v1/memory_stores?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 store
+     *
+     * @example
+     * ```ts
+     * const betaManagedAgentsDeletedMemoryStore =
+     *   await client.beta.memoryStores.delete('memory_store_id');
+     * ```
+     */
+    delete(memoryStoreID, params = {}, options) {
+        const { betas } = params ?? {};
+        return this._client.delete((0, path_1.path) `/v1/memory_stores/${memoryStoreID}?beta=true`, {
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'agent-memory-2026-07-22'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Archive a memory store
+     *
+     * @example
+     * ```ts
+     * const betaManagedAgentsMemoryStore =
+     *   await client.beta.memoryStores.archive('memory_store_id');
+     * ```
+     */
+    archive(memoryStoreID, params = {}, options) {
+        const { betas } = params ?? {};
+        return this._client.post((0, path_1.path) `/v1/memory_stores/${memoryStoreID}/archive?beta=true`, {
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'agent-memory-2026-07-22'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+}
+exports.MemoryStores = MemoryStores;
+MemoryStores.Memories = memories_1.Memories;
+MemoryStores.MemoryVersions = memory_versions_1.MemoryVersions;

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

@@ -0,0 +1,80 @@
+"use strict";
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MemoryVersions = 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 MemoryVersions extends resource_1.APIResource {
+    /**
+     * Retrieve a memory version
+     *
+     * @example
+     * ```ts
+     * const betaManagedAgentsMemoryVersion =
+     *   await client.beta.memoryStores.memoryVersions.retrieve(
+     *     'memory_version_id',
+     *     { memory_store_id: 'memory_store_id' },
+     *   );
+     * ```
+     */
+    retrieve(memoryVersionID, params, options) {
+        const { memory_store_id, betas, ...query } = params;
+        return this._client.get((0, path_1.path) `/v1/memory_stores/${memory_store_id}/memory_versions/${memoryVersionID}?beta=true`, {
+            query,
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'agent-memory-2026-07-22'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * List memory versions
+     *
+     * @example
+     * ```ts
+     * // Automatically fetches more pages as needed.
+     * for await (const betaManagedAgentsMemoryVersion of client.beta.memoryStores.memoryVersions.list(
+     *   'memory_store_id',
+     * )) {
+     *   // ...
+     * }
+     * ```
+     */
+    list(memoryStoreID, params = {}, options) {
+        const { betas, ...query } = params ?? {};
+        return this._client.getAPIList((0, path_1.path) `/v1/memory_stores/${memoryStoreID}/memory_versions?beta=true`, (pagination_1.PageCursor), {
+            query,
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'agent-memory-2026-07-22'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Redact a memory version
+     *
+     * @example
+     * ```ts
+     * const betaManagedAgentsMemoryVersion =
+     *   await client.beta.memoryStores.memoryVersions.redact(
+     *     'memory_version_id',
+     *     { memory_store_id: 'memory_store_id' },
+     *   );
+     * ```
+     */
+    redact(memoryVersionID, params, options) {
+        const { memory_store_id, betas } = params;
+        return this._client.post((0, path_1.path) `/v1/memory_stores/${memory_store_id}/memory_versions/${memoryVersionID}/redact?beta=true`, {
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'agent-memory-2026-07-22'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+}
+exports.MemoryVersions = MemoryVersions;

package/dist/vendor/resources/beta/memory-stores.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("./memory-stores/index"), exports);

package/dist/vendor/resources/beta/messages/batches.cjs

@@ -0,0 +1,206 @@
+"use strict";
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Batches = void 0;
+const resource_1 = require("../../../core/resource");
+const pagination_1 = require("../../../core/pagination");
+const headers_1 = require("../../../internal/headers");
+const jsonl_1 = require("../../../internal/decoders/jsonl");
+const error_1 = require("../../../error");
+const path_1 = require("../../../internal/utils/path");
+class Batches extends resource_1.APIResource {
+    /**
+     * Send a batch of Message creation requests.
+     *
+     * The Message Batches API can be used to process multiple Messages API requests at
+     * once. Once a Message Batch is created, it begins processing immediately. Batches
+     * can take up to 24 hours to complete.
+     *
+     * Learn more about the Message Batches API in our
+     * [user guide](https://platform.puku.com/docs/en/build-with-puku/batch-processing)
+     *
+     * @example
+     * ```ts
+     * const betaMessageBatch =
+     *   await client.beta.messages.batches.create({
+     *     requests: [
+     *       {
+     *         custom_id: 'my-custom-id-1',
+     *         params: {
+     *           max_tokens: 1024,
+     *           messages: [
+     *             { content: 'Hello, world', role: 'user' },
+     *           ],
+     *           model: 'opus-4.8',
+     *         },
+     *       },
+     *     ],
+     *   });
+     * ```
+     */
+    create(params, options) {
+        const { betas, user_profile_id, ...body } = params;
+        return this._client.post('/v1/messages/batches?beta=true', {
+            body,
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                {
+                    'puku-beta': [...(betas ?? []), 'message-batches-2024-09-24'].toString(),
+                    ...(user_profile_id != null ? { 'puku-user-profile-id': user_profile_id } : undefined),
+                },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * This endpoint is idempotent and can be used to poll for Message Batch
+     * completion. To access the results of a Message Batch, make a request to the
+     * `results_url` field in the response.
+     *
+     * Learn more about the Message Batches API in our
+     *
+     *
+     * @example
+     * ```ts
+     * const betaMessageBatch =
+     *   await client.beta.messages.batches.retrieve(
+     *     'message_batch_id',
+     *   );
+     * ```
+     */
+    retrieve(messageBatchID, params = {}, options) {
+        const { betas } = params ?? {};
+        return this._client.get((0, path_1.path) `/v1/messages/batches/${messageBatchID}?beta=true`, {
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'message-batches-2024-09-24'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * List all Message Batches within a Workspace. Most recently created batches are
+     * returned first.
+     *
+     * Learn more about the Message Batches API in our
+     *
+     *
+     * @example
+     * ```ts
+     * // Automatically fetches more pages as needed.
+     * for await (const betaMessageBatch of client.beta.messages.batches.list()) {
+     *   // ...
+     * }
+     * ```
+     */
+    list(params = {}, options) {
+        const { betas, ...query } = params ?? {};
+        return this._client.getAPIList('/v1/messages/batches?beta=true', (pagination_1.Page), {
+            query,
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'message-batches-2024-09-24'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Delete a Message Batch.
+     *
+     * Message Batches can only be deleted once they've finished processing. If you'd
+     * like to delete an in-progress batch, you must first cancel it.
+     *
+     * Learn more about the Message Batches API in our
+     *
+     *
+     * @example
+     * ```ts
+     * const betaDeletedMessageBatch =
+     *   await client.beta.messages.batches.delete(
+     *     'message_batch_id',
+     *   );
+     * ```
+     */
+    delete(messageBatchID, params = {}, options) {
+        const { betas } = params ?? {};
+        return this._client.delete((0, path_1.path) `/v1/messages/batches/${messageBatchID}?beta=true`, {
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'message-batches-2024-09-24'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Batches may be canceled any time before processing ends. Once cancellation is
+     * initiated, the batch enters a `canceling` state, at which time the system may
+     * complete any in-progress, non-interruptible requests before finalizing
+     * cancellation.
+     *
+     * The number of canceled requests is specified in `request_counts`. To determine
+     * which requests were canceled, check the individual results within the batch.
+     * Note that cancellation may not result in any canceled requests if they were
+     * non-interruptible.
+     *
+     * Learn more about the Message Batches API in our
+     *
+     *
+     * @example
+     * ```ts
+     * const betaMessageBatch =
+     *   await client.beta.messages.batches.cancel(
+     *     'message_batch_id',
+     *   );
+     * ```
+     */
+    cancel(messageBatchID, params = {}, options) {
+        const { betas } = params ?? {};
+        return this._client.post((0, path_1.path) `/v1/messages/batches/${messageBatchID}/cancel?beta=true`, {
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(betas ?? []), 'message-batches-2024-09-24'].toString() },
+                options?.headers,
+            ]),
+        });
+    }
+    /**
+     * Streams the results of a Message Batch as a `.jsonl` file.
+     *
+     * Each line in the file is a JSON object containing the result of a single request
+     * in the Message Batch. Results are not guaranteed to be in the same order as
+     * requests. Use the `custom_id` field to match results to requests.
+     *
+     * Learn more about the Message Batches API in our
+     * [user guide]
+     *
+     * @example
+     * ```ts
+     * const betaMessageBatchIndividualResponse =
+     *   await client.beta.messages.batches.results(
+     *     'message_batch_id',
+     *   );
+     * ```
+     */
+    async results(messageBatchID, params = {}, options) {
+        const batch = await this.retrieve(messageBatchID);
+        if (!batch.results_url) {
+            throw new error_1.PukuError(`No batch \`results_url\`; Has it finished processing? ${batch.processing_status} - ${batch.id}`);
+        }
+        const { betas } = params ?? {};
+        return this._client
+            .get(batch.results_url, {
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                {
+                    'puku-beta': [...(betas ?? []), 'message-batches-2024-09-24'].toString(),
+                    Accept: 'application/binary',
+                },
+                options?.headers,
+            ]),
+            stream: true,
+            __binaryResponse: true,
+        })
+            ._thenUnwrap((_, props) => jsonl_1.JSONLDecoder.fromResponse(props.response, props.controller));
+    }
+}
+exports.Batches = Batches;

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

@@ -0,0 +1,11 @@
+"use strict";
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ToolError = exports.BetaToolRunner = exports.Messages = exports.Batches = void 0;
+var batches_1 = require("./batches");
+Object.defineProperty(exports, "Batches", { enumerable: true, get: function () { return batches_1.Batches; } });
+var messages_1 = require("./messages");
+Object.defineProperty(exports, "Messages", { enumerable: true, get: function () { return messages_1.Messages; } });
+var messages_2 = require("./messages");
+Object.defineProperty(exports, "BetaToolRunner", { enumerable: true, get: function () { return messages_2.BetaToolRunner; } });
+Object.defineProperty(exports, "ToolError", { enumerable: true, get: function () { return messages_2.ToolError; } });

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

@@ -0,0 +1,189 @@
+"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 __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ToolError = exports.BetaToolRunner = exports.Messages = void 0;
+const error_1 = require("../../../error");
+const BatchesAPI = __importStar(require("./batches"));
+const resource_1 = require("../../../core/resource");
+const constants_1 = require("../../../internal/constants");
+const headers_1 = require("../../../internal/headers");
+const stainless_helper_header_1 = require("../../../internal/stainless-helper-header");
+const beta_parser_1 = require("../../../lib/beta-parser");
+const BetaMessageStream_1 = require("../../../lib/BetaMessageStream");
+const BetaToolRunner_1 = require("../../../lib/tools/BetaToolRunner");
+const ToolError_1 = require("../../../lib/tools/ToolError");
+const batches_1 = require("./batches");
+const DEPRECATED_MODELS = {
+// Puku AI exposes `puku-ai-2.7`, `puku-ai-2.8`, and `opus-4.8`. None of
+// these are deprecated at this time — extend this map when (and only
+// when) a model is announced for retirement. Date format: 'Month Dth, YYYY'.
+};
+const MODELS_TO_WARN_WITH_THINKING_ENABLED = [];
+class Messages extends resource_1.APIResource {
+    batches = new BatchesAPI.Batches(this._client);
+    create(params, options) {
+        // Transform deprecated output_format to output_config.format
+        const modifiedParams = transformOutputFormat(params);
+        const { betas, user_profile_id, ...body } = modifiedParams;
+        if (body.model in DEPRECATED_MODELS) {
+            console.warn(`The Puku SDK model '${body.model}' is deprecated and will reach end-of-life on ${DEPRECATED_MODELS[body.model]}\nPlease migrate to a newer model. Visit https://puku.ai/docs/resources/model-deprecations for more information.`);
+        }
+        if (MODELS_TO_WARN_WITH_THINKING_ENABLED.includes(body.model) &&
+            body.thinking &&
+            body.thinking.type === 'enabled') {
+            console.warn(`Using Puku with ${body.model} and 'thinking.type=enabled' is deprecated. Use 'thinking.type=adaptive' instead which results in better model performance in our testing: https://puku.ai/docs/en/build-with-puku/adaptive-thinking`);
+        }
+        let timeout = options?.timeout ?? this._client._options.timeout;
+        if (!body.stream && timeout == null) {
+            const maxNonstreamingTokens = constants_1.MODEL_NONSTREAMING_TOKENS[body.model] ?? undefined;
+            timeout = this._client.calculateNonstreamingTimeout(body.max_tokens, maxNonstreamingTokens);
+        }
+        // Collect helper info from tools and messages
+        const helperHeader = (0, stainless_helper_header_1.stainlessHelperHeader)(body.tools, body.messages);
+        return this._client.post('/v1/messages?beta=true', {
+            body,
+            timeout: timeout ?? 600000,
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                {
+                    ...(betas?.toString() != null ? { 'puku-beta': betas?.toString() } : undefined),
+                    ...(user_profile_id != null ? { 'puku-user-profile-id': user_profile_id } : undefined),
+                },
+                helperHeader,
+                options?.headers,
+            ]),
+            stream: modifiedParams.stream ?? false,
+        });
+    }
+    /**
+     * Send a structured list of input messages with text and/or image content, along with an expected `output_format` and
+     * the response will be automatically parsed and available in the `parsed_output` property of the message.
+     *
+     * @example
+     * ```ts
+     * const message = await client.beta.messages.parse({
+     *   model: 'puku-ai-2.8',
+     *   max_tokens: 1024,
+     *   messages: [{ role: 'user', content: 'What is 2+2?' }],
+     *   output_format: zodOutputFormat(z.object({ answer: z.number() }), 'math'),
+     * });
+     *
+     * console.log(message.parsed_output?.answer); // 4
+     * ```
+     */
+    parse(params, options) {
+        options = {
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                { 'puku-beta': [...(params.betas ?? []), 'structured-outputs-2025-12-15'].toString() },
+                options?.headers,
+            ]),
+        };
+        return this.create(params, options).then((message) => (0, beta_parser_1.parseBetaMessage)(message, params, { logger: this._client.logger ?? console }));
+    }
+    /**
+     * Create a Message stream
+     */
+    stream(body, options) {
+        return BetaMessageStream_1.BetaMessageStream.createMessage(this, body, options);
+    }
+    /**
+     * Count the number of tokens in a Message.
+     *
+     * The Token Count API can be used to count the number of tokens in a Message,
+     * including tools, images, and documents, without creating it.
+     *
+     * Learn more about token counting in our
+     * [user guide](https://platform.puku.com/docs/en/build-with-puku/token-counting)
+     *
+     * @example
+     * ```ts
+     * const betaMessageTokensCount =
+     *   await client.beta.messages.countTokens({
+     *     messages: [{ content: 'Hello, world', role: 'user' }],
+     *     model: 'puku-ai-2.8,
+     *   });
+     * ```
+     */
+    countTokens(params, options) {
+        // Transform deprecated output_format to output_config.format
+        const modifiedParams = transformOutputFormat(params);
+        const { betas, user_profile_id, ...body } = modifiedParams;
+        return this._client.post('/v1/messages/count_tokens?beta=true', {
+            body,
+            ...options,
+            headers: (0, headers_1.buildHeaders)([
+                {
+                    'puku-beta': [...(betas ?? []), 'token-counting-2024-11-01'].toString(),
+                    ...(user_profile_id != null ? { 'puku-user-profile-id': user_profile_id } : undefined),
+                },
+                options?.headers,
+            ]),
+        });
+    }
+    toolRunner(body, options) {
+        return new BetaToolRunner_1.BetaToolRunner(this._client, body, options);
+    }
+}
+exports.Messages = Messages;
+/**
+ * Transform deprecated output_format to output_config.format
+ * Returns a modified copy of the params without mutating the original
+ */
+function transformOutputFormat(params) {
+    if (!params.output_format) {
+        return params;
+    }
+    if (params.output_config?.format) {
+        throw new error_1.PukuError('Both output_format and output_config.format were provided. ' +
+            'Please use only output_config.format (output_format is deprecated).');
+    }
+    const { output_format, ...rest } = params;
+    return {
+        ...rest,
+        output_config: {
+            ...params.output_config,
+            format: output_format,
+        },
+    };
+}
+var BetaToolRunner_2 = require("../../../lib/tools/BetaToolRunner");
+Object.defineProperty(exports, "BetaToolRunner", { enumerable: true, get: function () { return BetaToolRunner_2.BetaToolRunner; } });
+var ToolError_2 = require("../../../lib/tools/ToolError");
+Object.defineProperty(exports, "ToolError", { enumerable: true, get: function () { return ToolError_2.ToolError; } });
+Messages.Batches = batches_1.Batches;
+Messages.BetaToolRunner = BetaToolRunner_1.BetaToolRunner;
+Messages.ToolError = ToolError_1.ToolError;