bedrock-agentcore 0.1.1 → 0.2.0

package/dist/src/identity/types.js

@@ -0,0 +1,5 @@
+/**
+ * Type definitions for AgentCore Identity SDK
+ */
+export {};
+//# sourceMappingURL=types.js.map

package/dist/src/identity/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/identity/types.ts"],"names":[],"mappings":"AAAA;;GAEG"}

package/dist/src/identity/wrappers.d.ts

@@ -0,0 +1,54 @@
+/**
+ * Higher-order functions for wrapping tools with automatic credential injection
+ */
+import type { OAuth2WrapperConfig, ApiKeyWrapperConfig } from './types.js';
+/**
+ * Helper type to extract all parameters except the last one (the token/apiKey)
+ */
+type InitParams<T extends unknown[]> = T extends [...infer Init, unknown] ? Init : never;
+/**
+ * Wraps an async function to automatically inject OAuth2 access token.
+ * The token is injected as the last parameter of the wrapped function.
+ *
+ * @param config - OAuth2 configuration
+ * @returns Function wrapper that injects token as last parameter
+ *
+ * @example
+ * ```typescript
+ * const myTool = withAccessToken({
+ *   workloadIdentityToken: token,
+ *   providerName: 'github',
+ *   scopes: ['repo'],
+ *   authFlow: 'M2M'
+ * })(async (input: string, token: string) => {
+ *   // Use token to call GitHub API
+ *   return { result: input };
+ * });
+ *
+ * await myTool('hello'); // token injected automatically
+ * ```
+ */
+export declare function withAccessToken(config: OAuth2WrapperConfig): <TParams extends [...unknown[], string], TReturn>(fn: (...args: TParams) => Promise<TReturn>) => ((...args: InitParams<TParams>) => Promise<TReturn>);
+/**
+ * Wraps an async function to automatically inject API key.
+ * The API key is injected as the last parameter of the wrapped function.
+ *
+ * @param config - API key configuration
+ * @returns Function wrapper that injects API key as last parameter
+ *
+ * @example
+ * ```typescript
+ * const myTool = withApiKey({
+ *   workloadIdentityToken: token,
+ *   providerName: 'openai'
+ * })(async (input: string, apiKey: string) => {
+ *   // Use API key to call OpenAI API
+ *   return { result: input };
+ * });
+ *
+ * await myTool('hello'); // apiKey injected automatically
+ * ```
+ */
+export declare function withApiKey(config: ApiKeyWrapperConfig): <TParams extends [...unknown[], string], TReturn>(fn: (...args: TParams) => Promise<TReturn>) => ((...args: InitParams<TParams>) => Promise<TReturn>);
+export {};
+//# sourceMappingURL=wrappers.d.ts.map

package/dist/src/identity/wrappers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"wrappers.d.ts","sourceRoot":"","sources":["../../../src/identity/wrappers.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,OAAO,KAAK,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAA;AAG1E;;GAEG;AACH,KAAK,UAAU,CAAC,CAAC,SAAS,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,MAAM,IAAI,EAAE,OAAO,CAAC,GAAG,IAAI,GAAG,KAAK,CAAA;AAExF;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,wBAAgB,eAAe,CAAC,MAAM,EAAE,mBAAmB,IAGjD,OAAO,SAAS,CAAC,GAAG,OAAO,EAAE,EAAE,MAAM,CAAC,EAAE,OAAO,EACrD,IAAI,CAAC,GAAG,IAAI,EAAE,OAAO,KAAK,OAAO,CAAC,OAAO,CAAC,KACzC,CAAC,CAAC,GAAG,IAAI,EAAE,UAAU,CAAC,OAAO,CAAC,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CA6BxD;AAED;;;;;;;;;;;;;;;;;;;GAmBG;AACH,wBAAgB,UAAU,CAAC,MAAM,EAAE,mBAAmB,IAG5C,OAAO,SAAS,CAAC,GAAG,OAAO,EAAE,EAAE,MAAM,CAAC,EAAE,OAAO,EACrD,IAAI,CAAC,GAAG,IAAI,EAAE,OAAO,KAAK,OAAO,CAAC,OAAO,CAAC,KACzC,CAAC,CAAC,GAAG,IAAI,EAAE,UAAU,CAAC,OAAO,CAAC,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAsBxD"}

