bedrock-agentcore 0.0.1 → 0.2.0

package/dist/src/runtime/app.js

@@ -0,0 +1,485 @@
+import { Buffer } from 'buffer';
+import { createRequire } from 'module';
+import { randomUUID } from 'crypto';
+import Fastify from 'fastify';
+import { z } from 'zod';
+import { runWithContext } from './context.js';
+const require = createRequire(import.meta.url);
+const fastifySse = require('@fastify/sse');
+const fastifyWebsocket = require('@fastify/websocket');
+/**
+ * Fastify-based HTTP server for hosting agents on AWS Bedrock AgentCore Runtime.
+ *
+ * This class provides an HTTP server that implements the AgentCore Runtime protocol
+ * with health check and invocation endpoints. The server runs on port 8080 and
+ * handles both JSON responses and Server-Sent Events (SSE) streaming.
+ *
+ * @example
+ * ```typescript
+ * const app = new BedrockAgentCoreApp({
+ *   invocationHandler: {
+ *     requestSchema: z.object({ message: z.string() }),
+ *     process: async (request, context) => {
+ *       console.log(`Processing request with session ${context.sessionId}`)
+ *       return `Hello ${request.message}!`
+ *     }
+ *   }
+ * })
+ *
+ * app.run()
+ * ```
+ */
+export class BedrockAgentCoreApp {
+    _app;
+    _config;
+    _handler;
+    _websocketHandler;
+    _activeTasksMap = new Map();
+    _taskCounter = 0;
+    _pingHandler;
+    _forcedPingStatus;
+    _lastStatusUpdateTime = Date.now();
+    _lastKnownStatus;
+    /**
+     * Creates a new BedrockAgentCoreApp instance.
+     *
+     * @param params - Configuration including handler and optional settings
+     */
+    constructor(params) {
+        if (!params ||
+            typeof params !== 'object' ||
+            !params.invocationHandler ||
+            typeof params.invocationHandler.process !== 'function') {
+            throw new Error('BedrockAgentCoreApp constructor requires an object with a handler property. ' +
+                'Usage: new BedrockAgentCoreApp({ invocationHandler: { process: handler } })');
+        }
+        this._handler = params.invocationHandler;
+        this._websocketHandler = params.websocketHandler ?? undefined;
+        this._config = params.config ?? {};
+        this._pingHandler = params.pingHandler ?? undefined;
+        // Configure Fastify logger based on BedrockAgentCoreAppConfig
+        const loggerConfig = this._getLoggerConfig();
+        this._app = Fastify({
+            logger: loggerConfig,
+            disableRequestLogging: this._config.logging?.disableRequestLogging ?? ((req) => req.url === '/ping'),
+        });
+    }
+    /**
+     * Starts the Fastify server on port 8080.
+     */
+    run() {
+        const PORT = 8080;
+        // Wait for Fastify to be ready (all plugins registered), setup routes, and start the server
+        Promise.resolve(this._registerPlugins())
+            .then(() => {
+            this._setupContentTypeParsers();
+            this._setupRoutes();
+            return this._app.listen({ port: PORT, host: '0.0.0.0' });
+        })
+            .then(() => {
+            this._app.log.info(`Server listening on port ${PORT}`);
+        })
+            .catch((error) => {
+            this._app.log.error(error);
+            process.exit(1);
+        });
+    }
+    /**
+     * Register an async task for health tracking.
+     *
+     * @param name - Human-readable task name
+     * @param metadata - Optional task metadata
+     * @returns Task ID for completion tracking
+     */
+    addAsyncTask(name, metadata) {
+        const taskId = ++this._taskCounter;
+        const taskInfo = {
+            name,
+            startTime: Date.now(),
+        };
+        if (metadata) {
+            taskInfo.metadata = metadata;
+        }
+        this._activeTasksMap.set(taskId, taskInfo);
+        return taskId;
+    }
+    /**
+     * Mark an async task as complete.
+     *
+     * @param taskId - Task ID from addAsyncTask
+     * @returns True if task was found and removed
+     */
+    completeAsyncTask(taskId) {
+        return this._activeTasksMap.delete(taskId);
+    }
+    /**
+     * Get current ping status based on priority system.
+     * Priority: Forced \> Custom Handler \> Automatic
+     *
+     * @returns Current health status
+     */
+    getCurrentPingStatus() {
+        // Priority 1: Forced status
+        if (this._forcedPingStatus) {
+            return this._forcedPingStatus;
+        }
+        // Priority 2: Custom handler
+        if (this._pingHandler) {
+            try {
+                const result = this._pingHandler();
+                // Handle both sync and async handlers
+                return result instanceof Promise ? 'Healthy' : result;
+            }
+            catch {
+                this._app.log.warn('Custom ping handler failed, falling back to automatic');
+            }
+        }
+        // Priority 3: Automatic based on active tasks
+        const status = this._activeTasksMap.size > 0 ? 'HealthyBusy' : 'Healthy';
+        // Track status changes
+        if (!this._lastKnownStatus || this._lastKnownStatus !== status) {
+            this._lastKnownStatus = status;
+            this._lastStatusUpdateTime = Date.now();
+        }
+        return status;
+    }
+    /**
+     * Decorator to automatically track async tasks.
+     * Status becomes HealthyBusy during execution.
+     *
+     * @param fn - Async function to wrap
+     * @returns Wrapped function with automatic task tracking
+     * @throws Error if fn is not an async function
+     */
+    asyncTask(fn) {
+        // Validate that fn is actually an async function
+        if (fn.constructor.name !== 'AsyncFunction') {
+            throw new Error('asyncTask can only be applied to async functions');
+        }
+        // eslint-disable-next-line @typescript-eslint/no-this-alias
+        const self = this;
+        const wrapped = async function (...args) {
+            const taskId = self.addAsyncTask(fn.name || 'anonymous');
+            try {
+                return await fn.apply(this, args);
+            }
+            finally {
+                self.completeAsyncTask(taskId);
+            }
+        };
+        Object.defineProperty(wrapped, 'name', { value: fn.name });
+        return wrapped;
+    }
+    /**
+     * Get information about currently running async tasks.
+     *
+     * @returns Task status with count and details
+     */
+    getAsyncTaskInfo() {
+        const now = Date.now();
+        const runningJobs = Array.from(this._activeTasksMap.values()).map((task) => ({
+            name: task.name,
+            duration: (now - task.startTime) / 1000, // Convert to seconds
+        }));
+        return {
+            activeCount: this._activeTasksMap.size,
+            runningJobs,
+        };
+    }
+    /**
+     * Sets up HTTP routes for the server.
+     */
+    _setupRoutes() {
+        // Health check endpoint
+        this._app.get('/ping', this._handlePing.bind(this));
+        // Invocation endpoint
+        this._app.post('/invocations', { sse: true }, this._handleInvocation.bind(this));
+        // WebSocket endpoint (only if handler exists)
+        if (this._websocketHandler) {
+            this._app.get('/ws', { websocket: true }, this._handleWebSocket.bind(this));
+        }
+    }
+    /**
+     * Register custom content type parsers using Fastify's native addContentTypeParser.
+     */
+    _setupContentTypeParsers() {
+        this._config.contentTypeParsers?.forEach((parserConfig) => {
+            const { contentType, parser, parseAs, bodyLimit } = parserConfig;
+            if (parseAs === 'stream') {
+                // Use FastifyContentTypeParser for stream mode (raw request payload)
+                const opts = bodyLimit ? { bodyLimit } : {};
+                this._app.addContentTypeParser(contentType, opts, parser);
+            }
+            else {
+                // Use FastifyBodyParser for string/buffer modes (default to 'string' when parseAs is undefined)
+                const opts = {
+                    parseAs: parseAs || 'string',
+                    ...(bodyLimit && { bodyLimit }),
+                };
+                this._app.addContentTypeParser(contentType, opts, parser);
+            }
+        });
+    }
+    /**
+     * Registers Fastify plugins.
+     */
+    async _registerPlugins() {
+        await this._app.register(fastifySse);
+        if (this._websocketHandler) {
+            await this._app.register(fastifyWebsocket);
+        }
+    }
+    /**
+     * Gets the logger configuration based on BedrockAgentCoreAppConfig.
+     *
+     * @returns Fastify logger configuration
+     */
+    _getLoggerConfig() {
+        const loggingConfig = this._config.logging;
+        // If no logging config provided, use default (enabled with info level)
+        if (!loggingConfig) {
+            return true;
+        }
+        // If logging is explicitly disabled, return false
+        if (loggingConfig.enabled === false) {
+            return false;
+        }
+        // If no options provided, use default level
+        if (!loggingConfig.options) {
+            return { level: 'info' };
+        }
+        // Return FastifyLoggerOptions directly
+        return loggingConfig.options;
+    }
+    /**
+     * Handles health check requests.
+     *
+     * @param request - Fastify request object
+     * @param reply - Fastify reply object
+     */
+    async _handlePing(request, reply) {
+        const status = this.getCurrentPingStatus();
+        const response = {
+            status,
+            time_of_last_update: new Date(this._lastStatusUpdateTime).toISOString(),
+        };
+        await reply.send(response);
+    }
+    /**
+     * Handles agent invocation requests.
+     *
+     * @param request - Fastify request object
+     * @param reply - Fastify reply object
+     */
+    async _handleInvocation(request, reply) {
+        try {
+            // Extract context
+            const context = this._extractContext(request);
+            // Validate sessionId
+            if (!context.sessionId) {
+                await reply.status(400).send({
+                    error: 'Missing sessionId. Provide via x-amzn-bedrock-agentcore-runtime-session-id header or request body.',
+                });
+                return;
+            }
+            // Validate request body with schema if provided
+            let handlerRequest;
+            if (this._handler.requestSchema) {
+                try {
+                    handlerRequest = await this._handler.requestSchema.parseAsync(request.body);
+                }
+                catch (error) {
+                    if (error instanceof z.ZodError) {
+                        await reply.status(400).send({
+                            error: 'Invalid request body format',
+                            details: error.issues,
+                        });
+                        return;
+                    }
+                    throw error;
+                }
+            }
+            else {
+                handlerRequest = request.body;
+            }
+            // Invoke handler with context
+            const result = await runWithContext(context, async () => {
+                return await this._handler.process(handlerRequest, context);
+            });
+            // Check if result is an async generator (streaming response)
+            if (this._isAsyncGenerator(result)) {
+                if (reply.sse) {
+                    // Wrap streaming in context so getContext() works during iteration
+                    await runWithContext(context, async () => {
+                        await this._handleStreamingResponse(reply, result);
+                    });
+                }
+                else {
+                    await reply.status(406).send({
+                        error: 'Streaming response requires Accept: text/event-stream header. Please include this header in your request to receive streaming data.',
+                    });
+                }
+            }
+            else {
+                // Return non-streaming response
+                if (reply.sse) {
+                    // SSE mode active but not streaming - respond based on client's Accept header
+                    const acceptHeader = request.headers.accept || 'application/json';
+                    if (acceptHeader.includes('application/json') || acceptHeader.includes('*/*')) {
+                        await reply.type('application/json').send(result);
+                    }
+                    else if (acceptHeader.includes('text/plain')) {
+                        const text = typeof result === 'string' ? result : JSON.stringify(result);
+                        await reply.type('text/plain').send(text);
+                    }
+                    else if (acceptHeader.includes('application/octet-stream')) {
+                        const buffer = Buffer.isBuffer(result) ? result : Buffer.from(JSON.stringify(result));
+                        await reply.type('application/octet-stream').send(buffer);
+                    }
+                    else {
+                        // Default to JSON
+                        await reply.type('application/json').send(result);
+                    }
+                }
+                else {
+                    await reply.send(result);
+                }
+            }
+        }
+        catch (error) {
+            // Handle errors
+            const errorMessage = error instanceof Error ? error.message : 'Unknown error';
+            await reply.status(500).send({
+                error: errorMessage,
+            });
+        }
+    }
+    /**
+     * Checks if a value is an async generator.
+     *
+     * @param value - Value to check
+     * @returns True if the value is an async generator
+     */
+    _isAsyncGenerator(value) {
+        return (value !== null &&
+            value !== undefined &&
+            typeof value === 'object' &&
+            'next' in value &&
+            typeof value.next === 'function' &&
+            'return' in value &&
+            typeof value.return === 'function' &&
+            'throw' in value &&
+            typeof value.throw === 'function' &&
+            Symbol.asyncIterator in value &&
+            typeof value[Symbol.asyncIterator] === 'function');
+    }
+    /**
+     * Handles streaming response using Server-Sent Events (SSE) via Fastify SSE plugin.
+     *
+     * @param reply - Fastify reply object
+     * @param generator - Async generator that yields data chunks
+     */
+    async _handleStreamingResponse(reply, generator) {
+        try {
+            await reply.sse.keepAlive();
+            // Stream data chunks
+            for await (const chunk of generator) {
+                // Stop if client disconnected
+                if (!reply.sse.isConnected) {
+                    break;
+                }
+                // Send SSE message
+                await reply.sse.send(chunk);
+            }
+        }
+        catch (error) {
+            // Send error event if still connected
+            if (reply.sse && reply.sse.isConnected) {
+                const errorMessage = error instanceof Error ? error.message : 'Unknown error';
+                await reply.sse.send({
+                    event: 'error',
+                    data: { error: errorMessage },
+                });
+            }
+            else {
+                this._app.log.error(error, 'Error during streaming SSE events');
+                throw new Error('Error during streaming SSE events');
+            }
+        }
+        finally {
+            // Close the SSE connection
+            if (reply.sse) {
+                reply.sse.close();
+            }
+        }
+    }
+    /**
+     * Handles WebSocket connections at /ws endpoint.
+     *
+     * @param connection - Fastify WebSocket connection
+     * @param request - Fastify request object
+     */
+    async _handleWebSocket(connection, request) {
+        try {
+            // Extract context from WebSocket request
+            const context = this._extractContext(request);
+            request.log.info({ sessionId: context.sessionId }, 'WebSocket connection established');
+            // Call the user's WebSocket handler with context (guaranteed to exist since route is conditionally registered)
+            await runWithContext(context, async () => {
+                return await this._websocketHandler(connection, context);
+            });
+        }
+        catch (error) {
+            request.log.error({ error: error instanceof Error ? error.message : String(error) }, 'WebSocket handler error');
+            try {
+                connection.close(1011, 'Internal server error');
+            }
+            catch (closeError) {
+                request.log.error({ closeError: closeError instanceof Error ? closeError.message : String(closeError) }, 'Error closing WebSocket');
+            }
+        }
+    }
+    /**
+     * Extracts request context from the incoming request.
+     *
+     * @param request - Fastify request object
+     * @returns Request context with sessionId, headers, and request-scoped logger
+     */
+    _extractContext(request) {
+        // Extract sessionId from AWS header or body
+        const sessionId = request.headers['x-amzn-bedrock-agentcore-runtime-session-id'] ||
+            request.body?.sessionId ||
+            '';
+        // Extract request ID or generate if missing
+        const requestId = request.headers['x-amzn-bedrock-agentcore-runtime-request-id'] || randomUUID();
+        // Extract OAuth2 callback URL
+        const oauth2CallbackUrl = request.headers['oauth2callbackurl'];
+        // Filter headers to include only Authorization and Custom-* headers
+        const filteredHeaders = {};
+        for (const [key, value] of Object.entries(request.headers)) {
+            const lowerKey = key.toLowerCase();
+            const stringValue = typeof value === 'string' ? value : Array.isArray(value) ? value.join(', ') : undefined;
+            if (stringValue) {
+                // Include Authorization header
+                if (lowerKey === 'authorization') {
+                    filteredHeaders[key] = stringValue;
+                }
+                // Include Custom-* headers
+                else if (lowerKey.startsWith('x-amzn-bedrock-agentcore-runtime-custom-')) {
+                    filteredHeaders[key] = stringValue;
+                }
+            }
+        }
+        // Extract workload token from header (if present)
+        const workloadAccessToken = request.headers['workloadaccesstoken'];
+        return {
+            sessionId,
+            headers: filteredHeaders,
+            workloadAccessToken,
+            requestId,
+            oauth2CallbackUrl,
+            log: request.log,
+        };
+    }
+}
+//# sourceMappingURL=app.js.map

