@aws/nx-plugin 1.0.0-rc.53 → 1.0.0-rc.55

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

@@ -2,17 +2,43 @@
  * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
  * SPDX-License-Identifier: Apache-2.0
  */ import { generateFiles, joinPathFragments, OverwriteStrategy, updateProjectConfiguration } from "@nx/devkit";
-import { AGENT_CONNECTION_PROJECT_DIR, addTypeScriptClientToAgent, addTypeScriptCoreClient, ensureTypeScriptAgentConnectionProject } from "../../../utils/agent-connection/agent-connection.js";
+import { addTsDependencies } from "../../../utils/add-dependencies.js";
+import { AGENT_CONNECTION_DEPENDENCIES, AGENT_CONNECTION_PROJECT_DIR, addTypeScriptClientToAgent, addTypeScriptCoreClient, ensureTypeScriptAgentConnectionProject } from "../../../utils/agent-connection/agent-connection.js";
 import { addDestructuredImport, addStarExport } from "../../../utils/ast.js";
-import { addDependenciesToPackageJson } from "../../../utils/dependencies.js";
+import { declareDependencies, ownedElsewhere } from "../../../utils/declared-dependencies.js";
 import { formatFilesInSubtree } from "../../../utils/format.js";
 import { installDependencies } from "../../../utils/install.js";
 import { addGeneratorMetricsIfApplicable } from "../../../utils/metrics.js";
 import { esmVars } from "../../../utils/module-format.js";
 import { kebabCase } from "../../../utils/names.js";
 import { getNpmScope } from "../../../utils/npm-scope.js";