package/dist/src/identity/wrappers.js

@@ -0,0 +1,95 @@
+/**
+ * Higher-order functions for wrapping tools with automatic credential injection
+ */
+import { IdentityClient } from './client.js';
+import { getContext } from '../runtime/context.js';
+/**
+ * Wraps an async function to automatically inject OAuth2 access token.
+ * The token is injected as the last parameter of the wrapped function.
+ *
+ * @param config - OAuth2 configuration
+ * @returns Function wrapper that injects token as last parameter
+ *
+ * @example
+ * ```typescript
+ * const myTool = withAccessToken({
+ *   workloadIdentityToken: token,
+ *   providerName: 'github',
+ *   scopes: ['repo'],
+ *   authFlow: 'M2M'
+ * })(async (input: string, token: string) => {
+ *   // Use token to call GitHub API
+ *   return { result: input };
+ * });
+ *
+ * await myTool('hello'); // token injected automatically
+ * ```
+ */
+export function withAccessToken(config) {
+    const client = new IdentityClient();
+    return (fn) => {
+        return async (...args) => {
+            // Get context and fallback to it if config doesn't provide token
+            const context = getContext();
+            const workloadToken = config.workloadIdentityToken ?? context?.workloadAccessToken;
+            if (!workloadToken) {
+                throw new Error('workloadIdentityToken not provided and no context available. ' +
+                    'Either pass workloadIdentityToken in config or call from within a request handler.');
+            }
+            const token = await client.getOAuth2Token({
+                providerName: config.providerName,
+                scopes: config.scopes,
+                authFlow: config.authFlow,
+                workloadIdentityToken: workloadToken,
+                onAuthUrl: config.onAuthUrl,
+                forceAuthentication: config.forceAuthentication,
+                callbackUrl: config.callbackUrl ?? context?.oauth2CallbackUrl,
+                customState: config.customState,
+                customParameters: config.customParameters,
+            });
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            return fn(...[...args, token]);
+        };
+    };
+}
+/**
+ * Wraps an async function to automatically inject API key.
+ * The API key is injected as the last parameter of the wrapped function.
+ *
+ * @param config - API key configuration
+ * @returns Function wrapper that injects API key as last parameter
+ *
+ * @example
+ * ```typescript
+ * const myTool = withApiKey({
+ *   workloadIdentityToken: token,
+ *   providerName: 'openai'
+ * })(async (input: string, apiKey: string) => {
+ *   // Use API key to call OpenAI API
+ *   return { result: input };
+ * });
+ *
+ * await myTool('hello'); // apiKey injected automatically
+ * ```
+ */
+export function withApiKey(config) {
+    const client = new IdentityClient();
+    return (fn) => {
+        return async (...args) => {
+            // Get context and fallback to it if config doesn't provide token
+            const context = getContext();
+            const workloadToken = config.workloadIdentityToken ?? context?.workloadAccessToken;
+            if (!workloadToken) {
+                throw new Error('workloadIdentityToken not provided and no context available. ' +
+                    'Either pass workloadIdentityToken in config or call from within a request handler.');
+            }
+            const apiKey = await client.getApiKey({
+                providerName: config.providerName,
+                workloadIdentityToken: workloadToken,
+            });
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            return fn(...[...args, apiKey]);
+        };
+    };
+}
+//# sourceMappingURL=wrappers.js.map