package/dist/src/runtime/app.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"app.js","sourceRoot":"","sources":["../../../src/runtime/app.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAA;AAC/B,OAAO,EAAE,aAAa,EAAE,MAAM,QAAQ,CAAA;AACtC,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAA;AACnC,OAAO,OAAO,MAAM,SAAS,CAAA;AAC7B,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAuBvB,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAA;AAE7C,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;AAC9C,MAAM,UAAU,GAAG,OAAO,CAAC,cAAc,CAAC,CAAA;AAC1C,MAAM,gBAAgB,GAAG,OAAO,CAAC,oBAAoB,CAAC,CAAA;AAEtD;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,OAAO,mBAAmB;IACb,IAAI,CAAiB;IACrB,OAAO,CAA2B;IAClC,QAAQ,CAA2E;IAC5F,iBAAiB,CAA8B;IACtC,eAAe,GAA+B,IAAI,GAAG,EAAE,CAAA;IAChE,YAAY,GAAW,CAAC,CAAA;IACxB,YAAY,CAA0D;IACtE,iBAAiB,CAAe;IAChC,qBAAqB,GAAW,IAAI,CAAC,GAAG,EAAE,CAAA;IAC1C,gBAAgB,CAAe;IAEvC;;;;OAIG;IACH,YAAY,MAA0C;QACpD,IACE,CAAC,MAAM;YACP,OAAO,MAAM,KAAK,QAAQ;YAC1B,CAAC,MAAM,CAAC,iBAAiB;YACzB,OAAO,MAAM,CAAC,iBAAiB,CAAC,OAAO,KAAK,UAAU,EACtD,CAAC;YACD,MAAM,IAAI,KAAK,CACb,8EAA8E;gBAC5E,6EAA6E,CAChF,CAAA;QACH,CAAC;QACD,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,iBAAiB,CAAA;QACxC,IAAI,CAAC,iBAAiB,GAAG,MAAM,CAAC,gBAAgB,IAAI,SAAS,CAAA;QAC7D,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,MAAM,IAAI,EAAE,CAAA;QAClC,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,WAAW,IAAI,SAAS,CAAA;QAEnD,8DAA8D;QAC9D,MAAM,YAAY,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAA;QAE5C,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC;YAClB,MAAM,EAAE,YAAY;YACpB,qBAAqB,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,qBAAqB,IAAI,CAAC,CAAC,GAAG,EAAW,EAAE,CAAC,GAAG,CAAC,GAAG,KAAK,OAAO,CAAC;SAC9G,CAAC,CAAA;IACJ,CAAC;IAED;;OAEG;IACH,GAAG;QACD,MAAM,IAAI,GAAG,IAAI,CAAA;QAEjB,4FAA4F;QAC5F,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;aACrC,IAAI,CAAC,GAAG,EAAE;YACT,IAAI,CAAC,wBAAwB,EAAE,CAAA;YAC/B,IAAI,CAAC,YAAY,EAAE,CAAA;YACnB,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAA;QAC1D,CAAC,CAAC;aACD,IAAI,CAAC,GAAG,EAAE;YACT,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,4BAA4B,IAAI,EAAE,CAAC,CAAA;QACxD,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,KAAY,EAAE,EAAE;YACtB,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;YAC1B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACjB,CAAC,CAAC,CAAA;IACN,CAAC;IAED;;;;;;OAMG;IACI,YAAY,CAAC,IAAY,EAAE,QAAkC;QAClE,MAAM,MAAM,GAAG,EAAE,IAAI,CAAC,YAAY,CAAA;QAClC,MAAM,QAAQ,GAAkB;YAC9B,IAAI;YACJ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAA;QACD,IAAI,QAAQ,EAAE,CAAC;YACb,QAAQ,CAAC,QAAQ,GAAG,QAAQ,CAAA;QAC9B,CAAC;QACD,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;QAC1C,OAAO,MAAM,CAAA;IACf,CAAC;IAED;;;;;OAKG;IACI,iBAAiB,CAAC,MAAc;QACrC,OAAO,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;IAC5C,CAAC;IAED;;;;;OAKG;IACI,oBAAoB;QACzB,4BAA4B;QAC5B,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,OAAO,IAAI,CAAC,iBAAiB,CAAA;QAC/B,CAAC;QAED,6BAA6B;QAC7B,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,EAAE,CAAA;gBAClC,sCAAsC;gBACtC,OAAO,MAAM,YAAY,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAA;YACvD,CAAC;YAAC,MAAM,CAAC;gBACP,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,uDAAuD,CAAC,CAAA;YAC7E,CAAC;QACH,CAAC;QAED,8CAA8C;QAC9C,MAAM,MAAM,GAAiB,IAAI,CAAC,eAAe,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAA;QAEtF,uBAAuB;QACvB,IAAI,CAAC,IAAI,CAAC,gBAAgB,IAAI,IAAI,CAAC,gBAAgB,KAAK,MAAM,EAAE,CAAC;YAC/D,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAA;YAC9B,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QACzC,CAAC;QAED,OAAO,MAAM,CAAA;IACf,CAAC;IAED;;;;;;;OAOG;IACI,SAAS,CAAqD,EAAK;QACxE,iDAAiD;QACjD,IAAI,EAAE,CAAC,WAAW,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;YAC5C,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAA;QACrE,CAAC;QAED,4DAA4D;QAC5D,MAAM,IAAI,GAAG,IAAI,CAAA;QACjB,MAAM,OAAO,GAAG,KAAK,WAA0B,GAAG,IAAe;YAC/D,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,IAAI,IAAI,WAAW,CAAC,CAAA;YACxD,IAAI,CAAC;gBACH,OAAO,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAA;YACnC,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAA;YAChC,CAAC;QACH,CAAC,CAAA;QACD,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC,CAAA;QAC1D,OAAO,OAAY,CAAA;IACrB,CAAC;IAED;;;;OAIG;IACI,gBAAgB;QACrB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QACtB,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC3E,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,QAAQ,EAAE,CAAC,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,IAAI,EAAE,qBAAqB;SAC/D,CAAC,CAAC,CAAA;QAEH,OAAO;YACL,WAAW,EAAE,IAAI,CAAC,eAAe,CAAC,IAAI;YACtC,WAAW;SACZ,CAAA;IACH,CAAC;IAED;;OAEG;IACK,YAAY;QAClB,wBAAwB;QACxB,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;QAEnD,sBAAsB;QACtB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;QAEhF,8CAA8C;QAC9C,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;QAC7E,CAAC;IACH,CAAC;IAED;;OAEG;IACK,wBAAwB;QAC9B,IAAI,CAAC,OAAO,CAAC,kBAAkB,EAAE,OAAO,CAAC,CAAC,YAAY,EAAE,EAAE;YACxD,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,YAAY,CAAA;YAEhE,IAAI,OAAO,KAAK,QAAQ,EAAE,CAAC;gBACzB,qEAAqE;gBACrE,MAAM,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAA;gBAC3C,IAAI,CAAC,IAAI,CAAC,oBAAoB,CAAC,WAAW,EAAE,IAAI,EAAE,MAAkC,CAAC,CAAA;YACvF,CAAC;iBAAM,CAAC;gBACN,gGAAgG;gBAChG,MAAM,IAAI,GAAG;oBACX,OAAO,EAAE,OAAO,IAAI,QAAQ;oBAC5B,GAAG,CAAC,SAAS,IAAI,EAAE,SAAS,EAAE,CAAC;iBAChC,CAAA;gBACD,IAAI,CAAC,IAAI,CAAC,oBAAoB,CAAC,WAAW,EAAE,IAAI,EAAE,MAA4C,CAAC,CAAA;YACjG,CAAC;QACH,CAAC,CAAC,CAAA;IACJ,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,gBAAgB;QAC5B,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAA;QACpC,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAA;QAC5C,CAAC;IACH,CAAC;IAED;;;;OAIG;IACK,gBAAgB;QACtB,MAAM,aAAa,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAA;QAE1C,uEAAuE;QACvE,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO,IAAI,CAAA;QACb,CAAC;QAED,kDAAkD;QAClD,IAAI,aAAa,CAAC,OAAO,KAAK,KAAK,EAAE,CAAC;YACpC,OAAO,KAAK,CAAA;QACd,CAAC;QAED,4CAA4C;QAC5C,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;YAC3B,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CAAA;QAC1B,CAAC;QAED,uCAAuC;QACvC,OAAO,aAAa,CAAC,OAAO,CAAA;IAC9B,CAAC;IAED;;;;;OAKG;IACK,KAAK,CAAC,WAAW,CAAC,OAAuB,EAAE,KAAmB;QACpE,MAAM,MAAM,GAAG,IAAI,CAAC,oBAAoB,EAAE,CAAA;QAC1C,MAAM,QAAQ,GAAwB;YACpC,MAAM;YACN,mBAAmB,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC,WAAW,EAAE;SACxE,CAAA;QACD,MAAM,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;IAC5B,CAAC;IAED;;;;;OAKG;IACK,KAAK,CAAC,iBAAiB,CAAC,OAAuB,EAAE,KAAmB;QAC1E,IAAI,CAAC;YACH,kBAAkB;YAClB,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAA;YAE7C,qBAAqB;YACrB,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;gBACvB,MAAM,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;oBAC3B,KAAK,EAAE,oGAAoG;iBAC5G,CAAC,CAAA;gBACF,OAAM;YACR,CAAC;YAED,gDAAgD;YAChD,IAAI,cAAgC,CAAA;YACpC,IAAI,IAAI,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;gBAChC,IAAI,CAAC;oBACH,cAAc,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;gBAC7E,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,KAAK,YAAY,CAAC,CAAC,QAAQ,EAAE,CAAC;wBAChC,MAAM,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;4BAC3B,KAAK,EAAE,6BAA6B;4BACpC,OAAO,EAAE,KAAK,CAAC,MAAM;yBACtB,CAAC,CAAA;wBACF,OAAM;oBACR,CAAC;oBACD,MAAM,KAAK,CAAA;gBACb,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,cAAc,GAAG,OAAO,CAAC,IAAwB,CAAA;YACnD,CAAC;YAED,8BAA8B;YAC9B,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,KAAK,IAAI,EAAE;gBACtD,OAAO,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,cAAc,EAAE,OAAO,CAAC,CAAA;YAC7D,CAAC,CAAC,CAAA;YAEF,6DAA6D;YAC7D,IAAI,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC;gBACnC,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;oBACd,mEAAmE;oBACnE,MAAM,cAAc,CAAC,OAAO,EAAE,KAAK,IAAI,EAAE;wBACvC,MAAM,IAAI,CAAC,wBAAwB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAA;oBACpD,CAAC,CAAC,CAAA;gBACJ,CAAC;qBAAM,CAAC;oBACN,MAAM,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;wBAC3B,KAAK,EACH,qIAAqI;qBACxI,CAAC,CAAA;gBACJ,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,gCAAgC;gBAChC,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;oBACd,8EAA8E;oBAC9E,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,IAAI,kBAAkB,CAAA;oBAEjE,IAAI,YAAY,CAAC,QAAQ,CAAC,kBAAkB,CAAC,IAAI,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;wBAC9E,MAAM,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;oBACnD,CAAC;yBAAM,IAAI,YAAY,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;wBAC/C,MAAM,IAAI,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;wBACzE,MAAM,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;oBAC3C,CAAC;yBAAM,IAAI,YAAY,CAAC,QAAQ,CAAC,0BAA0B,CAAC,EAAE,CAAC;wBAC7D,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAA;wBACrF,MAAM,KAAK,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;oBAC3D,CAAC;yBAAM,CAAC;wBACN,kBAAkB;wBAClB,MAAM,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;oBACnD,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,MAAM,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;gBAC1B,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,gBAAgB;YAChB,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAA;YAC7E,MAAM,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;gBAC3B,KAAK,EAAE,YAAY;aACpB,CAAC,CAAA;QACJ,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACK,iBAAiB,CAAC,KAAc;QACtC,OAAO,CACL,KAAK,KAAK,IAAI;YACd,KAAK,KAAK,SAAS;YACnB,OAAO,KAAK,KAAK,QAAQ;YACzB,MAAM,IAAI,KAAK;YACf,OAAO,KAAK,CAAC,IAAI,KAAK,UAAU;YAChC,QAAQ,IAAI,KAAK;YACjB,OAAO,KAAK,CAAC,MAAM,KAAK,UAAU;YAClC,OAAO,IAAI,KAAK;YAChB,OAAO,KAAK,CAAC,KAAK,KAAK,UAAU;YACjC,MAAM,CAAC,aAAa,IAAI,KAAK;YAC7B,OAAO,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,UAAU,CAClD,CAAA;IACH,CAAC;IAED;;;;;OAKG;IACK,KAAK,CAAC,wBAAwB,CAAC,KAAmB,EAAE,SAAoC;QAC9F,IAAI,CAAC;YACH,MAAM,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,CAAA;YAC3B,qBAAqB;YACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;gBACpC,8BAA8B;gBAC9B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;oBAC3B,MAAK;gBACP,CAAC;gBAED,mBAAmB;gBACnB,MAAM,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YAC7B,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,sCAAsC;YACtC,IAAI,KAAK,CAAC,GAAG,IAAI,KAAK,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;gBACvC,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAA;gBAC7E,MAAM,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC;oBACnB,KAAK,EAAE,OAAO;oBACd,IAAI,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE;iBAC9B,CAAC,CAAA;YACJ,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,mCAAmC,CAAC,CAAA;gBAC/D,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAA;YACtD,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,2BAA2B;YAC3B,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;gBACd,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,CAAA;YACnB,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACK,KAAK,CAAC,gBAAgB,CAAC,UAAqB,EAAE,OAAuB;QAC3E,IAAI,CAAC;YACH,yCAAyC;YACzC,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,CAAA;YAE7C,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,EAAE,kCAAkC,CAAC,CAAA;YAEtF,+GAA+G;YAC/G,MAAM,cAAc,CAAC,OAAO,EAAE,KAAK,IAAI,EAAE;gBACvC,OAAO,MAAM,IAAI,CAAC,iBAAkB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAA;YAC3D,CAAC,CAAC,CAAA;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EAAE,yBAAyB,CAAC,CAAA;YAC/G,IAAI,CAAC;gBACH,UAAU,CAAC,KAAK,CAAC,IAAI,EAAE,uBAAuB,CAAC,CAAA;YACjD,CAAC;YAAC,OAAO,UAAU,EAAE,CAAC;gBACpB,OAAO,CAAC,GAAG,CAAC,KAAK,CACf,EAAE,UAAU,EAAE,UAAU,YAAY,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,EACrF,yBAAyB,CAC1B,CAAA;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACK,eAAe,CAAC,OAAuB;QAC7C,4CAA4C;QAC5C,MAAM,SAAS,GACZ,OAAO,CAAC,OAAO,CAAC,6CAA6C,CAAY;YACxE,OAAO,CAAC,IAAgC,EAAE,SAAoB;YAChE,EAAE,CAAA;QAEJ,4CAA4C;QAC5C,MAAM,SAAS,GAAI,OAAO,CAAC,OAAO,CAAC,6CAA6C,CAAY,IAAI,UAAU,EAAE,CAAA;QAE5G,8BAA8B;QAC9B,MAAM,iBAAiB,GAAG,OAAO,CAAC,OAAO,CAAC,mBAAmB,CAAuB,CAAA;QAEpF,oEAAoE;QACpE,MAAM,eAAe,GAA2B,EAAE,CAAA;QAClD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3D,MAAM,QAAQ,GAAG,GAAG,CAAC,WAAW,EAAE,CAAA;YAClC,MAAM,WAAW,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;YAE3G,IAAI,WAAW,EAAE,CAAC;gBAChB,+BAA+B;gBAC/B,IAAI,QAAQ,KAAK,eAAe,EAAE,CAAC;oBACjC,eAAe,CAAC,GAAG,CAAC,GAAG,WAAW,CAAA;gBACpC,CAAC;gBACD,2BAA2B;qBACtB,IAAI,QAAQ,CAAC,UAAU,CAAC,0CAA0C,CAAC,EAAE,CAAC;oBACzE,eAAe,CAAC,GAAG,CAAC,GAAG,WAAW,CAAA;gBACpC,CAAC;YACH,CAAC;QACH,CAAC;QAED,kDAAkD;QAClD,MAAM,mBAAmB,GAAG,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAuB,CAAA;QAExF,OAAO;YACL,SAAS;YACT,OAAO,EAAE,eAAe;YACxB,mBAAmB;YACnB,SAAS;YACT,iBAAiB;YACjB,GAAG,EAAE,OAAO,CAAC,GAAG;SACjB,CAAA;IACH,CAAC;CACF"}

