@aws/nx-plugin 1.0.0-rc.23 → 1.0.0-rc.25

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

@@ -1 +1 @@
-{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../../../../../packages/nx-plugin/src/py/agent/react-connection/generator.ts"],"names":[],"mappings":";;;AAAA;;;GAGG;AACH,uCAOoB;AACpB,mFAAgG;AAEhG,wEAGkD;AAClD,oEAAiF;AACjF,kDAA6D;AAC7D,oDAAyE;AACzE,gDAA2E;AAC3E,0CAK2B;AAC3B,kDAAuD;AACvD,+CAGuB;AAEV,QAAA,wCAAwC,GACnD,IAAA,qBAAgB,EAAC,UAAU,CAAC,CAAC;AAExB,MAAM,+BAA+B,GAAG,KAAK,EAClD,IAAU,EACV,OAAkC,EAClC,EAAE;IACF,MAAM,qBAAqB,GAAG,IAAA,wCAAmC,EAC/D,IAAI,EACJ,OAAO,CAAC,aAAa,CACtB,CAAC;IACF,MAAM,kBAAkB,GAAG,IAAA,wCAAmC,EAC5D,IAAI,EACJ,OAAO,CAAC,aAAa,CACtB,CAAC;IAEF,MAAM,eAAe,GACnB,OAAO,CAAC,eAAe,CAAC;IAE1B,uEAAuE;IACvE,MAAM,QAAQ,GAAG,kBAAkB,CAAC,QAAe,CAAC;IACpD,MAAM,SAAS,GAAG,eAAe,EAAE,IAAI,IAAI,OAAO,CAAC;IACnD,MAAM,kBAAkB,GAAG,eAAe,EAAE,EAAE,IAAI,IAAA,mBAAW,EAAC,SAAS,CAAC,CAAC;IACzE,MAAM,SAAS,GAAG,eAAe,EAAE,IAAI,IAAI,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC;IACxE,MAAM,IAAI,GAAG,CAAC,eAAe,EAAE,IAAI,IAAI,QAAQ,EAAE,IAAI,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAC9E,MAAM,QAAQ,GAAG,CAAC,eAAe,EAAE,QAAQ,IAAI,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;IAErE,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;QACvB,MAAM,IAAI,KAAK,CACb,kDAAkD;YAChD,uEAAuE,CAC1E,CAAC;IACJ,CAAC;IAED,IAAI,wBAAwB,GAAa,EAAE,CAAC;IAE5C,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;QACzB,MAAM,IAAA,kCAAsB,EAAC,IAAI,EAAE;YACjC,qBAAqB;YACrB,SAAS;YACT,kBAAkB;YAClB,IAAI,EAAE,IAAgB;SACvB,CAAC,CAAC;IACL,CAAC;SAAM,CAAC;QACN,MAAM,UAAU,GAAG,aAAa,CAAC,kBAAkB,CAAC,CAAC;QACrD,MAAM,kBAAkB,GAAG,IAAA,mBAAW,EAAC,SAAS,CAAC,CAAC;QAClD,MAAM,iBAAiB,GAAG,eAAe,EAAE,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC;QAEtE,0DAA0D;QAC1D,IAAA,sBAAa,EACX,IAAI,EACJ,IAAA,0BAAiB,EAAC,SAAS,EAAE,aAAa,CAAC,EAC3C,kBAAkB,CAAC,IAAI,EACvB;YACE,UAAU;YACV,kBAAkB;SACnB,EACD;YACE,iBAAiB,EAAE,0BAAiB,CAAC,YAAY;SAClD,CACF,CAAC;QAEF,0EAA0E;QAC1E,MAAM,iBAAiB,GAAG,GAAG,iBAAiB,UAAU,CAAC;QACzD,MAAM,WAAW,GAAG,IAAA,0BAAiB,EACnC,MAAM,EACN,kBAAkB,CAAC,IAAI,EACvB,SAAS,EACT,kBAAkB,CACnB,CAAC;QACF,MAAM,QAAQ,GAAG,IAAA,0BAAiB,EAAC,WAAW,EAAE,cAAc,CAAC,CAAC;QAEhE,IAAA,mCAA0B,EAAC,IAAI,EAAE,kBAAkB,CAAC,IAAI,EAAE;YACxD,GAAG,kBAAkB;YACrB,OAAO,EAAE,IAAA,uBAAc,EAAC;gBACtB,GAAG,kBAAkB,CAAC,OAAO;gBAC7B,CAAC,iBAAiB,CAAC,EAAE;oBACnB,KAAK,EAAE,IAAI;oBACX,QAAQ,EAAE,iBAAiB;oBAC3B,OAAO,EAAE;wBACP,8CAA8C,kBAAkB,EAAE;qBACnE;oBACD,OAAO,EAAE;wBACP,QAAQ,EAAE;4BACR,uCAAuC,kBAAkB,2CAA2C,kBAAkB,gBAAgB;yBACvI;qBACF;iBACF;aACF,CAAC;SACH,CAAC,CAAC;QAEH,uFAAuF;QACvF,uFAAuF;QACvF,MAAM,IAAA,6BAAqB,EAAC,IAAI,EAAE;YAChC,OAAO,EAAE,kBAAkB;YAC3B,qBAAqB;YACrB,oBAAoB,EAAE,kBAAkB;YACxC,gBAAgB,EAAE,kBAAkB;YACpC,QAAQ;YACR,mBAAmB,EAAE,GAAG,kBAAkB,CAAC,IAAI,IAAI,iBAAiB,EAAE;YACtE,IAAI;YACJ,IAAI,EAAE,SAAS;YACf,cAAc,EAAE,IAAI;YACpB,cAAc,EAAE,IAAI;SACrB,CAAC,CAAC;QAEH,wBAAwB,GAAG,IAAA,yCAA2B,EAAC,kBAAkB,CAAC,CAAC;QAE3E,2EAA2E;QAC3E,0EAA0E;QAC1E,qEAAqE;QACrE,yDAAyD;QACzD,MAAM,IAAA,2DAAoC,EAAC,IAAI,EAAE;YAC/C,kBAAkB,EAAE,IAAA,iBAAS,EAAC,kBAAkB,CAAC;YACjD,kBAAkB;SACnB,CAAC,CAAC;IACL,CAAC;IAED,MAAM,IAAA,0CAA4B,EAChC,IAAI,EACJ,qBAAqB,CAAC,IAAI,EAC1B,kBAAkB,CAAC,IAAI,EACvB;QACE,SAAS;QACT,kBAAkB;QAClB,IAAI,EAAE,SAAS;QACf,eAAe;QACf,2BAA2B,EAAE,wBAAwB;KACtD,CACF,CAAC;IAEF,MAAM,IAAA,yCAA+B,EAAC,IAAI,EAAE;QAC1C,gDAAwC;KACzC,CAAC,CAAC;IAEH,MAAM,IAAA,6BAAoB,EAAC,IAAI,CAAC,CAAC;IACjC,OAAO,GAAG,EAAE;QACV,IAAA,4BAAmB,EAAC,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC;AACJ,CAAC,CAAC;AAxIW,QAAA,+BAA+B,mCAwI1C;AAEF;;GAEG;AACH,MAAM,aAAa,GAAG,CACpB,aAAqE,EAC7D,EAAE;IACV,IAAI,aAAa,CAAC,UAAU,EAAE,CAAC;QAC7B,MAAM,eAAe,GAAG,aAAa,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC5D,OAAO,eAAe,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACrD,CAAC;IACD,MAAM,IAAI,KAAK,CACb,8CAA8C,aAAa,CAAC,IAAI,EAAE,CACnE,CAAC;AACJ,CAAC,CAAC;AAEF,kBAAe,uCAA+B,CAAC"}
+{"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"}

package/src/py/agent/react-connection/{serve-local.d.ts → local-dev.d.ts}

@@ -4,7 +4,7 @@
  */
 import type { Tree } from '@nx/devkit';
 import type { ComponentMetadata } from '../../../utils/nx';
-export interface PyAgentServeLocalOptions {
+export interface PyAgentLocalDevOptions {
     agentName: string;
     agentNameClassName: string;
     port: number;
@@ -12,12 +12,12 @@ export interface PyAgentServeLocalOptions {
     additionalDependencyTargets?: string[];
 }
 /**
- * Adds the given py strands agent target project to the source project's serve-local target.
+ * Adds the given py strands agent target project to the source project's dev target.
  * Updates the runtime config provider (if it exists) to point to the local HTTP URL.
  */
-export declare const addPyAgentTargetToServeLocal: (tree: Tree, sourceProjectName: string, targetProjectName: string, options: PyAgentServeLocalOptions) => Promise<void>;
+export declare const addPyAgentTargetToLocalDev: (tree: Tree, sourceProjectName: string, targetProjectName: string, options: PyAgentLocalDevOptions) => Promise<void>;
 /**
  * Build the OpenAPI client generation target names for an HTTP agent so
  * that local API changes regenerate the client.
  */
-export declare const openApiClientServeLocalDeps: (agentNameClassName: string) => string[];
+export declare const openApiClientLocalDevDeps: (agentNameClassName: string) => string[];

package/src/py/agent/react-connection/local-dev.js

@@ -0,0 +1,36 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */ import { addAgentTargetToLocalDev } from "../../../connection/agent-local-dev.js";
+import { kebabCase } from "../../../utils/names.js";
+/**
+ * Adds the given py strands agent target project to the source project's dev target.
+ * Updates the runtime config provider (if it exists) to point to the local HTTP URL.
+ */ export const addPyAgentTargetToLocalDev = async (tree, sourceProjectName, targetProjectName, options)=>{
+    // AG-UI clients POST to the runtime config URL directly, so it must point at
+    // the agent's `/invocations` endpoint. HTTP agents go through the generated
+    // OpenAPI client, which appends the `/invocations` operation path itself, so
+    // its base URL must omit it (otherwise the request hits `/invocations/invocations`).
+    const protocol = (options.targetComponent?.protocol ?? 'http').toLowerCase();
+    const localUrl = protocol === 'ag-ui' ? `http://localhost:${options.port}/invocations` : `http://localhost:${options.port}`;
+    await addAgentTargetToLocalDev(tree, sourceProjectName, targetProjectName, {
+        agentNameClassName: options.agentNameClassName,
+        port: options.port,
+        targetComponent: options.targetComponent,
+        runtimeConfigNamespace: 'agentRuntimes',
+        localUrl,
+        additionalDependencyTargets: options.additionalDependencyTargets
+    });
+};
+/**
+ * Build the OpenAPI client generation target names for an HTTP agent so
+ * that local API changes regenerate the client.
+ */ export const openApiClientLocalDevDeps = (agentNameClassName)=>{
+    const clientGenTarget = `generate:${kebabCase(agentNameClassName)}-client`;
+    return [
+        clientGenTarget,
+        `watch-${clientGenTarget}`
+    ];
+};
+
+//# sourceMappingURL=local-dev.js.map

package/src/py/agent/react-connection/local-dev.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/py/agent/react-connection/local-dev.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nimport type { Tree } from '@nx/devkit';\nimport { addAgentTargetToLocalDev } from '../../../connection/agent-local-dev';\nimport { kebabCase } from '../../../utils/names';\nimport type { ComponentMetadata } from '../../../utils/nx';\n\nexport interface PyAgentLocalDevOptions {\n  agentName: string;\n  agentNameClassName: string;\n  port: number;\n  targetComponent?: ComponentMetadata;\n  additionalDependencyTargets?: string[];\n}\n\n/**\n * Adds the given py strands agent target project to the source project's dev target.\n * Updates the runtime config provider (if it exists) to point to the local HTTP URL.\n */\nexport const addPyAgentTargetToLocalDev = async (\n  tree: Tree,\n  sourceProjectName: string,\n  targetProjectName: string,\n  options: PyAgentLocalDevOptions,\n) => {\n  // AG-UI clients POST to the runtime config URL directly, so it must point at\n  // the agent's `/invocations` endpoint. HTTP agents go through the generated\n  // OpenAPI client, which appends the `/invocations` operation path itself, so\n  // its base URL must omit it (otherwise the request hits `/invocations/invocations`).\n  const protocol = (options.targetComponent?.protocol ?? 'http').toLowerCase();\n  const localUrl =\n    protocol === 'ag-ui'\n      ? `http://localhost:${options.port}/invocations`\n      : `http://localhost:${options.port}`;\n\n  await addAgentTargetToLocalDev(tree, sourceProjectName, targetProjectName, {\n    agentNameClassName: options.agentNameClassName,\n    port: options.port,\n    targetComponent: options.targetComponent,\n    runtimeConfigNamespace: 'agentRuntimes',\n    localUrl,\n    additionalDependencyTargets: options.additionalDependencyTargets,\n  });\n};\n\n/**\n * Build the OpenAPI client generation target names for an HTTP agent so\n * that local API changes regenerate the client.\n */\nexport const openApiClientLocalDevDeps = (\n  agentNameClassName: string,\n): string[] => {\n  const clientGenTarget = `generate:${kebabCase(agentNameClassName)}-client`;\n  return [clientGenTarget, `watch-${clientGenTarget}`];\n};\n"],"names":["addAgentTargetToLocalDev","kebabCase","addPyAgentTargetToLocalDev","tree","sourceProjectName","targetProjectName","options","protocol","targetComponent","toLowerCase","localUrl","port","agentNameClassName","runtimeConfigNamespace","additionalDependencyTargets","openApiClientLocalDevDeps","clientGenTarget"],"mappings":"AAAA;;;CAGC,GAED,SAASA,wBAAwB,QAAQ,yCAAsC;AAC/E,SAASC,SAAS,QAAQ,0BAAuB;AAWjD;;;CAGC,GACD,OAAO,MAAMC,6BAA6B,OACxCC,MACAC,mBACAC,mBACAC;IAEA,6EAA6E;IAC7E,4EAA4E;IAC5E,6EAA6E;IAC7E,qFAAqF;IACrF,MAAMC,WAAW,AAACD,CAAAA,QAAQE,eAAe,EAAED,YAAY,MAAK,EAAGE,WAAW;IAC1E,MAAMC,WACJH,aAAa,UACT,CAAC,iBAAiB,EAAED,QAAQK,IAAI,CAAC,YAAY,CAAC,GAC9C,CAAC,iBAAiB,EAAEL,QAAQK,IAAI,EAAE;IAExC,MAAMX,yBAAyBG,MAAMC,mBAAmBC,mBAAmB;QACzEO,oBAAoBN,QAAQM,kBAAkB;QAC9CD,MAAML,QAAQK,IAAI;QAClBH,iBAAiBF,QAAQE,eAAe;QACxCK,wBAAwB;QACxBH;QACAI,6BAA6BR,QAAQQ,2BAA2B;IAClE;AACF,EAAE;AAEF;;;CAGC,GACD,OAAO,MAAMC,4BAA4B,CACvCH;IAEA,MAAMI,kBAAkB,CAAC,SAAS,EAAEf,UAAUW,oBAAoB,OAAO,CAAC;IAC1E,OAAO;QAACI;QAAiB,CAAC,MAAM,EAAEA,iBAAiB;KAAC;AACtD,EAAE"}

package/src/py/agent/react-connection/schema.d.js

@@ -0,0 +1,8 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */ /**
+ * TypeScript types for options defined in schema.json
+ */ export { };
+
+//# sourceMappingURL=schema.d.js.map

package/src/py/agent/react-connection/schema.d.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/py/agent/react-connection/schema.d.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nimport { ComponentMetadata } from '../../../utils/nx';\n\n/**\n * TypeScript types for options defined in schema.json\n */\nexport interface PyAgentReactConnectionGeneratorSchema {\n  sourceProject: string;\n  targetProject: string;\n  sourceComponent?: ComponentMetadata;\n  targetComponent?: ComponentMetadata;\n  preferInstallDependencies?: boolean;\n}\n"],"names":[],"mappings":"AAAA;;;CAGC,GAGD;;CAEC,GACD,WAMC"}

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

@@ -12,4 +12,5 @@ export interface PyAgentReactConnectionGeneratorSchema {
   targetProject: string;
   sourceComponent?: ComponentMetadata;
   targetComponent?: ComponentMetadata;
+  preferInstallDependencies?: boolean;
 }

package/src/py/agent/react-connection/schema.json

@@ -20,6 +20,11 @@
     "targetComponent": {
       "type": "string",
       "description": "The agent component name"
+    },
+    "preferInstallDependencies": {
+      "type": "boolean",
+      "description": "Whether to prefer installing dependencies after the generator runs. Set to false to defer installing when batching multiple generators (an install still runs if needed so subsequent generators can compute the Nx project graph); install once at the end.",
+      "default": true
     }
   },
   "required": ["sourceProject", "targetProject"]

package/src/py/agent/schema.d.js

@@ -0,0 +1,6 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */ export { };
+
+//# sourceMappingURL=schema.d.js.map

package/src/py/agent/schema.d.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../../../packages/nx-plugin/src/py/agent/schema.d.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { IacOption } from '../../utils/iac';\n\nexport type PyAgentFramework = 'strands' | 'langchain';\nexport type PyAgentInfra = 'agentcore' | 'none';\nexport type AgentProtocol = 'http' | 'a2a' | 'ag-ui';\n\nexport type PyAgentAuth = 'iam' | 'cognito';\n\nexport interface PyAgentGeneratorSchema {\n  project: string;\n  framework?: PyAgentFramework;\n  name?: string;\n  infra?: PyAgentInfra;\n  auth?: PyAgentAuth;\n  protocol?: AgentProtocol;\n  iac: IacOption;\n  preferInstallDependencies?: boolean;\n}\n"],"names":[],"mappings":"AAAA;;;CAGC,GAUD,WASC"}

package/src/py/agent/schema.d.ts

@@ -19,4 +19,5 @@ export interface PyAgentGeneratorSchema {
   auth?: PyAgentAuth;
   protocol?: AgentProtocol;
   iac: IacOption;
+  preferInstallDependencies?: boolean;
 }

package/src/py/agent/schema.json

@@ -59,6 +59,11 @@
       "enum": ["agentcore", "none"],
       "default": "agentcore",
       "x-priority": "important"
+    },
+    "preferInstallDependencies": {
+      "type": "boolean",
+      "description": "Whether to prefer installing dependencies after the generator runs. Set to false to defer installing when batching multiple generators (an install still runs if needed so subsequent generators can compute the Nx project graph); install once at the end.",
+      "default": true
     }
   },
   "required": ["project"]

package/src/py/api/generator.js

@@ -1,15 +1,14 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.PY_API_GENERATOR_INFO = void 0;
-exports.pyApiGenerator = pyApiGenerator;
-const nx_1 = require("../../utils/nx");
-const generator_1 = require("../fast-api/generator");
-exports.PY_API_GENERATOR_INFO = (0, nx_1.getGeneratorInfo)(__filename);
-async function pyApiGenerator(tree, options) {
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */ import { getGeneratorInfo } from "../../utils/nx.js";
+import { pyFastApiProjectGenerator } from "../fast-api/generator.js";
+export const PY_API_GENERATOR_INFO = getGeneratorInfo(import.meta.filename);
+export async function pyApiGenerator(tree, options) {
     const framework = options.framework ?? 'fastapi';
-    switch (framework) {
+    switch(framework){
         case 'fastapi':
-            return (0, generator_1.pyFastApiProjectGenerator)(tree, {
+            return pyFastApiProjectGenerator(tree, {
                 name: options.name,
                 infra: options.infra,
                 integrationPattern: options.integrationPattern,
@@ -18,8 +17,10 @@ async function pyApiGenerator(tree, options) {
                 subDirectory: options.subDirectory,
                 moduleName: options.moduleName,
                 iac: options.iac,
+                preferInstallDependencies: options.preferInstallDependencies
             });
     }
 }
-exports.default = pyApiGenerator;
+export default pyApiGenerator;
+
 //# sourceMappingURL=generator.js.map

package/src/py/api/generator.js.map

@@ -1 +1 @@
-{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../../../../packages/nx-plugin/src/py/api/generator.ts"],"names":[],"mappings":";;;AAYA,wCAmBC;AA1BD,uCAAwE;AACxE,qDAAkE;AAGrD,QAAA,qBAAqB,GAChC,IAAA,qBAAgB,EAAC,UAAU,CAAC,CAAC;AAExB,KAAK,UAAU,cAAc,CAClC,IAAU,EACV,OAA6B;IAE7B,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,SAAS,CAAC;IAEjD,QAAQ,SAAS,EAAE,CAAC;QAClB,KAAK,SAAS;YACZ,OAAO,IAAA,qCAAyB,EAAC,IAAI,EAAE;gBACrC,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;gBAC9C,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,SAAS,EAAE,OAAO,CAAC,SAAS;gBAC5B,YAAY,EAAE,OAAO,CAAC,YAAY;gBAClC,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,GAAG,EAAE,OAAO,CAAC,GAAG;aACjB,CAAC,CAAC;IACP,CAAC;AACH,CAAC;AAED,kBAAe,cAAc,CAAC"}
+{"version":3,"sources":["../../../../../../packages/nx-plugin/src/py/api/generator.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nimport type { Tree } from '@nx/devkit';\nimport { getGeneratorInfo, type NxGeneratorInfo } from '../../utils/nx';\nimport { pyFastApiProjectGenerator } from '../fast-api/generator';\nimport type { PyApiGeneratorSchema } from './schema';\n\nexport const PY_API_GENERATOR_INFO: NxGeneratorInfo = getGeneratorInfo(\n  import.meta.filename,\n);\n\nexport async function pyApiGenerator(\n  tree: Tree,\n  options: PyApiGeneratorSchema,\n) {\n  const framework = options.framework ?? 'fastapi';\n\n  switch (framework) {\n    case 'fastapi':\n      return pyFastApiProjectGenerator(tree, {\n        name: options.name,\n        infra: options.infra,\n        integrationPattern: options.integrationPattern,\n        auth: options.auth,\n        directory: options.directory,\n        subDirectory: options.subDirectory,\n        moduleName: options.moduleName,\n        iac: options.iac,\n        preferInstallDependencies: options.preferInstallDependencies,\n      });\n  }\n}\n\nexport default pyApiGenerator;\n"],"names":["getGeneratorInfo","pyFastApiProjectGenerator","PY_API_GENERATOR_INFO","filename","pyApiGenerator","tree","options","framework","name","infra","integrationPattern","auth","directory","subDirectory","moduleName","iac","preferInstallDependencies"],"mappings":"AAAA;;;CAGC,GAED,SAASA,gBAAgB,QAA8B,oBAAiB;AACxE,SAASC,yBAAyB,QAAQ,2BAAwB;AAGlE,OAAO,MAAMC,wBAAyCF,iBACpD,YAAYG,QAAQ,EACpB;AAEF,OAAO,eAAeC,eACpBC,IAAU,EACVC,OAA6B;IAE7B,MAAMC,YAAYD,QAAQC,SAAS,IAAI;IAEvC,OAAQA;QACN,KAAK;YACH,OAAON,0BAA0BI,MAAM;gBACrCG,MAAMF,QAAQE,IAAI;gBAClBC,OAAOH,QAAQG,KAAK;gBACpBC,oBAAoBJ,QAAQI,kBAAkB;gBAC9CC,MAAML,QAAQK,IAAI;gBAClBC,WAAWN,QAAQM,SAAS;gBAC5BC,cAAcP,QAAQO,YAAY;gBAClCC,YAAYR,QAAQQ,UAAU;gBAC9BC,KAAKT,QAAQS,GAAG;gBAChBC,2BAA2BV,QAAQU,yBAAyB;YAC9D;IACJ;AACF;AAEA,eAAeZ,eAAe"}

package/src/py/api/schema.d.js

@@ -0,0 +1,6 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */ export { };
+
+//# sourceMappingURL=schema.d.js.map

package/src/py/api/schema.d.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../../../packages/nx-plugin/src/py/api/schema.d.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nimport { IacOption } from '../../utils/iac';\n\nexport interface PyApiGeneratorSchema {\n  readonly name: string;\n  readonly framework?: 'fastapi';\n  readonly infra: 'rest-lambda' | 'http-lambda' | 'none';\n  readonly integrationPattern?: 'isolated' | 'shared';\n  readonly auth: 'iam' | 'cognito' | 'custom';\n  readonly directory?: string;\n  readonly subDirectory?: string;\n  readonly moduleName?: string;\n  readonly iac: IacOption;\n  readonly preferInstallDependencies?: boolean;\n}\n"],"names":[],"mappings":"AAAA;;;CAGC,GAGD,WAWC"}

package/src/py/api/schema.d.ts

@@ -14,4 +14,5 @@ export interface PyApiGeneratorSchema {
   readonly subDirectory?: string;
   readonly moduleName?: string;
   readonly iac: IacOption;
+  readonly preferInstallDependencies?: boolean;
 }

package/src/py/api/schema.json

@@ -94,6 +94,11 @@
       "enum": ["rest-lambda", "http-lambda", "none"],
       "x-prompt": "What infrastructure would you like to deploy your API with?",
       "x-priority": "important"
+    },
+    "preferInstallDependencies": {
+      "type": "boolean",
+      "description": "Whether to prefer installing dependencies after the generator runs. Set to false to defer installing when batching multiple generators (an install still runs if needed so subsequent generators can compute the Nx project graph); install once at the end.",
+      "default": true
     }
   },
   "required": ["name"]

package/src/py/dynamodb/__snapshots__/generator.spec.ts.snap

@@ -3,7 +3,7 @@
 exports[`py#dynamodb generator > should generate scripts for finch engine 1`] = `
 "{
   "runtimeConfigKey": "MyTable",
-  "serveLocal": {
+  "localDev": {
     "port": 8000,
     "tableName": "proj-my-table",
     "image": "public.ecr.aws/aws-dynamodb-local/aws-dynamodb-local:latest",
@@ -35,7 +35,7 @@ exports[`py#dynamodb generator > should generate terraform modules when iac is t
   required_providers {
     aws = {
       source  = "hashicorp/aws"
-      version = "~> 6.0"
+      version = "6.52.0"
     }
   }
 }
@@ -192,7 +192,11 @@ exports[`py#dynamodb generator > should generate terraform modules when iac is t
   required_providers {
     aws = {
       source  = "hashicorp/aws"
-      version = "~> 6.0"
+      version = "6.52.0"
+    }
+    random = {
+      source  = "hashicorp/random"
+      version = "3.9.0"
     }
   }
 }
@@ -221,6 +225,12 @@ variable "enable_key_rotation" {
   default     = true
 }
 
+resource "random_string" "suffix" {
+  length  = 8
+  special = false
+  upper   = false
+}
+
 locals {
   dynamodb_config = jsondecode(file("\${path.module}/../../../../../../../packages/my_table/config.json"))
   global_secondary_indexes = [for gsi in local.dynamodb_config.tableConfig.globalSecondaryIndexes : {
@@ -233,7 +243,7 @@ locals {
 
 module "dynamodb_table" {
   source                         = "../../../core/dynamodb"
-  name                           = "proj-my-table"
+  name                           = "proj-my-table-\${random_string.suffix.result}"
   billing_mode                   = var.billing_mode
   point_in_time_recovery_enabled = var.point_in_time_recovery_enabled
   deletion_protection_enabled    = var.deletion_protection_enabled
@@ -271,10 +281,10 @@ exports[`py#dynamodb generator > should generate the dynamodb project > packages
 import { readFileSync } from 'fs';
 
 const {
-  serveLocal: { port, tableName },
+  localDev: { port, tableName },
   tableConfig: { globalSecondaryIndexes },
 } = JSON.parse(readFileSync('config.json', 'utf-8')) as {
-  serveLocal: { port: number; tableName: string };
+  localDev: { port: number; tableName: string };
   tableConfig: {
     globalSecondaryIndexes: {
       indexName: string;
@@ -462,7 +472,7 @@ import { readFileSync } from 'fs';
 
 const { containerEngine, image } = JSON.parse(
   readFileSync('config.json', 'utf-8'),
-).serveLocal;
+).localDev;
 
 const inspect = spawnSync(containerEngine, ['image', 'inspect', image], {
   stdio: 'ignore',
@@ -484,7 +494,7 @@ import { readFileSync } from 'fs';
 
 const { containerEngine, containerName, image, port } = JSON.parse(
   readFileSync('config.json', 'utf-8'),
-).serveLocal;
+).localDev;
 
 const containerExists = (): boolean =>
   spawnSync(containerEngine, ['container', 'inspect', containerName], {
@@ -500,12 +510,27 @@ const isRunning = (): boolean => {
   return result.status === 0 && result.stdout.trim() === 'true';
 };
 
-// Multiple dynamo table packages run serve-local in parallel sharing the same
+// Ensure the image is available locally before starting the container.
+const ensureImage = (): void => {
+  const inspect = spawnSync(containerEngine, ['image', 'inspect', image], {
+    stdio: 'ignore',
+  });
+  if (inspect.status !== 0) {
+    const pull = spawnSync(containerEngine, ['pull', image], {
+      stdio: 'inherit',
+    });
+    if (pull.status !== 0) process.exit(pull.status ?? 1);
+  }
+};
+
+// Multiple dynamo table packages run dev in parallel sharing the same
 // container. All processes stream logs, but only the one that created the
 // container stops it on exit — others just detach.
 let created = false;
 let runRetries = 0;
 
+ensureImage();
+
 for (;;) {
   if (containerExists()) {
     if (!isRunning()) {
@@ -605,15 +630,15 @@ runtime_config_key = "MyTable"
 
 
 # config.json is only present in local dev (not in the deployed bundle),
-# so calls to _serve_local must be guarded by is_local().
-def _serve_local() -> dict:
+# so calls to _local_dev must be guarded by is_local().
+def _local_dev() -> dict:
     config_path = os.path.join(os.path.dirname(__file__), "..", "config.json")
     with open(config_path) as f:
-        return json.load(f)["serveLocal"]
+        return json.load(f)["localDev"]
 
 
 def is_local() -> bool:
-    return os.environ.get("SERVE_LOCAL") == "true"
+    return os.environ.get("LOCAL_DEV") == "true"
 
 
 _table_name: str | None = None
@@ -622,7 +647,7 @@ _table_name: str | None = None
 def get_table_name() -> str:
     global _table_name
     if is_local():
-        return _serve_local()["tableName"]
+        return _local_dev()["tableName"]
     if _table_name is None:
         from aws_lambda_powertools.utilities.parameters import AppConfigProvider
 
@@ -654,7 +679,7 @@ from pynamodb.attributes import DiscriminatorAttribute, UnicodeAttribute
 from pynamodb.indexes import AllProjection, GlobalSecondaryIndex
 from pynamodb.models import Model
 
-from ..client import _serve_local, get_table_name, is_local
+from ..client import _local_dev, get_table_name, is_local
 
 
 class Gsi1Index(GlobalSecondaryIndex):
@@ -691,7 +716,7 @@ class BaseModel(Model):
     class Meta:
         table_name: str
         region = os.environ.get("AWS_REGION", "us-east-1")
-        host = f"http://localhost:{_serve_local()['port']}" if is_local() else None
+        host = f"http://localhost:{_local_dev()['port']}" if is_local() else None
         aws_access_key_id = "UNUSED" if is_local() else None
         aws_secret_access_key = "UNUSED" if is_local() else None
 
@@ -974,10 +999,10 @@ exports[`py#dynamodb generator > should generate with infra=none then upgrade to
 import { readFileSync } from 'fs';
 
 const {
-  serveLocal: { port, tableName },
+  localDev: { port, tableName },
   tableConfig: { globalSecondaryIndexes },
 } = JSON.parse(readFileSync('config.json', 'utf-8')) as {
-  serveLocal: { port: number; tableName: string };
+  localDev: { port: number; tableName: string };
   tableConfig: {
     globalSecondaryIndexes: {
       indexName: string;
@@ -1165,7 +1190,7 @@ import { readFileSync } from 'fs';
 
 const { containerEngine, image } = JSON.parse(
   readFileSync('config.json', 'utf-8'),
-).serveLocal;
+).localDev;
 
 const inspect = spawnSync(containerEngine, ['image', 'inspect', image], {
   stdio: 'ignore',
@@ -1187,7 +1212,7 @@ import { readFileSync } from 'fs';
 
 const { containerEngine, containerName, image, port } = JSON.parse(
   readFileSync('config.json', 'utf-8'),
-).serveLocal;
+).localDev;
 
 const containerExists = (): boolean =>
   spawnSync(containerEngine, ['container', 'inspect', containerName], {
@@ -1203,12 +1228,27 @@ const isRunning = (): boolean => {
   return result.status === 0 && result.stdout.trim() === 'true';
 };
 
-// Multiple dynamo table packages run serve-local in parallel sharing the same
+// Ensure the image is available locally before starting the container.
+const ensureImage = (): void => {
+  const inspect = spawnSync(containerEngine, ['image', 'inspect', image], {
+    stdio: 'ignore',
+  });
+  if (inspect.status !== 0) {
+    const pull = spawnSync(containerEngine, ['pull', image], {
+      stdio: 'inherit',
+    });
+    if (pull.status !== 0) process.exit(pull.status ?? 1);
+  }
+};
+
+// Multiple dynamo table packages run dev in parallel sharing the same
 // container. All processes stream logs, but only the one that created the
 // container stops it on exit — others just detach.
 let created = false;
 let runRetries = 0;
 
+ensureImage();
+
 for (;;) {
   if (containerExists()) {
     if (!isRunning()) {
@@ -1308,15 +1348,15 @@ runtime_config_key = "MyTable"
 
 
 # config.json is only present in local dev (not in the deployed bundle),
-# so calls to _serve_local must be guarded by is_local().
-def _serve_local() -> dict:
+# so calls to _local_dev must be guarded by is_local().
+def _local_dev() -> dict:
     config_path = os.path.join(os.path.dirname(__file__), "..", "config.json")
     with open(config_path) as f:
-        return json.load(f)["serveLocal"]
+        return json.load(f)["localDev"]
 
 
 def is_local() -> bool:
-    return os.environ.get("SERVE_LOCAL") == "true"
+    return os.environ.get("LOCAL_DEV") == "true"
 
 
 _table_name: str | None = None
@@ -1325,7 +1365,7 @@ _table_name: str | None = None
 def get_table_name() -> str:
     global _table_name
     if is_local():
-        return _serve_local()["tableName"]
+        return _local_dev()["tableName"]
     if _table_name is None:
         from aws_lambda_powertools.utilities.parameters import AppConfigProvider
 
@@ -1357,7 +1397,7 @@ from pynamodb.attributes import DiscriminatorAttribute, UnicodeAttribute
 from pynamodb.indexes import AllProjection, GlobalSecondaryIndex
 from pynamodb.models import Model
 
-from ..client import _serve_local, get_table_name, is_local
+from ..client import _local_dev, get_table_name, is_local
 
 
 class Gsi1Index(GlobalSecondaryIndex):
@@ -1394,7 +1434,7 @@ class BaseModel(Model):
     class Meta:
         table_name: str
         region = os.environ.get("AWS_REGION", "us-east-1")
-        host = f"http://localhost:{_serve_local()['port']}" if is_local() else None
+        host = f"http://localhost:{_local_dev()['port']}" if is_local() else None
         aws_access_key_id = "UNUSED" if is_local() else None
         aws_secret_access_key = "UNUSED" if is_local() else None
 

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

@@ -1,19 +1,19 @@
 // Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
 
-exports[`py#dynamodb agent-connection generator > should add dynamodb serve-local dependency to agent serve-local 1`] = `
+exports[`py#dynamodb agent-connection generator > should add dynamodb dev dependency to agent dev 1`] = `
 {
   "$schema": "../../node_modules/nx/schemas/project-schema.json",
   "name": "my-agent",
   "root": "packages/my-agent",
   "targets": {
-    "agent-serve-local": {
+    "agent-dev": {
       "continuous": true,
       "dependsOn": [
         {
           "projects": [
             "db",
           ],
-          "target": "serve-local",
+          "target": "dev",
         },
       ],
       "executor": "nx:run-commands",
@@ -28,14 +28,14 @@ exports[`py#dynamodb agent-connection generator > should use sourceComponent nam
   "name": "my-agent",
   "root": "packages/my-agent",
   "targets": {
-    "custom-agent-serve-local": {
+    "custom-agent-dev": {
       "continuous": true,
       "dependsOn": [
         {
           "projects": [
             "db",
           ],
-          "target": "serve-local",
+          "target": "dev",
         },
       ],
       "executor": "nx:run-commands",

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

@@ -1,39 +1,39 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.pyDynamoDBAgentConnectionGenerator = exports.PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO = void 0;
 /**
  * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
  * SPDX-License-Identifier: Apache-2.0
- */
-const devkit_1 = require("@nx/devkit");
-const format_1 = require("../../../utils/format");
-const metrics_1 = require("../../../utils/metrics");
-const nx_1 = require("../../../utils/nx");
-const nxlv_python_1 = require("../../../utils/nxlv-python");
-const py_1 = require("../../../utils/py");
-exports.PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO = (0, nx_1.getGeneratorInfo)(__filename);
-const pyDynamoDBAgentConnectionGenerator = async (tree, options) => {
-    const sourceProject = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.sourceProject);
-    const targetProject = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.targetProject);
-    (0, py_1.addWorkspaceDependencyToPyProject)(tree, sourceProject, targetProject);
+ */ import { updateProjectConfiguration } from "@nx/devkit";
+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 { addWorkspaceDependencyToPyProject } from "../../../utils/py.js";
+export const PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO = getGeneratorInfo(import.meta.filename);
+export const pyDynamoDBAgentConnectionGenerator = async (tree, options)=>{
+    const sourceProject = readProjectConfigurationUnqualified(tree, options.sourceProject);
+    const targetProject = readProjectConfigurationUnqualified(tree, options.targetProject);
+    addWorkspaceDependencyToPyProject(tree, sourceProject, targetProject);
     const agentName = options.sourceComponent?.name ?? 'agent';
-    const serveLocalTarget = `${agentName}-serve-local`;
-    if (sourceProject.targets?.[serveLocalTarget]) {
-        (0, nx_1.addDependencyToTargetIfNotPresent)(sourceProject, serveLocalTarget, {
-            projects: [targetProject.name],
-            target: 'serve-local',
+    const devTarget = `${agentName}-dev`;
+    if (sourceProject.targets?.[devTarget]) {
+        addDependencyToTargetIfNotPresent(sourceProject, devTarget, {
+            projects: [
+                targetProject.name
+            ],
+            target: 'dev'
         });
-        (0, devkit_1.updateProjectConfiguration)(tree, sourceProject.name, sourceProject);
+        updateProjectConfiguration(tree, sourceProject.name, sourceProject);
     }
-    await (0, metrics_1.addGeneratorMetricsIfApplicable)(tree, [
-        exports.PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO,
+    await addGeneratorMetricsIfApplicable(tree, [
+        PY_DYNAMODB_AGENT_CONNECTION_GENERATOR_INFO
     ]);
-    await (0, format_1.formatFilesInSubtree)(tree);
-    return async () => {
-        (0, devkit_1.installPackagesTask)(tree);
-        await new nxlv_python_1.UVProvider(tree.root, new nxlv_python_1.Logger(), tree).install();
-    };
+    await formatFilesInSubtree(tree);
+    return ()=>installDependencies(tree, options.preferInstallDependencies, {
+            languages: [
+                'typescript',
+                'python'
+            ]
+        });
 };
-exports.pyDynamoDBAgentConnectionGenerator = pyDynamoDBAgentConnectionGenerator;
-exports.default = exports.pyDynamoDBAgentConnectionGenerator;
+export default pyDynamoDBAgentConnectionGenerator;
+
 //# sourceMappingURL=generator.js.map