package/dist/src/identity/wrappers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"wrappers.js","sourceRoot":"","sources":["../../../src/identity/wrappers.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAA;AAE5C,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAA;AAOlD;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,UAAU,eAAe,CAAC,MAA2B;IACzD,MAAM,MAAM,GAAG,IAAI,cAAc,EAAE,CAAA;IAEnC,OAAO,CACL,EAA0C,EACY,EAAE;QACxD,OAAO,KAAK,EAAE,GAAG,IAAyB,EAAoB,EAAE;YAC9D,iEAAiE;YACjE,MAAM,OAAO,GAAG,UAAU,EAAE,CAAA;YAC5B,MAAM,aAAa,GAAG,MAAM,CAAC,qBAAqB,IAAI,OAAO,EAAE,mBAAmB,CAAA;YAElF,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,MAAM,IAAI,KAAK,CACb,+DAA+D;oBAC7D,oFAAoF,CACvF,CAAA;YACH,CAAC;YAED,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC;gBACxC,YAAY,EAAE,MAAM,CAAC,YAAY;gBACjC,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,qBAAqB,EAAE,aAAa;gBACpC,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,mBAAmB,EAAE,MAAM,CAAC,mBAAmB;gBAC/C,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,OAAO,EAAE,iBAAiB;gBAC7D,WAAW,EAAE,MAAM,CAAC,WAAW;gBAC/B,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;aAC1C,CAAC,CAAA;YAEF,8DAA8D;YAC9D,OAAO,EAAE,CAAC,GAAI,CAAC,GAAG,IAAI,EAAE,KAAK,CAAS,CAAC,CAAA;QACzC,CAAC,CAAA;IACH,CAAC,CAAA;AACH,CAAC;AAED;;;;;;;;;;;;;;;;;;;GAmBG;AACH,MAAM,UAAU,UAAU,CAAC,MAA2B;IACpD,MAAM,MAAM,GAAG,IAAI,cAAc,EAAE,CAAA;IAEnC,OAAO,CACL,EAA0C,EACY,EAAE;QACxD,OAAO,KAAK,EAAE,GAAG,IAAyB,EAAoB,EAAE;YAC9D,iEAAiE;YACjE,MAAM,OAAO,GAAG,UAAU,EAAE,CAAA;YAC5B,MAAM,aAAa,GAAG,MAAM,CAAC,qBAAqB,IAAI,OAAO,EAAE,mBAAmB,CAAA;YAElF,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,MAAM,IAAI,KAAK,CACb,+DAA+D;oBAC7D,oFAAoF,CACvF,CAAA;YACH,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC;gBACpC,YAAY,EAAE,MAAM,CAAC,YAAY;gBACjC,qBAAqB,EAAE,aAAa;aACrC,CAAC,CAAA;YAEF,8DAA8D;YAC9D,OAAO,EAAE,CAAC,GAAI,CAAC,GAAG,IAAI,EAAE,MAAM,CAAS,CAAC,CAAA;QAC1C,CAAC,CAAA;IACH,CAAC,CAAA;AACH,CAAC"}

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