package/dist/src/runtime/client.d.ts

@@ -0,0 +1,154 @@
+import type { RuntimeClientConfig, GenerateWsConnectionParams, GeneratePresignedUrlParams, GenerateWsConnectionOAuthParams, WebSocketConnection } from './types.js';
+/**
+ * Client for generating WebSocket authentication for AgentCore Runtime.
+ *
+ * This client provides authentication credentials for WebSocket connections
+ * to AgentCore Runtime endpoints, allowing applications to establish
+ * bidirectional streaming connections with agent runtimes.
+ *
+ * The client is stateless and does not manage session lifecycle. Each method
+ * call is independent and takes the runtime ARN as a parameter.
+ *
+ * @example
+ * ```typescript
+ * const client = new RuntimeClient({ region: 'us-west-2' })
+ *
+ * // Generate WebSocket connection with SigV4 headers
+ * const { url, headers } = await client.generateWsConnection({
+ *   runtimeArn: 'arn:aws:bedrock-agentcore:us-west-2:123456789012:runtime/my-runtime',
+ *   endpointName: 'DEFAULT'
+ * })
+ *
+ * // Generate presigned WebSocket URL
+ * const presignedUrl = await client.generatePresignedUrl({
+ *   runtimeArn: 'arn:aws:bedrock-agentcore:us-west-2:123456789012:runtime/my-runtime',
+ *   expires: 300
+ * })
+ * ```
+ */
+export declare class RuntimeClient {
+    readonly region: string;
+    private readonly credentialsProvider;
+    /**
+     * Creates a new RuntimeClient instance.
+     *
+     * @param config - Configuration options for the client
+     * @throws Error if region is not provided via config or AWS_REGION environment variable
+     */
+    constructor(config?: RuntimeClientConfig);
+    /**
+     * Parses runtime ARN and extracts components.
+     *
+     * @param runtimeArn - Full runtime ARN
+     * @returns Parsed ARN components
+     * @throws Error if ARN format is invalid
+     *
+     * @internal
+     */
+    private _parseRuntimeArn;
+    /**
+     * Builds WebSocket URL with query parameters.
+     *
+     * @param runtimeArn - Full runtime ARN
+     * @param endpointName - Optional endpoint name for qualifier param
+     * @param customHeaders - Optional custom query parameters
+     * @returns WebSocket URL with query parameters
+     *
+     * @internal
+     */
+    private _buildWebSocketUrl;
+    /**
+     * Generates WebSocket URL and SigV4 signed headers for runtime connection.
+     *
+     * This method creates authentication credentials for establishing a WebSocket
+     * connection to an AgentCore Runtime. The returned headers include AWS SigV4
+     * signature for authentication.
+     *
+     * @param params - Parameters for generating the connection
+     * @returns WebSocket URL and authentication headers
+     *
+     * @throws Error if runtime ARN format is invalid
+     * @throws Error if AWS credentials are not available
+     *
+     * @example
+     * ```typescript
+     * const client = new RuntimeClient({ region: 'us-west-2' })
+     *
+     * // With auto-generated session ID
+     * const { url, headers } = await client.generateWsConnection({
+     *   runtimeArn: 'arn:aws:bedrock-agentcore:us-west-2:123:runtime/my-runtime'
+     * })
+     *
+     * // With custom session ID and endpoint
+     * const connection = await client.generateWsConnection({
+     *   runtimeArn: 'arn:aws:bedrock-agentcore:us-west-2:123:runtime/my-runtime',
+     *   sessionId: 'my-session-123',
+     *   endpointName: 'DEFAULT'
+     * })
+     * ```
+     */
+    generateWsConnection(params: GenerateWsConnectionParams): Promise<WebSocketConnection>;
+    /**
+     * Generates a presigned WebSocket URL for runtime connection.
+     *
+     * Presigned URLs include authentication in query parameters, allowing
+     * frontend clients to connect without managing AWS credentials.
+     *
+     * @param params - Parameters for generating the presigned URL
+     * @returns Presigned WebSocket URL with authentication in query parameters
+     *
+     * @throws Error if expires exceeds maximum (300 seconds)
+     * @throws Error if runtime ARN format is invalid
+     * @throws Error if AWS credentials are not available
+     *
+     * @example
+     * ```typescript
+     * const client = new RuntimeClient({ region: 'us-west-2' })
+     *
+     * // Basic presigned URL
+     * const url = await client.generatePresignedUrl({
+     *   runtimeArn: 'arn:aws:bedrock-agentcore:us-west-2:123:runtime/my-runtime'
+     * })
+     *
+     * // With custom parameters
+     * const url = await client.generatePresignedUrl({
+     *   runtimeArn: 'arn:aws:bedrock-agentcore:us-west-2:123:runtime/my-runtime',
+     *   sessionId: 'my-session-123',
+     *   endpointName: 'DEFAULT',
+     *   customHeaders: { 'custom-param': 'value' },
+     *   expires: 300
+     * })
+     * ```
+     */
+    generatePresignedUrl(params: GeneratePresignedUrlParams): Promise<string>;
+    /**
+     * Generates WebSocket URL and OAuth headers for runtime connection.
+     *
+     * This method uses OAuth bearer token authentication instead of AWS SigV4.
+     * Suitable for scenarios where OAuth tokens are used for authentication.
+     * Does NOT require AWS credentials.
+     *
+     * @param params - Parameters for generating the connection
+     * @returns WebSocket URL and OAuth authentication headers
+     *
+     * @throws Error if bearer token is empty
+     * @throws Error if runtime ARN format is invalid
+     *
+     * @example
+     * ```typescript
+     * const client = new RuntimeClient({ region: 'us-west-2' })
+     *
+     * // With OAuth bearer token
+     * const { url, headers } = await client.generateWsConnectionOAuth({
+     *   runtimeArn: 'arn:aws:bedrock-agentcore:us-west-2:123:runtime/my-runtime',
+     *   bearerToken: 'eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9...',
+     *   endpointName: 'DEFAULT'
+     * })
+     *
+     * // Use with WebSocket client
+     * const ws = new WebSocket(url, { headers })
+     * ```
+     */
+    generateWsConnectionOAuth(params: GenerateWsConnectionOAuthParams): Promise<WebSocketConnection>;
+}
+//# sourceMappingURL=client.d.ts.map