-import { addDependencyToTargetIfNotPresent, getGeneratorInfo, readProjectConfigurationUnqualified } from "../../../utils/nx.js";
-import { withVersions } from "../../../utils/versions.js";
+import { addComponentGeneratorMetadata, addDependencyToTargetIfNotPresent, getGeneratorInfo, readProjectConfigurationUnqualified } from "../../../utils/nx.js";
+import { toProjectRelativePath } from "../../../utils/paths.js";
+// The MCP core client + vended client need these whatever the connection's
+// options, so no entry is conditional.
+export const DEPENDENCIES = declareDependencies()({
+    ts: [
+        {
+            name: '@modelcontextprotocol/sdk'
+        },
+        {
+            name: '@strands-agents/sdk'
+        },
+        {
+            name: '@aws-lambda-powertools/parameters'
+        },
+        {
+            name: '@aws-sdk/client-appconfigdata'
+        },
+        {
+            name: 'aws4fetch'
+        },
+        {
+            name: '@aws-sdk/credential-providers'
+        },
+        ...ownedElsewhere(AGENT_CONNECTION_DEPENDENCIES)
+    ]
+});
 export const TS_AGENT_MCP_CONNECTION_GENERATOR_INFO = getGeneratorInfo(import.meta.filename);
 export const tsAgentMcpConnectionGenerator = async (tree, options)=>{
     const sourceProject = readProjectConfigurationUnqualified(tree, options.sourceProject);
@@ -31,8 +57,8 @@ export const tsAgentMcpConnectionGenerator = async (tree, options)=>{
     const mcpServerPort = mcpComponent.port ?? 8000;
     const npmScope = getNpmScope(tree);
     // 1. Ensure the shared agent-connection project exists + has the MCP core client
-    await ensureTypeScriptAgentConnectionProject(tree);
-    await addTypeScriptCoreClient(tree, 'mcp');
+    await ensureTypeScriptAgentConnectionProject(tree, DEPENDENCIES);
+    await addTypeScriptCoreClient(tree, 'mcp', DEPENDENCIES);
     // 2. Generate the per-connection <Name>Client into app/
     generateFiles(tree, joinPathFragments(import.meta.dirname, 'files', 'agent-connection', 'app'), joinPathFragments(AGENT_CONNECTION_PROJECT_DIR, 'src', 'app'), {
         mcpServerKebabCase,
@@ -70,14 +96,15 @@ export const tsAgentMcpConnectionGenerator = async (tree, options)=>{
         updateProjectConfiguration(tree, sourceProject.name, sourceProject);
     }
     // 5. Add dependencies required by the MCP core client + vended client
-    addDependenciesToPackageJson(tree, withVersions([
-        '@modelcontextprotocol/sdk',
-        '@strands-agents/sdk',
-        '@aws-lambda-powertools/parameters',
-        '@aws-sdk/client-appconfigdata',
-        'aws4fetch',
-        '@aws-sdk/credential-providers'
-    ]), {}, joinPathFragments(sourceProject.root, 'package.json'));
+    addTsDependencies(tree, DEPENDENCIES, {
+        projectRoot: sourceProject.root
+    });
+    // Recorded so the version sync knows this connection's dependencies are ours.
+    addComponentGeneratorMetadata(tree, sourceProject.name, TS_AGENT_MCP_CONNECTION_GENERATOR_INFO, toProjectRelativePath(sourceProject, agentFilePath), mcpServerClassName, // The source component this connection is made from, so the pair is
+    // identifiable rather than just the two projects.
+    {
+        sourcePath: options.sourceComponent?.path
+    });
     await addGeneratorMetricsIfApplicable(tree, [
         TS_AGENT_MCP_CONNECTION_GENERATOR_INFO
     ]);

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/ts/agent/mcp-connection/generator.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nimport {\n  type GeneratorCallback,\n  generateFiles,\n  joinPathFragments,\n  OverwriteStrategy,\n  type Tree,\n  updateProjectConfiguration,\n} from '@nx/devkit';\nimport {\n  AGENT_CONNECTION_PROJECT_DIR,\n  addTypeScriptClientToAgent,\n  addTypeScriptCoreClient,\n  ensureTypeScriptAgentConnectionProject,\n} from '../../../utils/agent-connection/agent-connection';\nimport { addDestructuredImport, addStarExport } from '../../../utils/ast';\nimport { addDependenciesToPackageJson } from '../../../utils/dependencies';\nimport { formatFilesInSubtree } from '../../../utils/format';\nimport { installDependencies } from '../../../utils/install';\nimport { addGeneratorMetricsIfApplicable } from '../../../utils/metrics';\nimport { esmVars } from '../../../utils/module-format';\nimport { kebabCase } from '../../../utils/names';\nimport { getNpmScope } from '../../../utils/npm-scope';\nimport {\n  addDependencyToTargetIfNotPresent,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  readProjectConfigurationUnqualified,\n} from '../../../utils/nx';\nimport { withVersions } from '../../../utils/versions';\nimport type { TsAgentMcpConnectionGeneratorSchema } from './schema';\n\nexport const TS_AGENT_MCP_CONNECTION_GENERATOR_INFO: NxGeneratorInfo =\n  getGeneratorInfo(import.meta.filename);\n\nexport const tsAgentMcpConnectionGenerator = async (\n  tree: Tree,\n  options: TsAgentMcpConnectionGeneratorSchema,\n): Promise<GeneratorCallback> => {\n  const sourceProject = readProjectConfigurationUnqualified(\n    tree,\n    options.sourceProject,\n  );\n  const targetProject = readProjectConfigurationUnqualified(\n    tree,\n    options.targetProject,\n  );\n\n  const agentComponent = options.sourceComponent;\n  const mcpComponent = options.targetComponent;\n\n  if (!agentComponent || !mcpComponent) {\n    throw new Error(\n      'Both sourceComponent and targetComponent must be provided for ts#agent -> mcp connections',\n    );\n  }\n\n  if (mcpComponent.auth && mcpComponent.auth.toLowerCase() !== 'iam') {\n    throw new Error(\n      `MCP server connection currently only supports IAM authentication, but '${mcpComponent.name}' uses '${mcpComponent.auth}' authentication.`,\n    );\n  }\n\n  const mcpComponentName = mcpComponent.name ?? 'mcp-server';\n  const mcpServerClassName = mcpComponent.rc as string;\n  const mcpServerKebabCase = kebabCase(mcpServerClassName);\n  const mcpServerPort = mcpComponent.port ?? 8000;\n\n  const npmScope = getNpmScope(tree);\n\n  // 1. Ensure the shared agent-connection project exists + has the MCP core client\n  await ensureTypeScriptAgentConnectionProject(tree);\n  await addTypeScriptCoreClient(tree, 'mcp');\n\n  // 2. Generate the per-connection <Name>Client into app/\n  generateFiles(\n    tree,\n    joinPathFragments(import.meta.dirname, 'files', 'agent-connection', 'app'),\n    joinPathFragments(AGENT_CONNECTION_PROJECT_DIR, 'src', 'app'),\n    {\n      mcpServerKebabCase,\n      mcpServerClassName,\n      mcpServerPort,\n      ...esmVars(tree),\n    },\n    { overwriteStrategy: OverwriteStrategy.KeepExisting },\n  );\n\n  // Add re-export to index.ts\n  await addStarExport(\n    tree,\n    joinPathFragments(AGENT_CONNECTION_PROJECT_DIR, 'src', 'index.ts'),\n    `./app/${mcpServerKebabCase}-client-strands.js`,\n  );\n\n  // 3. AST transform agent.ts to add MCP tools\n  const agentSourceDir = joinPathFragments(\n    sourceProject.root,\n    agentComponent.path ?? 'src',\n  );\n  const agentFilePath = joinPathFragments(agentSourceDir, 'agent.ts');\n\n  if (tree.exists(agentFilePath)) {\n    const clientClassName = `${mcpServerClassName}ClientStrands`;\n    const clientVarName =\n      mcpServerClassName.charAt(0).toLowerCase() + mcpServerClassName.slice(1);\n\n    // Add import for the client\n    await addDestructuredImport(\n      tree,\n      agentFilePath,\n      [clientClassName],\n      `@${npmScope}/agent-connection`,\n    );\n\n    await addTypeScriptClientToAgent(\n      tree,\n      agentFilePath,\n      clientClassName,\n      clientVarName,\n    );\n  }\n\n  // 4. Set up dev target\n  const agentName = agentComponent.name ?? 'agent';\n  const devTargetName = `${agentName}-dev`;\n  const mcpDevTargetName = `${mcpComponentName}-dev`;\n\n  if (sourceProject.targets?.[devTargetName]) {\n    addDependencyToTargetIfNotPresent(sourceProject, devTargetName, {\n      projects: [targetProject.name],\n      target: mcpDevTargetName,\n    });\n    updateProjectConfiguration(tree, sourceProject.name, sourceProject);\n  }\n\n  // 5. Add dependencies required by the MCP core client + vended client\n  addDependenciesToPackageJson(\n    tree,\n    withVersions([\n      '@modelcontextprotocol/sdk',\n      '@strands-agents/sdk',\n      '@aws-lambda-powertools/parameters',\n      '@aws-sdk/client-appconfigdata',\n      'aws4fetch',\n      '@aws-sdk/credential-providers',\n    ]),\n    {},\n    joinPathFragments(sourceProject.root, 'package.json'),\n  );\n\n  await addGeneratorMetricsIfApplicable(tree, [\n    TS_AGENT_MCP_CONNECTION_GENERATOR_INFO,\n  ]);\n\n  await formatFilesInSubtree(tree);\n  return () =>\n    installDependencies(tree, options.preferInstallDependencies, {\n      languages: ['typescript'],\n    });\n};\n\nexport default tsAgentMcpConnectionGenerator;\n"],"names":["generateFiles","joinPathFragments","OverwriteStrategy","updateProjectConfiguration","AGENT_CONNECTION_PROJECT_DIR","addTypeScriptClientToAgent","addTypeScriptCoreClient","ensureTypeScriptAgentConnectionProject","addDestructuredImport","addStarExport","addDependenciesToPackageJson","formatFilesInSubtree","installDependencies","addGeneratorMetricsIfApplicable","esmVars","kebabCase","getNpmScope","addDependencyToTargetIfNotPresent","getGeneratorInfo","readProjectConfigurationUnqualified","withVersions","TS_AGENT_MCP_CONNECTION_GENERATOR_INFO","filename","tsAgentMcpConnectionGenerator","tree","options","sourceProject","targetProject","agentComponent","sourceComponent","mcpComponent","targetComponent","Error","auth","toLowerCase","name","mcpComponentName","mcpServerClassName","rc","mcpServerKebabCase","mcpServerPort","port","npmScope","dirname","overwriteStrategy","KeepExisting","agentSourceDir","root","path","agentFilePath","exists","clientClassName","clientVarName","charAt","slice","agentName","devTargetName","mcpDevTargetName","targets","projects","target","preferInstallDependencies","languages"],"mappings":"AAAA;;;CAGC,GACD,SAEEA,aAAa,EACbC,iBAAiB,EACjBC,iBAAiB,EAEjBC,0BAA0B,QACrB,aAAa;AACpB,SACEC,4BAA4B,EAC5BC,0BAA0B,EAC1BC,uBAAuB,EACvBC,sCAAsC,QACjC,sDAAmD;AAC1D,SAASC,qBAAqB,EAAEC,aAAa,QAAQ,wBAAqB;AAC1E,SAASC,4BAA4B,QAAQ,iCAA8B;AAC3E,SAASC,oBAAoB,QAAQ,2BAAwB;AAC7D,SAASC,mBAAmB,QAAQ,4BAAyB;AAC7D,SAASC,+BAA+B,QAAQ,4BAAyB;AACzE,SAASC,OAAO,QAAQ,kCAA+B;AACvD,SAASC,SAAS,QAAQ,0BAAuB;AACjD,SAASC,WAAW,QAAQ,8BAA2B;AACvD,SACEC,iCAAiC,EACjCC,gBAAgB,EAEhBC,mCAAmC,QAC9B,uBAAoB;AAC3B,SAASC,YAAY,QAAQ,6BAA0B;AAGvD,OAAO,MAAMC,yCACXH,iBAAiB,YAAYI,QAAQ,EAAE;AAEzC,OAAO,MAAMC,gCAAgC,OAC3CC,MACAC;IAEA,MAAMC,gBAAgBP,oCACpBK,MACAC,QAAQC,aAAa;IAEvB,MAAMC,gBAAgBR,oCACpBK,MACAC,QAAQE,aAAa;IAGvB,MAAMC,iBAAiBH,QAAQI,eAAe;IAC9C,MAAMC,eAAeL,QAAQM,eAAe;IAE5C,IAAI,CAACH,kBAAkB,CAACE,cAAc;QACpC,MAAM,IAAIE,MACR;IAEJ;IAEA,IAAIF,aAAaG,IAAI,IAAIH,aAAaG,IAAI,CAACC,WAAW,OAAO,OAAO;QAClE,MAAM,IAAIF,MACR,CAAC,uEAAuE,EAAEF,aAAaK,IAAI,CAAC,QAAQ,EAAEL,aAAaG,IAAI,CAAC,iBAAiB,CAAC;IAE9I;IAEA,MAAMG,mBAAmBN,aAAaK,IAAI,IAAI;IAC9C,MAAME,qBAAqBP,aAAaQ,EAAE;IAC1C,MAAMC,qBAAqBxB,UAAUsB;IACrC,MAAMG,gBAAgBV,aAAaW,IAAI,IAAI;IAE3C,MAAMC,WAAW1B,YAAYQ;IAE7B,iFAAiF;IACjF,MAAMjB,uCAAuCiB;IAC7C,MAAMlB,wBAAwBkB,MAAM;IAEpC,wDAAwD;IACxDxB,cACEwB,MACAvB,kBAAkB,YAAY0C,OAAO,EAAE,SAAS,oBAAoB,QACpE1C,kBAAkBG,8BAA8B,OAAO,QACvD;QACEmC;QACAF;QACAG;QACA,GAAG1B,QAAQU,KAAK;IAClB,GACA;QAAEoB,mBAAmB1C,kBAAkB2C,YAAY;IAAC;IAGtD,4BAA4B;IAC5B,MAAMpC,cACJe,MACAvB,kBAAkBG,8BAA8B,OAAO,aACvD,CAAC,MAAM,EAAEmC,mBAAmB,kBAAkB,CAAC;IAGjD,6CAA6C;IAC7C,MAAMO,iBAAiB7C,kBACrByB,cAAcqB,IAAI,EAClBnB,eAAeoB,IAAI,IAAI;IAEzB,MAAMC,gBAAgBhD,kBAAkB6C,gBAAgB;IAExD,IAAItB,KAAK0B,MAAM,CAACD,gBAAgB;QAC9B,MAAME,kBAAkB,GAAGd,mBAAmB,aAAa,CAAC;QAC5D,MAAMe,gBACJf,mBAAmBgB,MAAM,CAAC,GAAGnB,WAAW,KAAKG,mBAAmBiB,KAAK,CAAC;QAExE,4BAA4B;QAC5B,MAAM9C,sBACJgB,MACAyB,eACA;YAACE;SAAgB,EACjB,CAAC,CAAC,EAAET,SAAS,iBAAiB,CAAC;QAGjC,MAAMrC,2BACJmB,MACAyB,eACAE,iBACAC;IAEJ;IAEA,uBAAuB;IACvB,MAAMG,YAAY3B,eAAeO,IAAI,IAAI;IACzC,MAAMqB,gBAAgB,GAAGD,UAAU,IAAI,CAAC;IACxC,MAAME,mBAAmB,GAAGrB,iBAAiB,IAAI,CAAC;IAElD,IAAIV,cAAcgC,OAAO,EAAE,CAACF,cAAc,EAAE;QAC1CvC,kCAAkCS,eAAe8B,eAAe;YAC9DG,UAAU;gBAAChC,cAAcQ,IAAI;aAAC;YAC9ByB,QAAQH;QACV;QACAtD,2BAA2BqB,MAAME,cAAcS,IAAI,EAAET;IACvD;IAEA,sEAAsE;IACtEhB,6BACEc,MACAJ,aAAa;QACX;QACA;QACA;QACA;QACA;QACA;KACD,GACD,CAAC,GACDnB,kBAAkByB,cAAcqB,IAAI,EAAE;IAGxC,MAAMlC,gCAAgCW,MAAM;QAC1CH;KACD;IAED,MAAMV,qBAAqBa;IAC3B,OAAO,IACLZ,oBAAoBY,MAAMC,QAAQoC,yBAAyB,EAAE;YAC3DC,WAAW;gBAAC;aAAa;QAC3B;AACJ,EAAE;AAEF,eAAevC,8BAA8B"}
+{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/ts/agent/mcp-connection/generator.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nimport {\n  type GeneratorCallback,\n  generateFiles,\n  joinPathFragments,\n  OverwriteStrategy,\n  type Tree,\n  updateProjectConfiguration,\n} from '@nx/devkit';\nimport { addTsDependencies } from '../../../utils/add-dependencies';\nimport {\n  AGENT_CONNECTION_DEPENDENCIES,\n  AGENT_CONNECTION_PROJECT_DIR,\n  addTypeScriptClientToAgent,\n  addTypeScriptCoreClient,\n  ensureTypeScriptAgentConnectionProject,\n} from '../../../utils/agent-connection/agent-connection';\nimport { addDestructuredImport, addStarExport } from '../../../utils/ast';\nimport {\n  declareDependencies,\n  ownedElsewhere,\n} from '../../../utils/declared-dependencies';\nimport { formatFilesInSubtree } from '../../../utils/format';\nimport { installDependencies } from '../../../utils/install';\nimport { addGeneratorMetricsIfApplicable } from '../../../utils/metrics';\nimport { esmVars } from '../../../utils/module-format';\nimport { kebabCase } from '../../../utils/names';\nimport { getNpmScope } from '../../../utils/npm-scope';\nimport {\n  addComponentGeneratorMetadata,\n  addDependencyToTargetIfNotPresent,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  readProjectConfigurationUnqualified,\n} from '../../../utils/nx';\nimport { toProjectRelativePath } from '../../../utils/paths';\nimport type { TsAgentMcpConnectionGeneratorSchema } from './schema';\n\n// The MCP core client + vended client need these whatever the connection's\n// options, so no entry is conditional.\nexport const DEPENDENCIES = declareDependencies()({\n  ts: [\n    { name: '@modelcontextprotocol/sdk' },\n    { name: '@strands-agents/sdk' },\n    { name: '@aws-lambda-powertools/parameters' },\n    { name: '@aws-sdk/client-appconfigdata' },\n    { name: 'aws4fetch' },\n    { name: '@aws-sdk/credential-providers' },\n    ...ownedElsewhere(AGENT_CONNECTION_DEPENDENCIES),\n  ],\n});\n\nexport const TS_AGENT_MCP_CONNECTION_GENERATOR_INFO: NxGeneratorInfo =\n  getGeneratorInfo(import.meta.filename);\n\nexport const tsAgentMcpConnectionGenerator = async (\n  tree: Tree,\n  options: TsAgentMcpConnectionGeneratorSchema,\n): Promise<GeneratorCallback> => {\n  const sourceProject = readProjectConfigurationUnqualified(\n    tree,\n    options.sourceProject,\n  );\n  const targetProject = readProjectConfigurationUnqualified(\n    tree,\n    options.targetProject,\n  );\n\n  const agentComponent = options.sourceComponent;\n  const mcpComponent = options.targetComponent;\n\n  if (!agentComponent || !mcpComponent) {\n    throw new Error(\n      'Both sourceComponent and targetComponent must be provided for ts#agent -> mcp connections',\n    );\n  }\n\n  if (mcpComponent.auth && mcpComponent.auth.toLowerCase() !== 'iam') {\n    throw new Error(\n      `MCP server connection currently only supports IAM authentication, but '${mcpComponent.name}' uses '${mcpComponent.auth}' authentication.`,\n    );\n  }\n\n  const mcpComponentName = mcpComponent.name ?? 'mcp-server';\n  const mcpServerClassName = mcpComponent.rc as string;\n  const mcpServerKebabCase = kebabCase(mcpServerClassName);\n  const mcpServerPort = mcpComponent.port ?? 8000;\n\n  const npmScope = getNpmScope(tree);\n\n  // 1. Ensure the shared agent-connection project exists + has the MCP core client\n  await ensureTypeScriptAgentConnectionProject(tree, DEPENDENCIES);\n  await addTypeScriptCoreClient(tree, 'mcp', DEPENDENCIES);\n\n  // 2. Generate the per-connection <Name>Client into app/\n  generateFiles(\n    tree,\n    joinPathFragments(import.meta.dirname, 'files', 'agent-connection', 'app'),\n    joinPathFragments(AGENT_CONNECTION_PROJECT_DIR, 'src', 'app'),\n    {\n      mcpServerKebabCase,\n      mcpServerClassName,\n      mcpServerPort,\n      ...esmVars(tree),\n    },\n    { overwriteStrategy: OverwriteStrategy.KeepExisting },\n  );\n\n  // Add re-export to index.ts\n  await addStarExport(\n    tree,\n    joinPathFragments(AGENT_CONNECTION_PROJECT_DIR, 'src', 'index.ts'),\n    `./app/${mcpServerKebabCase}-client-strands.js`,\n  );\n\n  // 3. AST transform agent.ts to add MCP tools\n  const agentSourceDir = joinPathFragments(\n    sourceProject.root,\n    agentComponent.path ?? 'src',\n  );\n  const agentFilePath = joinPathFragments(agentSourceDir, 'agent.ts');\n\n  if (tree.exists(agentFilePath)) {\n    const clientClassName = `${mcpServerClassName}ClientStrands`;\n    const clientVarName =\n      mcpServerClassName.charAt(0).toLowerCase() + mcpServerClassName.slice(1);\n\n    // Add import for the client\n    await addDestructuredImport(\n      tree,\n      agentFilePath,\n      [clientClassName],\n      `@${npmScope}/agent-connection`,\n    );\n\n    await addTypeScriptClientToAgent(\n      tree,\n      agentFilePath,\n      clientClassName,\n      clientVarName,\n    );\n  }\n\n  // 4. Set up dev target\n  const agentName = agentComponent.name ?? 'agent';\n  const devTargetName = `${agentName}-dev`;\n  const mcpDevTargetName = `${mcpComponentName}-dev`;\n\n  if (sourceProject.targets?.[devTargetName]) {\n    addDependencyToTargetIfNotPresent(sourceProject, devTargetName, {\n      projects: [targetProject.name],\n      target: mcpDevTargetName,\n    });\n    updateProjectConfiguration(tree, sourceProject.name, sourceProject);\n  }\n\n  // 5. Add dependencies required by the MCP core client + vended client\n  addTsDependencies(tree, DEPENDENCIES, { projectRoot: sourceProject.root });\n\n  // Recorded so the version sync knows this connection's dependencies are ours.\n  addComponentGeneratorMetadata(\n    tree,\n    sourceProject.name,\n    TS_AGENT_MCP_CONNECTION_GENERATOR_INFO,\n    toProjectRelativePath(sourceProject, agentFilePath),\n    mcpServerClassName,\n    // The source component this connection is made from, so the pair is\n    // identifiable rather than just the two projects.\n    { sourcePath: options.sourceComponent?.path },\n  );\n\n  await addGeneratorMetricsIfApplicable(tree, [\n    TS_AGENT_MCP_CONNECTION_GENERATOR_INFO,\n  ]);\n\n  await formatFilesInSubtree(tree);\n  return () =>\n    installDependencies(tree, options.preferInstallDependencies, {\n      languages: ['typescript'],\n    });\n};\n\nexport default tsAgentMcpConnectionGenerator;\n"],"names":["generateFiles","joinPathFragments","OverwriteStrategy","updateProjectConfiguration","addTsDependencies","AGENT_CONNECTION_DEPENDENCIES","AGENT_CONNECTION_PROJECT_DIR","addTypeScriptClientToAgent","addTypeScriptCoreClient","ensureTypeScriptAgentConnectionProject","addDestructuredImport","addStarExport","declareDependencies","ownedElsewhere","formatFilesInSubtree","installDependencies","addGeneratorMetricsIfApplicable","esmVars","kebabCase","getNpmScope","addComponentGeneratorMetadata","addDependencyToTargetIfNotPresent","getGeneratorInfo","readProjectConfigurationUnqualified","toProjectRelativePath","DEPENDENCIES","ts","name","TS_AGENT_MCP_CONNECTION_GENERATOR_INFO","filename","tsAgentMcpConnectionGenerator","tree","options","sourceProject","targetProject","agentComponent","sourceComponent","mcpComponent","targetComponent","Error","auth","toLowerCase","mcpComponentName","mcpServerClassName","rc","mcpServerKebabCase","mcpServerPort","port","npmScope","dirname","overwriteStrategy","KeepExisting","agentSourceDir","root","path","agentFilePath","exists","clientClassName","clientVarName","charAt","slice","agentName","devTargetName","mcpDevTargetName","targets","projects","target","projectRoot","sourcePath","preferInstallDependencies","languages"],"mappings":"AAAA;;;CAGC,GACD,SAEEA,aAAa,EACbC,iBAAiB,EACjBC,iBAAiB,EAEjBC,0BAA0B,QACrB,aAAa;AACpB,SAASC,iBAAiB,QAAQ,qCAAkC;AACpE,SACEC,6BAA6B,EAC7BC,4BAA4B,EAC5BC,0BAA0B,EAC1BC,uBAAuB,EACvBC,sCAAsC,QACjC,sDAAmD;AAC1D,SAASC,qBAAqB,EAAEC,aAAa,QAAQ,wBAAqB;AAC1E,SACEC,mBAAmB,EACnBC,cAAc,QACT,0CAAuC;AAC9C,SAASC,oBAAoB,QAAQ,2BAAwB;AAC7D,SAASC,mBAAmB,QAAQ,4BAAyB;AAC7D,SAASC,+BAA+B,QAAQ,4BAAyB;AACzE,SAASC,OAAO,QAAQ,kCAA+B;AACvD,SAASC,SAAS,QAAQ,0BAAuB;AACjD,SAASC,WAAW,QAAQ,8BAA2B;AACvD,SACEC,6BAA6B,EAC7BC,iCAAiC,EACjCC,gBAAgB,EAEhBC,mCAAmC,QAC9B,uBAAoB;AAC3B,SAASC,qBAAqB,QAAQ,0BAAuB;AAG7D,2EAA2E;AAC3E,uCAAuC;AACvC,OAAO,MAAMC,eAAeb,sBAAsB;IAChDc,IAAI;QACF;YAAEC,MAAM;QAA4B;QACpC;YAAEA,MAAM;QAAsB;QAC9B;YAAEA,MAAM;QAAoC;QAC5C;YAAEA,MAAM;QAAgC;QACxC;YAAEA,MAAM;QAAY;QACpB;YAAEA,MAAM;QAAgC;WACrCd,eAAeR;KACnB;AACH,GAAG;AAEH,OAAO,MAAMuB,yCACXN,iBAAiB,YAAYO,QAAQ,EAAE;AAEzC,OAAO,MAAMC,gCAAgC,OAC3CC,MACAC;IAEA,MAAMC,gBAAgBV,oCACpBQ,MACAC,QAAQC,aAAa;IAEvB,MAAMC,gBAAgBX,oCACpBQ,MACAC,QAAQE,aAAa;IAGvB,MAAMC,iBAAiBH,QAAQI,eAAe;IAC9C,MAAMC,eAAeL,QAAQM,eAAe;IAE5C,IAAI,CAACH,kBAAkB,CAACE,cAAc;QACpC,MAAM,IAAIE,MACR;IAEJ;IAEA,IAAIF,aAAaG,IAAI,IAAIH,aAAaG,IAAI,CAACC,WAAW,OAAO,OAAO;QAClE,MAAM,IAAIF,MACR,CAAC,uEAAuE,EAAEF,aAAaV,IAAI,CAAC,QAAQ,EAAEU,aAAaG,IAAI,CAAC,iBAAiB,CAAC;IAE9I;IAEA,MAAME,mBAAmBL,aAAaV,IAAI,IAAI;IAC9C,MAAMgB,qBAAqBN,aAAaO,EAAE;IAC1C,MAAMC,qBAAqB3B,UAAUyB;IACrC,MAAMG,gBAAgBT,aAAaU,IAAI,IAAI;IAE3C,MAAMC,WAAW7B,YAAYY;IAE7B,iFAAiF;IACjF,MAAMtB,uCAAuCsB,MAAMN;IACnD,MAAMjB,wBAAwBuB,MAAM,OAAON;IAE3C,wDAAwD;IACxDzB,cACE+B,MACA9B,kBAAkB,YAAYgD,OAAO,EAAE,SAAS,oBAAoB,QACpEhD,kBAAkBK,8BAA8B,OAAO,QACvD;QACEuC;QACAF;QACAG;QACA,GAAG7B,QAAQc,KAAK;IAClB,GACA;QAAEmB,mBAAmBhD,kBAAkBiD,YAAY;IAAC;IAGtD,4BAA4B;IAC5B,MAAMxC,cACJoB,MACA9B,kBAAkBK,8BAA8B,OAAO,aACvD,CAAC,MAAM,EAAEuC,mBAAmB,kBAAkB,CAAC;IAGjD,6CAA6C;IAC7C,MAAMO,iBAAiBnD,kBACrBgC,cAAcoB,IAAI,EAClBlB,eAAemB,IAAI,IAAI;IAEzB,MAAMC,gBAAgBtD,kBAAkBmD,gBAAgB;IAExD,IAAIrB,KAAKyB,MAAM,CAACD,gBAAgB;QAC9B,MAAME,kBAAkB,GAAGd,mBAAmB,aAAa,CAAC;QAC5D,MAAMe,gBACJf,mBAAmBgB,MAAM,CAAC,GAAGlB,WAAW,KAAKE,mBAAmBiB,KAAK,CAAC;QAExE,4BAA4B;QAC5B,MAAMlD,sBACJqB,MACAwB,eACA;YAACE;SAAgB,EACjB,CAAC,CAAC,EAAET,SAAS,iBAAiB,CAAC;QAGjC,MAAMzC,2BACJwB,MACAwB,eACAE,iBACAC;IAEJ;IAEA,uBAAuB;IACvB,MAAMG,YAAY1B,eAAeR,IAAI,IAAI;IACzC,MAAMmC,gBAAgB,GAAGD,UAAU,IAAI,CAAC;IACxC,MAAME,mBAAmB,GAAGrB,iBAAiB,IAAI,CAAC;IAElD,IAAIT,cAAc+B,OAAO,EAAE,CAACF,cAAc,EAAE;QAC1CzC,kCAAkCY,eAAe6B,eAAe;YAC9DG,UAAU;gBAAC/B,cAAcP,IAAI;aAAC;YAC9BuC,QAAQH;QACV;QACA5D,2BAA2B4B,MAAME,cAAcN,IAAI,EAAEM;IACvD;IAEA,sEAAsE;IACtE7B,kBAAkB2B,MAAMN,cAAc;QAAE0C,aAAalC,cAAcoB,IAAI;IAAC;IAExE,8EAA8E;IAC9EjC,8BACEW,MACAE,cAAcN,IAAI,EAClBC,wCACAJ,sBAAsBS,eAAesB,gBACrCZ,oBACA,oEAAoE;IACpE,kDAAkD;IAClD;QAAEyB,YAAYpC,QAAQI,eAAe,EAAEkB;IAAK;IAG9C,MAAMtC,gCAAgCe,MAAM;QAC1CH;KACD;IAED,MAAMd,qBAAqBiB;IAC3B,OAAO,IACLhB,oBAAoBgB,MAAMC,QAAQqC,yBAAyB,EAAE;YAC3DC,WAAW;gBAAC;aAAa;QAC3B;AACJ,EAAE;AAEF,eAAexC,8BAA8B"}

package/src/ts/agent/react-connection/generator.d.ts

@@ -5,6 +5,192 @@
 import { type Tree } from '@nx/devkit';
 import type { ResolvedConnectionOptions } from '../../../connection/generator';
 import { type NxGeneratorInfo } from '../../../utils/nx';
+/** The metadata this generator records, which its predicates read. */
+export interface TsAgentReactConnectionMetadata {
+    readonly auth: string;
+    readonly protocol: string;
+    /** The AG-UI theme module, which the website's `ux` selects. */
+    readonly theme: string;
+}
+export declare const DEPENDENCIES: import("../../../utils/declared-dependencies").DependencyDeclaration<readonly [{
+    readonly name: "@trpc/client";
+    readonly when: (m: TsAgentReactConnectionMetadata) => boolean;
+}, {
+    readonly name: "@tanstack/react-query";
+    readonly when: (m: TsAgentReactConnectionMetadata) => boolean;
+}, {
+    readonly name: "@tanstack/react-query-devtools";
+    readonly when: (m: TsAgentReactConnectionMetadata) => boolean;
+}, {
+    readonly name: "@trpc/tanstack-react-query";
+    readonly when: (m: TsAgentReactConnectionMetadata) => boolean;
+}, {
+    readonly name: "oidc-client-ts";
+    readonly when: (m: TsAgentReactConnectionMetadata) => boolean;
+}, {
+    readonly name: "aws4fetch";
+    readonly when: (m: TsAgentReactConnectionMetadata) => boolean;
+}, {
+    readonly name: "@aws-sdk/credential-provider-cognito-identity";
+    readonly when: (m: TsAgentReactConnectionMetadata) => boolean;
+}, {
+    readonly name: "react-oidc-context";
+    readonly when: (m: TsAgentReactConnectionMetadata) => boolean;
+}, {
+    readonly name: "@smithy/types";
+    readonly when: (m: TsAgentReactConnectionMetadata) => boolean;
+    readonly dev: true;
+}, Omit<{
+    readonly name: "@copilotkit/react-core";
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "@ag-ui/client";
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "@cloudscape-design/chat-components";
+    readonly when: (m: import("../../react-website/agui/generator").AgUiMetadata) => boolean;
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "lucide-react";
+    readonly when: (m: import("../../react-website/agui/generator").AgUiMetadata) => boolean;
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "oidc-client-ts";
+    readonly when: (m: import("../../react-website/agui/generator").AgUiMetadata) => boolean;
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "aws4fetch";
+    readonly when: (m: import("../../react-website/agui/generator").AgUiMetadata) => boolean;
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "@aws-sdk/credential-provider-cognito-identity";
+    readonly when: (m: import("../../react-website/agui/generator").AgUiMetadata) => boolean;
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "react-oidc-context";
+    readonly when: (m: import("../../react-website/agui/generator").AgUiMetadata) => boolean;
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "@smithy/types";
+    readonly when: (m: import("../../react-website/agui/generator").AgUiMetadata) => boolean;
+    readonly dev: true;
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "react";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "react-dom";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "class-variance-authority";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "clsx";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "tailwind-merge";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "lucide-react";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "tw-animate-css";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "radix-ui";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "vite";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "vitest";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, Omit<{
+    readonly name: "@vitest/coverage-v8";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: TsAgentReactConnectionMetadata) => boolean;
+} & {
+    versionOnly: true;
+}], readonly import("../../../utils/declared-dependencies").DeclaredPyDependency<"a2a-sdk" | "ag-ui-langgraph" | "ag-ui-protocol" | "ag-ui-strands" | "aws-lambda-powertools" | "aws-lambda-powertools[tracer]" | "aws-lambda-powertools[parser]" | "aws-opentelemetry-distro" | "bedrock-agentcore" | "boto3" | "checkov" | "fastapi" | "fastapi[standard]" | "httpx" | "langchain" | "langchain-aws" | "langchain-mcp-adapters" | "langgraph" | "mcp" | "pip-check-updates" | "pip-licenses" | "ruff" | "strands-agents" | "strands-agents[a2a]" | "strands-agents-tools" | "ty" | "pynamodb" | "uvicorn" | "sqlmodel" | "alembic" | "aiomysql" | "asyncpg" | "greenlet", TsAgentReactConnectionMetadata>[]>;
 export declare const TS_AGENT_REACT_CONNECTION_GENERATOR_INFO: NxGeneratorInfo;
 export declare function tsAgentReactConnectionGenerator(tree: Tree, options: ResolvedConnectionOptions): Promise<() => Promise<void>>;
 export default tsAgentReactConnectionGenerator;

package/src/ts/agent/react-connection/generator.js

@@ -3,17 +3,68 @@
  * SPDX-License-Identifier: Apache-2.0
  */ import { generateFiles, joinPathFragments, OverwriteStrategy, updateJson } from "@nx/devkit";
 import { addAgentRuntimeToConnectionNamespace } from "../../../connection/agent-runtime-config.js";
+import { addTsDependencies } from "../../../utils/add-dependencies.js";
 import { addSingleImport, applyGritQL } from "../../../utils/ast.js";
-import { addDependenciesToPackageJson } from "../../../utils/dependencies.js";
+import { declareDependencies, onlyWhen, ownedElsewhere } from "../../../utils/declared-dependencies.js";
 import { formatFilesInSubtree } from "../../../utils/format.js";
 import { installDependencies } from "../../../utils/install.js";
 import { addGeneratorMetricsIfApplicable } from "../../../utils/metrics.js";
 import { kebabCase, toClassName } from "../../../utils/names.js";
-import { getGeneratorInfo, readProjectConfigurationUnqualified } from "../../../utils/nx.js";
-import { withVersions } from "../../../utils/versions.js";
-import { addAgUiReactConnection } from "../../react-website/agui/generator.js";
+import { addComponentGeneratorMetadata, getGeneratorInfo, readProjectConfigurationUnqualified } from "../../../utils/nx.js";
+import { toProjectRelativePath } from "../../../utils/paths.js";
+import { DEPENDENCIES as AGUI_DEPENDENCIES, addAgUiReactConnection, resolveAgUiTheme } from "../../react-website/agui/generator.js";
 import { runtimeConfigGenerator } from "../../react-website/runtime-config/generator.js";
 import { addTsAgentTargetToLocalDev } from "./local-dev.js";
+/** The tRPC-over-HTTP path, which the AG-UI path takes none of. */ const isHttp = (m)=>m.protocol !== 'ag-ui';
+/** The AG-UI path, whose packages `addAgUiReactConnection` adds. */ const isAgUi = (m)=>!isHttp(m);
+// Each entry names the protocol path it belongs to: the HTTP path's client and
+// auth packages are added here, the AG-UI path's by `addAgUiReactConnection` on
+// this generator's behalf.
+export const DEPENDENCIES = declareDependencies()({
+    ts: [
+        {
+            name: '@trpc/client',
+            when: isHttp
+        },
+        {
+            name: '@tanstack/react-query',
+            when: isHttp
+        },
+        {
+            name: '@tanstack/react-query-devtools',
+            when: isHttp
+        },
+        {
+            name: '@trpc/tanstack-react-query',
+            when: isHttp
+        },
+        {
+            name: 'oidc-client-ts',
+            when: (m)=>isHttp(m) && m.auth === 'iam'
+        },
+        {
+            name: 'aws4fetch',
+            when: (m)=>isHttp(m) && m.auth === 'iam'
+        },
+        {
+            name: '@aws-sdk/credential-provider-cognito-identity',
+            when: (m)=>isHttp(m) && m.auth === 'iam'
+        },
+        {
+            name: 'react-oidc-context',
+            when: (m)=>isHttp(m) && (m.auth === 'iam' || m.auth === 'cognito')
+        },
+        {
+            name: '@smithy/types',
+            when: isHttp,
+            dev: true
+        },
+        // `addAgUiReactConnection` adds these itself, so they are declared for
+        // ownership only. Gated on the AG-UI path, and within it on each entry's
+        // own theme or auth condition, which the metadata below records.
+        ...ownedElsewhere(onlyWhen(AGUI_DEPENDENCIES.ts, isAgUi))
+    ]
+});
 export const TS_AGENT_REACT_CONNECTION_GENERATOR_INFO = getGeneratorInfo(import.meta.filename);
 export async function tsAgentReactConnectionGenerator(tree, options) {
     const frontendProjectConfig = readProjectConfigurationUnqualified(tree, options.sourceProject);
@@ -27,6 +78,13 @@ export async function tsAgentReactConnectionGenerator(tree, options) {
     const auth = (targetComponent?.auth ?? metadata?.auth ?? 'iam').toLowerCase();
     const agentProjectAlias = agentProjectConfig.name;
     const agentPath = targetComponent?.path ?? 'src/agent';
+    // Recorded below and read by the declaration's predicates, so the packages
+    // added here are exactly the ones the version sync will own.
+    const connectionMetadata = {
+        auth,
+        protocol: targetComponent?.protocol ?? 'http',
+        theme: resolveAgUiTheme(frontendProjectConfig)
+    };
     if ((targetComponent?.protocol ?? '').toLowerCase() === 'a2a') {
         throw new Error(`Cannot connect a React website to an A2A agent. ` + `Consider generating an agent with the HTTP or AG-UI protocol instead.`);
     }
@@ -43,6 +101,8 @@ export async function tsAgentReactConnectionGenerator(tree, options) {
             port: agentPort,
             targetComponent
         });
+        // Recorded so the version sync knows this connection's dependencies are ours.
+        addComponentGeneratorMetadata(tree, frontendProjectConfig.name, TS_AGENT_REACT_CONNECTION_GENERATOR_INFO, toProjectRelativePath(frontendProjectConfig, joinPathFragments(frontendProjectConfig.sourceRoot, 'hooks', `useAgui${agentNameClassName}`)), agentNameClassName, connectionMetadata);
         await addGeneratorMetricsIfApplicable(tree, [
             TS_AGENT_REACT_CONNECTION_GENERATOR_INFO
         ]);
@@ -100,23 +160,12 @@ export async function tsAgentReactConnectionGenerator(tree, options) {
         agentNameKebabCase: kebabCase(agentNameClassName),
         agentNameClassName
     });
-    addDependenciesToPackageJson(tree, withVersions([
-        '@trpc/client',
-        '@tanstack/react-query',
-        '@tanstack/react-query-devtools',
-        '@trpc/tanstack-react-query',
-        ...auth === 'iam' ? [
-            'oidc-client-ts',
-            'aws4fetch',
-            '@aws-sdk/credential-provider-cognito-identity',
-            'react-oidc-context'
-        ] : [],
-        ...auth === 'cognito' ? [
-            'react-oidc-context'
-        ] : []
-    ]), withVersions([
-        '@smithy/types'
-    ]), joinPathFragments(frontendProjectConfig.root, 'package.json'));
+    addTsDependencies(tree, DEPENDENCIES, {
+        metadata: connectionMetadata,
+        projectRoot: frontendProjectConfig.root
+    });
+    // Recorded so the version sync knows this connection's dependencies are ours.
+    addComponentGeneratorMetadata(tree, frontendProjectConfig.name, TS_AGENT_REACT_CONNECTION_GENERATOR_INFO, toProjectRelativePath(frontendProjectConfig, joinPathFragments(frontendProjectConfig.sourceRoot, 'components', clientProviderName)), agentNameClassName, connectionMetadata);
     await addGeneratorMetricsIfApplicable(tree, [
         TS_AGENT_REACT_CONNECTION_GENERATOR_INFO
     ]);

package/src/ts/agent/react-connection/generator.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/ts/agent/react-connection/generator.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nimport {\n  generateFiles,\n  joinPathFragments,\n  OverwriteStrategy,\n  type Tree,\n  updateJson,\n} from '@nx/devkit';\nimport { addAgentRuntimeToConnectionNamespace } from '../../../connection/agent-runtime-config';\nimport type { ResolvedConnectionOptions } from '../../../connection/generator';\nimport { addSingleImport, applyGritQL } from '../../../utils/ast';\nimport { addDependenciesToPackageJson } from '../../../utils/dependencies';\nimport { formatFilesInSubtree } from '../../../utils/format';\nimport { installDependencies } from '../../../utils/install';\nimport { addGeneratorMetricsIfApplicable } from '../../../utils/metrics';\nimport { kebabCase, toClassName } from '../../../utils/names';\nimport {\n  type ComponentMetadata,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  readProjectConfigurationUnqualified,\n} from '../../../utils/nx';\nimport { withVersions } from '../../../utils/versions';\nimport {\n  type AgUiAuth,\n  addAgUiReactConnection,\n} from '../../react-website/agui/generator';\nimport { runtimeConfigGenerator } from '../../react-website/runtime-config/generator';\nimport { addTsAgentTargetToLocalDev } from './local-dev';\n\nexport const TS_AGENT_REACT_CONNECTION_GENERATOR_INFO: NxGeneratorInfo =\n  getGeneratorInfo(import.meta.filename);\n\nexport async function tsAgentReactConnectionGenerator(\n  tree: Tree,\n  options: ResolvedConnectionOptions,\n) {\n  const frontendProjectConfig = readProjectConfigurationUnqualified(\n    tree,\n    options.sourceProject,\n  );\n  const agentProjectConfig = readProjectConfigurationUnqualified(\n    tree,\n    options.targetProject,\n  );\n\n  const targetComponent: ComponentMetadata | undefined =\n    options.targetComponent;\n\n  // Extract agent metadata from the target component or project metadata\n  const metadata = agentProjectConfig.metadata as any;\n  const agentName = targetComponent?.name ?? 'agent';\n  const agentNameClassName = targetComponent?.rc ?? toClassName(agentName);\n  const agentPort = targetComponent?.port ?? metadata?.ports?.[0] ?? 8081;\n  const auth = (targetComponent?.auth ?? metadata?.auth ?? 'iam').toLowerCase();\n  const agentProjectAlias = agentProjectConfig.name;\n  const agentPath = targetComponent?.path ?? 'src/agent';\n\n  if ((targetComponent?.protocol ?? '').toLowerCase() === 'a2a') {\n    throw new Error(\n      `Cannot connect a React website to an A2A agent. ` +\n        `Consider generating an agent with the HTTP or AG-UI protocol instead.`,\n    );\n  }\n\n  if (targetComponent?.protocol === 'ag-ui') {\n    await addAgUiReactConnection(tree, {\n      frontendProjectConfig,\n      agentName,\n      agentNameClassName,\n      auth: auth as AgUiAuth,\n    });\n\n    await addTsAgentTargetToLocalDev(\n      tree,\n      frontendProjectConfig.name,\n      agentProjectConfig.name,\n      {\n        agentName,\n        agentNameClassName,\n        port: agentPort,\n        targetComponent,\n      },\n    );\n\n    await addGeneratorMetricsIfApplicable(tree, [\n      TS_AGENT_REACT_CONNECTION_GENERATOR_INFO,\n    ]);\n\n    await formatFilesInSubtree(tree);\n    return () =>\n      installDependencies(tree, options.preferInstallDependencies, {\n        languages: ['typescript'],\n      });\n  }\n\n  // Ensure the agent project has a wildcard path entry in tsconfig.base.json\n  // so that deep imports (e.g., for the router type) resolve correctly\n  ensureWildcardPathEntry(\n    tree,\n    agentProjectConfig.name,\n    agentProjectConfig.root,\n  );\n\n  generateFiles(\n    tree,\n    joinPathFragments(import.meta.dirname, 'files'),\n    frontendProjectConfig.root,\n    {\n      agentName,\n      agentNameClassName,\n      auth,\n      agentProjectAlias,\n      agentPath,\n    },\n    {\n      overwriteStrategy: OverwriteStrategy.KeepExisting,\n    },\n  );\n\n  // Generate the tanstack query provider if it does not exist already\n  generateFiles(\n    tree,\n    joinPathFragments(\n      import.meta.dirname,\n      '../../../utils/files/website/components/tanstack-query',\n    ),\n    joinPathFragments(frontendProjectConfig.sourceRoot, 'components'),\n    {},\n    {\n      overwriteStrategy: OverwriteStrategy.KeepExisting,\n    },\n  );\n\n  if (auth === 'iam') {\n    generateFiles(\n      tree,\n      joinPathFragments(\n        import.meta.dirname,\n        '../../../utils/files/website/hooks/sigv4',\n      ),\n      joinPathFragments(frontendProjectConfig.sourceRoot, 'hooks'),\n      {},\n      {\n        overwriteStrategy: OverwriteStrategy.KeepExisting,\n      },\n    );\n  }\n\n  await runtimeConfigGenerator(tree, {\n    project: frontendProjectConfig.name,\n    preferInstallDependencies: false,\n  });\n\n  // update main.tsx\n  const mainTsxPath = joinPathFragments(\n    frontendProjectConfig.sourceRoot,\n    'main.tsx',\n  );\n  await addSingleImport(\n    tree,\n    mainTsxPath,\n    'QueryClientProvider',\n    './components/QueryClientProvider',\n  );\n\n  const clientProviderName = `${agentNameClassName}AgentClientProvider`;\n  await addSingleImport(\n    tree,\n    mainTsxPath,\n    clientProviderName,\n    `./components/${clientProviderName}`,\n  );\n\n  // Wrap <App /> in QueryClientProvider if not already present\n  await applyGritQL(\n    tree,\n    mainTsxPath,\n    '`<App />` => `<QueryClientProvider><App /></QueryClientProvider>` where { $program <: not contains `<QueryClientProvider>$_</QueryClientProvider>` }',\n  );\n\n  // Wrap <App /> in the agent client provider if not already present\n  await applyGritQL(\n    tree,\n    mainTsxPath,\n    `\\`<App />\\` => \\`<${clientProviderName}><App /></${clientProviderName}>\\` where { $program <: not contains \\`<${clientProviderName}>$_</${clientProviderName}>\\` }`,\n  );\n\n  await addTsAgentTargetToLocalDev(\n    tree,\n    frontendProjectConfig.name,\n    agentProjectConfig.name,\n    {\n      agentName,\n      agentNameClassName,\n      port: agentPort,\n      targetComponent,\n    },\n  );\n\n  // Expose the agent's runtime ARN to the frontend via the 'connection'\n  // namespace (which is published to runtime-config.json). The agent construct\n  // itself only registers to the 'agentcore' namespace by default.\n  await addAgentRuntimeToConnectionNamespace(tree, {\n    agentNameKebabCase: kebabCase(agentNameClassName),\n    agentNameClassName,\n  });\n\n  addDependenciesToPackageJson(\n    tree,\n    withVersions([\n      '@trpc/client',\n      '@tanstack/react-query',\n      '@tanstack/react-query-devtools',\n      '@trpc/tanstack-react-query',\n      ...((auth === 'iam'\n        ? [\n            'oidc-client-ts',\n            'aws4fetch',\n            '@aws-sdk/credential-provider-cognito-identity',\n            'react-oidc-context',\n          ]\n        : []) as any),\n      ...((auth === 'cognito' ? ['react-oidc-context'] : []) as any),\n    ]),\n    withVersions(['@smithy/types']),\n    joinPathFragments(frontendProjectConfig.root, 'package.json'),\n  );\n\n  await addGeneratorMetricsIfApplicable(tree, [\n    TS_AGENT_REACT_CONNECTION_GENERATOR_INFO,\n  ]);\n\n  await formatFilesInSubtree(tree);\n  return () =>\n    installDependencies(tree, options.preferInstallDependencies, {\n      languages: ['typescript'],\n    });\n}\nexport default tsAgentReactConnectionGenerator;\n\n/**\n * Ensures a wildcard path entry exists in tsconfig.base.json for the given project,\n * allowing deep imports (e.g., `@scope/project/src/agent/router.js`).\n */\nfunction ensureWildcardPathEntry(\n  tree: Tree,\n  projectName: string,\n  projectRoot: string,\n) {\n  const tsconfigPath = ['tsconfig.base.json', 'tsconfig.json'].find((p) =>\n    tree.exists(p),\n  );\n  if (!tsconfigPath) return;\n\n  updateJson(tree, tsconfigPath, (json) => {\n    const paths = json.compilerOptions?.paths ?? {};\n    const wildcardKey = `${projectName}/*`;\n    if (!paths[wildcardKey]) {\n      paths[wildcardKey] = [`./${projectRoot}/*`];\n    }\n    json.compilerOptions = {\n      ...json.compilerOptions,\n      paths,\n    };\n    return json;\n  });\n}\n"],"names":["generateFiles","joinPathFragments","OverwriteStrategy","updateJson","addAgentRuntimeToConnectionNamespace","addSingleImport","applyGritQL","addDependenciesToPackageJson","formatFilesInSubtree","installDependencies","addGeneratorMetricsIfApplicable","kebabCase","toClassName","getGeneratorInfo","readProjectConfigurationUnqualified","withVersions","addAgUiReactConnection","runtimeConfigGenerator","addTsAgentTargetToLocalDev","TS_AGENT_REACT_CONNECTION_GENERATOR_INFO","filename","tsAgentReactConnectionGenerator","tree","options","frontendProjectConfig","sourceProject","agentProjectConfig","targetProject","targetComponent","metadata","agentName","name","agentNameClassName","rc","agentPort","port","ports","auth","toLowerCase","agentProjectAlias","agentPath","path","protocol","Error","preferInstallDependencies","languages","ensureWildcardPathEntry","root","dirname","overwriteStrategy","KeepExisting","sourceRoot","project","mainTsxPath","clientProviderName","agentNameKebabCase","projectName","projectRoot","tsconfigPath","find","p","exists","json","paths","compilerOptions","wildcardKey"],"mappings":"AAAA;;;CAGC,GACD,SACEA,aAAa,EACbC,iBAAiB,EACjBC,iBAAiB,EAEjBC,UAAU,QACL,aAAa;AACpB,SAASC,oCAAoC,QAAQ,8CAA2C;AAEhG,SAASC,eAAe,EAAEC,WAAW,QAAQ,wBAAqB;AAClE,SAASC,4BAA4B,QAAQ,iCAA8B;AAC3E,SAASC,oBAAoB,QAAQ,2BAAwB;AAC7D,SAASC,mBAAmB,QAAQ,4BAAyB;AAC7D,SAASC,+BAA+B,QAAQ,4BAAyB;AACzE,SAASC,SAAS,EAAEC,WAAW,QAAQ,0BAAuB;AAC9D,SAEEC,gBAAgB,EAEhBC,mCAAmC,QAC9B,uBAAoB;AAC3B,SAASC,YAAY,QAAQ,6BAA0B;AACvD,SAEEC,sBAAsB,QACjB,wCAAqC;AAC5C,SAASC,sBAAsB,QAAQ,kDAA+C;AACtF,SAASC,0BAA0B,QAAQ,iBAAc;AAEzD,OAAO,MAAMC,2CACXN,iBAAiB,YAAYO,QAAQ,EAAE;AAEzC,OAAO,eAAeC,gCACpBC,IAAU,EACVC,OAAkC;IAElC,MAAMC,wBAAwBV,oCAC5BQ,MACAC,QAAQE,aAAa;IAEvB,MAAMC,qBAAqBZ,oCACzBQ,MACAC,QAAQI,aAAa;IAGvB,MAAMC,kBACJL,QAAQK,eAAe;IAEzB,uEAAuE;IACvE,MAAMC,WAAWH,mBAAmBG,QAAQ;IAC5C,MAAMC,YAAYF,iBAAiBG,QAAQ;IAC3C,MAAMC,qBAAqBJ,iBAAiBK,MAAMrB,YAAYkB;IAC9D,MAAMI,YAAYN,iBAAiBO,QAAQN,UAAUO,OAAO,CAAC,EAAE,IAAI;IACnE,MAAMC,OAAO,AAACT,CAAAA,iBAAiBS,QAAQR,UAAUQ,QAAQ,KAAI,EAAGC,WAAW;IAC3E,MAAMC,oBAAoBb,mBAAmBK,IAAI;IACjD,MAAMS,YAAYZ,iBAAiBa,QAAQ;IAE3C,IAAI,AAACb,CAAAA,iBAAiBc,YAAY,EAAC,EAAGJ,WAAW,OAAO,OAAO;QAC7D,MAAM,IAAIK,MACR,CAAC,gDAAgD,CAAC,GAChD,CAAC,qEAAqE,CAAC;IAE7E;IAEA,IAAIf,iBAAiBc,aAAa,SAAS;QACzC,MAAM1B,uBAAuBM,MAAM;YACjCE;YACAM;YACAE;YACAK,MAAMA;QACR;QAEA,MAAMnB,2BACJI,MACAE,sBAAsBO,IAAI,EAC1BL,mBAAmBK,IAAI,EACvB;YACED;YACAE;YACAG,MAAMD;YACNN;QACF;QAGF,MAAMlB,gCAAgCY,MAAM;YAC1CH;SACD;QAED,MAAMX,qBAAqBc;QAC3B,OAAO,IACLb,oBAAoBa,MAAMC,QAAQqB,yBAAyB,EAAE;gBAC3DC,WAAW;oBAAC;iBAAa;YAC3B;IACJ;IAEA,2EAA2E;IAC3E,qEAAqE;IACrEC,wBACExB,MACAI,mBAAmBK,IAAI,EACvBL,mBAAmBqB,IAAI;IAGzB/C,cACEsB,MACArB,kBAAkB,YAAY+C,OAAO,EAAE,UACvCxB,sBAAsBuB,IAAI,EAC1B;QACEjB;QACAE;QACAK;QACAE;QACAC;IACF,GACA;QACES,mBAAmB/C,kBAAkBgD,YAAY;IACnD;IAGF,oEAAoE;IACpElD,cACEsB,MACArB,kBACE,YAAY+C,OAAO,EACnB,2DAEF/C,kBAAkBuB,sBAAsB2B,UAAU,EAAE,eACpD,CAAC,GACD;QACEF,mBAAmB/C,kBAAkBgD,YAAY;IACnD;IAGF,IAAIb,SAAS,OAAO;QAClBrC,cACEsB,MACArB,kBACE,YAAY+C,OAAO,EACnB,6CAEF/C,kBAAkBuB,sBAAsB2B,UAAU,EAAE,UACpD,CAAC,GACD;YACEF,mBAAmB/C,kBAAkBgD,YAAY;QACnD;IAEJ;IAEA,MAAMjC,uBAAuBK,MAAM;QACjC8B,SAAS5B,sBAAsBO,IAAI;QACnCa,2BAA2B;IAC7B;IAEA,kBAAkB;IAClB,MAAMS,cAAcpD,kBAClBuB,sBAAsB2B,UAAU,EAChC;IAEF,MAAM9C,gBACJiB,MACA+B,aACA,uBACA;IAGF,MAAMC,qBAAqB,GAAGtB,mBAAmB,mBAAmB,CAAC;IACrE,MAAM3B,gBACJiB,MACA+B,aACAC,oBACA,CAAC,aAAa,EAAEA,oBAAoB;IAGtC,6DAA6D;IAC7D,MAAMhD,YACJgB,MACA+B,aACA;IAGF,mEAAmE;IACnE,MAAM/C,YACJgB,MACA+B,aACA,CAAC,kBAAkB,EAAEC,mBAAmB,UAAU,EAAEA,mBAAmB,wCAAwC,EAAEA,mBAAmB,KAAK,EAAEA,mBAAmB,KAAK,CAAC;IAGtK,MAAMpC,2BACJI,MACAE,sBAAsBO,IAAI,EAC1BL,mBAAmBK,IAAI,EACvB;QACED;QACAE;QACAG,MAAMD;QACNN;IACF;IAGF,sEAAsE;IACtE,6EAA6E;IAC7E,iEAAiE;IACjE,MAAMxB,qCAAqCkB,MAAM;QAC/CiC,oBAAoB5C,UAAUqB;QAC9BA;IACF;IAEAzB,6BACEe,MACAP,aAAa;QACX;QACA;QACA;QACA;WACKsB,SAAS,QACV;YACE;YACA;YACA;YACA;SACD,GACD,EAAE;WACDA,SAAS,YAAY;YAAC;SAAqB,GAAG,EAAE;KACtD,GACDtB,aAAa;QAAC;KAAgB,GAC9Bd,kBAAkBuB,sBAAsBuB,IAAI,EAAE;IAGhD,MAAMrC,gCAAgCY,MAAM;QAC1CH;KACD;IAED,MAAMX,qBAAqBc;IAC3B,OAAO,IACLb,oBAAoBa,MAAMC,QAAQqB,yBAAyB,EAAE;YAC3DC,WAAW;gBAAC;aAAa;QAC3B;AACJ;AACA,eAAexB,gCAAgC;AAE/C;;;CAGC,GACD,SAASyB,wBACPxB,IAAU,EACVkC,WAAmB,EACnBC,WAAmB;IAEnB,MAAMC,eAAe;QAAC;QAAsB;KAAgB,CAACC,IAAI,CAAC,CAACC,IACjEtC,KAAKuC,MAAM,CAACD;IAEd,IAAI,CAACF,cAAc;IAEnBvD,WAAWmB,MAAMoC,cAAc,CAACI;QAC9B,MAAMC,QAAQD,KAAKE,eAAe,EAAED,SAAS,CAAC;QAC9C,MAAME,cAAc,GAAGT,YAAY,EAAE,CAAC;QACtC,IAAI,CAACO,KAAK,CAACE,YAAY,EAAE;YACvBF,KAAK,CAACE,YAAY,GAAG;gBAAC,CAAC,EAAE,EAAER,YAAY,EAAE,CAAC;aAAC;QAC7C;QACAK,KAAKE,eAAe,GAAG;YACrB,GAAGF,KAAKE,eAAe;YACvBD;QACF;QACA,OAAOD;IACT;AACF"}
+{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/ts/agent/react-connection/generator.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nimport {\n  generateFiles,\n  joinPathFragments,\n  OverwriteStrategy,\n  type Tree,\n  updateJson,\n} from '@nx/devkit';\nimport { addAgentRuntimeToConnectionNamespace } from '../../../connection/agent-runtime-config';\nimport type { ResolvedConnectionOptions } from '../../../connection/generator';\nimport { addTsDependencies } from '../../../utils/add-dependencies';\nimport { addSingleImport, applyGritQL } from '../../../utils/ast';\nimport {\n  declareDependencies,\n  onlyWhen,\n  ownedElsewhere,\n} from '../../../utils/declared-dependencies';\nimport { formatFilesInSubtree } from '../../../utils/format';\nimport { installDependencies } from '../../../utils/install';\nimport { addGeneratorMetricsIfApplicable } from '../../../utils/metrics';\nimport { kebabCase, toClassName } from '../../../utils/names';\nimport {\n  addComponentGeneratorMetadata,\n  type ComponentMetadata,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  readProjectConfigurationUnqualified,\n} from '../../../utils/nx';\nimport { toProjectRelativePath } from '../../../utils/paths';\nimport {\n  DEPENDENCIES as AGUI_DEPENDENCIES,\n  type AgUiAuth,\n  addAgUiReactConnection,\n  resolveAgUiTheme,\n} from '../../react-website/agui/generator';\nimport { runtimeConfigGenerator } from '../../react-website/runtime-config/generator';\nimport { addTsAgentTargetToLocalDev } from './local-dev';\n\n/** The metadata this generator records, which its predicates read. */\nexport interface TsAgentReactConnectionMetadata {\n  readonly auth: string;\n  readonly protocol: string;\n  /** The AG-UI theme module, which the website's `ux` selects. */\n  readonly theme: string;\n}\n\n/** The tRPC-over-HTTP path, which the AG-UI path takes none of. */\nconst isHttp = (m: TsAgentReactConnectionMetadata) => m.protocol !== 'ag-ui';\n\n/** The AG-UI path, whose packages `addAgUiReactConnection` adds. */\nconst isAgUi = (m: TsAgentReactConnectionMetadata) => !isHttp(m);\n\n// Each entry names the protocol path it belongs to: the HTTP path's client and\n// auth packages are added here, the AG-UI path's by `addAgUiReactConnection` on\n// this generator's behalf.\nexport const DEPENDENCIES =\n  declareDependencies<TsAgentReactConnectionMetadata>()({\n    ts: [\n      { name: '@trpc/client', when: isHttp },\n      { name: '@tanstack/react-query', when: isHttp },\n      { name: '@tanstack/react-query-devtools', when: isHttp },\n      { name: '@trpc/tanstack-react-query', when: isHttp },\n      { name: 'oidc-client-ts', when: (m) => isHttp(m) && m.auth === 'iam' },\n      { name: 'aws4fetch', when: (m) => isHttp(m) && m.auth === 'iam' },\n      {\n        name: '@aws-sdk/credential-provider-cognito-identity',\n        when: (m) => isHttp(m) && m.auth === 'iam',\n      },\n      {\n        name: 'react-oidc-context',\n        when: (m) => isHttp(m) && (m.auth === 'iam' || m.auth === 'cognito'),\n      },\n      { name: '@smithy/types', when: isHttp, dev: true },\n      // `addAgUiReactConnection` adds these itself, so they are declared for\n      // ownership only. Gated on the AG-UI path, and within it on each entry's\n      // own theme or auth condition, which the metadata below records.\n      ...ownedElsewhere(onlyWhen(AGUI_DEPENDENCIES.ts, isAgUi)),\n    ],\n  });\n\nexport const TS_AGENT_REACT_CONNECTION_GENERATOR_INFO: NxGeneratorInfo =\n  getGeneratorInfo(import.meta.filename);\n\nexport async function tsAgentReactConnectionGenerator(\n  tree: Tree,\n  options: ResolvedConnectionOptions,\n) {\n  const frontendProjectConfig = readProjectConfigurationUnqualified(\n    tree,\n    options.sourceProject,\n  );\n  const agentProjectConfig = readProjectConfigurationUnqualified(\n    tree,\n    options.targetProject,\n  );\n\n  const targetComponent: ComponentMetadata | undefined =\n    options.targetComponent;\n\n  // Extract agent metadata from the target component or project metadata\n  const metadata = agentProjectConfig.metadata as any;\n  const agentName = targetComponent?.name ?? 'agent';\n  const agentNameClassName = targetComponent?.rc ?? toClassName(agentName);\n  const agentPort = targetComponent?.port ?? metadata?.ports?.[0] ?? 8081;\n  const auth = (targetComponent?.auth ?? metadata?.auth ?? 'iam').toLowerCase();\n  const agentProjectAlias = agentProjectConfig.name;\n  const agentPath = targetComponent?.path ?? 'src/agent';\n\n  // Recorded below and read by the declaration's predicates, so the packages\n  // added here are exactly the ones the version sync will own.\n  const connectionMetadata: TsAgentReactConnectionMetadata = {\n    auth,\n    protocol: targetComponent?.protocol ?? 'http',\n    theme: resolveAgUiTheme(frontendProjectConfig),\n  };\n\n  if ((targetComponent?.protocol ?? '').toLowerCase() === 'a2a') {\n    throw new Error(\n      `Cannot connect a React website to an A2A agent. ` +\n        `Consider generating an agent with the HTTP or AG-UI protocol instead.`,\n    );\n  }\n\n  if (targetComponent?.protocol === 'ag-ui') {\n    await addAgUiReactConnection(tree, {\n      frontendProjectConfig,\n      agentName,\n      agentNameClassName,\n      auth: auth as AgUiAuth,\n    });\n\n    await addTsAgentTargetToLocalDev(\n      tree,\n      frontendProjectConfig.name,\n      agentProjectConfig.name,\n      {\n        agentName,\n        agentNameClassName,\n        port: agentPort,\n        targetComponent,\n      },\n    );\n\n    // Recorded so the version sync knows this connection's dependencies are ours.\n    addComponentGeneratorMetadata(\n      tree,\n      frontendProjectConfig.name,\n      TS_AGENT_REACT_CONNECTION_GENERATOR_INFO,\n      toProjectRelativePath(\n        frontendProjectConfig,\n        joinPathFragments(\n          frontendProjectConfig.sourceRoot,\n          'hooks',\n          `useAgui${agentNameClassName}`,\n        ),\n      ),\n      agentNameClassName,\n      connectionMetadata,\n    );\n\n    await addGeneratorMetricsIfApplicable(tree, [\n      TS_AGENT_REACT_CONNECTION_GENERATOR_INFO,\n    ]);\n\n    await formatFilesInSubtree(tree);\n    return () =>\n      installDependencies(tree, options.preferInstallDependencies, {\n        languages: ['typescript'],\n      });\n  }\n\n  // Ensure the agent project has a wildcard path entry in tsconfig.base.json\n  // so that deep imports (e.g., for the router type) resolve correctly\n  ensureWildcardPathEntry(\n    tree,\n    agentProjectConfig.name,\n    agentProjectConfig.root,\n  );\n\n  generateFiles(\n    tree,\n    joinPathFragments(import.meta.dirname, 'files'),\n    frontendProjectConfig.root,\n    {\n      agentName,\n      agentNameClassName,\n      auth,\n      agentProjectAlias,\n      agentPath,\n    },\n    {\n      overwriteStrategy: OverwriteStrategy.KeepExisting,\n    },\n  );\n\n  // Generate the tanstack query provider if it does not exist already\n  generateFiles(\n    tree,\n    joinPathFragments(\n      import.meta.dirname,\n      '../../../utils/files/website/components/tanstack-query',\n    ),\n    joinPathFragments(frontendProjectConfig.sourceRoot, 'components'),\n    {},\n    {\n      overwriteStrategy: OverwriteStrategy.KeepExisting,\n    },\n  );\n\n  if (auth === 'iam') {\n    generateFiles(\n      tree,\n      joinPathFragments(\n        import.meta.dirname,\n        '../../../utils/files/website/hooks/sigv4',\n      ),\n      joinPathFragments(frontendProjectConfig.sourceRoot, 'hooks'),\n      {},\n      {\n        overwriteStrategy: OverwriteStrategy.KeepExisting,\n      },\n    );\n  }\n\n  await runtimeConfigGenerator(tree, {\n    project: frontendProjectConfig.name,\n    preferInstallDependencies: false,\n  });\n\n  // update main.tsx\n  const mainTsxPath = joinPathFragments(\n    frontendProjectConfig.sourceRoot,\n    'main.tsx',\n  );\n  await addSingleImport(\n    tree,\n    mainTsxPath,\n    'QueryClientProvider',\n    './components/QueryClientProvider',\n  );\n\n  const clientProviderName = `${agentNameClassName}AgentClientProvider`;\n  await addSingleImport(\n    tree,\n    mainTsxPath,\n    clientProviderName,\n    `./components/${clientProviderName}`,\n  );\n\n  // Wrap <App /> in QueryClientProvider if not already present\n  await applyGritQL(\n    tree,\n    mainTsxPath,\n    '`<App />` => `<QueryClientProvider><App /></QueryClientProvider>` where { $program <: not contains `<QueryClientProvider>$_</QueryClientProvider>` }',\n  );\n\n  // Wrap <App /> in the agent client provider if not already present\n  await applyGritQL(\n    tree,\n    mainTsxPath,\n    `\\`<App />\\` => \\`<${clientProviderName}><App /></${clientProviderName}>\\` where { $program <: not contains \\`<${clientProviderName}>$_</${clientProviderName}>\\` }`,\n  );\n\n  await addTsAgentTargetToLocalDev(\n    tree,\n    frontendProjectConfig.name,\n    agentProjectConfig.name,\n    {\n      agentName,\n      agentNameClassName,\n      port: agentPort,\n      targetComponent,\n    },\n  );\n\n  // Expose the agent's runtime ARN to the frontend via the 'connection'\n  // namespace (which is published to runtime-config.json). The agent construct\n  // itself only registers to the 'agentcore' namespace by default.\n  await addAgentRuntimeToConnectionNamespace(tree, {\n    agentNameKebabCase: kebabCase(agentNameClassName),\n    agentNameClassName,\n  });\n\n  addTsDependencies(tree, DEPENDENCIES, {\n    metadata: connectionMetadata,\n    projectRoot: frontendProjectConfig.root,\n  });\n\n  // Recorded so the version sync knows this connection's dependencies are ours.\n  addComponentGeneratorMetadata(\n    tree,\n    frontendProjectConfig.name,\n    TS_AGENT_REACT_CONNECTION_GENERATOR_INFO,\n    toProjectRelativePath(\n      frontendProjectConfig,\n      joinPathFragments(\n        frontendProjectConfig.sourceRoot,\n        'components',\n        clientProviderName,\n      ),\n    ),\n    agentNameClassName,\n    connectionMetadata,\n  );\n\n  await addGeneratorMetricsIfApplicable(tree, [\n    TS_AGENT_REACT_CONNECTION_GENERATOR_INFO,\n  ]);\n\n  await formatFilesInSubtree(tree);\n  return () =>\n    installDependencies(tree, options.preferInstallDependencies, {\n      languages: ['typescript'],\n    });\n}\nexport default tsAgentReactConnectionGenerator;\n\n/**\n * Ensures a wildcard path entry exists in tsconfig.base.json for the given project,\n * allowing deep imports (e.g., `@scope/project/src/agent/router.js`).\n */\nfunction ensureWildcardPathEntry(\n  tree: Tree,\n  projectName: string,\n  projectRoot: string,\n) {\n  const tsconfigPath = ['tsconfig.base.json', 'tsconfig.json'].find((p) =>\n    tree.exists(p),\n  );\n  if (!tsconfigPath) return;\n\n  updateJson(tree, tsconfigPath, (json) => {\n    const paths = json.compilerOptions?.paths ?? {};\n    const wildcardKey = `${projectName}/*`;\n    if (!paths[wildcardKey]) {\n      paths[wildcardKey] = [`./${projectRoot}/*`];\n    }\n    json.compilerOptions = {\n      ...json.compilerOptions,\n      paths,\n    };\n    return json;\n  });\n}\n"],"names":["generateFiles","joinPathFragments","OverwriteStrategy","updateJson","addAgentRuntimeToConnectionNamespace","addTsDependencies","addSingleImport","applyGritQL","declareDependencies","onlyWhen","ownedElsewhere","formatFilesInSubtree","installDependencies","addGeneratorMetricsIfApplicable","kebabCase","toClassName","addComponentGeneratorMetadata","getGeneratorInfo","readProjectConfigurationUnqualified","toProjectRelativePath","DEPENDENCIES","AGUI_DEPENDENCIES","addAgUiReactConnection","resolveAgUiTheme","runtimeConfigGenerator","addTsAgentTargetToLocalDev","isHttp","m","protocol","isAgUi","ts","name","when","auth","dev","TS_AGENT_REACT_CONNECTION_GENERATOR_INFO","filename","tsAgentReactConnectionGenerator","tree","options","frontendProjectConfig","sourceProject","agentProjectConfig","targetProject","targetComponent","metadata","agentName","agentNameClassName","rc","agentPort","port","ports","toLowerCase","agentProjectAlias","agentPath","path","connectionMetadata","theme","Error","sourceRoot","preferInstallDependencies","languages","ensureWildcardPathEntry","root","dirname","overwriteStrategy","KeepExisting","project","mainTsxPath","clientProviderName","agentNameKebabCase","projectRoot","projectName","tsconfigPath","find","p","exists","json","paths","compilerOptions","wildcardKey"],"mappings":"AAAA;;;CAGC,GACD,SACEA,aAAa,EACbC,iBAAiB,EACjBC,iBAAiB,EAEjBC,UAAU,QACL,aAAa;AACpB,SAASC,oCAAoC,QAAQ,8CAA2C;AAEhG,SAASC,iBAAiB,QAAQ,qCAAkC;AACpE,SAASC,eAAe,EAAEC,WAAW,QAAQ,wBAAqB;AAClE,SACEC,mBAAmB,EACnBC,QAAQ,EACRC,cAAc,QACT,0CAAuC;AAC9C,SAASC,oBAAoB,QAAQ,2BAAwB;AAC7D,SAASC,mBAAmB,QAAQ,4BAAyB;AAC7D,SAASC,+BAA+B,QAAQ,4BAAyB;AACzE,SAASC,SAAS,EAAEC,WAAW,QAAQ,0BAAuB;AAC9D,SACEC,6BAA6B,EAE7BC,gBAAgB,EAEhBC,mCAAmC,QAC9B,uBAAoB;AAC3B,SAASC,qBAAqB,QAAQ,0BAAuB;AAC7D,SACEC,gBAAgBC,iBAAiB,EAEjCC,sBAAsB,EACtBC,gBAAgB,QACX,wCAAqC;AAC5C,SAASC,sBAAsB,QAAQ,kDAA+C;AACtF,SAASC,0BAA0B,QAAQ,iBAAc;AAUzD,iEAAiE,GACjE,MAAMC,SAAS,CAACC,IAAsCA,EAAEC,QAAQ,KAAK;AAErE,kEAAkE,GAClE,MAAMC,SAAS,CAACF,IAAsC,CAACD,OAAOC;AAE9D,+EAA+E;AAC/E,gFAAgF;AAChF,2BAA2B;AAC3B,OAAO,MAAMP,eACXZ,sBAAsD;IACpDsB,IAAI;QACF;YAAEC,MAAM;YAAgBC,MAAMN;QAAO;QACrC;YAAEK,MAAM;YAAyBC,MAAMN;QAAO;QAC9C;YAAEK,MAAM;YAAkCC,MAAMN;QAAO;QACvD;YAAEK,MAAM;YAA8BC,MAAMN;QAAO;QACnD;YAAEK,MAAM;YAAkBC,MAAM,CAACL,IAAMD,OAAOC,MAAMA,EAAEM,IAAI,KAAK;QAAM;QACrE;YAAEF,MAAM;YAAaC,MAAM,CAACL,IAAMD,OAAOC,MAAMA,EAAEM,IAAI,KAAK;QAAM;QAChE;YACEF,MAAM;YACNC,MAAM,CAACL,IAAMD,OAAOC,MAAMA,EAAEM,IAAI,KAAK;QACvC;QACA;YACEF,MAAM;YACNC,MAAM,CAACL,IAAMD,OAAOC,MAAOA,CAAAA,EAAEM,IAAI,KAAK,SAASN,EAAEM,IAAI,KAAK,SAAQ;QACpE;QACA;YAAEF,MAAM;YAAiBC,MAAMN;YAAQQ,KAAK;QAAK;QACjD,uEAAuE;QACvE,yEAAyE;QACzE,iEAAiE;WAC9DxB,eAAeD,SAASY,kBAAkBS,EAAE,EAAED;KAClD;AACH,GAAG;AAEL,OAAO,MAAMM,2CACXlB,iBAAiB,YAAYmB,QAAQ,EAAE;AAEzC,OAAO,eAAeC,gCACpBC,IAAU,EACVC,OAAkC;IAElC,MAAMC,wBAAwBtB,oCAC5BoB,MACAC,QAAQE,aAAa;IAEvB,MAAMC,qBAAqBxB,oCACzBoB,MACAC,QAAQI,aAAa;IAGvB,MAAMC,kBACJL,QAAQK,eAAe;IAEzB,uEAAuE;IACvE,MAAMC,WAAWH,mBAAmBG,QAAQ;IAC5C,MAAMC,YAAYF,iBAAiBb,QAAQ;IAC3C,MAAMgB,qBAAqBH,iBAAiBI,MAAMjC,YAAY+B;IAC9D,MAAMG,YAAYL,iBAAiBM,QAAQL,UAAUM,OAAO,CAAC,EAAE,IAAI;IACnE,MAAMlB,OAAO,AAACW,CAAAA,iBAAiBX,QAAQY,UAAUZ,QAAQ,KAAI,EAAGmB,WAAW;IAC3E,MAAMC,oBAAoBX,mBAAmBX,IAAI;IACjD,MAAMuB,YAAYV,iBAAiBW,QAAQ;IAE3C,2EAA2E;IAC3E,6DAA6D;IAC7D,MAAMC,qBAAqD;QACzDvB;QACAL,UAAUgB,iBAAiBhB,YAAY;QACvC6B,OAAOlC,iBAAiBiB;IAC1B;IAEA,IAAI,AAACI,CAAAA,iBAAiBhB,YAAY,EAAC,EAAGwB,WAAW,OAAO,OAAO;QAC7D,MAAM,IAAIM,MACR,CAAC,gDAAgD,CAAC,GAChD,CAAC,qEAAqE,CAAC;IAE7E;IAEA,IAAId,iBAAiBhB,aAAa,SAAS;QACzC,MAAMN,uBAAuBgB,MAAM;YACjCE;YACAM;YACAC;YACAd,MAAMA;QACR;QAEA,MAAMR,2BACJa,MACAE,sBAAsBT,IAAI,EAC1BW,mBAAmBX,IAAI,EACvB;YACEe;YACAC;YACAG,MAAMD;YACNL;QACF;QAGF,8EAA8E;QAC9E5B,8BACEsB,MACAE,sBAAsBT,IAAI,EAC1BI,0CACAhB,sBACEqB,uBACAvC,kBACEuC,sBAAsBmB,UAAU,EAChC,SACA,CAAC,OAAO,EAAEZ,oBAAoB,IAGlCA,oBACAS;QAGF,MAAM3C,gCAAgCyB,MAAM;YAC1CH;SACD;QAED,MAAMxB,qBAAqB2B;QAC3B,OAAO,IACL1B,oBAAoB0B,MAAMC,QAAQqB,yBAAyB,EAAE;gBAC3DC,WAAW;oBAAC;iBAAa;YAC3B;IACJ;IAEA,2EAA2E;IAC3E,qEAAqE;IACrEC,wBACExB,MACAI,mBAAmBX,IAAI,EACvBW,mBAAmBqB,IAAI;IAGzB/D,cACEsC,MACArC,kBAAkB,YAAY+D,OAAO,EAAE,UACvCxB,sBAAsBuB,IAAI,EAC1B;QACEjB;QACAC;QACAd;QACAoB;QACAC;IACF,GACA;QACEW,mBAAmB/D,kBAAkBgE,YAAY;IACnD;IAGF,oEAAoE;IACpElE,cACEsC,MACArC,kBACE,YAAY+D,OAAO,EACnB,2DAEF/D,kBAAkBuC,sBAAsBmB,UAAU,EAAE,eACpD,CAAC,GACD;QACEM,mBAAmB/D,kBAAkBgE,YAAY;IACnD;IAGF,IAAIjC,SAAS,OAAO;QAClBjC,cACEsC,MACArC,kBACE,YAAY+D,OAAO,EACnB,6CAEF/D,kBAAkBuC,sBAAsBmB,UAAU,EAAE,UACpD,CAAC,GACD;YACEM,mBAAmB/D,kBAAkBgE,YAAY;QACnD;IAEJ;IAEA,MAAM1C,uBAAuBc,MAAM;QACjC6B,SAAS3B,sBAAsBT,IAAI;QACnC6B,2BAA2B;IAC7B;IAEA,kBAAkB;IAClB,MAAMQ,cAAcnE,kBAClBuC,sBAAsBmB,UAAU,EAChC;IAEF,MAAMrD,gBACJgC,MACA8B,aACA,uBACA;IAGF,MAAMC,qBAAqB,GAAGtB,mBAAmB,mBAAmB,CAAC;IACrE,MAAMzC,gBACJgC,MACA8B,aACAC,oBACA,CAAC,aAAa,EAAEA,oBAAoB;IAGtC,6DAA6D;IAC7D,MAAM9D,YACJ+B,MACA8B,aACA;IAGF,mEAAmE;IACnE,MAAM7D,YACJ+B,MACA8B,aACA,CAAC,kBAAkB,EAAEC,mBAAmB,UAAU,EAAEA,mBAAmB,wCAAwC,EAAEA,mBAAmB,KAAK,EAAEA,mBAAmB,KAAK,CAAC;IAGtK,MAAM5C,2BACJa,MACAE,sBAAsBT,IAAI,EAC1BW,mBAAmBX,IAAI,EACvB;QACEe;QACAC;QACAG,MAAMD;QACNL;IACF;IAGF,sEAAsE;IACtE,6EAA6E;IAC7E,iEAAiE;IACjE,MAAMxC,qCAAqCkC,MAAM;QAC/CgC,oBAAoBxD,UAAUiC;QAC9BA;IACF;IAEA1C,kBAAkBiC,MAAMlB,cAAc;QACpCyB,UAAUW;QACVe,aAAa/B,sBAAsBuB,IAAI;IACzC;IAEA,8EAA8E;IAC9E/C,8BACEsB,MACAE,sBAAsBT,IAAI,EAC1BI,0CACAhB,sBACEqB,uBACAvC,kBACEuC,sBAAsBmB,UAAU,EAChC,cACAU,sBAGJtB,oBACAS;IAGF,MAAM3C,gCAAgCyB,MAAM;QAC1CH;KACD;IAED,MAAMxB,qBAAqB2B;IAC3B,OAAO,IACL1B,oBAAoB0B,MAAMC,QAAQqB,yBAAyB,EAAE;YAC3DC,WAAW;gBAAC;aAAa;QAC3B;AACJ;AACA,eAAexB,gCAAgC;AAE/C;;;CAGC,GACD,SAASyB,wBACPxB,IAAU,EACVkC,WAAmB,EACnBD,WAAmB;IAEnB,MAAME,eAAe;QAAC;QAAsB;KAAgB,CAACC,IAAI,CAAC,CAACC,IACjErC,KAAKsC,MAAM,CAACD;IAEd,IAAI,CAACF,cAAc;IAEnBtE,WAAWmC,MAAMmC,cAAc,CAACI;QAC9B,MAAMC,QAAQD,KAAKE,eAAe,EAAED,SAAS,CAAC;QAC9C,MAAME,cAAc,GAAGR,YAAY,EAAE,CAAC;QACtC,IAAI,CAACM,KAAK,CAACE,YAAY,EAAE;YACvBF,KAAK,CAACE,YAAY,GAAG;gBAAC,CAAC,EAAE,EAAET,YAAY,EAAE,CAAC;aAAC;QAC7C;QACAM,KAAKE,eAAe,GAAG;YACrB,GAAGF,KAAKE,eAAe;YACvBD;QACF;QACA,OAAOD;IACT;AACF"}

package/src/ts/astro-docs/generator.d.ts

@@ -5,6 +5,44 @@
 import { type GeneratorCallback, type Tree } from '@nx/devkit';
 import { type NxGeneratorInfo } from '../../utils/nx';
 import type { TsAstroDocsGeneratorSchema } from './schema';
+/** The metadata this generator records, which its predicates read. */
+export interface TsAstroDocsMetadata {
+    readonly framework: string;
+    readonly includeTranslation: boolean;
+    readonly includeBlog: boolean;
+}
+export declare const DEPENDENCIES: import("../../utils/declared-dependencies").DependencyDeclaration<readonly [{
+    readonly name: "astro";
+}, {
+    readonly name: "@astrojs/starlight";
+}, {
+    readonly name: "starlight-blog";
+    readonly when: (m: TsAstroDocsMetadata) => boolean;
+}, {
+    readonly name: "@strands-agents/sdk";
+    readonly when: (m: TsAstroDocsMetadata) => boolean;
+}, {
+    readonly name: "commander";
+    readonly when: (m: TsAstroDocsMetadata) => boolean;
+}, {
+    readonly name: "fast-glob";
+    readonly when: (m: TsAstroDocsMetadata) => boolean;
+}, {
+    readonly name: "fs-extra";
+    readonly when: (m: TsAstroDocsMetadata) => boolean;
+}, {
+    readonly name: "simple-git";
+    readonly when: (m: TsAstroDocsMetadata) => boolean;
+}, {
+    readonly name: "@types/fs-extra";
+    readonly when: (m: TsAstroDocsMetadata) => boolean;
+    readonly dev: true;
+}, {
+    readonly name: "tsx";
+    readonly when: (m: TsAstroDocsMetadata) => boolean;
+    readonly dev: true;
+    readonly root: true;
+}], readonly import("../../utils/declared-dependencies").DeclaredPyDependency<"a2a-sdk" | "ag-ui-langgraph" | "ag-ui-protocol" | "ag-ui-strands" | "aws-lambda-powertools" | "aws-lambda-powertools[tracer]" | "aws-lambda-powertools[parser]" | "aws-opentelemetry-distro" | "bedrock-agentcore" | "boto3" | "checkov" | "fastapi" | "fastapi[standard]" | "httpx" | "langchain" | "langchain-aws" | "langchain-mcp-adapters" | "langgraph" | "mcp" | "pip-check-updates" | "pip-licenses" | "ruff" | "strands-agents" | "strands-agents[a2a]" | "strands-agents-tools" | "ty" | "pynamodb" | "uvicorn" | "sqlmodel" | "alembic" | "aiomysql" | "asyncpg" | "greenlet", TsAstroDocsMetadata>[]>;
 export declare const TS_ASTRO_DOCS_GENERATOR_INFO: NxGeneratorInfo;
 export declare const tsAstroDocsGenerator: (tree: Tree, schema: TsAstroDocsGeneratorSchema) => Promise<GeneratorCallback>;
 export default tsAstroDocsGenerator;