@@ -0,0 +1,144 @@
+import { z } from 'zod';
+import type { BedrockAgentCoreAppParams, AsyncTaskStatus, HealthStatus } from './types.js';
+/**
+ * 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 declare class BedrockAgentCoreApp<TSchema extends z.ZodSchema = z.ZodSchema<unknown>> {
+    private readonly _app;
+    private readonly _config;
+    private readonly _handler;
+    private _websocketHandler;
+    private readonly _activeTasksMap;
+    private _taskCounter;
+    private _pingHandler;
+    private _forcedPingStatus?;
+    private _lastStatusUpdateTime;
+    private _lastKnownStatus?;
+    /**
+     * Creates a new BedrockAgentCoreApp instance.
+     *
+     * @param params - Configuration including handler and optional settings
+     */
+    constructor(params: BedrockAgentCoreAppParams<TSchema>);
+    /**
+     * Starts the Fastify server on port 8080.
+     */
+    run(): void;
+    /**
+     * 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: string, metadata?: Record<string, unknown>): number;
+    /**
+     * Mark an async task as complete.
+     *
+     * @param taskId - Task ID from addAsyncTask
+     * @returns True if task was found and removed
+     */
+    completeAsyncTask(taskId: number): boolean;
+    /**
+     * Get current ping status based on priority system.
+     * Priority: Forced \> Custom Handler \> Automatic
+     *
+     * @returns Current health status
+     */
+    getCurrentPingStatus(): HealthStatus;
+    /**
+     * 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<T extends (...args: unknown[]) => Promise<unknown>>(fn: T): T;
+    /**
+     * Get information about currently running async tasks.
+     *
+     * @returns Task status with count and details
+     */
+    getAsyncTaskInfo(): AsyncTaskStatus;
+    /**
+     * Sets up HTTP routes for the server.
+     */
+    private _setupRoutes;
+    /**
+     * Register custom content type parsers using Fastify's native addContentTypeParser.
+     */
+    private _setupContentTypeParsers;
+    /**
+     * Registers Fastify plugins.
+     */
+    private _registerPlugins;
+    /**
+     * Gets the logger configuration based on BedrockAgentCoreAppConfig.
+     *
+     * @returns Fastify logger configuration
+     */
+    private _getLoggerConfig;
+    /**
+     * Handles health check requests.
+     *
+     * @param request - Fastify request object
+     * @param reply - Fastify reply object
+     */
+    private _handlePing;
+    /**
+     * Handles agent invocation requests.
+     *
+     * @param request - Fastify request object
+     * @param reply - Fastify reply object
+     */
+    private _handleInvocation;
+    /**
+     * Checks if a value is an async generator.
+     *
+     * @param value - Value to check
+     * @returns True if the value is an async generator
+     */
+    private _isAsyncGenerator;
+    /**
+     * 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
+     */
+    private _handleStreamingResponse;
+    /**
+     * Handles WebSocket connections at /ws endpoint.
+     *
+     * @param connection - Fastify WebSocket connection
+     * @param request - Fastify request object
+     */
+    private _handleWebSocket;
+    /**
+     * Extracts request context from the incoming request.
+     *
+     * @param request - Fastify request object
+     * @returns Request context with sessionId, headers, and request-scoped logger
+     */
+    private _extractContext;
+}
+//# sourceMappingURL=app.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"app.d.ts","sourceRoot":"","sources":["../../../src/runtime/app.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAYvB,OAAO,KAAK,EACV,yBAAyB,EAOzB,eAAe,EACf,YAAY,EACb,MAAM,YAAY,CAAA;AAOnB;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,qBAAa,mBAAmB,CAAC,OAAO,SAAS,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC;IACjF,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAiB;IACtC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAA2B;IACnD,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAA2E;IACpG,OAAO,CAAC,iBAAiB,CAA8B;IACvD,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAwC;IACxE,OAAO,CAAC,YAAY,CAAY;IAChC,OAAO,CAAC,YAAY,CAA0D;IAC9E,OAAO,CAAC,iBAAiB,CAAC,CAAc;IACxC,OAAO,CAAC,qBAAqB,CAAqB;IAClD,OAAO,CAAC,gBAAgB,CAAC,CAAc;IAEvC;;;;OAIG;gBACS,MAAM,EAAE,yBAAyB,CAAC,OAAO,CAAC;IA0BtD;;OAEG;IACH,GAAG,IAAI,IAAI;IAmBX;;;;;;OAMG;IACI,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM;IAa7E;;;;;OAKG;IACI,iBAAiB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO;IAIjD;;;;;OAKG;IACI,oBAAoB,IAAI,YAAY;IA6B3C;;;;;;;OAOG;IACI,SAAS,CAAC,CAAC,SAAS,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC;IAoB9E;;;;OAIG;IACI,gBAAgB,IAAI,eAAe;IAa1C;;OAEG;IACH,OAAO,CAAC,YAAY;IAapB;;OAEG;IACH,OAAO,CAAC,wBAAwB;IAmBhC;;OAEG;YACW,gBAAgB;IAO9B;;;;OAIG;IACH,OAAO,CAAC,gBAAgB;IAsBxB;;;;;OAKG;YACW,WAAW;IASzB;;;;;OAKG;YACW,iBAAiB;IAiF/B;;;;;OAKG;IACH,OAAO,CAAC,iBAAiB;IAgBzB;;;;;OAKG;YACW,wBAAwB;IAiCtC;;;;;OAKG;YACW,gBAAgB;IAwB9B;;;;;OAKG;IACH,OAAO,CAAC,eAAe;CA2CxB"}