package/dist/src/runtime/client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../../src/runtime/client.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,EACV,mBAAmB,EACnB,0BAA0B,EAC1B,0BAA0B,EAC1B,+BAA+B,EAC/B,mBAAmB,EAEpB,MAAM,YAAY,CAAA;AAGnB;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AACH,qBAAa,aAAa;IACxB,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IACvB,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAA+B;IAEnE;;;;;OAKG;gBACS,MAAM,GAAE,mBAAwB;IAS5C;;;;;;;;OAQG;IACH,OAAO,CAAC,gBAAgB;IAiBxB;;;;;;;;;OASG;IACH,OAAO,CAAC,kBAAkB;IAyB1B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA6BG;IACG,oBAAoB,CAAC,MAAM,EAAE,0BAA0B,GAAG,OAAO,CAAC,mBAAmB,CAAC;IA8C5F;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA+BG;IACG,oBAAoB,CAAC,MAAM,EAAE,0BAA0B,GAAG,OAAO,CAAC,MAAM,CAAC;IAuE/E;;;;;;;;;;;;;;;;;;;;;;;;;;;OA2BG;IACG,yBAAyB,CAAC,MAAM,EAAE,+BAA+B,GAAG,OAAO,CAAC,mBAAmB,CAAC;CAuCvG"}