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

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/py/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  addPythonCoreClient,\n  addPythonReExport,\n  ensurePythonAgentConnectionProject,\n  getPythonAgentConnectionModuleName,\n  getPythonAgentConnectionProject,\n  getPythonAgentConnectionProjectDir,\n  PY_MCP_FAMILY_CONNECTIONS,\n  resolveAgentFramework,\n} from '../../../utils/agent-connection/agent-connection';\nimport { addPythonDestructuredImport } from '../../../utils/ast';\nimport { formatFilesInSubtree } from '../../../utils/format';\nimport { installDependencies } from '../../../utils/install';\nimport { addGeneratorMetricsIfApplicable } from '../../../utils/metrics';\nimport { snakeCase } from '../../../utils/names';\nimport {\n  addDependencyToTargetIfNotPresent,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  readProjectConfigurationUnqualified,\n} from '../../../utils/nx';\nimport {\n  addDependenciesToPyProjectToml,\n  addWorkspaceDependencyToPyProject,\n} from '../../../utils/py';\nimport type { PyAgentMcpConnectionGeneratorSchema } from './schema';\n\nexport const PY_AGENT_MCP_CONNECTION_GENERATOR_INFO: NxGeneratorInfo =\n  getGeneratorInfo(import.meta.filename);\n\nexport const pyAgentMcpConnectionGenerator = async (\n  tree: Tree,\n  options: PyAgentMcpConnectionGeneratorSchema,\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 py#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 mcpServerSnakeCase = snakeCase(mcpServerClassName);\n  const mcpServerPort = mcpComponent.port ?? 8000;\n\n  // The source agent's framework selects the client shape, dependencies and the\n  // agent.py transform — see PY_MCP_FAMILY_CONNECTIONS (keyed by framework, not a\n  // boolean, so a third framework is an additive entry there).\n  const framework = resolveAgentFramework(agentComponent.framework);\n  const connection = PY_MCP_FAMILY_CONNECTIONS[framework];\n\n  // 1. Ensure the shared Python agent-connection project exists + has the\n  //    MCP core client (framework Layer-2) and its shared SigV4 auth helper.\n  await ensurePythonAgentConnectionProject(tree);\n  await addPythonCoreClient(tree, 'mcp', framework);\n\n  const agentConnectionProjectDir = getPythonAgentConnectionProjectDir(tree);\n  const agentConnectionModuleName = getPythonAgentConnectionModuleName(tree);\n\n  // Shared MCP transport + signed httpx auth deps, plus whatever extra deps the\n  // framework's MCP client needs (e.g. langchain-mcp-adapters for LangChain).\n  addDependenciesToPyProjectToml(tree, agentConnectionProjectDir, [\n    'boto3',\n    'httpx',\n    'mcp',\n    ...connection.deps,\n  ]);\n\n  // 2. Generate the per-connection client into the shared agent-connection project\n  const appDir = joinPathFragments(\n    agentConnectionProjectDir,\n    agentConnectionModuleName,\n    'app',\n  );\n  generateFiles(\n    tree,\n    joinPathFragments(\n      import.meta.dirname,\n      'files',\n      connection.appTemplateSubdir,\n    ),\n    appDir,\n    {\n      mcpServerSnakeCase,\n      mcpServerClassName,\n      mcpServerPort,\n      agentConnectionModuleName,\n    },\n    { overwriteStrategy: OverwriteStrategy.KeepExisting },\n  );\n\n  // Ensure app/__init__.py exists\n  const appInitPath = joinPathFragments(appDir, '__init__.py');\n  if (!tree.exists(appInitPath)) {\n    tree.write(appInitPath, '');\n  }\n\n  // Add re-export to module __init__.py\n  const moduleInitPath = joinPathFragments(\n    agentConnectionProjectDir,\n    agentConnectionModuleName,\n    '__init__.py',\n  );\n  await addPythonReExport(\n    tree,\n    moduleInitPath,\n    `.app.${mcpServerSnakeCase}_client_${connection.clientModuleSuffix}`,\n    `${mcpServerClassName}Client${connection.clientClassSuffix}`,\n  );\n\n  // 3. Transform agent.py to add MCP client import and usage\n  const agentSourceDir = joinPathFragments(\n    sourceProject.root,\n    agentComponent.path ?? 'src',\n  );\n  const agentFilePath = joinPathFragments(agentSourceDir, 'agent.py');\n\n  if (tree.exists(agentFilePath)) {\n    const clientClassName = `${mcpServerClassName}Client${connection.clientClassSuffix}`;\n    const clientVarName = mcpServerSnakeCase;\n\n    await addPythonDestructuredImport(\n      tree,\n      agentFilePath,\n      [clientClassName],\n      agentConnectionModuleName,\n    );\n    await connection.wireClientIntoAgent(\n      tree,\n      agentFilePath,\n      clientClassName,\n      clientVarName,\n    );\n  }\n\n  // 4. Add workspace dependency from agent project to agent-connection project\n  addWorkspaceDependencyToPyProject(\n    tree,\n    sourceProject,\n    getPythonAgentConnectionProject(tree),\n  );\n\n  // 5. Set up dev target dependencies\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  await addGeneratorMetricsIfApplicable(tree, [\n    PY_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 pyAgentMcpConnectionGenerator;\n"],"names":["generateFiles","joinPathFragments","OverwriteStrategy","updateProjectConfiguration","addPythonCoreClient","addPythonReExport","ensurePythonAgentConnectionProject","getPythonAgentConnectionModuleName","getPythonAgentConnectionProject","getPythonAgentConnectionProjectDir","PY_MCP_FAMILY_CONNECTIONS","resolveAgentFramework","addPythonDestructuredImport","formatFilesInSubtree","installDependencies","addGeneratorMetricsIfApplicable","snakeCase","addDependencyToTargetIfNotPresent","getGeneratorInfo","readProjectConfigurationUnqualified","addDependenciesToPyProjectToml","addWorkspaceDependencyToPyProject","PY_AGENT_MCP_CONNECTION_GENERATOR_INFO","filename","pyAgentMcpConnectionGenerator","tree","options","sourceProject","targetProject","agentComponent","sourceComponent","mcpComponent","targetComponent","Error","auth","toLowerCase","name","mcpComponentName","mcpServerClassName","rc","mcpServerSnakeCase","mcpServerPort","port","framework","connection","agentConnectionProjectDir","agentConnectionModuleName","deps","appDir","dirname","appTemplateSubdir","overwriteStrategy","KeepExisting","appInitPath","exists","write","moduleInitPath","clientModuleSuffix","clientClassSuffix","agentSourceDir","root","path","agentFilePath","clientClassName","clientVarName","wireClientIntoAgent","agentName","devTargetName","mcpDevTargetName","targets","projects","target","preferInstallDependencies","languages"],"mappings":"AAAA;;;CAGC,GACD,SAEEA,aAAa,EACbC,iBAAiB,EACjBC,iBAAiB,EAEjBC,0BAA0B,QACrB,aAAa;AACpB,SACEC,mBAAmB,EACnBC,iBAAiB,EACjBC,kCAAkC,EAClCC,kCAAkC,EAClCC,+BAA+B,EAC/BC,kCAAkC,EAClCC,yBAAyB,EACzBC,qBAAqB,QAChB,sDAAmD;AAC1D,SAASC,2BAA2B,QAAQ,wBAAqB;AACjE,SAASC,oBAAoB,QAAQ,2BAAwB;AAC7D,SAASC,mBAAmB,QAAQ,4BAAyB;AAC7D,SAASC,+BAA+B,QAAQ,4BAAyB;AACzE,SAASC,SAAS,QAAQ,0BAAuB;AACjD,SACEC,iCAAiC,EACjCC,gBAAgB,EAEhBC,mCAAmC,QAC9B,uBAAoB;AAC3B,SACEC,8BAA8B,EAC9BC,iCAAiC,QAC5B,uBAAoB;AAG3B,OAAO,MAAMC,yCACXJ,iBAAiB,YAAYK,QAAQ,EAAE;AAEzC,OAAO,MAAMC,gCAAgC,OAC3CC,MACAC;IAEA,MAAMC,gBAAgBR,oCACpBM,MACAC,QAAQC,aAAa;IAEvB,MAAMC,gBAAgBT,oCACpBM,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,8EAA8E;IAC9E,gFAAgF;IAChF,6DAA6D;IAC7D,MAAMC,YAAYhC,sBAAsBkB,eAAec,SAAS;IAChE,MAAMC,aAAalC,yBAAyB,CAACiC,UAAU;IAEvD,wEAAwE;IACxE,2EAA2E;IAC3E,MAAMrC,mCAAmCmB;IACzC,MAAMrB,oBAAoBqB,MAAM,OAAOkB;IAEvC,MAAME,4BAA4BpC,mCAAmCgB;IACrE,MAAMqB,4BAA4BvC,mCAAmCkB;IAErE,8EAA8E;IAC9E,4EAA4E;IAC5EL,+BAA+BK,MAAMoB,2BAA2B;QAC9D;QACA;QACA;WACGD,WAAWG,IAAI;KACnB;IAED,iFAAiF;IACjF,MAAMC,SAAS/C,kBACb4C,2BACAC,2BACA;IAEF9C,cACEyB,MACAxB,kBACE,YAAYgD,OAAO,EACnB,SACAL,WAAWM,iBAAiB,GAE9BF,QACA;QACER;QACAF;QACAG;QACAK;IACF,GACA;QAAEK,mBAAmBjD,kBAAkBkD,YAAY;IAAC;IAGtD,gCAAgC;IAChC,MAAMC,cAAcpD,kBAAkB+C,QAAQ;IAC9C,IAAI,CAACvB,KAAK6B,MAAM,CAACD,cAAc;QAC7B5B,KAAK8B,KAAK,CAACF,aAAa;IAC1B;IAEA,sCAAsC;IACtC,MAAMG,iBAAiBvD,kBACrB4C,2BACAC,2BACA;IAEF,MAAMzC,kBACJoB,MACA+B,gBACA,CAAC,KAAK,EAAEhB,mBAAmB,QAAQ,EAAEI,WAAWa,kBAAkB,EAAE,EACpE,GAAGnB,mBAAmB,MAAM,EAAEM,WAAWc,iBAAiB,EAAE;IAG9D,2DAA2D;IAC3D,MAAMC,iBAAiB1D,kBACrB0B,cAAciC,IAAI,EAClB/B,eAAegC,IAAI,IAAI;IAEzB,MAAMC,gBAAgB7D,kBAAkB0D,gBAAgB;IAExD,IAAIlC,KAAK6B,MAAM,CAACQ,gBAAgB;QAC9B,MAAMC,kBAAkB,GAAGzB,mBAAmB,MAAM,EAAEM,WAAWc,iBAAiB,EAAE;QACpF,MAAMM,gBAAgBxB;QAEtB,MAAM5B,4BACJa,MACAqC,eACA;YAACC;SAAgB,EACjBjB;QAEF,MAAMF,WAAWqB,mBAAmB,CAClCxC,MACAqC,eACAC,iBACAC;IAEJ;IAEA,6EAA6E;IAC7E3C,kCACEI,MACAE,eACAnB,gCAAgCiB;IAGlC,oCAAoC;IACpC,MAAMyC,YAAYrC,eAAeO,IAAI,IAAI;IACzC,MAAM+B,gBAAgB,GAAGD,UAAU,IAAI,CAAC;IACxC,MAAME,mBAAmB,GAAG/B,iBAAiB,IAAI,CAAC;IAElD,IAAIV,cAAc0C,OAAO,EAAE,CAACF,cAAc,EAAE;QAC1ClD,kCAAkCU,eAAewC,eAAe;YAC9DG,UAAU;gBAAC1C,cAAcQ,IAAI;aAAC;YAC9BmC,QAAQH;QACV;QACAjE,2BAA2BsB,MAAME,cAAcS,IAAI,EAAET;IACvD;IAEA,MAAMZ,gCAAgCU,MAAM;QAC1CH;KACD;IAED,MAAMT,qBAAqBY;IAC3B,OAAO,IACLX,oBAAoBW,MAAMC,QAAQ8C,yBAAyB,EAAE;YAC3DC,WAAW;gBAAC;aAAa;QAC3B;AACJ,EAAE;AAEF,eAAejD,8BAA8B"}
+{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/py/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 { addPyDependencies } from '../../../utils/add-dependencies';\nimport {\n  AGENT_CONNECTION_PY_DEPENDENCIES,\n  type AgentFramework,\n  addPythonCoreClient,\n  addPythonReExport,\n  ensurePythonAgentConnectionProject,\n  getPythonAgentConnectionModuleName,\n  getPythonAgentConnectionProject,\n  getPythonAgentConnectionProjectDir,\n  PY_MCP_FAMILY_CONNECTIONS,\n  resolveAgentFramework,\n} from '../../../utils/agent-connection/agent-connection';\nimport { addPythonDestructuredImport } 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 { snakeCase } from '../../../utils/names';\nimport {\n  addComponentGeneratorMetadata,\n  addDependencyToTargetIfNotPresent,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  readProjectConfigurationUnqualified,\n} from '../../../utils/nx';\nimport { toProjectRelativePath } from '../../../utils/paths';\nimport { addWorkspaceDependencyToPyProject } from '../../../utils/py';\nimport type { PyAgentMcpConnectionGeneratorSchema } from './schema';\n\n/** The metadata this generator records, which its predicates read. */\nexport interface PyAgentMcpConnectionMetadata {\n  readonly framework: AgentFramework;\n}\n\n// The framework picks the MCP client's extra deps, so it names its branch here.\nexport const DEPENDENCIES = declareDependencies<PyAgentMcpConnectionMetadata>()(\n  {\n    py: [\n      { name: 'boto3' },\n      { name: 'httpx' },\n      { name: 'mcp' },\n      // Backs the LangChain client; it must not pull Strands in.\n      {\n        name: 'langchain-mcp-adapters',\n        when: (m) => m.framework === 'langchain',\n      },\n      ...ownedElsewhere(AGENT_CONNECTION_PY_DEPENDENCIES),\n    ],\n  },\n);\n\nexport const PY_AGENT_MCP_CONNECTION_GENERATOR_INFO: NxGeneratorInfo =\n  getGeneratorInfo(import.meta.filename);\n\nexport const pyAgentMcpConnectionGenerator = async (\n  tree: Tree,\n  options: PyAgentMcpConnectionGeneratorSchema,\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 py#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 mcpServerSnakeCase = snakeCase(mcpServerClassName);\n  const mcpServerPort = mcpComponent.port ?? 8000;\n\n  // The source agent's framework selects the client shape, dependencies and the\n  // agent.py transform — see PY_MCP_FAMILY_CONNECTIONS (keyed by framework, not a\n  // boolean, so a third framework is an additive entry there).\n  const framework = resolveAgentFramework(agentComponent.framework);\n  const connection = PY_MCP_FAMILY_CONNECTIONS[framework];\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 metadata: PyAgentMcpConnectionMetadata = { framework };\n\n  // 1. Ensure the shared Python agent-connection project exists + has the\n  //    MCP core client (framework Layer-2) and its shared SigV4 auth helper.\n  await ensurePythonAgentConnectionProject(tree, DEPENDENCIES);\n  await addPythonCoreClient(tree, 'mcp', DEPENDENCIES, framework);\n\n  const agentConnectionProjectDir = getPythonAgentConnectionProjectDir(tree);\n  const agentConnectionModuleName = getPythonAgentConnectionModuleName(tree);\n\n  // Shared MCP transport + signed httpx auth deps, plus whatever extra deps the\n  // framework's MCP client needs (e.g. langchain-mcp-adapters for LangChain).\n  // These belong to the shared agent-connection project, which holds the client.\n  addPyDependencies(tree, DEPENDENCIES, {\n    metadata,\n    projectRoot: agentConnectionProjectDir,\n  });\n\n  // 2. Generate the per-connection client into the shared agent-connection project\n  const appDir = joinPathFragments(\n    agentConnectionProjectDir,\n    agentConnectionModuleName,\n    'app',\n  );\n  generateFiles(\n    tree,\n    joinPathFragments(\n      import.meta.dirname,\n      'files',\n      connection.appTemplateSubdir,\n    ),\n    appDir,\n    {\n      mcpServerSnakeCase,\n      mcpServerClassName,\n      mcpServerPort,\n      agentConnectionModuleName,\n    },\n    { overwriteStrategy: OverwriteStrategy.KeepExisting },\n  );\n\n  // Ensure app/__init__.py exists\n  const appInitPath = joinPathFragments(appDir, '__init__.py');\n  if (!tree.exists(appInitPath)) {\n    tree.write(appInitPath, '');\n  }\n\n  // Add re-export to module __init__.py\n  const moduleInitPath = joinPathFragments(\n    agentConnectionProjectDir,\n    agentConnectionModuleName,\n    '__init__.py',\n  );\n  await addPythonReExport(\n    tree,\n    moduleInitPath,\n    `.app.${mcpServerSnakeCase}_client_${connection.clientModuleSuffix}`,\n    `${mcpServerClassName}Client${connection.clientClassSuffix}`,\n  );\n\n  // 3. Transform agent.py to add MCP client import and usage\n  const agentSourceDir = joinPathFragments(\n    sourceProject.root,\n    agentComponent.path ?? 'src',\n  );\n  const agentFilePath = joinPathFragments(agentSourceDir, 'agent.py');\n\n  if (tree.exists(agentFilePath)) {\n    const clientClassName = `${mcpServerClassName}Client${connection.clientClassSuffix}`;\n    const clientVarName = mcpServerSnakeCase;\n\n    await addPythonDestructuredImport(\n      tree,\n      agentFilePath,\n      [clientClassName],\n      agentConnectionModuleName,\n    );\n    await connection.wireClientIntoAgent(\n      tree,\n      agentFilePath,\n      clientClassName,\n      clientVarName,\n    );\n  }\n\n  // 4. Add workspace dependency from agent project to agent-connection project\n  addWorkspaceDependencyToPyProject(\n    tree,\n    sourceProject,\n    getPythonAgentConnectionProject(tree),\n  );\n\n  // 5. Set up dev target dependencies\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  // Recorded so the version sync knows this connection's dependencies are ours.\n  addComponentGeneratorMetadata(\n    tree,\n    sourceProject.name,\n    PY_AGENT_MCP_CONNECTION_GENERATOR_INFO,\n    toProjectRelativePath(sourceProject, agentFilePath),\n    mcpServerClassName,\n    // `sourcePath` names the source component this connection is made from, so\n    // the pair is identifiable rather than just the two projects.\n    { ...metadata, sourcePath: options.sourceComponent?.path },\n  );\n\n  await addGeneratorMetricsIfApplicable(tree, [\n    PY_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 pyAgentMcpConnectionGenerator;\n"],"names":["generateFiles","joinPathFragments","OverwriteStrategy","updateProjectConfiguration","addPyDependencies","AGENT_CONNECTION_PY_DEPENDENCIES","addPythonCoreClient","addPythonReExport","ensurePythonAgentConnectionProject","getPythonAgentConnectionModuleName","getPythonAgentConnectionProject","getPythonAgentConnectionProjectDir","PY_MCP_FAMILY_CONNECTIONS","resolveAgentFramework","addPythonDestructuredImport","declareDependencies","ownedElsewhere","formatFilesInSubtree","installDependencies","addGeneratorMetricsIfApplicable","snakeCase","addComponentGeneratorMetadata","addDependencyToTargetIfNotPresent","getGeneratorInfo","readProjectConfigurationUnqualified","toProjectRelativePath","addWorkspaceDependencyToPyProject","DEPENDENCIES","py","name","when","m","framework","PY_AGENT_MCP_CONNECTION_GENERATOR_INFO","filename","pyAgentMcpConnectionGenerator","tree","options","sourceProject","targetProject","agentComponent","sourceComponent","mcpComponent","targetComponent","Error","auth","toLowerCase","mcpComponentName","mcpServerClassName","rc","mcpServerSnakeCase","mcpServerPort","port","connection","metadata","agentConnectionProjectDir","agentConnectionModuleName","projectRoot","appDir","dirname","appTemplateSubdir","overwriteStrategy","KeepExisting","appInitPath","exists","write","moduleInitPath","clientModuleSuffix","clientClassSuffix","agentSourceDir","root","path","agentFilePath","clientClassName","clientVarName","wireClientIntoAgent","agentName","devTargetName","mcpDevTargetName","targets","projects","target","sourcePath","preferInstallDependencies","languages"],"mappings":"AAAA;;;CAGC,GACD,SAEEA,aAAa,EACbC,iBAAiB,EACjBC,iBAAiB,EAEjBC,0BAA0B,QACrB,aAAa;AACpB,SAASC,iBAAiB,QAAQ,qCAAkC;AACpE,SACEC,gCAAgC,EAEhCC,mBAAmB,EACnBC,iBAAiB,EACjBC,kCAAkC,EAClCC,kCAAkC,EAClCC,+BAA+B,EAC/BC,kCAAkC,EAClCC,yBAAyB,EACzBC,qBAAqB,QAChB,sDAAmD;AAC1D,SAASC,2BAA2B,QAAQ,wBAAqB;AACjE,SACEC,mBAAmB,EACnBC,cAAc,QACT,0CAAuC;AAC9C,SAASC,oBAAoB,QAAQ,2BAAwB;AAC7D,SAASC,mBAAmB,QAAQ,4BAAyB;AAC7D,SAASC,+BAA+B,QAAQ,4BAAyB;AACzE,SAASC,SAAS,QAAQ,0BAAuB;AACjD,SACEC,6BAA6B,EAC7BC,iCAAiC,EACjCC,gBAAgB,EAEhBC,mCAAmC,QAC9B,uBAAoB;AAC3B,SAASC,qBAAqB,QAAQ,0BAAuB;AAC7D,SAASC,iCAAiC,QAAQ,uBAAoB;AAQtE,gFAAgF;AAChF,OAAO,MAAMC,eAAeZ,sBAC1B;IACEa,IAAI;QACF;YAAEC,MAAM;QAAQ;QAChB;YAAEA,MAAM;QAAQ;QAChB;YAAEA,MAAM;QAAM;QACd,2DAA2D;QAC3D;YACEA,MAAM;YACNC,MAAM,CAACC,IAAMA,EAAEC,SAAS,KAAK;QAC/B;WACGhB,eAAeX;KACnB;AACH,GACA;AAEF,OAAO,MAAM4B,yCACXV,iBAAiB,YAAYW,QAAQ,EAAE;AAEzC,OAAO,MAAMC,gCAAgC,OAC3CC,MACAC;IAEA,MAAMC,gBAAgBd,oCACpBY,MACAC,QAAQC,aAAa;IAEvB,MAAMC,gBAAgBf,oCACpBY,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,aAAab,IAAI,CAAC,QAAQ,EAAEa,aAAaG,IAAI,CAAC,iBAAiB,CAAC;IAE9I;IAEA,MAAME,mBAAmBL,aAAab,IAAI,IAAI;IAC9C,MAAMmB,qBAAqBN,aAAaO,EAAE;IAC1C,MAAMC,qBAAqB9B,UAAU4B;IACrC,MAAMG,gBAAgBT,aAAaU,IAAI,IAAI;IAE3C,8EAA8E;IAC9E,gFAAgF;IAChF,6DAA6D;IAC7D,MAAMpB,YAAYnB,sBAAsB2B,eAAeR,SAAS;IAChE,MAAMqB,aAAazC,yBAAyB,CAACoB,UAAU;IAEvD,2EAA2E;IAC3E,6DAA6D;IAC7D,MAAMsB,WAAyC;QAAEtB;IAAU;IAE3D,wEAAwE;IACxE,2EAA2E;IAC3E,MAAMxB,mCAAmC4B,MAAMT;IAC/C,MAAMrB,oBAAoB8B,MAAM,OAAOT,cAAcK;IAErD,MAAMuB,4BAA4B5C,mCAAmCyB;IACrE,MAAMoB,4BAA4B/C,mCAAmC2B;IAErE,8EAA8E;IAC9E,4EAA4E;IAC5E,+EAA+E;IAC/EhC,kBAAkBgC,MAAMT,cAAc;QACpC2B;QACAG,aAAaF;IACf;IAEA,iFAAiF;IACjF,MAAMG,SAASzD,kBACbsD,2BACAC,2BACA;IAEFxD,cACEoC,MACAnC,kBACE,YAAY0D,OAAO,EACnB,SACAN,WAAWO,iBAAiB,GAE9BF,QACA;QACER;QACAF;QACAG;QACAK;IACF,GACA;QAAEK,mBAAmB3D,kBAAkB4D,YAAY;IAAC;IAGtD,gCAAgC;IAChC,MAAMC,cAAc9D,kBAAkByD,QAAQ;IAC9C,IAAI,CAACtB,KAAK4B,MAAM,CAACD,cAAc;QAC7B3B,KAAK6B,KAAK,CAACF,aAAa;IAC1B;IAEA,sCAAsC;IACtC,MAAMG,iBAAiBjE,kBACrBsD,2BACAC,2BACA;IAEF,MAAMjD,kBACJ6B,MACA8B,gBACA,CAAC,KAAK,EAAEhB,mBAAmB,QAAQ,EAAEG,WAAWc,kBAAkB,EAAE,EACpE,GAAGnB,mBAAmB,MAAM,EAAEK,WAAWe,iBAAiB,EAAE;IAG9D,2DAA2D;IAC3D,MAAMC,iBAAiBpE,kBACrBqC,cAAcgC,IAAI,EAClB9B,eAAe+B,IAAI,IAAI;IAEzB,MAAMC,gBAAgBvE,kBAAkBoE,gBAAgB;IAExD,IAAIjC,KAAK4B,MAAM,CAACQ,gBAAgB;QAC9B,MAAMC,kBAAkB,GAAGzB,mBAAmB,MAAM,EAAEK,WAAWe,iBAAiB,EAAE;QACpF,MAAMM,gBAAgBxB;QAEtB,MAAMpC,4BACJsB,MACAoC,eACA;YAACC;SAAgB,EACjBjB;QAEF,MAAMH,WAAWsB,mBAAmB,CAClCvC,MACAoC,eACAC,iBACAC;IAEJ;IAEA,6EAA6E;IAC7EhD,kCACEU,MACAE,eACA5B,gCAAgC0B;IAGlC,oCAAoC;IACpC,MAAMwC,YAAYpC,eAAeX,IAAI,IAAI;IACzC,MAAMgD,gBAAgB,GAAGD,UAAU,IAAI,CAAC;IACxC,MAAME,mBAAmB,GAAG/B,iBAAiB,IAAI,CAAC;IAElD,IAAIT,cAAcyC,OAAO,EAAE,CAACF,cAAc,EAAE;QAC1CvD,kCAAkCgB,eAAeuC,eAAe;YAC9DG,UAAU;gBAACzC,cAAcV,IAAI;aAAC;YAC9BoD,QAAQH;QACV;QACA3E,2BAA2BiC,MAAME,cAAcT,IAAI,EAAES;IACvD;IAEA,8EAA8E;IAC9EjB,8BACEe,MACAE,cAAcT,IAAI,EAClBI,wCACAR,sBAAsBa,eAAekC,gBACrCxB,oBACA,2EAA2E;IAC3E,8DAA8D;IAC9D;QAAE,GAAGM,QAAQ;QAAE4B,YAAY7C,QAAQI,eAAe,EAAE8B;IAAK;IAG3D,MAAMpD,gCAAgCiB,MAAM;QAC1CH;KACD;IAED,MAAMhB,qBAAqBmB;IAC3B,OAAO,IACLlB,oBAAoBkB,MAAMC,QAAQ8C,yBAAyB,EAAE;YAC3DC,WAAW;gBAAC;aAAa;QAC3B;AACJ,EAAE;AAEF,eAAejD,8BAA8B"}

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

@@ -5,6 +5,152 @@
 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 PyAgentReactConnectionMetadata {
+    readonly protocol: string;
+    readonly auth: 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 [Omit<{
+    readonly name: "oidc-client-ts";
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "react-oidc-context";
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "@aws-sdk/credential-provider-cognito-identity";
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "aws4fetch";
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "@tanstack/react-query";
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "@tanstack/react-query-devtools";
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "@smithy/types";
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "@copilotkit/react-core";
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "@ag-ui/client";
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "@cloudscape-design/chat-components";
+    readonly when: (m: import("../../../ts/react-website/agui/generator").AgUiMetadata) => boolean;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "lucide-react";
+    readonly when: (m: import("../../../ts/react-website/agui/generator").AgUiMetadata) => boolean;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "oidc-client-ts";
+    readonly when: (m: import("../../../ts/react-website/agui/generator").AgUiMetadata) => boolean;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "aws4fetch";
+    readonly when: (m: import("../../../ts/react-website/agui/generator").AgUiMetadata) => boolean;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "@aws-sdk/credential-provider-cognito-identity";
+    readonly when: (m: import("../../../ts/react-website/agui/generator").AgUiMetadata) => boolean;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "react-oidc-context";
+    readonly when: (m: import("../../../ts/react-website/agui/generator").AgUiMetadata) => boolean;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "@smithy/types";
+    readonly when: (m: import("../../../ts/react-website/agui/generator").AgUiMetadata) => boolean;
+    readonly dev: true;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "react";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "react-dom";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "class-variance-authority";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "clsx";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "tailwind-merge";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "lucide-react";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "tw-animate-css";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "radix-ui";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "vite";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "vitest";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "@vitest/coverage-v8";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}], 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", PyAgentReactConnectionMetadata>[]>;
 export declare const PY_AGENT_REACT_CONNECTION_GENERATOR_INFO: NxGeneratorInfo;
 export declare const pyAgentReactConnectionGenerator: (tree: Tree, options: ResolvedConnectionOptions) => Promise<() => Promise<void>>;
 export default pyAgentReactConnectionGenerator;

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

@@ -3,15 +3,29 @@
  * SPDX-License-Identifier: Apache-2.0
  */ import { generateFiles, joinPathFragments, OverwriteStrategy, updateProjectConfiguration } from "@nx/devkit";
 import { addAgentRuntimeToConnectionNamespace } from "../../../connection/agent-runtime-config.js";
-import { addAgUiReactConnection } from "../../../ts/react-website/agui/generator.js";
-import { addOpenApiReactClient } from "../../../utils/connection/open-api/react.js";
+import { DEPENDENCIES as AGUI_DEPENDENCIES, addAgUiReactConnection, resolveAgUiTheme } from "../../../ts/react-website/agui/generator.js";
+import { addOpenApiReactClient, OPEN_API_REACT_DEPENDENCIES } from "../../../utils/connection/open-api/react.js";
+import { declareDependencies, onlyWhen } 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, toSnakeCase } from "../../../utils/names.js";
-import { getGeneratorInfo, readProjectConfigurationUnqualified } from "../../../utils/nx.js";
+import { addComponentGeneratorMetadata, getGeneratorInfo, readProjectConfigurationUnqualified } from "../../../utils/nx.js";
 import { sortObjectKeys } from "../../../utils/object.js";
+import { toProjectRelativePath } from "../../../utils/paths.js";
 import { addPyAgentTargetToLocalDev, openApiClientLocalDevDeps } from "./local-dev.js";
+/** The OpenAPI-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 OpenAPI
+// client packages are added by `addOpenApiReactClient`, the AG-UI path's by
+// `addAgUiReactConnection`, both on this generator's behalf. The AG-UI entries
+// keep their own theme and auth conditions, which the metadata below records.
+export const DEPENDENCIES = declareDependencies()({
+    ts: [
+        ...onlyWhen(OPEN_API_REACT_DEPENDENCIES, isHttp),
+        ...onlyWhen(AGUI_DEPENDENCIES.ts, isAgUi)
+    ]
+});
 export const PY_AGENT_REACT_CONNECTION_GENERATOR_INFO = getGeneratorInfo(import.meta.filename);
 export const pyAgentReactConnectionGenerator = async (tree, options)=>{
     const frontendProjectConfig = readProjectConfigurationUnqualified(tree, options.sourceProject);
@@ -24,6 +38,13 @@ export const pyAgentReactConnectionGenerator = async (tree, options)=>{
     const agentPort = targetComponent?.port ?? metadata?.ports?.[0] ?? 8081;
     const auth = (targetComponent?.auth ?? metadata?.auth ?? 'iam').toLowerCase();
     const protocol = (targetComponent?.protocol ?? 'http').toLowerCase();
+    // 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 = {
+        protocol,
+        auth,
+        theme: resolveAgUiTheme(frontendProjectConfig)
+    };
     if (protocol === '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.`);
     }
@@ -81,7 +102,7 @@ export const pyAgentReactConnectionGenerator = async (tree, options)=>{
             port: agentPort,
             isAgentRuntime: true,
             skipLocalDev: true
-        });
+        }, DEPENDENCIES);
         additionalLocalDevDeps = openApiClientLocalDevDeps(agentNameClassName);
         // HTTP only — the AG-UI branch handles this inside addAgUiReactConnection.
         // Agent constructs publish their runtime ARN to the 'agentcore' namespace
@@ -99,6 +120,8 @@ export const pyAgentReactConnectionGenerator = async (tree, options)=>{
         targetComponent,
         additionalDependencyTargets: additionalLocalDevDeps
     });
+    // Recorded so the version sync knows this connection's dependencies are ours.
+    addComponentGeneratorMetadata(tree, frontendProjectConfig.name, PY_AGENT_REACT_CONNECTION_GENERATOR_INFO, toProjectRelativePath(frontendProjectConfig, protocol === 'ag-ui' ? joinPathFragments(frontendProjectConfig.sourceRoot, 'hooks', `useAgui${agentNameClassName}`) : joinPathFragments(frontendProjectConfig.sourceRoot, 'components', `${agentNameClassName}Provider`)), agentNameClassName, connectionMetadata);
     await addGeneratorMetricsIfApplicable(tree, [
         PY_AGENT_REACT_CONNECTION_GENERATOR_INFO
     ]);

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/py/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  updateProjectConfiguration,\n} from '@nx/devkit';\nimport { addAgentRuntimeToConnectionNamespace } from '../../../connection/agent-runtime-config';\nimport type { ResolvedConnectionOptions } from '../../../connection/generator';\nimport {\n  type AgUiAuth,\n  addAgUiReactConnection,\n} from '../../../ts/react-website/agui/generator';\nimport { addOpenApiReactClient } from '../../../utils/connection/open-api/react';\nimport { formatFilesInSubtree } from '../../../utils/format';\nimport { installDependencies } from '../../../utils/install';\nimport { addGeneratorMetricsIfApplicable } from '../../../utils/metrics';\nimport { kebabCase, toClassName, toSnakeCase } from '../../../utils/names';\nimport {\n  type ComponentMetadata,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  readProjectConfigurationUnqualified,\n} from '../../../utils/nx';\nimport { sortObjectKeys } from '../../../utils/object';\nimport {\n  addPyAgentTargetToLocalDev,\n  openApiClientLocalDevDeps,\n} from './local-dev';\n\nexport const PY_AGENT_REACT_CONNECTION_GENERATOR_INFO: NxGeneratorInfo =\n  getGeneratorInfo(import.meta.filename);\n\nexport const pyAgentReactConnectionGenerator = async (\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 protocol = (targetComponent?.protocol ?? 'http').toLowerCase();\n\n  if (protocol === '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  let additionalLocalDevDeps: string[] = [];\n\n  if (protocol === 'ag-ui') {\n    await addAgUiReactConnection(tree, {\n      frontendProjectConfig,\n      agentName,\n      agentNameClassName,\n      auth: auth as AgUiAuth,\n    });\n  } else {\n    const moduleName = getModuleName(agentProjectConfig);\n    const agentNameSnakeCase = toSnakeCase(agentName);\n    const agentTargetPrefix = targetComponent?.name ? agentName : 'agent';\n\n    // Add OpenAPI spec generation script scoped to this agent\n    generateFiles(\n      tree,\n      joinPathFragments(import.meta.dirname, 'files/agent'),\n      agentProjectConfig.root,\n      {\n        moduleName,\n        agentNameSnakeCase,\n      },\n      {\n        overwriteStrategy: OverwriteStrategy.KeepExisting,\n      },\n    );\n\n    // Instrument the OpenAPI spec generation as a target on the agent project\n    const openApiTargetName = `${agentTargetPrefix}-openapi`;\n    const openApiDist = joinPathFragments(\n      'dist',\n      agentProjectConfig.root,\n      'openapi',\n      agentNameSnakeCase,\n    );\n    const specPath = joinPathFragments(openApiDist, 'openapi.json');\n\n    updateProjectConfiguration(tree, agentProjectConfig.name, {\n      ...agentProjectConfig,\n      targets: sortObjectKeys({\n        ...agentProjectConfig.targets,\n        [openApiTargetName]: {\n          cache: true,\n          executor: 'nx:run-commands',\n          outputs: [\n            `{workspaceRoot}/dist/{projectRoot}/openapi/${agentNameSnakeCase}`,\n          ],\n          options: {\n            commands: [\n              `uv run python {projectRoot}/scripts/${agentNameSnakeCase}_openapi.py \"dist/{projectRoot}/openapi/${agentNameSnakeCase}/openapi.json\"`,\n            ],\n          },\n        },\n      }),\n    });\n\n    // Use the shared OpenAPI react client utility for hooks, providers, and build targets.\n    // The dev target is handled separately below using the agent-specific dev target.\n    await addOpenApiReactClient(tree, {\n      apiName: agentNameClassName,\n      frontendProjectConfig,\n      backendProjectConfig: agentProjectConfig,\n      specBuildProject: agentProjectConfig,\n      specPath,\n      specBuildTargetName: `${agentProjectConfig.name}:${openApiTargetName}`,\n      auth,\n      port: agentPort,\n      isAgentRuntime: true,\n      skipLocalDev: true,\n    });\n\n    additionalLocalDevDeps = openApiClientLocalDevDeps(agentNameClassName);\n\n    // HTTP only — the AG-UI branch handles this inside addAgUiReactConnection.\n    // Agent constructs publish their runtime ARN to the 'agentcore' namespace\n    // by default, which isn't exposed to the website; patch them to also\n    // publish under 'connection' so the browser can read it.\n    await addAgentRuntimeToConnectionNamespace(tree, {\n      agentNameKebabCase: kebabCase(agentNameClassName),\n      agentNameClassName,\n    });\n  }\n\n  await addPyAgentTargetToLocalDev(\n    tree,\n    frontendProjectConfig.name,\n    agentProjectConfig.name,\n    {\n      agentName,\n      agentNameClassName,\n      port: agentPort,\n      targetComponent,\n      additionalDependencyTargets: additionalLocalDevDeps,\n    },\n  );\n\n  await addGeneratorMetricsIfApplicable(tree, [\n    PY_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/**\n * Determine the Python module name from the project configuration\n */\nconst getModuleName = (\n  projectConfig: ReturnType<typeof readProjectConfigurationUnqualified>,\n): string => {\n  if (projectConfig.sourceRoot) {\n    const sourceRootParts = projectConfig.sourceRoot.split('/');\n    return sourceRootParts[sourceRootParts.length - 1];\n  }\n  throw new Error(\n    `Could not determine sourceRoot for project ${projectConfig.name}`,\n  );\n};\n\nexport default pyAgentReactConnectionGenerator;\n"],"names":["generateFiles","joinPathFragments","OverwriteStrategy","updateProjectConfiguration","addAgentRuntimeToConnectionNamespace","addAgUiReactConnection","addOpenApiReactClient","formatFilesInSubtree","installDependencies","addGeneratorMetricsIfApplicable","kebabCase","toClassName","toSnakeCase","getGeneratorInfo","readProjectConfigurationUnqualified","sortObjectKeys","addPyAgentTargetToLocalDev","openApiClientLocalDevDeps","PY_AGENT_REACT_CONNECTION_GENERATOR_INFO","filename","pyAgentReactConnectionGenerator","tree","options","frontendProjectConfig","sourceProject","agentProjectConfig","targetProject","targetComponent","metadata","agentName","name","agentNameClassName","rc","agentPort","port","ports","auth","toLowerCase","protocol","Error","additionalLocalDevDeps","moduleName","getModuleName","agentNameSnakeCase","agentTargetPrefix","dirname","root","overwriteStrategy","KeepExisting","openApiTargetName","openApiDist","specPath","targets","cache","executor","outputs","commands","apiName","backendProjectConfig","specBuildProject","specBuildTargetName","isAgentRuntime","skipLocalDev","agentNameKebabCase","additionalDependencyTargets","preferInstallDependencies","languages","projectConfig","sourceRoot","sourceRootParts","split","length"],"mappings":"AAAA;;;CAGC,GACD,SACEA,aAAa,EACbC,iBAAiB,EACjBC,iBAAiB,EAEjBC,0BAA0B,QACrB,aAAa;AACpB,SAASC,oCAAoC,QAAQ,8CAA2C;AAEhG,SAEEC,sBAAsB,QACjB,8CAA2C;AAClD,SAASC,qBAAqB,QAAQ,8CAA2C;AACjF,SAASC,oBAAoB,QAAQ,2BAAwB;AAC7D,SAASC,mBAAmB,QAAQ,4BAAyB;AAC7D,SAASC,+BAA+B,QAAQ,4BAAyB;AACzE,SAASC,SAAS,EAAEC,WAAW,EAAEC,WAAW,QAAQ,0BAAuB;AAC3E,SAEEC,gBAAgB,EAEhBC,mCAAmC,QAC9B,uBAAoB;AAC3B,SAASC,cAAc,QAAQ,2BAAwB;AACvD,SACEC,0BAA0B,EAC1BC,yBAAyB,QACpB,iBAAc;AAErB,OAAO,MAAMC,2CACXL,iBAAiB,YAAYM,QAAQ,EAAE;AAEzC,OAAO,MAAMC,kCAAkC,OAC7CC,MACAC;IAEA,MAAMC,wBAAwBT,oCAC5BO,MACAC,QAAQE,aAAa;IAEvB,MAAMC,qBAAqBX,oCACzBO,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,WAAW,AAACX,CAAAA,iBAAiBW,YAAY,MAAK,EAAGD,WAAW;IAElE,IAAIC,aAAa,OAAO;QACtB,MAAM,IAAIC,MACR,CAAC,gDAAgD,CAAC,GAChD,CAAC,qEAAqE,CAAC;IAE7E;IAEA,IAAIC,yBAAmC,EAAE;IAEzC,IAAIF,aAAa,SAAS;QACxB,MAAMjC,uBAAuBgB,MAAM;YACjCE;YACAM;YACAE;YACAK,MAAMA;QACR;IACF,OAAO;QACL,MAAMK,aAAaC,cAAcjB;QACjC,MAAMkB,qBAAqB/B,YAAYiB;QACvC,MAAMe,oBAAoBjB,iBAAiBG,OAAOD,YAAY;QAE9D,0DAA0D;QAC1D7B,cACEqB,MACApB,kBAAkB,YAAY4C,OAAO,EAAE,gBACvCpB,mBAAmBqB,IAAI,EACvB;YACEL;YACAE;QACF,GACA;YACEI,mBAAmB7C,kBAAkB8C,YAAY;QACnD;QAGF,0EAA0E;QAC1E,MAAMC,oBAAoB,GAAGL,kBAAkB,QAAQ,CAAC;QACxD,MAAMM,cAAcjD,kBAClB,QACAwB,mBAAmBqB,IAAI,EACvB,WACAH;QAEF,MAAMQ,WAAWlD,kBAAkBiD,aAAa;QAEhD/C,2BAA2BkB,MAAMI,mBAAmBK,IAAI,EAAE;YACxD,GAAGL,kBAAkB;YACrB2B,SAASrC,eAAe;gBACtB,GAAGU,mBAAmB2B,OAAO;gBAC7B,CAACH,kBAAkB,EAAE;oBACnBI,OAAO;oBACPC,UAAU;oBACVC,SAAS;wBACP,CAAC,2CAA2C,EAAEZ,oBAAoB;qBACnE;oBACDrB,SAAS;wBACPkC,UAAU;4BACR,CAAC,oCAAoC,EAAEb,mBAAmB,wCAAwC,EAAEA,mBAAmB,cAAc,CAAC;yBACvI;oBACH;gBACF;YACF;QACF;QAEA,uFAAuF;QACvF,kFAAkF;QAClF,MAAMrC,sBAAsBe,MAAM;YAChCoC,SAAS1B;YACTR;YACAmC,sBAAsBjC;YACtBkC,kBAAkBlC;YAClB0B;YACAS,qBAAqB,GAAGnC,mBAAmBK,IAAI,CAAC,CAAC,EAAEmB,mBAAmB;YACtEb;YACAF,MAAMD;YACN4B,gBAAgB;YAChBC,cAAc;QAChB;QAEAtB,yBAAyBvB,0BAA0Bc;QAEnD,2EAA2E;QAC3E,0EAA0E;QAC1E,qEAAqE;QACrE,yDAAyD;QACzD,MAAM3B,qCAAqCiB,MAAM;YAC/C0C,oBAAoBrD,UAAUqB;YAC9BA;QACF;IACF;IAEA,MAAMf,2BACJK,MACAE,sBAAsBO,IAAI,EAC1BL,mBAAmBK,IAAI,EACvB;QACED;QACAE;QACAG,MAAMD;QACNN;QACAqC,6BAA6BxB;IAC/B;IAGF,MAAM/B,gCAAgCY,MAAM;QAC1CH;KACD;IAED,MAAMX,qBAAqBc;IAC3B,OAAO,IACLb,oBAAoBa,MAAMC,QAAQ2C,yBAAyB,EAAE;YAC3DC,WAAW;gBAAC;aAAa;QAC3B;AACJ,EAAE;AAEF;;CAEC,GACD,MAAMxB,gBAAgB,CACpByB;IAEA,IAAIA,cAAcC,UAAU,EAAE;QAC5B,MAAMC,kBAAkBF,cAAcC,UAAU,CAACE,KAAK,CAAC;QACvD,OAAOD,eAAe,CAACA,gBAAgBE,MAAM,GAAG,EAAE;IACpD;IACA,MAAM,IAAIhC,MACR,CAAC,2CAA2C,EAAE4B,cAAcrC,IAAI,EAAE;AAEtE;AAEA,eAAeV,gCAAgC"}
+{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/py/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  updateProjectConfiguration,\n} from '@nx/devkit';\nimport { addAgentRuntimeToConnectionNamespace } from '../../../connection/agent-runtime-config';\nimport type { ResolvedConnectionOptions } from '../../../connection/generator';\nimport {\n  DEPENDENCIES as AGUI_DEPENDENCIES,\n  type AgUiAuth,\n  addAgUiReactConnection,\n  resolveAgUiTheme,\n} from '../../../ts/react-website/agui/generator';\nimport {\n  addOpenApiReactClient,\n  OPEN_API_REACT_DEPENDENCIES,\n} from '../../../utils/connection/open-api/react';\nimport {\n  declareDependencies,\n  onlyWhen,\n} from '../../../utils/declared-dependencies';\nimport { formatFilesInSubtree } from '../../../utils/format';\nimport { installDependencies } from '../../../utils/install';\nimport { addGeneratorMetricsIfApplicable } from '../../../utils/metrics';\nimport { kebabCase, toClassName, toSnakeCase } from '../../../utils/names';\nimport {\n  addComponentGeneratorMetadata,\n  type ComponentMetadata,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  readProjectConfigurationUnqualified,\n} from '../../../utils/nx';\nimport { sortObjectKeys } from '../../../utils/object';\nimport { toProjectRelativePath } from '../../../utils/paths';\nimport {\n  addPyAgentTargetToLocalDev,\n  openApiClientLocalDevDeps,\n} from './local-dev';\n\n/** The metadata this generator records, which its predicates read. */\nexport interface PyAgentReactConnectionMetadata {\n  readonly protocol: string;\n  readonly auth: string;\n  /** The AG-UI theme module, which the website's `ux` selects. */\n  readonly theme: string;\n}\n\n/** The OpenAPI-over-HTTP path, which the AG-UI path takes none of. */\nconst isHttp = (m: PyAgentReactConnectionMetadata) => m.protocol !== 'ag-ui';\n\n/** The AG-UI path, whose packages `addAgUiReactConnection` adds. */\nconst isAgUi = (m: PyAgentReactConnectionMetadata) => !isHttp(m);\n\n// Each entry names the protocol path it belongs to: the HTTP path's OpenAPI\n// client packages are added by `addOpenApiReactClient`, the AG-UI path's by\n// `addAgUiReactConnection`, both on this generator's behalf. The AG-UI entries\n// keep their own theme and auth conditions, which the metadata below records.\nexport const DEPENDENCIES =\n  declareDependencies<PyAgentReactConnectionMetadata>()({\n    ts: [\n      ...onlyWhen(OPEN_API_REACT_DEPENDENCIES, isHttp),\n      ...onlyWhen(AGUI_DEPENDENCIES.ts, isAgUi),\n    ],\n  });\n\nexport const PY_AGENT_REACT_CONNECTION_GENERATOR_INFO: NxGeneratorInfo =\n  getGeneratorInfo(import.meta.filename);\n\nexport const pyAgentReactConnectionGenerator = async (\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 protocol = (targetComponent?.protocol ?? 'http').toLowerCase();\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: PyAgentReactConnectionMetadata = {\n    protocol,\n    auth,\n    theme: resolveAgUiTheme(frontendProjectConfig),\n  };\n\n  if (protocol === '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  let additionalLocalDevDeps: string[] = [];\n\n  if (protocol === 'ag-ui') {\n    await addAgUiReactConnection(tree, {\n      frontendProjectConfig,\n      agentName,\n      agentNameClassName,\n      auth: auth as AgUiAuth,\n    });\n  } else {\n    const moduleName = getModuleName(agentProjectConfig);\n    const agentNameSnakeCase = toSnakeCase(agentName);\n    const agentTargetPrefix = targetComponent?.name ? agentName : 'agent';\n\n    // Add OpenAPI spec generation script scoped to this agent\n    generateFiles(\n      tree,\n      joinPathFragments(import.meta.dirname, 'files/agent'),\n      agentProjectConfig.root,\n      {\n        moduleName,\n        agentNameSnakeCase,\n      },\n      {\n        overwriteStrategy: OverwriteStrategy.KeepExisting,\n      },\n    );\n\n    // Instrument the OpenAPI spec generation as a target on the agent project\n    const openApiTargetName = `${agentTargetPrefix}-openapi`;\n    const openApiDist = joinPathFragments(\n      'dist',\n      agentProjectConfig.root,\n      'openapi',\n      agentNameSnakeCase,\n    );\n    const specPath = joinPathFragments(openApiDist, 'openapi.json');\n\n    updateProjectConfiguration(tree, agentProjectConfig.name, {\n      ...agentProjectConfig,\n      targets: sortObjectKeys({\n        ...agentProjectConfig.targets,\n        [openApiTargetName]: {\n          cache: true,\n          executor: 'nx:run-commands',\n          outputs: [\n            `{workspaceRoot}/dist/{projectRoot}/openapi/${agentNameSnakeCase}`,\n          ],\n          options: {\n            commands: [\n              `uv run python {projectRoot}/scripts/${agentNameSnakeCase}_openapi.py \"dist/{projectRoot}/openapi/${agentNameSnakeCase}/openapi.json\"`,\n            ],\n          },\n        },\n      }),\n    });\n\n    // Use the shared OpenAPI react client utility for hooks, providers, and build targets.\n    // The dev target is handled separately below using the agent-specific dev target.\n    await addOpenApiReactClient(\n      tree,\n      {\n        apiName: agentNameClassName,\n        frontendProjectConfig,\n        backendProjectConfig: agentProjectConfig,\n        specBuildProject: agentProjectConfig,\n        specPath,\n        specBuildTargetName: `${agentProjectConfig.name}:${openApiTargetName}`,\n        auth,\n        port: agentPort,\n        isAgentRuntime: true,\n        skipLocalDev: true,\n      },\n      DEPENDENCIES,\n    );\n\n    additionalLocalDevDeps = openApiClientLocalDevDeps(agentNameClassName);\n\n    // HTTP only — the AG-UI branch handles this inside addAgUiReactConnection.\n    // Agent constructs publish their runtime ARN to the 'agentcore' namespace\n    // by default, which isn't exposed to the website; patch them to also\n    // publish under 'connection' so the browser can read it.\n    await addAgentRuntimeToConnectionNamespace(tree, {\n      agentNameKebabCase: kebabCase(agentNameClassName),\n      agentNameClassName,\n    });\n  }\n\n  await addPyAgentTargetToLocalDev(\n    tree,\n    frontendProjectConfig.name,\n    agentProjectConfig.name,\n    {\n      agentName,\n      agentNameClassName,\n      port: agentPort,\n      targetComponent,\n      additionalDependencyTargets: additionalLocalDevDeps,\n    },\n  );\n\n  // Recorded so the version sync knows this connection's dependencies are ours.\n  addComponentGeneratorMetadata(\n    tree,\n    frontendProjectConfig.name,\n    PY_AGENT_REACT_CONNECTION_GENERATOR_INFO,\n    toProjectRelativePath(\n      frontendProjectConfig,\n      protocol === 'ag-ui'\n        ? joinPathFragments(\n            frontendProjectConfig.sourceRoot,\n            'hooks',\n            `useAgui${agentNameClassName}`,\n          )\n        : joinPathFragments(\n            frontendProjectConfig.sourceRoot,\n            'components',\n            `${agentNameClassName}Provider`,\n          ),\n    ),\n    agentNameClassName,\n    connectionMetadata,\n  );\n\n  await addGeneratorMetricsIfApplicable(tree, [\n    PY_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/**\n * Determine the Python module name from the project configuration\n */\nconst getModuleName = (\n  projectConfig: ReturnType<typeof readProjectConfigurationUnqualified>,\n): string => {\n  if (projectConfig.sourceRoot) {\n    const sourceRootParts = projectConfig.sourceRoot.split('/');\n    return sourceRootParts[sourceRootParts.length - 1];\n  }\n  throw new Error(\n    `Could not determine sourceRoot for project ${projectConfig.name}`,\n  );\n};\n\nexport default pyAgentReactConnectionGenerator;\n"],"names":["generateFiles","joinPathFragments","OverwriteStrategy","updateProjectConfiguration","addAgentRuntimeToConnectionNamespace","DEPENDENCIES","AGUI_DEPENDENCIES","addAgUiReactConnection","resolveAgUiTheme","addOpenApiReactClient","OPEN_API_REACT_DEPENDENCIES","declareDependencies","onlyWhen","formatFilesInSubtree","installDependencies","addGeneratorMetricsIfApplicable","kebabCase","toClassName","toSnakeCase","addComponentGeneratorMetadata","getGeneratorInfo","readProjectConfigurationUnqualified","sortObjectKeys","toProjectRelativePath","addPyAgentTargetToLocalDev","openApiClientLocalDevDeps","isHttp","m","protocol","isAgUi","ts","PY_AGENT_REACT_CONNECTION_GENERATOR_INFO","filename","pyAgentReactConnectionGenerator","tree","options","frontendProjectConfig","sourceProject","agentProjectConfig","targetProject","targetComponent","metadata","agentName","name","agentNameClassName","rc","agentPort","port","ports","auth","toLowerCase","connectionMetadata","theme","Error","additionalLocalDevDeps","moduleName","getModuleName","agentNameSnakeCase","agentTargetPrefix","dirname","root","overwriteStrategy","KeepExisting","openApiTargetName","openApiDist","specPath","targets","cache","executor","outputs","commands","apiName","backendProjectConfig","specBuildProject","specBuildTargetName","isAgentRuntime","skipLocalDev","agentNameKebabCase","additionalDependencyTargets","sourceRoot","preferInstallDependencies","languages","projectConfig","sourceRootParts","split","length"],"mappings":"AAAA;;;CAGC,GACD,SACEA,aAAa,EACbC,iBAAiB,EACjBC,iBAAiB,EAEjBC,0BAA0B,QACrB,aAAa;AACpB,SAASC,oCAAoC,QAAQ,8CAA2C;AAEhG,SACEC,gBAAgBC,iBAAiB,EAEjCC,sBAAsB,EACtBC,gBAAgB,QACX,8CAA2C;AAClD,SACEC,qBAAqB,EACrBC,2BAA2B,QACtB,8CAA2C;AAClD,SACEC,mBAAmB,EACnBC,QAAQ,QACH,0CAAuC;AAC9C,SAASC,oBAAoB,QAAQ,2BAAwB;AAC7D,SAASC,mBAAmB,QAAQ,4BAAyB;AAC7D,SAASC,+BAA+B,QAAQ,4BAAyB;AACzE,SAASC,SAAS,EAAEC,WAAW,EAAEC,WAAW,QAAQ,0BAAuB;AAC3E,SACEC,6BAA6B,EAE7BC,gBAAgB,EAEhBC,mCAAmC,QAC9B,uBAAoB;AAC3B,SAASC,cAAc,QAAQ,2BAAwB;AACvD,SAASC,qBAAqB,QAAQ,0BAAuB;AAC7D,SACEC,0BAA0B,EAC1BC,yBAAyB,QACpB,iBAAc;AAUrB,oEAAoE,GACpE,MAAMC,SAAS,CAACC,IAAsCA,EAAEC,QAAQ,KAAK;AAErE,kEAAkE,GAClE,MAAMC,SAAS,CAACF,IAAsC,CAACD,OAAOC;AAE9D,4EAA4E;AAC5E,4EAA4E;AAC5E,+EAA+E;AAC/E,8EAA8E;AAC9E,OAAO,MAAMtB,eACXM,sBAAsD;IACpDmB,IAAI;WACClB,SAASF,6BAA6BgB;WACtCd,SAASN,kBAAkBwB,EAAE,EAAED;KACnC;AACH,GAAG;AAEL,OAAO,MAAME,2CACXX,iBAAiB,YAAYY,QAAQ,EAAE;AAEzC,OAAO,MAAMC,kCAAkC,OAC7CC,MACAC;IAEA,MAAMC,wBAAwBf,oCAC5Ba,MACAC,QAAQE,aAAa;IAEvB,MAAMC,qBAAqBjB,oCACzBa,MACAC,QAAQI,aAAa;IAGvB,MAAMC,kBACJL,QAAQK,eAAe;IAEzB,uEAAuE;IACvE,MAAMC,WAAWH,mBAAmBG,QAAQ;IAC5C,MAAMC,YAAYF,iBAAiBG,QAAQ;IAC3C,MAAMC,qBAAqBJ,iBAAiBK,MAAM5B,YAAYyB;IAC9D,MAAMI,YAAYN,iBAAiBO,QAAQN,UAAUO,OAAO,CAAC,EAAE,IAAI;IACnE,MAAMC,OAAO,AAACT,CAAAA,iBAAiBS,QAAQR,UAAUQ,QAAQ,KAAI,EAAGC,WAAW;IAC3E,MAAMtB,WAAW,AAACY,CAAAA,iBAAiBZ,YAAY,MAAK,EAAGsB,WAAW;IAElE,2EAA2E;IAC3E,6DAA6D;IAC7D,MAAMC,qBAAqD;QACzDvB;QACAqB;QACAG,OAAO5C,iBAAiB4B;IAC1B;IAEA,IAAIR,aAAa,OAAO;QACtB,MAAM,IAAIyB,MACR,CAAC,gDAAgD,CAAC,GAChD,CAAC,qEAAqE,CAAC;IAE7E;IAEA,IAAIC,yBAAmC,EAAE;IAEzC,IAAI1B,aAAa,SAAS;QACxB,MAAMrB,uBAAuB2B,MAAM;YACjCE;YACAM;YACAE;YACAK,MAAMA;QACR;IACF,OAAO;QACL,MAAMM,aAAaC,cAAclB;QACjC,MAAMmB,qBAAqBvC,YAAYwB;QACvC,MAAMgB,oBAAoBlB,iBAAiBG,OAAOD,YAAY;QAE9D,0DAA0D;QAC1D1C,cACEkC,MACAjC,kBAAkB,YAAY0D,OAAO,EAAE,gBACvCrB,mBAAmBsB,IAAI,EACvB;YACEL;YACAE;QACF,GACA;YACEI,mBAAmB3D,kBAAkB4D,YAAY;QACnD;QAGF,0EAA0E;QAC1E,MAAMC,oBAAoB,GAAGL,kBAAkB,QAAQ,CAAC;QACxD,MAAMM,cAAc/D,kBAClB,QACAqC,mBAAmBsB,IAAI,EACvB,WACAH;QAEF,MAAMQ,WAAWhE,kBAAkB+D,aAAa;QAEhD7D,2BAA2B+B,MAAMI,mBAAmBK,IAAI,EAAE;YACxD,GAAGL,kBAAkB;YACrB4B,SAAS5C,eAAe;gBACtB,GAAGgB,mBAAmB4B,OAAO;gBAC7B,CAACH,kBAAkB,EAAE;oBACnBI,OAAO;oBACPC,UAAU;oBACVC,SAAS;wBACP,CAAC,2CAA2C,EAAEZ,oBAAoB;qBACnE;oBACDtB,SAAS;wBACPmC,UAAU;4BACR,CAAC,oCAAoC,EAAEb,mBAAmB,wCAAwC,EAAEA,mBAAmB,cAAc,CAAC;yBACvI;oBACH;gBACF;YACF;QACF;QAEA,uFAAuF;QACvF,kFAAkF;QAClF,MAAMhD,sBACJyB,MACA;YACEqC,SAAS3B;YACTR;YACAoC,sBAAsBlC;YACtBmC,kBAAkBnC;YAClB2B;YACAS,qBAAqB,GAAGpC,mBAAmBK,IAAI,CAAC,CAAC,EAAEoB,mBAAmB;YACtEd;YACAF,MAAMD;YACN6B,gBAAgB;YAChBC,cAAc;QAChB,GACAvE;QAGFiD,yBAAyB7B,0BAA0BmB;QAEnD,2EAA2E;QAC3E,0EAA0E;QAC1E,qEAAqE;QACrE,yDAAyD;QACzD,MAAMxC,qCAAqC8B,MAAM;YAC/C2C,oBAAoB7D,UAAU4B;YAC9BA;QACF;IACF;IAEA,MAAMpB,2BACJU,MACAE,sBAAsBO,IAAI,EAC1BL,mBAAmBK,IAAI,EACvB;QACED;QACAE;QACAG,MAAMD;QACNN;QACAsC,6BAA6BxB;IAC/B;IAGF,8EAA8E;IAC9EnC,8BACEe,MACAE,sBAAsBO,IAAI,EAC1BZ,0CACAR,sBACEa,uBACAR,aAAa,UACT3B,kBACEmC,sBAAsB2C,UAAU,EAChC,SACA,CAAC,OAAO,EAAEnC,oBAAoB,IAEhC3C,kBACEmC,sBAAsB2C,UAAU,EAChC,cACA,GAAGnC,mBAAmB,QAAQ,CAAC,IAGvCA,oBACAO;IAGF,MAAMpC,gCAAgCmB,MAAM;QAC1CH;KACD;IAED,MAAMlB,qBAAqBqB;IAC3B,OAAO,IACLpB,oBAAoBoB,MAAMC,QAAQ6C,yBAAyB,EAAE;YAC3DC,WAAW;gBAAC;aAAa;QAC3B;AACJ,EAAE;AAEF;;CAEC,GACD,MAAMzB,gBAAgB,CACpB0B;IAEA,IAAIA,cAAcH,UAAU,EAAE;QAC5B,MAAMI,kBAAkBD,cAAcH,UAAU,CAACK,KAAK,CAAC;QACvD,OAAOD,eAAe,CAACA,gBAAgBE,MAAM,GAAG,EAAE;IACpD;IACA,MAAM,IAAIhC,MACR,CAAC,2CAA2C,EAAE6B,cAAcvC,IAAI,EAAE;AAEtE;AAEA,eAAeV,gCAAgC"}

package/src/py/dynamodb/agent-connection/__snapshots__/generator.spec.ts.snap

@@ -3,6 +3,15 @@
 exports[`py#dynamodb agent-connection generator > should add dynamodb dev dependency to agent dev 1`] = `
 {
   "$schema": "../../node_modules/nx/schemas/project-schema.json",
+  "metadata": {
+    "components": [
+      {
+        "generator": "py#dynamodb#agent-connection",
+        "name": "agent-db",
+        "path": "packages/db",
+      },
+    ],
+  },
   "name": "my-agent",
   "root": "packages/my-agent",
   "targets": {
@@ -25,6 +34,15 @@ exports[`py#dynamodb agent-connection generator > should add dynamodb dev depend
 exports[`py#dynamodb agent-connection generator > should use sourceComponent name when provided 1`] = `
 {
   "$schema": "../../node_modules/nx/schemas/project-schema.json",
+  "metadata": {
+    "components": [
+      {
+        "generator": "py#dynamodb#agent-connection",
+        "name": "custom-agent-db",
+        "path": "packages/db",
+      },
+    ],
+  },
   "name": "my-agent",
   "root": "packages/my-agent",
   "targets": {

package/src/py/dynamodb/agent-connection/generator.js

@@ -5,7 +5,7 @@
 import { formatFilesInSubtree } from "../../../utils/format.js";
 import { installDependencies } from "../../../utils/install.js";
 import { addGeneratorMetricsIfApplicable } from "../../../utils/metrics.js";
-import { addDependencyToTargetIfNotPresent, getGeneratorInfo, readProjectConfigurationUnqualified } from "../../../utils/nx.js";
+import { addComponentGeneratorMetadata, addDependencyToTargetIfNotPresent, getGeneratorInfo, readProjectConfigurationUnqualified } from "../../../utils/nx.js";
 import { addWorkspaceDependencyToPyProject } from "../../../utils/py.js";
 export const PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO = getGeneratorInfo(import.meta.filename);
 export const pyDynamoDBAgentConnectionGenerator = async (tree, options)=>{
@@ -23,6 +23,12 @@ export const pyDynamoDBAgentConnectionGenerator = async (tree, options)=>{
         });
         updateProjectConfiguration(tree, sourceProject.name, sourceProject);
     }
+    // Recorded so the version sync can identify this connection.
+    addComponentGeneratorMetadata(tree, sourceProject.name, PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO, targetProject.root, `${agentName}-${targetProject.name}`, // 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, [
         PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO
     ]);

package/src/py/dynamodb/agent-connection/generator.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/py/dynamodb/agent-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  type Tree,\n  updateProjectConfiguration,\n} from '@nx/devkit';\nimport { formatFilesInSubtree } from '../../../utils/format';\nimport { installDependencies } from '../../../utils/install';\nimport { addGeneratorMetricsIfApplicable } from '../../../utils/metrics';\nimport {\n  addDependencyToTargetIfNotPresent,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  readProjectConfigurationUnqualified,\n} from '../../../utils/nx';\nimport { addWorkspaceDependencyToPyProject } from '../../../utils/py';\nimport type { PyDynamoDBAgentConnectionGeneratorSchema } from './schema';\n\nexport const PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO: NxGeneratorInfo =\n  getGeneratorInfo(import.meta.filename);\n\nexport const pyDynamoDBAgentConnectionGenerator = async (\n  tree: Tree,\n  options: PyDynamoDBAgentConnectionGeneratorSchema,\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  addWorkspaceDependencyToPyProject(tree, sourceProject, targetProject);\n\n  const agentName = options.sourceComponent?.name ?? 'agent';\n  const devTarget = `${agentName}-dev`;\n\n  if (sourceProject.targets?.[devTarget]) {\n    addDependencyToTargetIfNotPresent(sourceProject, devTarget, {\n      projects: [targetProject.name],\n      target: 'dev',\n    });\n    updateProjectConfiguration(tree, sourceProject.name, sourceProject);\n  }\n\n  await addGeneratorMetricsIfApplicable(tree, [\n    PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO,\n  ]);\n  await formatFilesInSubtree(tree);\n  return () =>\n    installDependencies(tree, options.preferInstallDependencies, {\n      languages: ['typescript', 'python'],\n    });\n};\n\nexport default pyDynamoDBAgentConnectionGenerator;\n"],"names":["updateProjectConfiguration","formatFilesInSubtree","installDependencies","addGeneratorMetricsIfApplicable","addDependencyToTargetIfNotPresent","getGeneratorInfo","readProjectConfigurationUnqualified","addWorkspaceDependencyToPyProject","PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO","filename","pyDynamoDBAgentConnectionGenerator","tree","options","sourceProject","targetProject","agentName","sourceComponent","name","devTarget","targets","projects","target","preferInstallDependencies","languages"],"mappings":"AAAA;;;CAGC,GACD,SAGEA,0BAA0B,QACrB,aAAa;AACpB,SAASC,oBAAoB,QAAQ,2BAAwB;AAC7D,SAASC,mBAAmB,QAAQ,4BAAyB;AAC7D,SAASC,+BAA+B,QAAQ,4BAAyB;AACzE,SACEC,iCAAiC,EACjCC,gBAAgB,EAEhBC,mCAAmC,QAC9B,uBAAoB;AAC3B,SAASC,iCAAiC,QAAQ,uBAAoB;AAGtE,OAAO,MAAMC,8CACXH,iBAAiB,YAAYI,QAAQ,EAAE;AAEzC,OAAO,MAAMC,qCAAqC,OAChDC,MACAC;IAEA,MAAMC,gBAAgBP,oCACpBK,MACAC,QAAQC,aAAa;IAEvB,MAAMC,gBAAgBR,oCACpBK,MACAC,QAAQE,aAAa;IAGvBP,kCAAkCI,MAAME,eAAeC;IAEvD,MAAMC,YAAYH,QAAQI,eAAe,EAAEC,QAAQ;IACnD,MAAMC,YAAY,GAAGH,UAAU,IAAI,CAAC;IAEpC,IAAIF,cAAcM,OAAO,EAAE,CAACD,UAAU,EAAE;QACtCd,kCAAkCS,eAAeK,WAAW;YAC1DE,UAAU;gBAACN,cAAcG,IAAI;aAAC;YAC9BI,QAAQ;QACV;QACArB,2BAA2BW,MAAME,cAAcI,IAAI,EAAEJ;IACvD;IAEA,MAAMV,gCAAgCQ,MAAM;QAC1CH;KACD;IACD,MAAMP,qBAAqBU;IAC3B,OAAO,IACLT,oBAAoBS,MAAMC,QAAQU,yBAAyB,EAAE;YAC3DC,WAAW;gBAAC;gBAAc;aAAS;QACrC;AACJ,EAAE;AAEF,eAAeb,mCAAmC"}
+{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/py/dynamodb/agent-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  type Tree,\n  updateProjectConfiguration,\n} from '@nx/devkit';\nimport { formatFilesInSubtree } from '../../../utils/format';\nimport { installDependencies } from '../../../utils/install';\nimport { addGeneratorMetricsIfApplicable } from '../../../utils/metrics';\nimport {\n  addComponentGeneratorMetadata,\n  addDependencyToTargetIfNotPresent,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  readProjectConfigurationUnqualified,\n} from '../../../utils/nx';\nimport { addWorkspaceDependencyToPyProject } from '../../../utils/py';\nimport type { PyDynamoDBAgentConnectionGeneratorSchema } from './schema';\n\nexport const PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO: NxGeneratorInfo =\n  getGeneratorInfo(import.meta.filename);\n\nexport const pyDynamoDBAgentConnectionGenerator = async (\n  tree: Tree,\n  options: PyDynamoDBAgentConnectionGeneratorSchema,\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  addWorkspaceDependencyToPyProject(tree, sourceProject, targetProject);\n\n  const agentName = options.sourceComponent?.name ?? 'agent';\n  const devTarget = `${agentName}-dev`;\n\n  if (sourceProject.targets?.[devTarget]) {\n    addDependencyToTargetIfNotPresent(sourceProject, devTarget, {\n      projects: [targetProject.name],\n      target: 'dev',\n    });\n    updateProjectConfiguration(tree, sourceProject.name, sourceProject);\n  }\n\n  // Recorded so the version sync can identify this connection.\n  addComponentGeneratorMetadata(\n    tree,\n    sourceProject.name,\n    PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO,\n    targetProject.root,\n    `${agentName}-${targetProject.name}`,\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    PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO,\n  ]);\n  await formatFilesInSubtree(tree);\n  return () =>\n    installDependencies(tree, options.preferInstallDependencies, {\n      languages: ['typescript', 'python'],\n    });\n};\n\nexport default pyDynamoDBAgentConnectionGenerator;\n"],"names":["updateProjectConfiguration","formatFilesInSubtree","installDependencies","addGeneratorMetricsIfApplicable","addComponentGeneratorMetadata","addDependencyToTargetIfNotPresent","getGeneratorInfo","readProjectConfigurationUnqualified","addWorkspaceDependencyToPyProject","PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO","filename","pyDynamoDBAgentConnectionGenerator","tree","options","sourceProject","targetProject","agentName","sourceComponent","name","devTarget","targets","projects","target","root","sourcePath","path","preferInstallDependencies","languages"],"mappings":"AAAA;;;CAGC,GACD,SAGEA,0BAA0B,QACrB,aAAa;AACpB,SAASC,oBAAoB,QAAQ,2BAAwB;AAC7D,SAASC,mBAAmB,QAAQ,4BAAyB;AAC7D,SAASC,+BAA+B,QAAQ,4BAAyB;AACzE,SACEC,6BAA6B,EAC7BC,iCAAiC,EACjCC,gBAAgB,EAEhBC,mCAAmC,QAC9B,uBAAoB;AAC3B,SAASC,iCAAiC,QAAQ,uBAAoB;AAGtE,OAAO,MAAMC,8CACXH,iBAAiB,YAAYI,QAAQ,EAAE;AAEzC,OAAO,MAAMC,qCAAqC,OAChDC,MACAC;IAEA,MAAMC,gBAAgBP,oCACpBK,MACAC,QAAQC,aAAa;IAEvB,MAAMC,gBAAgBR,oCACpBK,MACAC,QAAQE,aAAa;IAGvBP,kCAAkCI,MAAME,eAAeC;IAEvD,MAAMC,YAAYH,QAAQI,eAAe,EAAEC,QAAQ;IACnD,MAAMC,YAAY,GAAGH,UAAU,IAAI,CAAC;IAEpC,IAAIF,cAAcM,OAAO,EAAE,CAACD,UAAU,EAAE;QACtCd,kCAAkCS,eAAeK,WAAW;YAC1DE,UAAU;gBAACN,cAAcG,IAAI;aAAC;YAC9BI,QAAQ;QACV;QACAtB,2BAA2BY,MAAME,cAAcI,IAAI,EAAEJ;IACvD;IAEA,6DAA6D;IAC7DV,8BACEQ,MACAE,cAAcI,IAAI,EAClBT,6CACAM,cAAcQ,IAAI,EAClB,GAAGP,UAAU,CAAC,EAAED,cAAcG,IAAI,EAAE,EACpC,oEAAoE;IACpE,kDAAkD;IAClD;QAAEM,YAAYX,QAAQI,eAAe,EAAEQ;IAAK;IAG9C,MAAMtB,gCAAgCS,MAAM;QAC1CH;KACD;IACD,MAAMR,qBAAqBW;IAC3B,OAAO,IACLV,oBAAoBU,MAAMC,QAAQa,yBAAyB,EAAE;YAC3DC,WAAW;gBAAC;gBAAc;aAAS;QACrC;AACJ,EAAE;AAEF,eAAehB,mCAAmC"}

package/src/py/dynamodb/fast-api-connection/__snapshots__/generator.spec.ts.snap

@@ -3,6 +3,15 @@
 exports[`py#dynamodb fast-api-connection generator > should add dynamodb dev dependency to fast-api dev 1`] = `
 {
   "$schema": "../../node_modules/nx/schemas/project-schema.json",
+  "metadata": {
+    "components": [
+      {
+        "generator": "py#dynamodb#fast-api-connection",
+        "name": "db",
+        "path": "packages/db",
+      },
+    ],
+  },
   "name": "api",
   "root": "packages/api",
   "targets": {

package/src/py/dynamodb/fast-api-connection/generator.js

@@ -5,7 +5,7 @@
 import { formatFilesInSubtree } from "../../../utils/format.js";
 import { installDependencies } from "../../../utils/install.js";
 import { addGeneratorMetricsIfApplicable } from "../../../utils/metrics.js";
-import { addDependencyToTargetIfNotPresent, getGeneratorInfo, readProjectConfigurationUnqualified } from "../../../utils/nx.js";
+import { addComponentGeneratorMetadata, addDependencyToTargetIfNotPresent, getGeneratorInfo, readProjectConfigurationUnqualified } from "../../../utils/nx.js";
 import { addWorkspaceDependencyToPyProject } from "../../../utils/py.js";
 export const PY_DYNAMODB_FAST_API_CONNECTION_GENERATOR_INFO = getGeneratorInfo(import.meta.filename);
 export const pyDynamoDBFastApiConnectionGenerator = async (tree, options)=>{
@@ -21,6 +21,8 @@ export const pyDynamoDBFastApiConnectionGenerator = async (tree, options)=>{
         });
         updateProjectConfiguration(tree, sourceProject.name, sourceProject);
     }
+    // Recorded so the version sync can identify this connection.
+    addComponentGeneratorMetadata(tree, sourceProject.name, PY_DYNAMODB_FAST_API_CONNECTION_GENERATOR_INFO, targetProject.root, targetProject.name);
     await addGeneratorMetricsIfApplicable(tree, [
         PY_DYNAMODB_FAST_API_CONNECTION_GENERATOR_INFO
     ]);

package/src/py/dynamodb/fast-api-connection/generator.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/py/dynamodb/fast-api-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  type Tree,\n  updateProjectConfiguration,\n} from '@nx/devkit';\nimport { formatFilesInSubtree } from '../../../utils/format';\nimport { installDependencies } from '../../../utils/install';\nimport { addGeneratorMetricsIfApplicable } from '../../../utils/metrics';\nimport {\n  addDependencyToTargetIfNotPresent,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  readProjectConfigurationUnqualified,\n} from '../../../utils/nx';\nimport { addWorkspaceDependencyToPyProject } from '../../../utils/py';\nimport type { PyDynamoDBFastApiConnectionGeneratorSchema } from './schema';\n\nexport const PY_DYNAMODB_FAST_API_CONNECTION_GENERATOR_INFO: NxGeneratorInfo =\n  getGeneratorInfo(import.meta.filename);\n\nexport const pyDynamoDBFastApiConnectionGenerator = async (\n  tree: Tree,\n  options: PyDynamoDBFastApiConnectionGeneratorSchema,\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  addWorkspaceDependencyToPyProject(tree, sourceProject, targetProject);\n\n  if (sourceProject.targets?.['dev']) {\n    addDependencyToTargetIfNotPresent(sourceProject, 'dev', {\n      projects: [targetProject.name],\n      target: 'dev',\n    });\n    updateProjectConfiguration(tree, sourceProject.name, sourceProject);\n  }\n\n  await addGeneratorMetricsIfApplicable(tree, [\n    PY_DYNAMODB_FAST_API_CONNECTION_GENERATOR_INFO,\n  ]);\n  await formatFilesInSubtree(tree);\n  return () =>\n    installDependencies(tree, options.preferInstallDependencies, {\n      languages: ['typescript', 'python'],\n    });\n};\n\nexport default pyDynamoDBFastApiConnectionGenerator;\n"],"names":["updateProjectConfiguration","formatFilesInSubtree","installDependencies","addGeneratorMetricsIfApplicable","addDependencyToTargetIfNotPresent","getGeneratorInfo","readProjectConfigurationUnqualified","addWorkspaceDependencyToPyProject","PY_DYNAMODB_FAST_API_CONNECTION_GENERATOR_INFO","filename","pyDynamoDBFastApiConnectionGenerator","tree","options","sourceProject","targetProject","targets","projects","name","target","preferInstallDependencies","languages"],"mappings":"AAAA;;;CAGC,GACD,SAGEA,0BAA0B,QACrB,aAAa;AACpB,SAASC,oBAAoB,QAAQ,2BAAwB;AAC7D,SAASC,mBAAmB,QAAQ,4BAAyB;AAC7D,SAASC,+BAA+B,QAAQ,4BAAyB;AACzE,SACEC,iCAAiC,EACjCC,gBAAgB,EAEhBC,mCAAmC,QAC9B,uBAAoB;AAC3B,SAASC,iCAAiC,QAAQ,uBAAoB;AAGtE,OAAO,MAAMC,iDACXH,iBAAiB,YAAYI,QAAQ,EAAE;AAEzC,OAAO,MAAMC,uCAAuC,OAClDC,MACAC;IAEA,MAAMC,gBAAgBP,oCACpBK,MACAC,QAAQC,aAAa;IAEvB,MAAMC,gBAAgBR,oCACpBK,MACAC,QAAQE,aAAa;IAGvBP,kCAAkCI,MAAME,eAAeC;IAEvD,IAAID,cAAcE,OAAO,EAAE,CAAC,MAAM,EAAE;QAClCX,kCAAkCS,eAAe,OAAO;YACtDG,UAAU;gBAACF,cAAcG,IAAI;aAAC;YAC9BC,QAAQ;QACV;QACAlB,2BAA2BW,MAAME,cAAcI,IAAI,EAAEJ;IACvD;IAEA,MAAMV,gCAAgCQ,MAAM;QAC1CH;KACD;IACD,MAAMP,qBAAqBU;IAC3B,OAAO,IACLT,oBAAoBS,MAAMC,QAAQO,yBAAyB,EAAE;YAC3DC,WAAW;gBAAC;gBAAc;aAAS;QACrC;AACJ,EAAE;AAEF,eAAeV,qCAAqC"}
+{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/py/dynamodb/fast-api-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  type Tree,\n  updateProjectConfiguration,\n} from '@nx/devkit';\nimport { formatFilesInSubtree } from '../../../utils/format';\nimport { installDependencies } from '../../../utils/install';\nimport { addGeneratorMetricsIfApplicable } from '../../../utils/metrics';\nimport {\n  addComponentGeneratorMetadata,\n  addDependencyToTargetIfNotPresent,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  readProjectConfigurationUnqualified,\n} from '../../../utils/nx';\nimport { addWorkspaceDependencyToPyProject } from '../../../utils/py';\nimport type { PyDynamoDBFastApiConnectionGeneratorSchema } from './schema';\n\nexport const PY_DYNAMODB_FAST_API_CONNECTION_GENERATOR_INFO: NxGeneratorInfo =\n  getGeneratorInfo(import.meta.filename);\n\nexport const pyDynamoDBFastApiConnectionGenerator = async (\n  tree: Tree,\n  options: PyDynamoDBFastApiConnectionGeneratorSchema,\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  addWorkspaceDependencyToPyProject(tree, sourceProject, targetProject);\n\n  if (sourceProject.targets?.['dev']) {\n    addDependencyToTargetIfNotPresent(sourceProject, 'dev', {\n      projects: [targetProject.name],\n      target: 'dev',\n    });\n    updateProjectConfiguration(tree, sourceProject.name, sourceProject);\n  }\n\n  // Recorded so the version sync can identify this connection.\n  addComponentGeneratorMetadata(\n    tree,\n    sourceProject.name,\n    PY_DYNAMODB_FAST_API_CONNECTION_GENERATOR_INFO,\n    targetProject.root,\n    targetProject.name,\n  );\n\n  await addGeneratorMetricsIfApplicable(tree, [\n    PY_DYNAMODB_FAST_API_CONNECTION_GENERATOR_INFO,\n  ]);\n  await formatFilesInSubtree(tree);\n  return () =>\n    installDependencies(tree, options.preferInstallDependencies, {\n      languages: ['typescript', 'python'],\n    });\n};\n\nexport default pyDynamoDBFastApiConnectionGenerator;\n"],"names":["updateProjectConfiguration","formatFilesInSubtree","installDependencies","addGeneratorMetricsIfApplicable","addComponentGeneratorMetadata","addDependencyToTargetIfNotPresent","getGeneratorInfo","readProjectConfigurationUnqualified","addWorkspaceDependencyToPyProject","PY_DYNAMODB_FAST_API_CONNECTION_GENERATOR_INFO","filename","pyDynamoDBFastApiConnectionGenerator","tree","options","sourceProject","targetProject","targets","projects","name","target","root","preferInstallDependencies","languages"],"mappings":"AAAA;;;CAGC,GACD,SAGEA,0BAA0B,QACrB,aAAa;AACpB,SAASC,oBAAoB,QAAQ,2BAAwB;AAC7D,SAASC,mBAAmB,QAAQ,4BAAyB;AAC7D,SAASC,+BAA+B,QAAQ,4BAAyB;AACzE,SACEC,6BAA6B,EAC7BC,iCAAiC,EACjCC,gBAAgB,EAEhBC,mCAAmC,QAC9B,uBAAoB;AAC3B,SAASC,iCAAiC,QAAQ,uBAAoB;AAGtE,OAAO,MAAMC,iDACXH,iBAAiB,YAAYI,QAAQ,EAAE;AAEzC,OAAO,MAAMC,uCAAuC,OAClDC,MACAC;IAEA,MAAMC,gBAAgBP,oCACpBK,MACAC,QAAQC,aAAa;IAEvB,MAAMC,gBAAgBR,oCACpBK,MACAC,QAAQE,aAAa;IAGvBP,kCAAkCI,MAAME,eAAeC;IAEvD,IAAID,cAAcE,OAAO,EAAE,CAAC,MAAM,EAAE;QAClCX,kCAAkCS,eAAe,OAAO;YACtDG,UAAU;gBAACF,cAAcG,IAAI;aAAC;YAC9BC,QAAQ;QACV;QACAnB,2BAA2BY,MAAME,cAAcI,IAAI,EAAEJ;IACvD;IAEA,6DAA6D;IAC7DV,8BACEQ,MACAE,cAAcI,IAAI,EAClBT,gDACAM,cAAcK,IAAI,EAClBL,cAAcG,IAAI;IAGpB,MAAMf,gCAAgCS,MAAM;QAC1CH;KACD;IACD,MAAMR,qBAAqBW;IAC3B,OAAO,IACLV,oBAAoBU,MAAMC,QAAQQ,yBAAyB,EAAE;YAC3DC,WAAW;gBAAC;gBAAc;aAAS;QACrC;AACJ,EAAE;AAEF,eAAeX,qCAAqC"}

package/src/py/dynamodb/generator.d.ts

@@ -5,6 +5,35 @@
 import { type GeneratorCallback, type Tree } from '@nx/devkit';
 import { type NxGeneratorInfo } from '../../utils/nx';
 import type { PyDynamoDBGeneratorSchema } from './schema';
+export declare const DEPENDENCIES: import("../../utils/declared-dependencies").DependencyDeclaration<readonly [{
+    readonly name: "@aws-sdk/client-dynamodb";
+} & {
+    versionOnly: true;
+}, {
+    readonly name: "tsx";
+} & {
+    versionOnly: true;
+}, {
+    readonly name: "constructs";
+    readonly when: (m: import("../../utils/shared-constructs-constants").IacMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, {
+    readonly name: "aws-cdk-lib";
+    readonly when: (m: import("../../utils/shared-constructs-constants").IacMetadata) => boolean;
+} & {
+    versionOnly: true;
+}, {
+    readonly name: "@types/node";
+} & {
+    versionOnly: true;
+}], readonly [{
+    readonly name: "pynamodb";
+}, {
+    readonly name: "boto3";
+}, {
+    readonly name: "aws-lambda-powertools";
+}]>;
 export declare const PY_DYNAMODB_GENERATOR_INFO: NxGeneratorInfo;
 export declare const pyDynamoDBGenerator: (tree: Tree, options: PyDynamoDBGeneratorSchema) => Promise<GeneratorCallback>;
 export default pyDynamoDBGenerator;

package/src/py/dynamodb/generator.js

@@ -3,7 +3,9 @@
  * SPDX-License-Identifier: Apache-2.0
  */ import { relative } from "node:path";
 import { generateFiles, joinPathFragments, readProjectConfiguration, updateProjectConfiguration } from "@nx/devkit";
+import { addPyDependencies } from "../../utils/add-dependencies.js";
 import { resolveContainers } from "../../utils/containers.js";
+import { declareDependencies, ownedElsewhere } from "../../utils/declared-dependencies.js";
 import { addDynamoDBInfra } from "../../utils/dynamodb-constructs/dynamodb-constructs.js";
 import { formatFilesInSubtree } from "../../utils/format.js";
 import { resolveIac } from "../../utils/iac.js";
@@ -13,11 +15,27 @@ import { kebabCase, toClassName } from "../../utils/names.js";
 import { getNpmScope } from "../../utils/npm-scope.js";
 import { addGeneratorMetadata, getGeneratorInfo, projectExists } from "../../utils/nx.js";
 import { assignSharedPort } from "../../utils/port.js";
-import { addDependenciesToPyProjectToml } from "../../utils/py.js";
-import { sharedConstructsGenerator } from "../../utils/shared-constructs.js";
+import { SHARED_CONSTRUCTS_DEPENDENCIES, sharedConstructsGenerator } from "../../utils/shared-constructs.js";
 import { DYNAMODB_GENERATOR_IDS, PACKAGES_DIR, SHARED_SCRIPTS_DIR } from "../../utils/shared-constructs-constants.js";
-import { sharedDynamoDBScriptsGenerator } from "../../utils/shared-dynamodb-scripts.js";
+import { SHARED_DYNAMODB_SCRIPTS_DEPENDENCIES, sharedDynamoDBScriptsGenerator } from "../../utils/shared-dynamodb-scripts.js";
 import pyProjectGenerator, { getPyProjectDetails } from "../project/generator.js";
+export const DEPENDENCIES = declareDependencies()({
+    ts: [
+        ...ownedElsewhere(SHARED_DYNAMODB_SCRIPTS_DEPENDENCIES),
+        ...ownedElsewhere(SHARED_CONSTRUCTS_DEPENDENCIES)
+    ],
+    py: [
+        {
+            name: 'pynamodb'
+        },
+        {
+            name: 'boto3'
+        },
+        {
+            name: 'aws-lambda-powertools'
+        }
+    ]
+});
 export const PY_DYNAMODB_GENERATOR_INFO = getGeneratorInfo(import.meta.filename);
 export const pyDynamoDBGenerator = async (tree, options)=>{
     const nameClassName = toClassName(options.name);
@@ -49,7 +67,7 @@ export const pyDynamoDBGenerator = async (tree, options)=>{
         containerEngine
     };
     generateFiles(tree, joinPathFragments(import.meta.dirname, 'files'), dir, templateOptions);
-    await sharedDynamoDBScriptsGenerator(tree);
+    await sharedDynamoDBScriptsGenerator(tree, DEPENDENCIES);
     const scriptsDir = relative(dir, joinPathFragments(PACKAGES_DIR, SHARED_SCRIPTS_DIR, 'src', 'dynamodb'));
     projectConfig.targets['pull-image'] = {
         executor: 'nx:run-commands',
@@ -74,12 +92,16 @@ export const pyDynamoDBGenerator = async (tree, options)=>{
         }
     };
     updateProjectConfiguration(tree, fullyQualifiedName, projectConfig);
-    addGeneratorMetadata(tree, fullyQualifiedName, PY_DYNAMODB_GENERATOR_INFO);
+    // Recorded so the version sync can tell a CDK project from a Terraform one;
+    // undefined when no infrastructure was generated.
+    const iac = options.infra !== 'none' ? await resolveIac(tree, options.iac) : undefined;
+    addGeneratorMetadata(tree, fullyQualifiedName, PY_DYNAMODB_GENERATOR_INFO, iac ? {
+        iac
+    } : {});
     if (options.infra !== 'none') {
-        const iac = await resolveIac(tree, options.iac);
         await sharedConstructsGenerator(tree, {
             iac
-        });
+        }, DEPENDENCIES);
         await addDynamoDBInfra(tree, {
             iac,
             projectName: fullyQualifiedName,
@@ -89,11 +111,9 @@ export const pyDynamoDBGenerator = async (tree, options)=>{
             projectRoot: dir
         });
     }
-    addDependenciesToPyProjectToml(tree, dir, [
-        'pynamodb',
-        'boto3',
-        'aws-lambda-powertools'
-    ]);
+    addPyDependencies(tree, DEPENDENCIES, {
+        projectRoot: dir
+    });
     await addGeneratorMetricsIfApplicable(tree, [
         PY_DYNAMODB_GENERATOR_INFO
     ]);