@aws/nx-plugin 0.83.1 → 0.84.1

package/src/ts/strands-agent/files/app/agent-core-trpc-client.ts.template

@@ -1,3 +1,4 @@
+import { randomUUID } from 'node:crypto';
 import { fromNodeProviderChain } from '@aws-sdk/credential-providers';
 import { createTRPCClient, createWSClient, WebSocketLinkOptions, wsLink } from '@trpc/client';
 import { AwsClient } from 'aws4fetch';
@@ -9,6 +10,11 @@ export interface WebSocketTrpcClientOptions {
    * URL of the websocket server, eg http://localhost:8080/ws
    */
   url: string;
+  /**
+   * Session ID to include in the WebSocket handshake headers.
+   * Defaults to a random UUID if not provided.
+   */
+  sessionId?: string;
   /**
    * Factory for creating headers to include in the WebSocket handshake
    */
@@ -24,12 +30,16 @@ export class WebSocketTrpcClient {
    */
   public static create = <TRouter extends AnyTRPCRouter>({
     url,
+    sessionId = randomUUID(),
     buildHeaders = async () => ({}),
   }: WebSocketTrpcClientOptions) => {
     let headers = {};
     const client = createWSClient({
       url: async () => {
-        headers = await buildHeaders();
+        headers = {
+          ...(await buildHeaders()),
+          'X-Amzn-Bedrock-AgentCore-Runtime-Session-Id': sessionId,
+        };
         return url;
       },
       WebSocket: class extends WebSocket {
@@ -50,6 +60,11 @@ export interface AgentCoreTrpcClientOptions {
    * The ARN of the Bedrock AgentCore Runtime to connect to
    */
   agentRuntimeArn: string;
+  /**
+   * Session ID to include in the WebSocket handshake headers.
+   * Defaults to a random UUID if not provided.
+   */
+  sessionId?: string;
 }
 
 export interface AgentCoreTrpcClientIamOptions extends AgentCoreTrpcClientOptions {
@@ -76,7 +91,7 @@ export class AgentCoreTrpcClient {
    */
   private static buildUrl = (agentRuntimeArn: string) => {
     const region = agentRuntimeArn.split(':')[3];
-    const url = `wss://bedrock-agentcore.${region}.amazonaws.com/runtimes/${agentRuntimeArn.replace(/:/g, '%3A').replace(/\//g, '%2F')}/ws`;
+    const url = `wss://bedrock-agentcore.${region}.amazonaws.com/runtimes/${encodeURIComponent(agentRuntimeArn)}/ws`;
     return { region, url };
   };
 
@@ -93,6 +108,7 @@ export class AgentCoreTrpcClient {
 
     return WebSocketTrpcClient.create<TRouter>({
       url,
+      sessionId: options.sessionId,
       buildHeaders: async () =>
         Object.fromEntries(
           (
@@ -116,6 +132,7 @@ export class AgentCoreTrpcClient {
 
     return WebSocketTrpcClient.create<TRouter>({
       url,
+      sessionId: options.sessionId,
       buildHeaders: async () => ({
         Authorization: `Bearer ${await options.accessTokenProvider()}`,
       }),

package/src/ts/strands-agent/files/app/agent.ts.template

@@ -1,20 +1,22 @@
 import { Agent, tool } from '@strands-agents/sdk';
 import { z } from 'zod';
 
-const add = tool({
-  name: 'Add',
-  description: 'Add two numbers',
+const multiply = tool({
+  name: 'Multiply',
+  description: 'Multiply two numbers',
   inputSchema: z.object({
     a: z.number(),
     b: z.number(),
   }),
-  callback: ({ a, b }) => a + b,
+  callback: ({ a, b }) => a * b,
 });
 
-export const getAgent = () =>
-  new Agent({
-    systemPrompt: `You are an addition wizard.
-  Use the add tool for addition tasks.
+export const getAgent = async (sessionId: string) => {
+  console.log(`Creating agent for session ${sessionId}`);
+  return new Agent({
+    systemPrompt: `You are a mathematical wizard.
+  Use your tools for mathematical tasks.
   Refer to tools as your 'spellbook'.`,
-    tools: [add],
+    tools: [multiply],
   });
+};

package/src/ts/strands-agent/files/app/index.ts.template

@@ -1,3 +1,4 @@
+import { randomUUID } from 'node:crypto';
 import { createServer } from 'http';
 import {
   CreateHTTPContextOptions,
@@ -16,7 +17,15 @@ const PORT = parseInt(process.env.PORT || '8080');
 
 const createContext = (
   opts: CreateHTTPContextOptions | CreateWSSContextFnOptions,
-): Context => ({});
+): Context => {
+  const sessionId =
+    ('req' in opts
+      ? opts.req.headers['x-amzn-bedrock-agentcore-runtime-session-id']
+      : undefined) ?? randomUUID();
+  return {
+    sessionId: Array.isArray(sessionId) ? sessionId[0] : sessionId,
+  };
+};
 
 const handler = createHTTPHandler({
   router: appRouter,

package/src/ts/strands-agent/files/app/init.ts.template

@@ -1,7 +1,8 @@
 import { initTRPC } from '@trpc/server';
 
-// eslint-disable-next-line
-export interface Context {}
+export interface Context {
+  sessionId: string;
+}
 
 export const t = initTRPC.context<Context>().create();
 

package/src/ts/strands-agent/files/app/router.ts.template

@@ -15,7 +15,7 @@ export const appRouter = router({
       }),
     )
     .subscription(async function* (opts) {
-      const agent = getAgent();
+      const agent = await getAgent(opts.ctx.sessionId);
 
       for await (const event of agent.stream(opts.input.message)) {
         if (

package/src/ts/strands-agent/files/deploy/Dockerfile.template

@@ -4,7 +4,7 @@ WORKDIR /app
 
 # Add AWS Distro for OpenTelemetry for observability
 # https://docs.aws.amazon.com/bedrock-agentcore/latest/devguide/observability-configure.html
-RUN npm install @aws/aws-distro-opentelemetry-node-autoinstrumentation@^0.7.0
+RUN npm install @aws/aws-distro-opentelemetry-node-autoinstrumentation@<%- adotVersion %>
 
 # Copy bundled agent
 COPY --from=workspace <%- distDir %>/bundle/agent/<%- name %>/index.js /app

package/src/ts/strands-agent/generator.js

@@ -51,6 +51,7 @@ const tsStrandsAgentGenerator = (tree, options) => tslib_1.__awaiter(void 0, voi
         (0, devkit_1.generateFiles)(tree, (0, devkit_1.joinPathFragments)(__dirname, 'files', 'deploy'), targetSourceDir, {
             distDir,
             name,
+            adotVersion: versions_1.TS_VERSIONS['@aws/aws-distro-opentelemetry-node-autoinstrumentation'],
         }, { overwriteStrategy: devkit_1.OverwriteStrategy.KeepExisting });
         const dockerTargetName = `${agentTargetPrefix}-docker`;
         project.targets[dockerTargetName] = {
@@ -102,8 +103,19 @@ const tsStrandsAgentGenerator = (tree, options) => tslib_1.__awaiter(void 0, voi
                     },
                 },
                 continuous: true,
+            }, [`${agentTargetPrefix}-serve-local`]: {
+                executor: 'nx:run-commands',
+                options: {
+                    commands: [`tsx --watch ${relativeSourceDir}/index.ts`],
+                    cwd: '{projectRoot}',
+                    env: {
+                        PORT: `${localDevPort}`,
+                        SERVE_LOCAL: 'true',
+                    },
+                },
+                continuous: true,
             } }) }));
-    (0, nx_1.addComponentGeneratorMetadata)(tree, project.name, exports.TS_STRANDS_AGENT_GENERATOR_INFO, targetSourceDirRelativeToProjectRoot, agentTargetPrefix, { port: localDevPort });
+    (0, nx_1.addComponentGeneratorMetadata)(tree, project.name, exports.TS_STRANDS_AGENT_GENERATOR_INFO, targetSourceDirRelativeToProjectRoot, agentTargetPrefix, { port: localDevPort, rc: agentNameClassName });
     yield (0, metrics_1.addGeneratorMetricsIfApplicable)(tree, [
         exports.TS_STRANDS_AGENT_GENERATOR_INFO,
     ]);

package/src/ts/strands-agent/generator.js.map

@@ -1 +1 @@
-{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../../../../packages/nx-plugin/src/ts/strands-agent/generator.ts"],"names":[],"mappings":";;;;AAAA;;;GAGG;AACH,uCASoB;AAEpB,uCAMwB;AACxB,iDAAsE;AACtE,+CAA0D;AAC1D,6CAA2D;AAC3D,mDAAoD;AACpD,qDAAoD;AACpD,sDAAsE;AACtE,yCAAqD;AACrD,qEAA0E;AAC1E,mGAAwF;AACxF,2CAA8C;AAEjC,QAAA,+BAA+B,GAC1C,IAAA,qBAAgB,EAAC,UAAU,CAAC,CAAC;AAExB,MAAM,uBAAuB,GAAG,CACrC,IAAU,EACV,OAAsC,EACV,EAAE;;IAC9B,MAAM,OAAO,GAAG,IAAA,wCAAmC,EAAC,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;IAE3E,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAA,0BAAiB,EAAC,OAAO,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC,EAAE,CAAC;QACnE,MAAM,IAAI,KAAK,CACb,uBAAuB,OAAO,CAAC,OAAO,wDAAwD,CAC/F,CAAC;IACJ,CAAC;IAED,MAAM,WAAW,GAAG,GAAG,IAAA,iBAAS,EAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC;IACxE,MAAM,IAAI,GAAG,IAAA,iBAAS,EAAC,OAAO,CAAC,IAAI,IAAI,WAAW,CAAC,CAAC;IACpD,MAAM,kBAAkB,GAAG,IAAA,mBAAW,EAAC,IAAI,CAAC,CAAC;IAC7C,MAAM,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;IACxD,MAAM,oCAAoC,GAAG,IAAA,0BAAiB,EAC5D,KAAK,EACL,iBAAiB,CAClB,CAAC;IACF,MAAM,eAAe,GAAG,IAAA,0BAAiB,EACvC,OAAO,CAAC,IAAI,EACZ,oCAAoC,CACrC,CAAC;IACF,MAAM,iBAAiB,GAAG,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,GAAG,GAAG,EAAE,IAAI,CAAC,CAAC;IAC5E,MAAM,OAAO,GAAG,IAAA,0BAAiB,EAAC,MAAM,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;IAExD,MAAM,WAAW,GAAG,MAAA,OAAO,CAAC,WAAW,mCAAI,yBAAyB,CAAC;IAErE,yBAAyB;IACzB,IAAA,sBAAa,EACX,IAAI,EACJ,IAAA,0BAAiB,EAAC,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,EAC5C,eAAe,EACf;QACE,IAAI;QACJ,kBAAkB;QAClB,OAAO;KACR,EACD,EAAE,iBAAiB,EAAE,0BAAiB,CAAC,YAAY,EAAE,CACtD,CAAC;IAEF,IAAI,WAAW,KAAK,yBAAyB,EAAE,CAAC;QAC9C,MAAM,cAAc,GAAG,GAAG,IAAA,uBAAW,EAAC,IAAI,CAAC,IAAI,IAAI,SAAS,CAAC;QAE7D,oBAAoB;QACpB,IAAA,kCAAyB,EAAC,IAAI,EAAE,OAAO,EAAE;YACvC,cAAc,EAAE,GAAG,oCAAoC,WAAW;YAClE,eAAe,EAAE,IAAA,0BAAiB,EAAC,OAAO,EAAE,IAAI,CAAC;SAClD,CAAC,CAAC;QAEH,qBAAqB;QACrB,IAAA,sBAAa,EACX,IAAI,EACJ,IAAA,0BAAiB,EAAC,SAAS,EAAE,OAAO,EAAE,QAAQ,CAAC,EAC/C,eAAe,EACf;YACE,OAAO;YACP,IAAI;SACL,EACD,EAAE,iBAAiB,EAAE,0BAAiB,CAAC,YAAY,EAAE,CACtD,CAAC;QAEF,MAAM,gBAAgB,GAAG,GAAG,iBAAiB,SAAS,CAAC;QAEvD,OAAO,CAAC,OAAO,CAAC,gBAAgB,CAAC,GAAG;YAClC,KAAK,EAAE,IAAI;YACX,QAAQ,EAAE,iBAAiB;YAC3B,OAAO,EAAE;gBACP,OAAO,EAAE,0CAA0C,cAAc,IAAI,eAAe,8BAA8B;aACnH;YACD,SAAS,EAAE,CAAC,QAAQ,CAAC;SACtB,CAAC;QAEF,IAAA,sCAAiC,EAAC,OAAO,EAAE,QAAQ,EAAE,gBAAgB,CAAC,CAAC;QACvE,IAAA,sCAAiC,EAAC,OAAO,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;QAE9D,wBAAwB;QACxB,MAAM,WAAW,GAAG,MAAM,IAAA,wBAAkB,EAAC,IAAI,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;QACxE,MAAM,IAAA,6CAAyB,EAAC,IAAI,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC;QAEvD,IAAA,qCAAa,EAAC,IAAI,EAAE;YAClB,kBAAkB,EAAE,IAAI;YACxB,kBAAkB;YAClB,WAAW,EAAE,OAAO,CAAC,IAAI;YACzB,cAAc;YACd,WAAW;SACZ,CAAC,CAAC;IACL,CAAC;IAED,mBAAmB;IACnB,IAAA,qCAA4B,EAC1B,IAAI,EACJ,IAAA,uBAAY,EAAC;QACX,cAAc;QACd,cAAc;QACd,KAAK;QACL,qBAAqB;QACrB,IAAI;QACJ,MAAM;QACN,+BAA+B;QAC/B,+BAA+B;QAC/B,mCAAmC;QACnC,WAAW;QACX,2BAA2B;KAC5B,CAAC,EACF,IAAA,uBAAY,EAAC,CAAC,KAAK,EAAE,WAAW,EAAE,aAAa,EAAE,aAAa,CAAC,CAAC,CACjE,CAAC;IAEF,gGAAgG;IAChG,+FAA+F;IAC/F,sCAAsC;IACtC,MAAM,YAAY,GAAG,IAAA,iBAAU,EAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;IAErD,IAAA,mCAA0B,EAAC,IAAI,EAAE,OAAO,CAAC,IAAI,kCACxC,OAAO,KACV,OAAO,kCACF,OAAO,CAAC,OAAO,KAClB,CAAC,GAAG,iBAAiB,QAAQ,CAAC,EAAE;gBAC9B,QAAQ,EAAE,iBAAiB;gBAC3B,OAAO,EAAE;oBACP,QAAQ,EAAE,CAAC,eAAe,iBAAiB,WAAW,CAAC;oBACvD,GAAG,EAAE,eAAe;oBACpB,GAAG,EAAE;wBACH,IAAI,EAAE,GAAG,YAAY,EAAE;qBACxB;iBACF;gBACD,UAAU,EAAE,IAAI;aACjB,OAEH,CAAC;IAEH,IAAA,kCAA6B,EAC3B,IAAI,EACJ,OAAO,CAAC,IAAI,EACZ,uCAA+B,EAC/B,oCAAoC,EACpC,iBAAiB,EACjB,EAAE,IAAI,EAAE,YAAY,EAAE,CACvB,CAAC;IAEF,MAAM,IAAA,yCAA+B,EAAC,IAAI,EAAE;QAC1C,uCAA+B;KAChC,CAAC,CAAC;IAEH,MAAM,IAAA,6BAAoB,EAAC,IAAI,CAAC,CAAC;IACjC,OAAO,GAAG,EAAE;QACV,IAAA,4BAAmB,EAAC,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC;AACJ,CAAC,CAAA,CAAC;AArJW,QAAA,uBAAuB,2BAqJlC;AAEF,kBAAe,+BAAuB,CAAC"}
+{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../../../../packages/nx-plugin/src/ts/strands-agent/generator.ts"],"names":[],"mappings":";;;;AAAA;;;GAGG;AACH,uCASoB;AAEpB,uCAMwB;AACxB,iDAAsE;AACtE,+CAA0D;AAC1D,6CAA2D;AAC3D,mDAAiE;AACjE,qDAAoD;AACpD,sDAAsE;AACtE,yCAAqD;AACrD,qEAA0E;AAC1E,mGAAwF;AACxF,2CAA8C;AAEjC,QAAA,+BAA+B,GAC1C,IAAA,qBAAgB,EAAC,UAAU,CAAC,CAAC;AAExB,MAAM,uBAAuB,GAAG,CACrC,IAAU,EACV,OAAsC,EACV,EAAE;;IAC9B,MAAM,OAAO,GAAG,IAAA,wCAAmC,EAAC,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;IAE3E,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAA,0BAAiB,EAAC,OAAO,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC,EAAE,CAAC;QACnE,MAAM,IAAI,KAAK,CACb,uBAAuB,OAAO,CAAC,OAAO,wDAAwD,CAC/F,CAAC;IACJ,CAAC;IAED,MAAM,WAAW,GAAG,GAAG,IAAA,iBAAS,EAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC;IACxE,MAAM,IAAI,GAAG,IAAA,iBAAS,EAAC,OAAO,CAAC,IAAI,IAAI,WAAW,CAAC,CAAC;IACpD,MAAM,kBAAkB,GAAG,IAAA,mBAAW,EAAC,IAAI,CAAC,CAAC;IAC7C,MAAM,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;IACxD,MAAM,oCAAoC,GAAG,IAAA,0BAAiB,EAC5D,KAAK,EACL,iBAAiB,CAClB,CAAC;IACF,MAAM,eAAe,GAAG,IAAA,0BAAiB,EACvC,OAAO,CAAC,IAAI,EACZ,oCAAoC,CACrC,CAAC;IACF,MAAM,iBAAiB,GAAG,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,GAAG,GAAG,EAAE,IAAI,CAAC,CAAC;IAC5E,MAAM,OAAO,GAAG,IAAA,0BAAiB,EAAC,MAAM,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;IAExD,MAAM,WAAW,GAAG,MAAA,OAAO,CAAC,WAAW,mCAAI,yBAAyB,CAAC;IAErE,yBAAyB;IACzB,IAAA,sBAAa,EACX,IAAI,EACJ,IAAA,0BAAiB,EAAC,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,EAC5C,eAAe,EACf;QACE,IAAI;QACJ,kBAAkB;QAClB,OAAO;KACR,EACD,EAAE,iBAAiB,EAAE,0BAAiB,CAAC,YAAY,EAAE,CACtD,CAAC;IAEF,IAAI,WAAW,KAAK,yBAAyB,EAAE,CAAC;QAC9C,MAAM,cAAc,GAAG,GAAG,IAAA,uBAAW,EAAC,IAAI,CAAC,IAAI,IAAI,SAAS,CAAC;QAE7D,oBAAoB;QACpB,IAAA,kCAAyB,EAAC,IAAI,EAAE,OAAO,EAAE;YACvC,cAAc,EAAE,GAAG,oCAAoC,WAAW;YAClE,eAAe,EAAE,IAAA,0BAAiB,EAAC,OAAO,EAAE,IAAI,CAAC;SAClD,CAAC,CAAC;QAEH,qBAAqB;QACrB,IAAA,sBAAa,EACX,IAAI,EACJ,IAAA,0BAAiB,EAAC,SAAS,EAAE,OAAO,EAAE,QAAQ,CAAC,EAC/C,eAAe,EACf;YACE,OAAO;YACP,IAAI;YACJ,WAAW,EACT,sBAAW,CACT,wDAAwD,CACzD;SACJ,EACD,EAAE,iBAAiB,EAAE,0BAAiB,CAAC,YAAY,EAAE,CACtD,CAAC;QAEF,MAAM,gBAAgB,GAAG,GAAG,iBAAiB,SAAS,CAAC;QAEvD,OAAO,CAAC,OAAO,CAAC,gBAAgB,CAAC,GAAG;YAClC,KAAK,EAAE,IAAI;YACX,QAAQ,EAAE,iBAAiB;YAC3B,OAAO,EAAE;gBACP,OAAO,EAAE,0CAA0C,cAAc,IAAI,eAAe,8BAA8B;aACnH;YACD,SAAS,EAAE,CAAC,QAAQ,CAAC;SACtB,CAAC;QAEF,IAAA,sCAAiC,EAAC,OAAO,EAAE,QAAQ,EAAE,gBAAgB,CAAC,CAAC;QACvE,IAAA,sCAAiC,EAAC,OAAO,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;QAE9D,wBAAwB;QACxB,MAAM,WAAW,GAAG,MAAM,IAAA,wBAAkB,EAAC,IAAI,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;QACxE,MAAM,IAAA,6CAAyB,EAAC,IAAI,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC;QAEvD,IAAA,qCAAa,EAAC,IAAI,EAAE;YAClB,kBAAkB,EAAE,IAAI;YACxB,kBAAkB;YAClB,WAAW,EAAE,OAAO,CAAC,IAAI;YACzB,cAAc;YACd,WAAW;SACZ,CAAC,CAAC;IACL,CAAC;IAED,mBAAmB;IACnB,IAAA,qCAA4B,EAC1B,IAAI,EACJ,IAAA,uBAAY,EAAC;QACX,cAAc;QACd,cAAc;QACd,KAAK;QACL,qBAAqB;QACrB,IAAI;QACJ,MAAM;QACN,+BAA+B;QAC/B,+BAA+B;QAC/B,mCAAmC;QACnC,WAAW;QACX,2BAA2B;KAC5B,CAAC,EACF,IAAA,uBAAY,EAAC,CAAC,KAAK,EAAE,WAAW,EAAE,aAAa,EAAE,aAAa,CAAC,CAAC,CACjE,CAAC;IAEF,gGAAgG;IAChG,+FAA+F;IAC/F,sCAAsC;IACtC,MAAM,YAAY,GAAG,IAAA,iBAAU,EAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;IAErD,IAAA,mCAA0B,EAAC,IAAI,EAAE,OAAO,CAAC,IAAI,kCACxC,OAAO,KACV,OAAO,kCACF,OAAO,CAAC,OAAO,KAClB,CAAC,GAAG,iBAAiB,QAAQ,CAAC,EAAE;gBAC9B,QAAQ,EAAE,iBAAiB;gBAC3B,OAAO,EAAE;oBACP,QAAQ,EAAE,CAAC,eAAe,iBAAiB,WAAW,CAAC;oBACvD,GAAG,EAAE,eAAe;oBACpB,GAAG,EAAE;wBACH,IAAI,EAAE,GAAG,YAAY,EAAE;qBACxB;iBACF;gBACD,UAAU,EAAE,IAAI;aACjB,EACD,CAAC,GAAG,iBAAiB,cAAc,CAAC,EAAE;gBACpC,QAAQ,EAAE,iBAAiB;gBAC3B,OAAO,EAAE;oBACP,QAAQ,EAAE,CAAC,eAAe,iBAAiB,WAAW,CAAC;oBACvD,GAAG,EAAE,eAAe;oBACpB,GAAG,EAAE;wBACH,IAAI,EAAE,GAAG,YAAY,EAAE;wBACvB,WAAW,EAAE,MAAM;qBACpB;iBACF;gBACD,UAAU,EAAE,IAAI;aACjB,OAEH,CAAC;IAEH,IAAA,kCAA6B,EAC3B,IAAI,EACJ,OAAO,CAAC,IAAI,EACZ,uCAA+B,EAC/B,oCAAoC,EACpC,iBAAiB,EACjB,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,kBAAkB,EAAE,CAC/C,CAAC;IAEF,MAAM,IAAA,yCAA+B,EAAC,IAAI,EAAE;QAC1C,uCAA+B;KAChC,CAAC,CAAC;IAEH,MAAM,IAAA,6BAAoB,EAAC,IAAI,CAAC,CAAC;IACjC,OAAO,GAAG,EAAE;QACV,IAAA,4BAAmB,EAAC,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC;AACJ,CAAC,CAAA,CAAC;AArKW,QAAA,uBAAuB,2BAqKlC;AAEF,kBAAe,+BAAuB,CAAC"}

package/src/ts/strands-agent/mcp-connection/__snapshots__/generator.spec.ts.snap

@@ -0,0 +1,231 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`ts#strands-agent#mcp-connection generator > should match snapshot for agent-connection src files > agent-connection-index.ts 1`] = `
+"export * from './app/inventory-mcp-client.js';
+// Export your library code here
+"
+`;
+
+exports[`ts#strands-agent#mcp-connection generator > should match snapshot for agent-connection src files > agentcore-mcp-client.ts 1`] = `
+"import { randomUUID } from 'node:crypto';
+import { McpClient } from '@strands-agents/sdk';
+import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
+import { AwsClient } from 'aws4fetch';
+import { fromNodeProviderChain } from '@aws-sdk/credential-providers';
+
+/**
+ * Common options for creating an AgentCore MCP client
+ */
+export interface AgentCoreMcpClientOptions {
+  /**
+   * The ARN of the Bedrock AgentCore Runtime to connect to
+   */
+  agentRuntimeArn: string;
+  /**
+   * Session ID to include in request headers.
+   * Defaults to a random UUID if not provided.
+   */
+  sessionId?: string;
+}
+
+/**
+ * Options for creating an AgentCore MCP client with IAM authentication
+ */
+export interface AgentCoreMcpClientIamOptions extends AgentCoreMcpClientOptions {
+  /**
+   * Optional AWS credential provider. If not provided, uses the default
+   * credential provider chain from the AWS SDK.
+   */
+  credentialProvider?: ReturnType<typeof fromNodeProviderChain>;
+}
+
+/**
+ * Options for creating an AgentCore MCP client with JWT authentication
+ */
+export interface AgentCoreMcpClientJwtOptions extends AgentCoreMcpClientOptions {
+  /**
+   * A function which returns the JWT access token used to authenticate
+   */
+  accessTokenProvider: () => Promise<string>;
+}
+
+/**
+ * Internal options for creating an AgentCore MCP client
+ */
+interface CreateOptions {
+  url: string;
+  sessionId?: string;
+  buildHeaders?: () => Promise<Record<string, string>>;
+  fetch?: typeof fetch;
+}
+
+/**
+ * Factory for clients to call MCP servers hosted on Bedrock AgentCore Runtime
+ */
+export class AgentCoreMcpClient {
+  /**
+   * Construct the invocation URL for an AgentCore runtime ARN and extract the region.
+   * ARN format: arn:partition:service:region:account-id:resource
+   */
+  private static buildUrl(agentRuntimeArn: string): {
+    region: string;
+    url: string;
+  } {
+    const region = agentRuntimeArn.split(':')[3];
+    const url = \`https://bedrock-agentcore.\${region}.amazonaws.com/runtimes/\${encodeURIComponent(agentRuntimeArn)}/invocations?qualifier=DEFAULT\`;
+    return { region, url };
+  }
+
+  /**
+   * Internal method to create an MCP client
+   */
+  private static create(options: CreateOptions): McpClient {
+    const {
+      url,
+      sessionId = randomUUID(),
+      buildHeaders = async () => ({}),
+      fetch: customFetch,
+    } = options;
+
+    // Create a custom fetch that adds required headers
+    const fetchWithHeaders: typeof fetch = async (input, init) => {
+      const headers = {
+        'X-Amzn-Bedrock-AgentCore-Runtime-Session-Id': sessionId,
+        ...(await buildHeaders()),
+      };
+      const existingHeaders = init?.headers;
+
+      const mergedInit = {
+        ...init,
+        headers: !existingHeaders
+          ? headers
+          : existingHeaders instanceof Headers
+            ? (() => {
+                const h = new Headers(existingHeaders);
+                Object.entries(headers).forEach(([k, v]) => h.append(k, v));
+                return h;
+              })()
+            : Array.isArray(existingHeaders)
+              ? [...existingHeaders, ...Object.entries(headers)]
+              : { ...existingHeaders, ...headers },
+      };
+      return (customFetch || fetch)(input, mergedInit);
+    };
+
+    // Create the transport
+    const transport = new StreamableHTTPClientTransport(new URL(url), {
+      fetch: fetchWithHeaders,
+    });
+
+    // Create and return the MCP client
+    return new McpClient({ transport });
+  }
+
+  /**
+   * Create an MCP client with IAM (SigV4) authentication
+   */
+  static withIamAuth(options: AgentCoreMcpClientIamOptions): McpClient {
+    const { agentRuntimeArn, sessionId } = options;
+    const credentialProvider =
+      options.credentialProvider ?? fromNodeProviderChain();
+    const { region, url } = AgentCoreMcpClient.buildUrl(agentRuntimeArn);
+
+    // Create a SigV4 signing fetch function
+    const sigv4Fetch: typeof fetch = async (...args) => {
+      const credentials = await credentialProvider();
+      const client = new AwsClient({
+        ...credentials,
+        service: 'bedrock-agentcore',
+        region,
+      });
+      return client.fetch(...args);
+    };
+
+    return AgentCoreMcpClient.create({
+      url,
+      sessionId,
+      fetch: sigv4Fetch,
+    });
+  }
+
+  /**
+   * Create an MCP client with JWT authentication
+   */
+  static withJwtAuth(options: AgentCoreMcpClientJwtOptions): McpClient {
+    const { agentRuntimeArn, accessTokenProvider, sessionId } = options;
+    const { url } = AgentCoreMcpClient.buildUrl(agentRuntimeArn);
+
+    return AgentCoreMcpClient.create({
+      url,
+      sessionId,
+      buildHeaders: async () => ({
+        Authorization: \`Bearer \${await accessTokenProvider()}\`,
+      }),
+    });
+  }
+}
+"
+`;
+
+exports[`ts#strands-agent#mcp-connection generator > should match snapshot for agent-connection src files > inventory-mcp-client.ts 1`] = `
+"import { McpClient } from '@strands-agents/sdk';
+import { AgentCoreMcpClient } from '../core/agentcore-mcp-client.js';
+import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
+import { getAppConfig } from '@aws-lambda-powertools/parameters/appconfig';
+
+/**
+ * Client for connecting to the InventoryMcp MCP server.
+ *
+ * In deployed mode: resolves the AgentCore runtime ARN from runtime config
+ * and uses SigV4-authenticated AgentCore transport.
+ *
+ * In local mode (SERVE_LOCAL env var set): connects directly
+ * to the local MCP server via Streamable HTTP.
+ */
+export class InventoryMcpClient {
+  static async create(sessionId: string): Promise<McpClient> {
+    if (process.env.SERVE_LOCAL === 'true') {
+      // Local development: connect directly via Streamable HTTP
+      const transport = new StreamableHTTPClientTransport(
+        new URL('http://localhost:8082/mcp'),
+      );
+      return new McpClient({ transport });
+    }
+
+    // Deployed: resolve ARN from runtime config, use AgentCore transport
+    const config = (await getAppConfig('connection', {
+      application: process.env.RUNTIME_CONFIG_APP_ID!,
+      environment: 'default',
+      transform: 'json',
+    })) as Record<string, any>;
+
+    return AgentCoreMcpClient.withIamAuth({
+      agentRuntimeArn: config.agentRuntimes.InventoryMcp,
+      sessionId,
+    });
+  }
+}
+"
+`;
+
+exports[`ts#strands-agent#mcp-connection generator > should match snapshot for agent-connection src files > transformed-agent.ts 1`] = `
+"import { InventoryMcpClient } from ':proj/agent-connection';
+import { Agent, tool } from '@strands-agents/sdk';
+import { z } from 'zod';
+
+const multiply = tool({
+  name: 'Multiply',
+  description: 'Multiply two numbers',
+  inputSchema: z.object({ a: z.number(), b: z.number() }),
+  callback: ({ a, b }) => a * b,
+});
+
+export const getAgent = async (sessionId: string) => {
+  const inventoryMcp = await InventoryMcpClient.create(sessionId);
+  return new Agent({
+    systemPrompt: 'You are a mathematical wizard.',
+    tools: [inventoryMcp, multiply],
+  });
+};
+"
+`;

package/src/ts/strands-agent/mcp-connection/files/agent-connection/app/__mcpServerKebabCase__-client.ts.template

@@ -0,0 +1,37 @@
+import { McpClient } from '@strands-agents/sdk';
+import { AgentCoreMcpClient } from '../core/agentcore-mcp-client.js';
+import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
+import { getAppConfig } from '@aws-lambda-powertools/parameters/appconfig';
+
+/**
+ * Client for connecting to the <%- mcpServerClassName %> MCP server.
+ *
+ * In deployed mode: resolves the AgentCore runtime ARN from runtime config
+ * and uses SigV4-authenticated AgentCore transport.
+ *
+ * In local mode (SERVE_LOCAL env var set): connects directly
+ * to the local MCP server via Streamable HTTP.
+ */
+export class <%- mcpServerClassName %>Client {
+  static async create(sessionId: string): Promise<McpClient> {
+    if (process.env.SERVE_LOCAL === 'true') {
+      // Local development: connect directly via Streamable HTTP
+      const transport = new StreamableHTTPClientTransport(
+        new URL('http://localhost:<%- mcpServerPort %>/mcp'),
+      );
+      return new McpClient({ transport });
+    }
+
+    // Deployed: resolve ARN from runtime config, use AgentCore transport
+    const config = (await getAppConfig('connection', {
+      application: process.env.RUNTIME_CONFIG_APP_ID!,
+      environment: 'default',
+      transform: 'json',
+    })) as Record<string, any>;
+
+    return AgentCoreMcpClient.withIamAuth({
+      agentRuntimeArn: config.agentRuntimes.<%- mcpServerClassName %>,
+      sessionId,
+    });
+  }
+}

package/src/ts/strands-agent/mcp-connection/generator.d.ts

@@ -0,0 +1,10 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { GeneratorCallback, Tree } from '@nx/devkit';
+import { TsStrandsAgentMcpConnectionGeneratorSchema } from './schema';
+import { NxGeneratorInfo } from '../../../utils/nx';
+export declare const TS_STRANDS_AGENT_MCP_CONNECTION_GENERATOR_INFO: NxGeneratorInfo;
+export declare const tsStrandsAgentMcpConnectionGenerator: (tree: Tree, options: TsStrandsAgentMcpConnectionGeneratorSchema) => Promise<GeneratorCallback>;
+export default tsStrandsAgentMcpConnectionGenerator;

package/src/ts/strands-agent/mcp-connection/generator.js

@@ -0,0 +1,124 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.tsStrandsAgentMcpConnectionGenerator = exports.TS_STRANDS_AGENT_MCP_CONNECTION_GENERATOR_INFO = void 0;
+const tslib_1 = require("tslib");
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+const devkit_1 = require("@nx/devkit");
+const nx_1 = require("../../../utils/nx");
+const metrics_1 = require("../../../utils/metrics");
+const format_1 = require("../../../utils/format");
+const names_1 = require("../../../utils/names");
+const versions_1 = require("../../../utils/versions");
+const npm_scope_1 = require("../../../utils/npm-scope");
+const ast_1 = require("../../../utils/ast");
+const agent_connection_1 = require("../../../utils/agent-connection/agent-connection");
+const typescript_1 = tslib_1.__importStar(require("typescript"));
+exports.TS_STRANDS_AGENT_MCP_CONNECTION_GENERATOR_INFO = (0, nx_1.getGeneratorInfo)(__filename);
+const tsStrandsAgentMcpConnectionGenerator = (tree, options) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+    var _a, _b, _c, _d, _e, _f;
+    const sourceProject = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.sourceProject);
+    const targetProject = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.targetProject);
+    const agentComponent = options.sourceComponent;
+    const mcpComponent = options.targetComponent;
+    if (!agentComponent || !mcpComponent) {
+        throw new Error('Both sourceComponent and targetComponent must be provided for ts#strands-agent -> mcp connections');
+    }
+    const mcpComponentName = (_a = mcpComponent.name) !== null && _a !== void 0 ? _a : 'mcp-server';
+    const mcpServerClassName = mcpComponent.rc;
+    const mcpServerKebabCase = (0, names_1.kebabCase)(mcpServerClassName);
+    const mcpServerPort = (_b = mcpComponent.port) !== null && _b !== void 0 ? _b : 8000;
+    const npmScope = (0, npm_scope_1.getNpmScope)(tree);
+    // 1. Ensure the shared agent-connection project exists
+    yield (0, agent_connection_1.ensureTypeScriptAgentConnectionProject)(tree);
+    // 2. Generate the per-connection <Name>Client into app/
+    (0, devkit_1.generateFiles)(tree, (0, devkit_1.joinPathFragments)(__dirname, 'files', 'agent-connection', 'app'), (0, devkit_1.joinPathFragments)(agent_connection_1.AGENT_CONNECTION_PROJECT_DIR, 'src', 'app'), {
+        mcpServerKebabCase,
+        mcpServerClassName,
+        mcpServerPort,
+    }, { overwriteStrategy: devkit_1.OverwriteStrategy.KeepExisting });
+    // Add re-export to index.ts
+    (0, ast_1.addStarExport)(tree, (0, devkit_1.joinPathFragments)(agent_connection_1.AGENT_CONNECTION_PROJECT_DIR, 'src', 'index.ts'), `./app/${mcpServerKebabCase}-client.js`);
+    // 3. AST transform agent.ts to add MCP tools
+    const agentSourceDir = (0, devkit_1.joinPathFragments)(sourceProject.root, (_c = agentComponent.path) !== null && _c !== void 0 ? _c : 'src');
+    const agentFilePath = (0, devkit_1.joinPathFragments)(agentSourceDir, 'agent.ts');
+    if (tree.exists(agentFilePath)) {
+        const clientClassName = `${mcpServerClassName}Client`;
+        const clientVarName = mcpServerClassName.charAt(0).toLowerCase() + mcpServerClassName.slice(1);
+        // Add import for the client
+        (0, ast_1.addDestructuredImport)(tree, agentFilePath, [clientClassName], `:${npmScope}/agent-connection`);
+        // Create: const <clientVarName> = await <ClientClassName>.create(sessionId);
+        const clientCreationStatement = typescript_1.factory.createVariableStatement(undefined, typescript_1.factory.createVariableDeclarationList([
+            typescript_1.factory.createVariableDeclaration(clientVarName, undefined, undefined, typescript_1.factory.createAwaitExpression(typescript_1.factory.createCallExpression(typescript_1.factory.createPropertyAccessExpression(typescript_1.factory.createIdentifier(clientClassName), 'create'), undefined, [typescript_1.factory.createIdentifier('sessionId')]))),
+        ], 2));
+        // Transform the arrow function that contains `new Agent` to have a block body
+        // with the client creation statement before the return
+        (0, ast_1.replaceIfExists)(tree, agentFilePath, 'ArrowFunction:has(NewExpression[expression.name="Agent"])', (node) => {
+            const arrow = node;
+            // Check if already transformed
+            if (arrow.getText().includes(clientClassName))
+                return node;
+            // If the body is an expression (not a block), wrap it in a block with return
+            if (!typescript_1.default.isBlock(arrow.body)) {
+                return typescript_1.factory.updateArrowFunction(arrow, arrow.modifiers, arrow.typeParameters, arrow.parameters, arrow.type, arrow.equalsGreaterThanToken, typescript_1.factory.createBlock([
+                    clientCreationStatement,
+                    typescript_1.factory.createReturnStatement(arrow.body),
+                ], true));
+            }
+            // Body is already a block — insert before the statement containing `new Agent`
+            const b = arrow.body;
+            const agentStatementIndex = b.statements.findIndex((s) => s.getText().includes('new Agent('));
+            if (agentStatementIndex === -1)
+                return node;
+            const newStatements = [...b.statements];
+            newStatements.splice(agentStatementIndex, 0, clientCreationStatement);
+            return typescript_1.factory.updateArrowFunction(arrow, arrow.modifiers, arrow.typeParameters, arrow.parameters, arrow.type, arrow.equalsGreaterThanToken, typescript_1.factory.updateBlock(b, newStatements));
+        });
+        // Add client to the tools array: tools: [clientVarName, ...existing]
+        (0, ast_1.replaceIfExists)(tree, agentFilePath, 'NewExpression[expression.name="Agent"] ObjectLiteralExpression PropertyAssignment[name.name="tools"] ArrayLiteralExpression', (node) => {
+            const arr = node;
+            // Check if already added
+            if (arr.getText().includes(clientVarName))
+                return node;
+            return typescript_1.factory.updateArrayLiteralExpression(arr, [
+                typescript_1.factory.createIdentifier(clientVarName),
+                ...arr.elements,
+            ]);
+        });
+    }
+    // 4. Set up serve-local target
+    const agentName = (_d = agentComponent.name) !== null && _d !== void 0 ? _d : 'agent';
+    const serveLocalTargetName = `${agentName}-serve-local`;
+    const mcpServeLocalTargetName = `${mcpComponentName}-serve-local`;
+    if ((_e = sourceProject.targets) === null || _e === void 0 ? void 0 : _e[serveLocalTargetName]) {
+        const target = sourceProject.targets[serveLocalTargetName];
+        target.dependsOn = [
+            ...((_f = target.dependsOn) !== null && _f !== void 0 ? _f : []),
+            {
+                projects: [targetProject.name],
+                target: mcpServeLocalTargetName,
+            },
+        ];
+        (0, devkit_1.updateProjectConfiguration)(tree, sourceProject.name, sourceProject);
+    }
+    // 5. Add dependencies
+    (0, devkit_1.addDependenciesToPackageJson)(tree, (0, versions_1.withVersions)([
+        '@modelcontextprotocol/sdk',
+        '@aws-lambda-powertools/parameters',
+        '@aws-sdk/client-appconfigdata',
+        'aws4fetch',
+        '@aws-sdk/credential-providers',
+    ]), {});
+    yield (0, metrics_1.addGeneratorMetricsIfApplicable)(tree, [
+        exports.TS_STRANDS_AGENT_MCP_CONNECTION_GENERATOR_INFO,
+    ]);
+    yield (0, format_1.formatFilesInSubtree)(tree);
+    return () => {
+        (0, devkit_1.installPackagesTask)(tree);
+    };
+});
+exports.tsStrandsAgentMcpConnectionGenerator = tsStrandsAgentMcpConnectionGenerator;
+exports.default = exports.tsStrandsAgentMcpConnectionGenerator;
+//# sourceMappingURL=generator.js.map

package/src/ts/strands-agent/mcp-connection/generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../../../../../packages/nx-plugin/src/ts/strands-agent/mcp-connection/generator.ts"],"names":[],"mappings":";;;;AAAA;;;GAGG;AACH,uCASoB;AAEpB,0CAI2B;AAC3B,oDAAyE;AACzE,kDAA6D;AAC7D,gDAAiD;AACjD,sDAAuD;AACvD,wDAAuD;AACvD,4CAI4B;AAC5B,uFAG0D;AAC1D,iEAA4F;AAE/E,QAAA,8CAA8C,GACzD,IAAA,qBAAgB,EAAC,UAAU,CAAC,CAAC;AAExB,MAAM,oCAAoC,GAAG,CAClD,IAAU,EACV,OAAmD,EACvB,EAAE;;IAC9B,MAAM,aAAa,GAAG,IAAA,wCAAmC,EACvD,IAAI,EACJ,OAAO,CAAC,aAAa,CACtB,CAAC;IACF,MAAM,aAAa,GAAG,IAAA,wCAAmC,EACvD,IAAI,EACJ,OAAO,CAAC,aAAa,CACtB,CAAC;IAEF,MAAM,cAAc,GAAG,OAAO,CAAC,eAAe,CAAC;IAC/C,MAAM,YAAY,GAAG,OAAO,CAAC,eAAe,CAAC;IAE7C,IAAI,CAAC,cAAc,IAAI,CAAC,YAAY,EAAE,CAAC;QACrC,MAAM,IAAI,KAAK,CACb,mGAAmG,CACpG,CAAC;IACJ,CAAC;IAED,MAAM,gBAAgB,GAAG,MAAA,YAAY,CAAC,IAAI,mCAAI,YAAY,CAAC;IAC3D,MAAM,kBAAkB,GAAG,YAAY,CAAC,EAAY,CAAC;IACrD,MAAM,kBAAkB,GAAG,IAAA,iBAAS,EAAC,kBAAkB,CAAC,CAAC;IACzD,MAAM,aAAa,GAAG,MAAA,YAAY,CAAC,IAAI,mCAAI,IAAI,CAAC;IAEhD,MAAM,QAAQ,GAAG,IAAA,uBAAW,EAAC,IAAI,CAAC,CAAC;IAEnC,uDAAuD;IACvD,MAAM,IAAA,yDAAsC,EAAC,IAAI,CAAC,CAAC;IAEnD,wDAAwD;IACxD,IAAA,sBAAa,EACX,IAAI,EACJ,IAAA,0BAAiB,EAAC,SAAS,EAAE,OAAO,EAAE,kBAAkB,EAAE,KAAK,CAAC,EAChE,IAAA,0BAAiB,EAAC,+CAA4B,EAAE,KAAK,EAAE,KAAK,CAAC,EAC7D;QACE,kBAAkB;QAClB,kBAAkB;QAClB,aAAa;KACd,EACD,EAAE,iBAAiB,EAAE,0BAAiB,CAAC,YAAY,EAAE,CACtD,CAAC;IAEF,4BAA4B;IAC5B,IAAA,mBAAa,EACX,IAAI,EACJ,IAAA,0BAAiB,EAAC,+CAA4B,EAAE,KAAK,EAAE,UAAU,CAAC,EAClE,SAAS,kBAAkB,YAAY,CACxC,CAAC;IAEF,6CAA6C;IAC7C,MAAM,cAAc,GAAG,IAAA,0BAAiB,EACtC,aAAa,CAAC,IAAI,EAClB,MAAA,cAAc,CAAC,IAAI,mCAAI,KAAK,CAC7B,CAAC;IACF,MAAM,aAAa,GAAG,IAAA,0BAAiB,EAAC,cAAc,EAAE,UAAU,CAAC,CAAC;IAEpE,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;QAC/B,MAAM,eAAe,GAAG,GAAG,kBAAkB,QAAQ,CAAC;QACtD,MAAM,aAAa,GACjB,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAE3E,4BAA4B;QAC5B,IAAA,2BAAqB,EACnB,IAAI,EACJ,aAAa,EACb,CAAC,eAAe,CAAC,EACjB,IAAI,QAAQ,mBAAmB,CAChC,CAAC;QAEF,6EAA6E;QAC7E,MAAM,uBAAuB,GAAG,oBAAO,CAAC,uBAAuB,CAC7D,SAAS,EACT,oBAAO,CAAC,6BAA6B,CACnC;YACE,oBAAO,CAAC,yBAAyB,CAC/B,aAAa,EACb,SAAS,EACT,SAAS,EACT,oBAAO,CAAC,qBAAqB,CAC3B,oBAAO,CAAC,oBAAoB,CAC1B,oBAAO,CAAC,8BAA8B,CACpC,oBAAO,CAAC,gBAAgB,CAAC,eAAe,CAAC,EACzC,QAAQ,CACT,EACD,SAAS,EACT,CAAC,oBAAO,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC,CACxC,CACF,CACF;SACF,EACD,CAAC,CACF,CACF,CAAC;QAEF,8EAA8E;QAC9E,uDAAuD;QACvD,IAAA,qBAAe,EACb,IAAI,EACJ,aAAa,EACb,2DAA2D,EAC3D,CAAC,IAAI,EAAE,EAAE;YACP,MAAM,KAAK,GAAG,IAAqB,CAAC;YAEpC,+BAA+B;YAC/B,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,eAAe,CAAC;gBAAE,OAAO,IAAI,CAAC;YAE3D,6EAA6E;YAC7E,IAAI,CAAC,oBAAE,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5B,OAAO,oBAAO,CAAC,mBAAmB,CAChC,KAAK,EACL,KAAK,CAAC,SAAS,EACf,KAAK,CAAC,cAAc,EACpB,KAAK,CAAC,UAAU,EAChB,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,sBAAsB,EAC5B,oBAAO,CAAC,WAAW,CACjB;oBACE,uBAAuB;oBACvB,oBAAO,CAAC,qBAAqB,CAC3B,KAAK,CAAC,IAAkB,CACzB;iBACF,EACD,IAAI,CACL,CACF,CAAC;YACJ,CAAC;YAED,+EAA+E;YAC/E,MAAM,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC;YACrB,MAAM,mBAAmB,GAAG,CAAC,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CACvD,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,CACnC,CAAC;YACF,IAAI,mBAAmB,KAAK,CAAC,CAAC;gBAAE,OAAO,IAAI,CAAC;YAE5C,MAAM,aAAa,GAAG,CAAC,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC;YACxC,aAAa,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,EAAE,uBAAuB,CAAC,CAAC;YACtE,OAAO,oBAAO,CAAC,mBAAmB,CAChC,KAAK,EACL,KAAK,CAAC,SAAS,EACf,KAAK,CAAC,cAAc,EACpB,KAAK,CAAC,UAAU,EAChB,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,sBAAsB,EAC5B,oBAAO,CAAC,WAAW,CAAC,CAAC,EAAE,aAAa,CAAC,CACtC,CAAC;QACJ,CAAC,CACF,CAAC;QAEF,qEAAqE;QACrE,IAAA,qBAAe,EACb,IAAI,EACJ,aAAa,EACb,6HAA6H,EAC7H,CAAC,IAAI,EAAE,EAAE;YACP,MAAM,GAAG,GAAG,IAA8B,CAAC;YAC3C,yBAAyB;YACzB,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC;gBAAE,OAAO,IAAI,CAAC;YAEvD,OAAO,oBAAO,CAAC,4BAA4B,CAAC,GAAG,EAAE;gBAC/C,oBAAO,CAAC,gBAAgB,CAAC,aAAa,CAAC;gBACvC,GAAG,GAAG,CAAC,QAAQ;aAChB,CAAC,CAAC;QACL,CAAC,CACF,CAAC;IACJ,CAAC;IAED,+BAA+B;IAC/B,MAAM,SAAS,GAAG,MAAA,cAAc,CAAC,IAAI,mCAAI,OAAO,CAAC;IACjD,MAAM,oBAAoB,GAAG,GAAG,SAAS,cAAc,CAAC;IACxD,MAAM,uBAAuB,GAAG,GAAG,gBAAgB,cAAc,CAAC;IAElE,IAAI,MAAA,aAAa,CAAC,OAAO,0CAAG,oBAAoB,CAAC,EAAE,CAAC;QAClD,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC;QAC3D,MAAM,CAAC,SAAS,GAAG;YACjB,GAAG,CAAC,MAAA,MAAM,CAAC,SAAS,mCAAI,EAAE,CAAC;YAC3B;gBACE,QAAQ,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC;gBAC9B,MAAM,EAAE,uBAAuB;aAChC;SACF,CAAC;QACF,IAAA,mCAA0B,EAAC,IAAI,EAAE,aAAa,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;IACtE,CAAC;IAED,sBAAsB;IACtB,IAAA,qCAA4B,EAC1B,IAAI,EACJ,IAAA,uBAAY,EAAC;QACX,2BAA2B;QAC3B,mCAAmC;QACnC,+BAA+B;QAC/B,WAAW;QACX,+BAA+B;KAChC,CAAC,EACF,EAAE,CACH,CAAC;IAEF,MAAM,IAAA,yCAA+B,EAAC,IAAI,EAAE;QAC1C,sDAA8C;KAC/C,CAAC,CAAC;IAEH,MAAM,IAAA,6BAAoB,EAAC,IAAI,CAAC,CAAC;IACjC,OAAO,GAAG,EAAE;QACV,IAAA,4BAAmB,EAAC,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC;AACJ,CAAC,CAAA,CAAC;AA/MW,QAAA,oCAAoC,wCA+M/C;AAEF,kBAAe,4CAAoC,CAAC"}

package/src/ts/strands-agent/mcp-connection/schema.d.ts

@@ -0,0 +1,15 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { ComponentMetadata } from '../../../utils/nx';
+
+/**
+ * TypeScript types for options defined in schema.json
+ */
+export interface TsStrandsAgentMcpConnectionGeneratorSchema {
+  sourceProject: string;
+  targetProject: string;
+  sourceComponent?: ComponentMetadata;
+  targetComponent?: ComponentMetadata;
+}