@aws-blocks/bb-agent 0.3.4 → 0.3.5

package/DESIGN.md

@@ -42,6 +42,7 @@ The CDK class mirrors the runtime's BB creation:
 - **DistributedTable:** `${id}-messages` — conversation history (only when `inferenceOnly: false`)
 - **Realtime:** `${id}-rt` — streaming namespace `chunks`
 - **AsyncJob:** `${id}-job` — job payload: `{ message, conversationId?, channelId }`
+  - Event source uses `batchSize: 1` / `maxBatchingWindowSeconds: 0`: the caller is blocked on the job starting, so the Agent opts out of AsyncJob's batching defaults (10 / 5s) rather than add up to 5s of latency to an interactive turn.
 
 > **Note:** Internal Building Blocks are created on the parent scope (not `this`) to ensure correct nested-scope resolution on AWS.
 

package/dist/agent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,KAAK,EAA6C,MAAM,kBAAkB,CAAC;AACpF,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAIpD,OAAO,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAExD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAGzD,OAAO,KAAK,EAAyB,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAIlF,OAAO,KAAK,EAAE,WAAW,EAAoB,iBAAiB,EAAE,aAAa,EAAE,OAAO,EAAE,YAAY,EAAyC,WAAW,EAAa,iBAAiB,EAAE,kBAAkB,EAA6B,MAAM,YAAY,CAAC;AAmF1P;;;;;;;;;GASG;AACH,qBAAa,SAAS,CAAC,QAAQ,GAAG,kBAAkB,CAAE,SAAQ,KAAK;IAClE,4CAA4C;IAC5C,OAAO,CAAC,MAAM,CAAwB;IACtC,yFAAyF;IACzF,OAAO,CAAC,OAAO,CAAmC;IAClD,mCAAmC;IACnC,OAAO,CAAC,aAAa,CAAC,CAA8G;IACpI,6BAA6B;IAC7B,OAAO,CAAC,QAAQ,CAAC,CAA4G;IAC7H,oDAAoD;IACpD,OAAO,CAAC,EAAE,CAAgC;IAC1C,mFAAmF;IACnF,OAAO,CAAC,GAAG,CAA6C;IACxD,mCAAmC;IACnC,OAAO,CAAC,WAAW,CAA0C;IAC7D,wDAAwD;IACxD,OAAO,CAAC,eAAe,CAAkB;IACzC,+CAA+C;IAC/C,OAAO,CAAC,aAAa,CAAa;IAClC,2FAA2F;IAC3F,SAAS,CAAC,GAAG,EAAE,WAAW,CAAC;IAE3B;;;;;;OAMG;gBACS,KAAK,EAAE,WAAW,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,CAAC,QAAQ,CAAC,EAAE,WAAW,EAAE,WAAW,GAAG,WAAW,EAAE,GAAG,SAAS,EAAE,qBAAqB,EAAE,CAAC,MAAM,EAAE,UAAU,KAAK,eAAe;IAgE/L;;;;;;;;OAQG;YACW,QAAQ;YA2GR,kBAAkB;IAgGhC;;;;;;;;OAQG;IACG,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,aAAa,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC,iBAAiB,CAAC;IA+B5F;;;;OAIG;IACG,MAAM,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,CAAC,iBAAiB,CAAC,EAAE,OAAO,CAAC,EAAE;QAAE,cAAc,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,QAAQ,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAoC/J;;;OAGG;IACH,OAAO,CAAC,cAAc;IAUtB,yEAAyE;IACnE,oBAAoB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAQ3D,kGAAkG;IAClG,UAAU,CAAC,SAAS,EAAE,MAAM;IAI5B;;;;;;;;OAQG;IACG,oBAAoB,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,GAAG,CAAA;KAAE,CAAC,CAAC;IAoB9G,yCAAyC;IACnC,iBAAiB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAShE;;;;;;;;;;;;OAYG;IACG,eAAe,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAyBnF,iDAAiD;IAC3C,kBAAkB,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAuBnE"}
+{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,KAAK,EAA6C,MAAM,kBAAkB,CAAC;AACpF,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAIpD,OAAO,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAExD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AAGzD,OAAO,KAAK,EAAyB,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAIlF,OAAO,KAAK,EAAE,WAAW,EAAoB,iBAAiB,EAAE,aAAa,EAAE,OAAO,EAAE,YAAY,EAAyC,WAAW,EAAa,iBAAiB,EAAE,kBAAkB,EAA6B,MAAM,YAAY,CAAC;AAoF1P;;;;;;;;;GASG;AACH,qBAAa,SAAS,CAAC,QAAQ,GAAG,kBAAkB,CAAE,SAAQ,KAAK;IAClE,4CAA4C;IAC5C,OAAO,CAAC,MAAM,CAAwB;IACtC,yFAAyF;IACzF,OAAO,CAAC,OAAO,CAAmC;IAClD,mCAAmC;IACnC,OAAO,CAAC,aAAa,CAAC,CAA8G;IACpI,6BAA6B;IAC7B,OAAO,CAAC,QAAQ,CAAC,CAA4G;IAC7H,oDAAoD;IACpD,OAAO,CAAC,EAAE,CAAgC;IAC1C,mFAAmF;IACnF,OAAO,CAAC,GAAG,CAA6C;IACxD,mCAAmC;IACnC,OAAO,CAAC,WAAW,CAA0C;IAC7D,wDAAwD;IACxD,OAAO,CAAC,eAAe,CAAkB;IACzC,+CAA+C;IAC/C,OAAO,CAAC,aAAa,CAAa;IAClC,2FAA2F;IAC3F,SAAS,CAAC,GAAG,EAAE,WAAW,CAAC;IAE3B;;;;;;OAMG;gBACS,KAAK,EAAE,WAAW,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,CAAC,QAAQ,CAAC,EAAE,WAAW,EAAE,WAAW,GAAG,WAAW,EAAE,GAAG,SAAS,EAAE,qBAAqB,EAAE,CAAC,MAAM,EAAE,UAAU,KAAK,eAAe;IAiE/L;;;;;;;;OAQG;YACW,QAAQ;YA2GR,kBAAkB;IAgGhC;;;;;;;;OAQG;IACG,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,aAAa,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC,iBAAiB,CAAC;IA+B5F;;;;OAIG;IACG,MAAM,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,CAAC,iBAAiB,CAAC,EAAE,OAAO,CAAC,EAAE;QAAE,cAAc,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,QAAQ,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAoC/J;;;OAGG;IACH,OAAO,CAAC,cAAc;IAUtB,yEAAyE;IACnE,oBAAoB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAQ3D,kGAAkG;IAClG,UAAU,CAAC,SAAS,EAAE,MAAM;IAI5B;;;;;;;;OAQG;IACG,oBAAoB,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,GAAG,CAAA;KAAE,CAAC,CAAC;IAoB9G,yCAAyC;IACnC,iBAAiB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAShE;;;;;;;;;;;;OAYG;IACG,eAAe,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAyBnF,iDAAiD;IAC3C,kBAAkB,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAuBnE"}

package/dist/agent.js

@@ -10,6 +10,7 @@ import { z } from 'zod';
 import { createStrandsModel, checkModelHealth } from './model-factory.js';
 import { messageSchema, conversationSchema, agentStreamChunkSchema } from './schemas.js';
 import { AgentErrors, blocksAgentError, InterruptError } from './errors.js';
+import { INTERACTIVE_JOB_EVENT_SOURCE } from './job-event-source.js';
 import { BB_NAME, BB_VERSION } from './version.js';
 import { ulid } from 'ulid';
 /** Payload submitted to the internal AsyncJob BB. */
@@ -147,6 +148,7 @@ export class AgentBase extends Scope {
         });
         this.job = new AsyncJob(this, 'job', {
             schema: jobPayloadSchema,
+            ...INTERACTIVE_JOB_EVENT_SOURCE,
             handler: async (payload) => {
                 try {
                     await this.runAgent(payload.message, payload.conversationId, payload.channelId, payload.userId, payload.interruptResponses, payload.context);

package/dist/index.cdk.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.cdk.d.ts","sourceRoot":"","sources":["../src/index.cdk.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,KAAK,EAAE,MAAM,sBAAsB,CAAC;AAC7C,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAQpD,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAO1D,qBAAa,KAAM,SAAQ,KAAK;IAC/B;;;;;;OAMG;gBACS,KAAK,EAAE,WAAW,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,GAAG;CAwCxD"}
+{"version":3,"file":"index.cdk.d.ts","sourceRoot":"","sources":["../src/index.cdk.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,KAAK,EAAE,MAAM,sBAAsB,CAAC;AAC7C,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AASpD,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAO1D,qBAAa,KAAM,SAAQ,KAAK;IAC/B;;;;;;OAMG;gBACS,KAAK,EAAE,WAAW,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,GAAG;CAyCxD"}

package/dist/index.cdk.js

@@ -7,6 +7,7 @@ import { Realtime } from '@aws-blocks/bb-realtime';
 import { AsyncJob } from '@aws-blocks/bb-async-job';
 import { FileBucket } from '@aws-blocks/bb-file-bucket';
 import { messageSchema, conversationSchema, agentStreamChunkSchema } from './schemas.js';
+import { INTERACTIVE_JOB_EVENT_SOURCE } from './job-event-source.js';
 import { z } from 'zod';
 export { AgentErrors } from './errors.js';
 export { BedrockModels, OllamaModels } from './models.js';
@@ -24,7 +25,7 @@ export class Agent extends Scope {
      */
     constructor(scope, id, config) {
         super(id, { parent: scope });
-        this.handler.addToRolePolicy(new PolicyStatement({
+        this.executionRole.addToPrincipalPolicy(new PolicyStatement({
             actions: ['bedrock:InvokeModel', 'bedrock:InvokeModelWithResponseStream', 'bedrock:GetFoundationModel', 'bedrock:ListFoundationModels', 'bedrock:GetInferenceProfile'],
             resources: [
                 'arn:aws:bedrock:*::foundation-model/*',
@@ -54,6 +55,7 @@ export class Agent extends Scope {
         });
         new AsyncJob(this, 'job', {
             schema: jobPayloadSchema,
+            ...INTERACTIVE_JOB_EVENT_SOURCE,
             handler: async () => { },
         });
     }

package/dist/index.cdk.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=index.cdk.test.d.ts.map

package/dist/index.cdk.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.cdk.test.d.ts","sourceRoot":"","sources":["../src/index.cdk.test.ts"],"names":[],"mappings":""}

package/dist/index.cdk.test.js

@@ -0,0 +1,66 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+/**
+ * CDK-side regression tests for the Agent's internal AsyncJob event source.
+ *
+ * `stream()` submits one job per interactive turn (and a second on HITL resume),
+ * so the caller is blocked on that job starting. AsyncJob's defaults
+ * (batchSize 10 / maxBatchingWindowSeconds 5) would add up to 5s of SQS
+ * batching delay to that human-blocking path, so the Agent opts out at both
+ * construction sites. These tests pin the opt-out to the synthesized template:
+ * if the defaults are ever inherited again, the latency regression fails here
+ * instead of surfacing as a slow agent in production.
+ *
+ * Must run under `--conditions=cdk`; otherwise the internal BBs resolve to
+ * their mock implementations and no CloudFormation resources are produced.
+ */
+import { test, afterEach } from 'node:test';
+import assert from 'node:assert';
+import * as cdk from 'aws-cdk-lib';
+import { Template } from 'aws-cdk-lib/assertions';
+import { Scope, DEFAULT_NODE_RUNTIME } from '@aws-blocks/core/cdk';
+import { Agent } from './index.cdk.js';
+class StubBlocksStack extends cdk.Stack {
+    handler;
+    executionRole;
+    id;
+    constructor(scope, id) {
+        super(scope, id);
+        this.id = id;
+        globalThis.CURRENT_BLOCKS_STACK = this;
+        this.executionRole = new cdk.aws_iam.Role(this, 'BlocksRole', {
+            assumedBy: new cdk.aws_iam.ServicePrincipal('lambda.amazonaws.com'),
+        });
+        this.handler = new cdk.aws_lambda.Function(this, 'StubHandler', {
+            runtime: DEFAULT_NODE_RUNTIME,
+            handler: 'index.handler',
+            code: cdk.aws_lambda.Code.fromInline('exports.handler = async () => {};'),
+            role: this.executionRole,
+        });
+    }
+}
+// synthAgent() installs its own stack as the ambient CURRENT_BLOCKS_STACK; clear it
+// afterwards so no test observes a stack left behind by the previous one (node
+// --test isolates files by default, so this is hygiene against future sharing).
+afterEach(() => {
+    delete globalThis.CURRENT_BLOCKS_STACK;
+});
+function synthAgent() {
+    const app = new cdk.App();
+    const stack = new StubBlocksStack(app, 'teststack');
+    const parent = new Scope('app');
+    new Agent(parent, 'agent', { inferenceOnly: true });
+    const mappings = Template.fromStack(stack).findResources('AWS::Lambda::EventSourceMapping');
+    const keys = Object.keys(mappings);
+    assert.strictEqual(keys.length, 1, 'exactly one event source mapping expected for the agent job');
+    return mappings[keys[0]].Properties;
+}
+test('CDK: the Agent job takes one message per invocation (no batching on the interactive path)', () => {
+    assert.strictEqual(synthAgent().BatchSize, 1);
+});
+test('CDK: the Agent job has no SQS batching window (no added latency for the caller)', () => {
+    assert.strictEqual(synthAgent().MaximumBatchingWindowInSeconds, 0);
+});
+test('CDK: the Agent job still reports partial batch failures', () => {
+    assert.deepStrictEqual(synthAgent().FunctionResponseTypes, ['ReportBatchItemFailures']);
+});

package/dist/job-event-source.d.ts

@@ -0,0 +1,19 @@
+/**
+ * SQS event source options for the Agent's internal AsyncJob, shared by the
+ * runtime (`agent.ts`) and CDK (`index.cdk.ts`) construction sites so an edit to
+ * one site's numbers cannot silently miss the other. Only the CDK site's values
+ * reach a real event source mapping today — the runtime entry points do not read
+ * these options — so this is a single source of truth for the intent, not a
+ * runtime/CDK sync guarantee.
+ *
+ * `stream()` submits one job per interactive turn (and a second on HITL
+ * resume) while the caller is blocked on that job starting, so AsyncJob's
+ * batching defaults would add up to 5s of delivery latency to a human-facing
+ * path. `batchSize: 1` also keeps one failing turn out of a shared batch,
+ * which matters because the handler is not idempotent.
+ */
+export declare const INTERACTIVE_JOB_EVENT_SOURCE: {
+    readonly batchSize: 1;
+    readonly maxBatchingWindowSeconds: 0;
+};
+//# sourceMappingURL=job-event-source.d.ts.map

package/dist/job-event-source.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"job-event-source.d.ts","sourceRoot":"","sources":["../src/job-event-source.ts"],"names":[],"mappings":"AAGA;;;;;;;;;;;;;GAaG;AACH,eAAO,MAAM,4BAA4B;;;CAG/B,CAAC"}

package/dist/job-event-source.js

@@ -0,0 +1,20 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+/**
+ * SQS event source options for the Agent's internal AsyncJob, shared by the
+ * runtime (`agent.ts`) and CDK (`index.cdk.ts`) construction sites so an edit to
+ * one site's numbers cannot silently miss the other. Only the CDK site's values
+ * reach a real event source mapping today — the runtime entry points do not read
+ * these options — so this is a single source of truth for the intent, not a
+ * runtime/CDK sync guarantee.
+ *
+ * `stream()` submits one job per interactive turn (and a second on HITL
+ * resume) while the caller is blocked on that job starting, so AsyncJob's
+ * batching defaults would add up to 5s of delivery latency to a human-facing
+ * path. `batchSize: 1` also keeps one failing turn out of a shared batch,
+ * which matters because the handler is not idempotent.
+ */
+export const INTERACTIVE_JOB_EVENT_SOURCE = {
+    batchSize: 1,
+    maxBatchingWindowSeconds: 0,
+};

package/dist/version.d.ts

@@ -1,3 +1,3 @@
 export declare const BB_NAME = "Agent";
-export declare const BB_VERSION = "0.3.4";
+export declare const BB_VERSION = "0.3.5";
 //# sourceMappingURL=version.d.ts.map

package/dist/version.js

@@ -1,3 +1,3 @@
 // Auto-generated by scripts/generate-version.mjs — do not edit manually
 export const BB_NAME = 'Agent';
-export const BB_VERSION = '0.3.4';
+export const BB_VERSION = '0.3.5';

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aws-blocks/bb-agent",
-  "version": "0.3.4",
+  "version": "0.3.5",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/aws-devtools-labs/aws-blocks.git",
@@ -40,15 +40,15 @@
   "scripts": {
     "prebuild": "node ../../scripts/generate-version.mjs Agent",
     "build": "tsc --build",
-    "test": "node --test dist/index.test.js"
+    "test": "node --test dist/index.test.js && node --conditions=cdk --test dist/index.cdk.test.js"
   },
   "dependencies": {
-    "@aws-blocks/bb-async-job": "^0.1.4",
-    "@aws-blocks/bb-distributed-table": "^0.1.5",
-    "@aws-blocks/bb-file-bucket": "^0.1.4",
-    "@aws-blocks/bb-logger": "^0.1.4",
-    "@aws-blocks/bb-realtime": "^0.1.4",
-    "@aws-blocks/core": "^0.2.0",
+    "@aws-blocks/bb-async-job": "^0.1.5",
+    "@aws-blocks/bb-distributed-table": "^0.1.6",
+    "@aws-blocks/bb-file-bucket": "^0.1.5",
+    "@aws-blocks/bb-logger": "^0.1.5",
+    "@aws-blocks/bb-realtime": "^0.1.5",
+    "@aws-blocks/core": "^0.3.0",
     "@aws-sdk/client-bedrock": "^3.700.0",
     "@opentelemetry/api": "^1.9.0",
     "@strands-agents/sdk": "~1.3.0",

package/src/agent.ts

@@ -17,6 +17,7 @@ import { createStrandsModel, checkModelHealth } from './model-factory.js';
 import { messageSchema, conversationSchema, agentStreamChunkSchema } from './schemas.js';
 import type { AgentConfig, AgentStreamChunk, AgentStreamResult, StreamOptions, Message, Conversation, TokenUsage, ConversationManagerConfig, ModelConfig, JSONValue, InterruptResponse, DefaultToolContext, AgentTool, ToolDefinition } from './types.js';
 import { AgentErrors, blocksAgentError, InterruptError } from './errors.js';
+import { INTERACTIVE_JOB_EVENT_SOURCE } from './job-event-source.js';
 import { BB_NAME, BB_VERSION } from './version.js';
 import { ulid } from 'ulid';
 
@@ -167,6 +168,7 @@ export class AgentBase<TContext = DefaultToolContext> extends Scope {
 
 		this.job = new AsyncJob(this, 'job', {
 			schema: jobPayloadSchema,
+			...INTERACTIVE_JOB_EVENT_SOURCE,
 			handler: async (payload) => {
 				try {
 					await this.runAgent(payload.message, payload.conversationId, payload.channelId, payload.userId, payload.interruptResponses, payload.context);

package/src/index.cdk.test.ts

@@ -0,0 +1,75 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+/**
+ * CDK-side regression tests for the Agent's internal AsyncJob event source.
+ *
+ * `stream()` submits one job per interactive turn (and a second on HITL resume),
+ * so the caller is blocked on that job starting. AsyncJob's defaults
+ * (batchSize 10 / maxBatchingWindowSeconds 5) would add up to 5s of SQS
+ * batching delay to that human-blocking path, so the Agent opts out at both
+ * construction sites. These tests pin the opt-out to the synthesized template:
+ * if the defaults are ever inherited again, the latency regression fails here
+ * instead of surfacing as a slow agent in production.
+ *
+ * Must run under `--conditions=cdk`; otherwise the internal BBs resolve to
+ * their mock implementations and no CloudFormation resources are produced.
+ */
+import { test, afterEach } from 'node:test';
+import assert from 'node:assert';
+import * as cdk from 'aws-cdk-lib';
+import type { Construct } from 'constructs';
+import { Template } from 'aws-cdk-lib/assertions';
+import { Scope, DEFAULT_NODE_RUNTIME } from '@aws-blocks/core/cdk';
+import { Agent } from './index.cdk.js';
+
+class StubBlocksStack extends cdk.Stack {
+	public readonly handler: cdk.aws_lambda.Function;
+	public readonly executionRole: cdk.aws_iam.IRole;
+	public readonly id: string;
+	constructor(scope: Construct, id: string) {
+		super(scope, id);
+		this.id = id;
+		(globalThis as any).CURRENT_BLOCKS_STACK = this;
+		this.executionRole = new cdk.aws_iam.Role(this, 'BlocksRole', {
+			assumedBy: new cdk.aws_iam.ServicePrincipal('lambda.amazonaws.com'),
+		});
+		this.handler = new cdk.aws_lambda.Function(this, 'StubHandler', {
+			runtime: DEFAULT_NODE_RUNTIME,
+			handler: 'index.handler',
+			code: cdk.aws_lambda.Code.fromInline('exports.handler = async () => {};'),
+			role: this.executionRole,
+		});
+	}
+}
+
+// synthAgent() installs its own stack as the ambient CURRENT_BLOCKS_STACK; clear it
+// afterwards so no test observes a stack left behind by the previous one (node
+// --test isolates files by default, so this is hygiene against future sharing).
+afterEach(() => {
+	delete (globalThis as any).CURRENT_BLOCKS_STACK;
+});
+
+function synthAgent(): any {
+	const app = new cdk.App();
+	const stack = new StubBlocksStack(app, 'teststack');
+	const parent = new Scope('app');
+	new Agent(parent, 'agent', { inferenceOnly: true });
+
+	const mappings = Template.fromStack(stack).findResources('AWS::Lambda::EventSourceMapping');
+	const keys = Object.keys(mappings);
+	assert.strictEqual(keys.length, 1, 'exactly one event source mapping expected for the agent job');
+	return mappings[keys[0]].Properties;
+}
+
+test('CDK: the Agent job takes one message per invocation (no batching on the interactive path)', () => {
+	assert.strictEqual(synthAgent().BatchSize, 1);
+});
+
+test('CDK: the Agent job has no SQS batching window (no added latency for the caller)', () => {
+	assert.strictEqual(synthAgent().MaximumBatchingWindowInSeconds, 0);
+});
+
+test('CDK: the Agent job still reports partial batch failures', () => {
+	assert.deepStrictEqual(synthAgent().FunctionResponseTypes, ['ReportBatchItemFailures']);
+});

package/src/index.cdk.ts

@@ -9,6 +9,7 @@ import { Realtime } from '@aws-blocks/bb-realtime';
 import { AsyncJob } from '@aws-blocks/bb-async-job';
 import { FileBucket } from '@aws-blocks/bb-file-bucket';
 import { messageSchema, conversationSchema, agentStreamChunkSchema } from './schemas.js';
+import { INTERACTIVE_JOB_EVENT_SOURCE } from './job-event-source.js';
 import { z } from 'zod';
 
 export { AgentErrors } from './errors.js';
@@ -30,7 +31,7 @@ export class Agent extends Scope {
 	constructor(scope: ScopeParent, id: string, config?: any) {
 		super(id, { parent: scope });
 
-		this.handler.addToRolePolicy(new PolicyStatement({
+		this.executionRole.addToPrincipalPolicy(new PolicyStatement({
 			actions: ['bedrock:InvokeModel', 'bedrock:InvokeModelWithResponseStream', 'bedrock:GetFoundationModel', 'bedrock:ListFoundationModels', 'bedrock:GetInferenceProfile'],
 			resources: [
 				'arn:aws:bedrock:*::foundation-model/*',
@@ -64,6 +65,7 @@ export class Agent extends Scope {
 
 		new AsyncJob(this, 'job', {
 			schema: jobPayloadSchema,
+			...INTERACTIVE_JOB_EVENT_SOURCE,
 			handler: async () => {},
 		});
 	}

package/src/job-event-source.ts

@@ -0,0 +1,21 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+/**
+ * SQS event source options for the Agent's internal AsyncJob, shared by the
+ * runtime (`agent.ts`) and CDK (`index.cdk.ts`) construction sites so an edit to
+ * one site's numbers cannot silently miss the other. Only the CDK site's values
+ * reach a real event source mapping today — the runtime entry points do not read
+ * these options — so this is a single source of truth for the intent, not a
+ * runtime/CDK sync guarantee.
+ *
+ * `stream()` submits one job per interactive turn (and a second on HITL
+ * resume) while the caller is blocked on that job starting, so AsyncJob's
+ * batching defaults would add up to 5s of delivery latency to a human-facing
+ * path. `batchSize: 1` also keeps one failing turn out of a shared batch,
+ * which matters because the handler is not idempotent.
+ */
+export const INTERACTIVE_JOB_EVENT_SOURCE = {
+	batchSize: 1,
+	maxBatchingWindowSeconds: 0,
+} as const;

package/src/version.ts

@@ -1,3 +1,3 @@
 // Auto-generated by scripts/generate-version.mjs — do not edit manually
 export const BB_NAME = 'Agent';
-export const BB_VERSION = '0.3.4';
+export const BB_VERSION = '0.3.5';