@aws/nx-plugin 1.0.0-rc.95 → 1.0.0-rc.97

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

@@ -10,7 +10,7 @@ import type { PyAgentA2aConnectionGeneratorSchema } from './schema';
 export interface PyAgentA2aConnectionMetadata {
     readonly framework: AgentFramework;
 }
-export declare const DEPENDENCIES: import("../../../utils/declared-dependencies.js").DependencyDeclaration<readonly import("../../../utils/declared-dependencies.js").DeclaredTsDependency<"@ag-ui/client" | "@ag-ui/core" | "@ag-ui/encoder" | "@biomejs/biome" | "@a2a-js/sdk" | "@aws/aws-distro-opentelemetry-node-autoinstrumentation" | "@opentelemetry/propagator-jaeger" | "minimatch" | "@aws-sdk/client-dynamodb" | "@aws-sdk/client-api-gateway" | "@aws-sdk/client-iam" | "@aws-sdk/client-bedrock-agentcore" | "@aws-sdk/client-bedrock-runtime" | "@aws-sdk/client-s3" | "@aws-sdk/client-sts" | "@aws-sdk/credential-providers" | "@aws-sdk/credential-provider-cognito-identity" | "@aws-sdk/client-secrets-manager" | "@aws-sdk/rds-signer" | "@smithy/server-apigateway" | "@smithy/server-node" | "@aws-lambda-powertools/logger" | "@aws-lambda-powertools/metrics" | "@aws-lambda-powertools/parameters" | "@aws-lambda-powertools/tracer" | "@aws-lambda-powertools/parser" | "@aws-sdk/client-appconfigdata" | "@middy/core" | "@nxlv/python" | "@nx-extend/terraform" | "nx" | "@nx/devkit" | "@nx/js" | "@nx/react" | "@nx/vite" | "@nx/vitest" | "@nx/workspace" | "create-nx-workspace" | "@swc-node/register" | "@swc/core" | "@modelcontextprotocol/sdk" | "@modelcontextprotocol/inspector" | "@ag-ui/a2ui-toolkit" | "@ag-ui/aws-strands" | "agent-chat-cli" | "@copilotkit/react-core" | "rxjs" | "@strands-agents/sdk" | "@tanstack/react-router" | "@tanstack/router-plugin" | "@tanstack/router-generator" | "@tanstack/virtual-file-routes" | "@tanstack/router-utils" | "@cloudscape-design/board-components" | "@cloudscape-design/chat-components" | "@cloudscape-design/components" | "@cloudscape-design/global-styles" | "@tanstack/react-query" | "@tanstack/react-query-devtools" | "@trpc/tanstack-react-query" | "@trpc/client" | "@trpc/server" | "@types/node" | "@types/aws-lambda" | "@types/cors" | "@types/pg" | "@types/ws" | "@types/express" | "@smithy/config-resolver" | "@smithy/node-config-provider" | "@smithy/node-http-handler" | "@smithy/types" | "@vitest/coverage-v8" | "@vitest/ui" | "@astrojs/react" | "@astrojs/starlight" | "astro" | "cookie" | "aws4fetch" | "aws-cdk" | "aws-cdk-lib" | "aws-xray-sdk-core" | "constructs" | "cors" | "chalk" | "class-variance-authority" | "clsx" | "commander" | "electrodb" | "esbuild" | "event-source-polyfill" | "@types/event-source-polyfill" | "@prisma/adapter-mariadb" | "@prisma/adapter-pg" | "@prisma/client" | "ejs" | "@types/ejs" | "express" | "fast-glob" | "husky" | "fs-extra" | "@types/fs-extra" | "mariadb" | "mise" | "npm" | "npm-check-updates" | "oidc-client-ts" | "pg" | "prisma" | "react-oidc-context" | "react" | "react-dom" | "rolldown" | "rolldown-plugin-dts" | "shx" | "simple-git" | "source-map-support" | "starlight-blog" | "tailwindcss" | "@tailwindcss/vite" | "tsx" | "lucide-react" | "radix-ui" | "shadcn" | "tw-animate-css" | "tailwind-merge" | "vite" | "typescript" | "vitest" | "zod" | "ws", PyAgentA2aConnectionMetadata>[], readonly [{
+export declare const DEPENDENCIES: import("../../../utils/declared-dependencies.js").DependencyDeclaration<readonly import("../../../utils/declared-dependencies.js").DeclaredTsDependency<"@ag-ui/client" | "@ag-ui/core" | "@ag-ui/encoder" | "@biomejs/biome" | "@a2a-js/sdk" | "@aws/aws-distro-opentelemetry-node-autoinstrumentation" | "@opentelemetry/propagator-jaeger" | "minimatch" | "deepmerge-ts" | "mysql2" | "@aws-sdk/client-dynamodb" | "@aws-sdk/client-api-gateway" | "@aws-sdk/client-iam" | "@aws-sdk/client-bedrock-agentcore" | "@aws-sdk/client-bedrock-runtime" | "@aws-sdk/client-s3" | "@aws-sdk/client-sts" | "@aws-sdk/client-cognito-identity-provider" | "@aws-sdk/credential-providers" | "@aws-sdk/credential-provider-cognito-identity" | "@aws-sdk/client-secrets-manager" | "@aws-sdk/rds-signer" | "@smithy/server-apigateway" | "@smithy/server-node" | "@aws-lambda-powertools/logger" | "@aws-lambda-powertools/metrics" | "@aws-lambda-powertools/parameters" | "@aws-lambda-powertools/tracer" | "@aws-lambda-powertools/parser" | "@aws-sdk/client-appconfigdata" | "@middy/core" | "@nxlv/python" | "@nx-extend/terraform" | "nx" | "@nx/devkit" | "@nx/js" | "@nx/react" | "@nx/vite" | "@nx/vitest" | "@nx/workspace" | "create-nx-workspace" | "@swc-node/register" | "@swc/core" | "@modelcontextprotocol/sdk" | "@modelcontextprotocol/inspector" | "@ag-ui/a2ui-toolkit" | "@ag-ui/aws-strands" | "agent-chat-cli" | "@copilotkit/react-core" | "rxjs" | "@strands-agents/sdk" | "@tanstack/react-router" | "@tanstack/router-plugin" | "@tanstack/router-generator" | "@tanstack/virtual-file-routes" | "@tanstack/router-utils" | "@cloudscape-design/board-components" | "@cloudscape-design/chat-components" | "@cloudscape-design/components" | "@cloudscape-design/global-styles" | "@tanstack/react-query" | "@tanstack/react-query-devtools" | "@trpc/tanstack-react-query" | "@trpc/client" | "@trpc/server" | "@types/node" | "@types/aws-lambda" | "@types/cors" | "@types/pg" | "@types/ws" | "@types/express" | "@smithy/config-resolver" | "@smithy/node-config-provider" | "@smithy/node-http-handler" | "@smithy/types" | "@vitest/coverage-v8" | "@vitest/ui" | "@astrojs/react" | "@astrojs/starlight" | "astro" | "cookie" | "aws4fetch" | "aws-cdk" | "aws-cdk-lib" | "aws-xray-sdk-core" | "constructs" | "cors" | "chalk" | "class-variance-authority" | "clsx" | "commander" | "electrodb" | "esbuild" | "event-source-polyfill" | "@types/event-source-polyfill" | "@prisma/adapter-mariadb" | "@prisma/adapter-pg" | "@prisma/client" | "ejs" | "@types/ejs" | "express" | "fast-glob" | "husky" | "fs-extra" | "@types/fs-extra" | "mariadb" | "mise" | "npm" | "npm-check-updates" | "oidc-client-ts" | "pg" | "prisma" | "react-oidc-context" | "react" | "react-dom" | "rolldown" | "rolldown-plugin-dts" | "shx" | "simple-git" | "source-map-support" | "starlight-blog" | "tailwindcss" | "@tailwindcss/vite" | "tsx" | "lucide-react" | "radix-ui" | "shadcn" | "tw-animate-css" | "tailwind-merge" | "vite" | "typescript" | "vitest" | "jsdom" | "zod" | "ws", PyAgentA2aConnectionMetadata>[], readonly [{
     readonly name: "boto3";
 }, {
     readonly name: "httpx";

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

@@ -10,7 +10,7 @@ import type { PyAgentGatewayConnectionGeneratorSchema } from './schema';
 export interface PyAgentGatewayConnectionMetadata {
     readonly framework: AgentFramework;
 }
-export declare const DEPENDENCIES: import("../../../utils/declared-dependencies.js").DependencyDeclaration<readonly import("../../../utils/declared-dependencies.js").DeclaredTsDependency<"@ag-ui/client" | "@ag-ui/core" | "@ag-ui/encoder" | "@biomejs/biome" | "@a2a-js/sdk" | "@aws/aws-distro-opentelemetry-node-autoinstrumentation" | "@opentelemetry/propagator-jaeger" | "minimatch" | "@aws-sdk/client-dynamodb" | "@aws-sdk/client-api-gateway" | "@aws-sdk/client-iam" | "@aws-sdk/client-bedrock-agentcore" | "@aws-sdk/client-bedrock-runtime" | "@aws-sdk/client-s3" | "@aws-sdk/client-sts" | "@aws-sdk/credential-providers" | "@aws-sdk/credential-provider-cognito-identity" | "@aws-sdk/client-secrets-manager" | "@aws-sdk/rds-signer" | "@smithy/server-apigateway" | "@smithy/server-node" | "@aws-lambda-powertools/logger" | "@aws-lambda-powertools/metrics" | "@aws-lambda-powertools/parameters" | "@aws-lambda-powertools/tracer" | "@aws-lambda-powertools/parser" | "@aws-sdk/client-appconfigdata" | "@middy/core" | "@nxlv/python" | "@nx-extend/terraform" | "nx" | "@nx/devkit" | "@nx/js" | "@nx/react" | "@nx/vite" | "@nx/vitest" | "@nx/workspace" | "create-nx-workspace" | "@swc-node/register" | "@swc/core" | "@modelcontextprotocol/sdk" | "@modelcontextprotocol/inspector" | "@ag-ui/a2ui-toolkit" | "@ag-ui/aws-strands" | "agent-chat-cli" | "@copilotkit/react-core" | "rxjs" | "@strands-agents/sdk" | "@tanstack/react-router" | "@tanstack/router-plugin" | "@tanstack/router-generator" | "@tanstack/virtual-file-routes" | "@tanstack/router-utils" | "@cloudscape-design/board-components" | "@cloudscape-design/chat-components" | "@cloudscape-design/components" | "@cloudscape-design/global-styles" | "@tanstack/react-query" | "@tanstack/react-query-devtools" | "@trpc/tanstack-react-query" | "@trpc/client" | "@trpc/server" | "@types/node" | "@types/aws-lambda" | "@types/cors" | "@types/pg" | "@types/ws" | "@types/express" | "@smithy/config-resolver" | "@smithy/node-config-provider" | "@smithy/node-http-handler" | "@smithy/types" | "@vitest/coverage-v8" | "@vitest/ui" | "@astrojs/react" | "@astrojs/starlight" | "astro" | "cookie" | "aws4fetch" | "aws-cdk" | "aws-cdk-lib" | "aws-xray-sdk-core" | "constructs" | "cors" | "chalk" | "class-variance-authority" | "clsx" | "commander" | "electrodb" | "esbuild" | "event-source-polyfill" | "@types/event-source-polyfill" | "@prisma/adapter-mariadb" | "@prisma/adapter-pg" | "@prisma/client" | "ejs" | "@types/ejs" | "express" | "fast-glob" | "husky" | "fs-extra" | "@types/fs-extra" | "mariadb" | "mise" | "npm" | "npm-check-updates" | "oidc-client-ts" | "pg" | "prisma" | "react-oidc-context" | "react" | "react-dom" | "rolldown" | "rolldown-plugin-dts" | "shx" | "simple-git" | "source-map-support" | "starlight-blog" | "tailwindcss" | "@tailwindcss/vite" | "tsx" | "lucide-react" | "radix-ui" | "shadcn" | "tw-animate-css" | "tailwind-merge" | "vite" | "typescript" | "vitest" | "zod" | "ws", PyAgentGatewayConnectionMetadata>[], readonly [{
+export declare const DEPENDENCIES: import("../../../utils/declared-dependencies.js").DependencyDeclaration<readonly import("../../../utils/declared-dependencies.js").DeclaredTsDependency<"@ag-ui/client" | "@ag-ui/core" | "@ag-ui/encoder" | "@biomejs/biome" | "@a2a-js/sdk" | "@aws/aws-distro-opentelemetry-node-autoinstrumentation" | "@opentelemetry/propagator-jaeger" | "minimatch" | "deepmerge-ts" | "mysql2" | "@aws-sdk/client-dynamodb" | "@aws-sdk/client-api-gateway" | "@aws-sdk/client-iam" | "@aws-sdk/client-bedrock-agentcore" | "@aws-sdk/client-bedrock-runtime" | "@aws-sdk/client-s3" | "@aws-sdk/client-sts" | "@aws-sdk/client-cognito-identity-provider" | "@aws-sdk/credential-providers" | "@aws-sdk/credential-provider-cognito-identity" | "@aws-sdk/client-secrets-manager" | "@aws-sdk/rds-signer" | "@smithy/server-apigateway" | "@smithy/server-node" | "@aws-lambda-powertools/logger" | "@aws-lambda-powertools/metrics" | "@aws-lambda-powertools/parameters" | "@aws-lambda-powertools/tracer" | "@aws-lambda-powertools/parser" | "@aws-sdk/client-appconfigdata" | "@middy/core" | "@nxlv/python" | "@nx-extend/terraform" | "nx" | "@nx/devkit" | "@nx/js" | "@nx/react" | "@nx/vite" | "@nx/vitest" | "@nx/workspace" | "create-nx-workspace" | "@swc-node/register" | "@swc/core" | "@modelcontextprotocol/sdk" | "@modelcontextprotocol/inspector" | "@ag-ui/a2ui-toolkit" | "@ag-ui/aws-strands" | "agent-chat-cli" | "@copilotkit/react-core" | "rxjs" | "@strands-agents/sdk" | "@tanstack/react-router" | "@tanstack/router-plugin" | "@tanstack/router-generator" | "@tanstack/virtual-file-routes" | "@tanstack/router-utils" | "@cloudscape-design/board-components" | "@cloudscape-design/chat-components" | "@cloudscape-design/components" | "@cloudscape-design/global-styles" | "@tanstack/react-query" | "@tanstack/react-query-devtools" | "@trpc/tanstack-react-query" | "@trpc/client" | "@trpc/server" | "@types/node" | "@types/aws-lambda" | "@types/cors" | "@types/pg" | "@types/ws" | "@types/express" | "@smithy/config-resolver" | "@smithy/node-config-provider" | "@smithy/node-http-handler" | "@smithy/types" | "@vitest/coverage-v8" | "@vitest/ui" | "@astrojs/react" | "@astrojs/starlight" | "astro" | "cookie" | "aws4fetch" | "aws-cdk" | "aws-cdk-lib" | "aws-xray-sdk-core" | "constructs" | "cors" | "chalk" | "class-variance-authority" | "clsx" | "commander" | "electrodb" | "esbuild" | "event-source-polyfill" | "@types/event-source-polyfill" | "@prisma/adapter-mariadb" | "@prisma/adapter-pg" | "@prisma/client" | "ejs" | "@types/ejs" | "express" | "fast-glob" | "husky" | "fs-extra" | "@types/fs-extra" | "mariadb" | "mise" | "npm" | "npm-check-updates" | "oidc-client-ts" | "pg" | "prisma" | "react-oidc-context" | "react" | "react-dom" | "rolldown" | "rolldown-plugin-dts" | "shx" | "simple-git" | "source-map-support" | "starlight-blog" | "tailwindcss" | "@tailwindcss/vite" | "tsx" | "lucide-react" | "radix-ui" | "shadcn" | "tw-animate-css" | "tailwind-merge" | "vite" | "typescript" | "vitest" | "jsdom" | "zod" | "ws", PyAgentGatewayConnectionMetadata>[], readonly [{
     readonly name: "boto3";
 }, {
     readonly name: "httpx";

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

@@ -10,7 +10,7 @@ import type { PyAgentMcpConnectionGeneratorSchema } from './schema';
 export interface PyAgentMcpConnectionMetadata {
     readonly framework: AgentFramework;
 }
-export declare const DEPENDENCIES: import("../../../utils/declared-dependencies.js").DependencyDeclaration<readonly import("../../../utils/declared-dependencies.js").DeclaredTsDependency<"@ag-ui/client" | "@ag-ui/core" | "@ag-ui/encoder" | "@biomejs/biome" | "@a2a-js/sdk" | "@aws/aws-distro-opentelemetry-node-autoinstrumentation" | "@opentelemetry/propagator-jaeger" | "minimatch" | "@aws-sdk/client-dynamodb" | "@aws-sdk/client-api-gateway" | "@aws-sdk/client-iam" | "@aws-sdk/client-bedrock-agentcore" | "@aws-sdk/client-bedrock-runtime" | "@aws-sdk/client-s3" | "@aws-sdk/client-sts" | "@aws-sdk/credential-providers" | "@aws-sdk/credential-provider-cognito-identity" | "@aws-sdk/client-secrets-manager" | "@aws-sdk/rds-signer" | "@smithy/server-apigateway" | "@smithy/server-node" | "@aws-lambda-powertools/logger" | "@aws-lambda-powertools/metrics" | "@aws-lambda-powertools/parameters" | "@aws-lambda-powertools/tracer" | "@aws-lambda-powertools/parser" | "@aws-sdk/client-appconfigdata" | "@middy/core" | "@nxlv/python" | "@nx-extend/terraform" | "nx" | "@nx/devkit" | "@nx/js" | "@nx/react" | "@nx/vite" | "@nx/vitest" | "@nx/workspace" | "create-nx-workspace" | "@swc-node/register" | "@swc/core" | "@modelcontextprotocol/sdk" | "@modelcontextprotocol/inspector" | "@ag-ui/a2ui-toolkit" | "@ag-ui/aws-strands" | "agent-chat-cli" | "@copilotkit/react-core" | "rxjs" | "@strands-agents/sdk" | "@tanstack/react-router" | "@tanstack/router-plugin" | "@tanstack/router-generator" | "@tanstack/virtual-file-routes" | "@tanstack/router-utils" | "@cloudscape-design/board-components" | "@cloudscape-design/chat-components" | "@cloudscape-design/components" | "@cloudscape-design/global-styles" | "@tanstack/react-query" | "@tanstack/react-query-devtools" | "@trpc/tanstack-react-query" | "@trpc/client" | "@trpc/server" | "@types/node" | "@types/aws-lambda" | "@types/cors" | "@types/pg" | "@types/ws" | "@types/express" | "@smithy/config-resolver" | "@smithy/node-config-provider" | "@smithy/node-http-handler" | "@smithy/types" | "@vitest/coverage-v8" | "@vitest/ui" | "@astrojs/react" | "@astrojs/starlight" | "astro" | "cookie" | "aws4fetch" | "aws-cdk" | "aws-cdk-lib" | "aws-xray-sdk-core" | "constructs" | "cors" | "chalk" | "class-variance-authority" | "clsx" | "commander" | "electrodb" | "esbuild" | "event-source-polyfill" | "@types/event-source-polyfill" | "@prisma/adapter-mariadb" | "@prisma/adapter-pg" | "@prisma/client" | "ejs" | "@types/ejs" | "express" | "fast-glob" | "husky" | "fs-extra" | "@types/fs-extra" | "mariadb" | "mise" | "npm" | "npm-check-updates" | "oidc-client-ts" | "pg" | "prisma" | "react-oidc-context" | "react" | "react-dom" | "rolldown" | "rolldown-plugin-dts" | "shx" | "simple-git" | "source-map-support" | "starlight-blog" | "tailwindcss" | "@tailwindcss/vite" | "tsx" | "lucide-react" | "radix-ui" | "shadcn" | "tw-animate-css" | "tailwind-merge" | "vite" | "typescript" | "vitest" | "zod" | "ws", PyAgentMcpConnectionMetadata>[], readonly [{
+export declare const DEPENDENCIES: import("../../../utils/declared-dependencies.js").DependencyDeclaration<readonly import("../../../utils/declared-dependencies.js").DeclaredTsDependency<"@ag-ui/client" | "@ag-ui/core" | "@ag-ui/encoder" | "@biomejs/biome" | "@a2a-js/sdk" | "@aws/aws-distro-opentelemetry-node-autoinstrumentation" | "@opentelemetry/propagator-jaeger" | "minimatch" | "deepmerge-ts" | "mysql2" | "@aws-sdk/client-dynamodb" | "@aws-sdk/client-api-gateway" | "@aws-sdk/client-iam" | "@aws-sdk/client-bedrock-agentcore" | "@aws-sdk/client-bedrock-runtime" | "@aws-sdk/client-s3" | "@aws-sdk/client-sts" | "@aws-sdk/client-cognito-identity-provider" | "@aws-sdk/credential-providers" | "@aws-sdk/credential-provider-cognito-identity" | "@aws-sdk/client-secrets-manager" | "@aws-sdk/rds-signer" | "@smithy/server-apigateway" | "@smithy/server-node" | "@aws-lambda-powertools/logger" | "@aws-lambda-powertools/metrics" | "@aws-lambda-powertools/parameters" | "@aws-lambda-powertools/tracer" | "@aws-lambda-powertools/parser" | "@aws-sdk/client-appconfigdata" | "@middy/core" | "@nxlv/python" | "@nx-extend/terraform" | "nx" | "@nx/devkit" | "@nx/js" | "@nx/react" | "@nx/vite" | "@nx/vitest" | "@nx/workspace" | "create-nx-workspace" | "@swc-node/register" | "@swc/core" | "@modelcontextprotocol/sdk" | "@modelcontextprotocol/inspector" | "@ag-ui/a2ui-toolkit" | "@ag-ui/aws-strands" | "agent-chat-cli" | "@copilotkit/react-core" | "rxjs" | "@strands-agents/sdk" | "@tanstack/react-router" | "@tanstack/router-plugin" | "@tanstack/router-generator" | "@tanstack/virtual-file-routes" | "@tanstack/router-utils" | "@cloudscape-design/board-components" | "@cloudscape-design/chat-components" | "@cloudscape-design/components" | "@cloudscape-design/global-styles" | "@tanstack/react-query" | "@tanstack/react-query-devtools" | "@trpc/tanstack-react-query" | "@trpc/client" | "@trpc/server" | "@types/node" | "@types/aws-lambda" | "@types/cors" | "@types/pg" | "@types/ws" | "@types/express" | "@smithy/config-resolver" | "@smithy/node-config-provider" | "@smithy/node-http-handler" | "@smithy/types" | "@vitest/coverage-v8" | "@vitest/ui" | "@astrojs/react" | "@astrojs/starlight" | "astro" | "cookie" | "aws4fetch" | "aws-cdk" | "aws-cdk-lib" | "aws-xray-sdk-core" | "constructs" | "cors" | "chalk" | "class-variance-authority" | "clsx" | "commander" | "electrodb" | "esbuild" | "event-source-polyfill" | "@types/event-source-polyfill" | "@prisma/adapter-mariadb" | "@prisma/adapter-pg" | "@prisma/client" | "ejs" | "@types/ejs" | "express" | "fast-glob" | "husky" | "fs-extra" | "@types/fs-extra" | "mariadb" | "mise" | "npm" | "npm-check-updates" | "oidc-client-ts" | "pg" | "prisma" | "react-oidc-context" | "react" | "react-dom" | "rolldown" | "rolldown-plugin-dts" | "shx" | "simple-git" | "source-map-support" | "starlight-blog" | "tailwindcss" | "@tailwindcss/vite" | "tsx" | "lucide-react" | "radix-ui" | "shadcn" | "tw-animate-css" | "tailwind-merge" | "vite" | "typescript" | "vitest" | "jsdom" | "zod" | "ws", PyAgentMcpConnectionMetadata>[], readonly [{
     readonly name: "boto3";
 }, {
     readonly name: "httpx";

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

@@ -311,6 +311,9 @@ class SigV4HttpAgent extends HttpAgent {
   }
 }
 
+// The id this agent is registered under, used as \`agentId\` by CopilotKit components.
+export const TEST_AGENT_ID = 'agent';
+
 export const useAguiTestAgent = (): Record<string, AbstractAgent> => {
   const runtimeConfig = useRuntimeConfig();
   const agentRuntimeValue = runtimeConfig.agentRuntimes.TestAgent;
@@ -327,7 +330,7 @@ export const useAguiTestAgent = (): Record<string, AbstractAgent> => {
   return useMemo((): Record<string, AbstractAgent> => {
     const agent = new SigV4HttpAgent({ url }, sigv4);
 
-    return { agent: agent };
+    return { [TEST_AGENT_ID]: agent };
   }, [url, sigv4]);
 };
 "

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

@@ -151,6 +151,35 @@ export declare const DEPENDENCIES: import("../../../utils/declared-dependencies.
     versionOnly: true;
 }, "when"> & {
     when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "jsdom";
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "@nx/vitest";
+    readonly dev: true;
+    readonly root: true;
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "@types/node";
+    readonly dev: true;
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
+}, Omit<{
+    readonly name: "@types/node";
+    readonly dev: true;
+    readonly root: true;
+} & {
+    versionOnly: true;
+}, "when"> & {
+    when: (m: PyAgentReactConnectionMetadata) => boolean;
 }], readonly import("../../../utils/declared-dependencies.js").DeclaredPyDependency<"ruff" | "a2a-sdk" | "ag-ui-langgraph" | "ag-ui-protocol" | "ag-ui-strands" | "aiosqlite" | "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" | "langgraph-checkpoint-aws" | "langgraph-checkpoint-sqlite" | "mcp" | "pip-check-updates" | "pip-licenses" | "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 & {

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

@@ -674,12 +674,11 @@ logs.on('exit', cleanup);
 `;
 
 exports[`py#dynamodb generator > should generate the dynamodb project > packages/my_table/proj_my_table/__init__.py 1`] = `
-"from .entities import BaseModel, ExampleModel
+""""Automatically generated by Nx."""
 
-__all__ = [
-    "BaseModel",
-    "ExampleModel",
-]
+from .entities import BaseModel, ExampleModel
+
+__all__ = ["BaseModel", "ExampleModel"]
 "
 `;
 
@@ -1395,12 +1394,11 @@ logs.on('exit', cleanup);
 `;
 
 exports[`py#dynamodb generator > should generate with infra=none then upgrade to infra=dynamodb > packages/my_table/proj_my_table/__init__.py 1`] = `
-"from .entities import BaseModel, ExampleModel
+""""Automatically generated by Nx."""
 
-__all__ = [
-    "BaseModel",
-    "ExampleModel",
-]
+from .entities import BaseModel, ExampleModel
+
+__all__ = ["BaseModel", "ExampleModel"]
 "
 `;
 

package/src/py/dynamodb/generator.js

@@ -2,10 +2,12 @@
  * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
  * SPDX-License-Identifier: Apache-2.0
  */ import { relative } from "node:path";
-import { generateFiles, joinPathFragments, readProjectConfiguration, updateProjectConfiguration } from "@nx/devkit";
+import { generateFiles, joinPathFragments, OverwriteStrategy, readProjectConfiguration, updateProjectConfiguration } from "@nx/devkit";
 import { addPyDependencies } from "../../utils/add-dependencies.js";
+import { addPythonReExport } from "../../utils/agent-connection/agent-connection.js";
 import { resolveContainers } from "../../utils/containers.js";
 import { declareDependencies, ownedElsewhere } from "../../utils/declared-dependencies.js";
+import { DYNAMODB_LOCAL_IMAGE, writeDynamoDBConfig } from "../../utils/dynamodb-config.js";
 import { addDynamoDBInfra } from "../../utils/dynamodb-constructs/dynamodb-constructs.js";
 import { formatFilesInSubtree } from "../../utils/format.js";
 import { resolveIac } from "../../utils/iac.js";
@@ -60,13 +62,38 @@ export const pyDynamoDBGenerator = async (tree, options)=>{
     const containerName = `${getNpmScope(tree)}-dynamodb`;
     const templateOptions = {
         name: normalizedModuleName,
-        runtimeConfigKey: nameClassName,
-        localDynamoDBPort,
-        localTableName,
-        containerName,
-        containerEngine
+        runtimeConfigKey: nameClassName
     };
-    generateFiles(tree, joinPathFragments(import.meta.dirname, 'files'), dir, templateOptions);
+    // The entities and the client are the user's to author — the guide walks
+    // through adding entity modules under `entities/` — so a re-run leaves them
+    // alone.
+    generateFiles(tree, joinPathFragments(import.meta.dirname, 'files'), dir, templateOptions, {
+        overwriteStrategy: OverwriteStrategy.KeepExisting
+    });
+    // The barrels re-export the generated base and example models. Adding to
+    // whatever is already there registers them without discarding the user's own
+    // entity exports on a re-run — which matters beyond the barrel itself, since
+    // PynamoDB only registers a model's discriminator once it is imported.
+    const entitiesInitPath = joinPathFragments(dir, normalizedModuleName, 'entities', '__init__.py');
+    await addPythonReExport(tree, entitiesInitPath, '.base', 'BaseModel');
+    await addPythonReExport(tree, entitiesInitPath, '.example', 'ExampleModel');
+    const packageInitPath = joinPathFragments(dir, normalizedModuleName, '__init__.py');
+    for (const importName of [
+        'BaseModel',
+        'ExampleModel'
+    ]){
+        await addPythonReExport(tree, packageInitPath, '.entities', importName);
+    }
+    writeDynamoDBConfig(tree, dir, {
+        runtimeConfigKey: nameClassName,
+        localDev: {
+            port: localDynamoDBPort,
+            tableName: localTableName,
+            image: DYNAMODB_LOCAL_IMAGE,
+            containerName,
+            containerEngine
+        }
+    });
     await sharedDynamoDBScriptsGenerator(tree, DEPENDENCIES);
     const scriptsDir = relative(dir, joinPathFragments(PACKAGES_DIR, SHARED_SCRIPTS_DIR, 'src', 'dynamodb'));
     projectConfig.targets['pull-image'] = {

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../../../../packages/nx-plugin/src/py/dynamodb/generator.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { relative } from 'node:path';\nimport {\n  type GeneratorCallback,\n  generateFiles,\n  joinPathFragments,\n  readProjectConfiguration,\n  type Tree,\n  updateProjectConfiguration,\n} from '@nx/devkit';\nimport { addPyDependencies } from '../../utils/add-dependencies.js';\nimport { resolveContainers } from '../../utils/containers.js';\nimport {\n  declareDependencies,\n  ownedElsewhere,\n} from '../../utils/declared-dependencies.js';\nimport { addDynamoDBInfra } from '../../utils/dynamodb-constructs/dynamodb-constructs.js';\nimport { formatFilesInSubtree } from '../../utils/format.js';\nimport { resolveIac } from '../../utils/iac.js';\nimport { installDependencies } from '../../utils/install.js';\nimport { addGeneratorMetricsIfApplicable } from '../../utils/metrics.js';\nimport { kebabCase, toClassName } from '../../utils/names.js';\nimport { getNpmScope } from '../../utils/npm-scope.js';\nimport {\n  addGeneratorMetadata,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  projectExists,\n} from '../../utils/nx.js';\nimport { assignSharedPort } from '../../utils/port.js';\nimport {\n  SHARED_CONSTRUCTS_DEPENDENCIES,\n  sharedConstructsGenerator,\n} from '../../utils/shared-constructs.js';\nimport {\n  DYNAMODB_GENERATOR_IDS,\n  PACKAGES_DIR,\n  SHARED_SCRIPTS_DIR,\n} from '../../utils/shared-constructs-constants.js';\nimport {\n  SHARED_DYNAMODB_SCRIPTS_DEPENDENCIES,\n  sharedDynamoDBScriptsGenerator,\n} from '../../utils/shared-dynamodb-scripts.js';\nimport pyProjectGenerator, {\n  getPyProjectDetails,\n} from '../project/generator.js';\nimport type { PyDynamoDBGeneratorSchema } from './schema';\n\nexport const DEPENDENCIES = declareDependencies()({\n  ts: [\n    ...ownedElsewhere(SHARED_DYNAMODB_SCRIPTS_DEPENDENCIES),\n    ...ownedElsewhere(SHARED_CONSTRUCTS_DEPENDENCIES),\n  ],\n  py: [\n    { name: 'pynamodb' },\n    { name: 'boto3' },\n    { name: 'aws-lambda-powertools' },\n  ],\n});\n\nexport const PY_DYNAMODB_GENERATOR_INFO: NxGeneratorInfo = getGeneratorInfo(\n  import.meta.filename,\n);\n\nexport const pyDynamoDBGenerator = async (\n  tree: Tree,\n  options: PyDynamoDBGeneratorSchema,\n): Promise<GeneratorCallback> => {\n  const nameClassName = toClassName(options.name);\n  const localTableName = `${getNpmScope(tree)}-${kebabCase(options.tableName ?? options.name)}`;\n  const containerEngine = await resolveContainers(tree, 'inherit');\n  const { fullyQualifiedName, dir, normalizedModuleName } = getPyProjectDetails(\n    tree,\n    {\n      name: options.name,\n      directory: options.directory,\n      subDirectory: options.subDirectory,\n    },\n  );\n\n  if (!projectExists(tree, fullyQualifiedName)) {\n    await pyProjectGenerator(tree, {\n      name: options.name,\n      directory: options.directory,\n      subDirectory: options.subDirectory,\n      type: 'library',\n      preferInstallDependencies: false,\n    });\n  }\n\n  const projectConfig = readProjectConfiguration(tree, fullyQualifiedName);\n\n  const localDynamoDBPort = assignSharedPort(\n    tree,\n    projectConfig,\n    DYNAMODB_GENERATOR_IDS,\n    8000,\n  );\n\n  const containerName = `${getNpmScope(tree)}-dynamodb`;\n\n  const templateOptions = {\n    name: normalizedModuleName,\n    runtimeConfigKey: nameClassName,\n    localDynamoDBPort,\n    localTableName,\n    containerName,\n    containerEngine,\n  };\n\n  generateFiles(\n    tree,\n    joinPathFragments(import.meta.dirname, 'files'),\n    dir,\n    templateOptions,\n  );\n\n  await sharedDynamoDBScriptsGenerator(tree, DEPENDENCIES);\n\n  const scriptsDir = relative(\n    dir,\n    joinPathFragments(PACKAGES_DIR, SHARED_SCRIPTS_DIR, 'src', 'dynamodb'),\n  );\n\n  projectConfig.targets['pull-image'] = {\n    executor: 'nx:run-commands',\n    options: {\n      command: `tsx ${scriptsDir}/pull-image.ts`,\n      cwd: '{projectRoot}',\n    },\n  };\n  projectConfig.targets['dev'] = {\n    executor: 'nx:run-commands',\n    continuous: true,\n    dependsOn: ['pull-image'],\n    options: {\n      commands: [\n        `tsx ${scriptsDir}/start-container.ts`,\n        `tsx ${scriptsDir}/create-local-table.ts`,\n      ],\n      parallel: true,\n      cwd: '{projectRoot}',\n    },\n  };\n\n  updateProjectConfiguration(tree, fullyQualifiedName, projectConfig);\n  // Recorded so the version sync can tell a CDK project from a Terraform one;\n  // undefined when no infrastructure was generated.\n  const iac =\n    options.infra !== 'none' ? await resolveIac(tree, options.iac) : undefined;\n\n  addGeneratorMetadata(\n    tree,\n    fullyQualifiedName,\n    PY_DYNAMODB_GENERATOR_INFO,\n    iac ? { iac } : {},\n  );\n\n  if (options.infra !== 'none') {\n    await sharedConstructsGenerator(tree, { iac }, DEPENDENCIES);\n    await addDynamoDBInfra(tree, {\n      iac,\n      projectName: fullyQualifiedName,\n      nameClassName,\n      nameKebabCase: kebabCase(options.name),\n      tableName: localTableName,\n      projectRoot: dir,\n    });\n  }\n\n  addPyDependencies(tree, DEPENDENCIES, { projectRoot: dir });\n\n  await addGeneratorMetricsIfApplicable(tree, [PY_DYNAMODB_GENERATOR_INFO]);\n\n  await formatFilesInSubtree(tree);\n  return () =>\n    installDependencies(tree, options.preferInstallDependencies, {\n      languages: ['typescript', 'python'],\n    });\n};\n\nexport default pyDynamoDBGenerator;\n"],"names":["relative","generateFiles","joinPathFragments","readProjectConfiguration","updateProjectConfiguration","addPyDependencies","resolveContainers","declareDependencies","ownedElsewhere","addDynamoDBInfra","formatFilesInSubtree","resolveIac","installDependencies","addGeneratorMetricsIfApplicable","kebabCase","toClassName","getNpmScope","addGeneratorMetadata","getGeneratorInfo","projectExists","assignSharedPort","SHARED_CONSTRUCTS_DEPENDENCIES","sharedConstructsGenerator","DYNAMODB_GENERATOR_IDS","PACKAGES_DIR","SHARED_SCRIPTS_DIR","SHARED_DYNAMODB_SCRIPTS_DEPENDENCIES","sharedDynamoDBScriptsGenerator","pyProjectGenerator","getPyProjectDetails","DEPENDENCIES","ts","py","name","PY_DYNAMODB_GENERATOR_INFO","filename","pyDynamoDBGenerator","tree","options","nameClassName","localTableName","tableName","containerEngine","fullyQualifiedName","dir","normalizedModuleName","directory","subDirectory","type","preferInstallDependencies","projectConfig","localDynamoDBPort","containerName","templateOptions","runtimeConfigKey","dirname","scriptsDir","targets","executor","command","cwd","continuous","dependsOn","commands","parallel","iac","infra","undefined","projectName","nameKebabCase","projectRoot","languages"],"mappings":"AAAA;;;CAGC,GAED,SAASA,QAAQ,QAAQ,YAAY;AACrC,SAEEC,aAAa,EACbC,iBAAiB,EACjBC,wBAAwB,EAExBC,0BAA0B,QACrB,aAAa;AACpB,SAASC,iBAAiB,QAAQ,kCAAkC;AACpE,SAASC,iBAAiB,QAAQ,4BAA4B;AAC9D,SACEC,mBAAmB,EACnBC,cAAc,QACT,uCAAuC;AAC9C,SAASC,gBAAgB,QAAQ,yDAAyD;AAC1F,SAASC,oBAAoB,QAAQ,wBAAwB;AAC7D,SAASC,UAAU,QAAQ,qBAAqB;AAChD,SAASC,mBAAmB,QAAQ,yBAAyB;AAC7D,SAASC,+BAA+B,QAAQ,yBAAyB;AACzE,SAASC,SAAS,EAAEC,WAAW,QAAQ,uBAAuB;AAC9D,SAASC,WAAW,QAAQ,2BAA2B;AACvD,SACEC,oBAAoB,EACpBC,gBAAgB,EAEhBC,aAAa,QACR,oBAAoB;AAC3B,SAASC,gBAAgB,QAAQ,sBAAsB;AACvD,SACEC,8BAA8B,EAC9BC,yBAAyB,QACpB,mCAAmC;AAC1C,SACEC,sBAAsB,EACtBC,YAAY,EACZC,kBAAkB,QACb,6CAA6C;AACpD,SACEC,oCAAoC,EACpCC,8BAA8B,QACzB,yCAAyC;AAChD,OAAOC,sBACLC,mBAAmB,QACd,0BAA0B;AAGjC,OAAO,MAAMC,eAAevB,sBAAsB;IAChDwB,IAAI;WACCvB,eAAekB;WACflB,eAAea;KACnB;IACDW,IAAI;QACF;YAAEC,MAAM;QAAW;QACnB;YAAEA,MAAM;QAAQ;QAChB;YAAEA,MAAM;QAAwB;KACjC;AACH,GAAG;AAEH,OAAO,MAAMC,6BAA8ChB,iBACzD,YAAYiB,QAAQ,EACpB;AAEF,OAAO,MAAMC,sBAAsB,OACjCC,MACAC;IAEA,MAAMC,gBAAgBxB,YAAYuB,QAAQL,IAAI;IAC9C,MAAMO,iBAAiB,GAAGxB,YAAYqB,MAAM,CAAC,EAAEvB,UAAUwB,QAAQG,SAAS,IAAIH,QAAQL,IAAI,GAAG;IAC7F,MAAMS,kBAAkB,MAAMpC,kBAAkB+B,MAAM;IACtD,MAAM,EAAEM,kBAAkB,EAAEC,GAAG,EAAEC,oBAAoB,EAAE,GAAGhB,oBACxDQ,MACA;QACEJ,MAAMK,QAAQL,IAAI;QAClBa,WAAWR,QAAQQ,SAAS;QAC5BC,cAAcT,QAAQS,YAAY;IACpC;IAGF,IAAI,CAAC5B,cAAckB,MAAMM,qBAAqB;QAC5C,MAAMf,mBAAmBS,MAAM;YAC7BJ,MAAMK,QAAQL,IAAI;YAClBa,WAAWR,QAAQQ,SAAS;YAC5BC,cAAcT,QAAQS,YAAY;YAClCC,MAAM;YACNC,2BAA2B;QAC7B;IACF;IAEA,MAAMC,gBAAgB/C,yBAAyBkC,MAAMM;IAErD,MAAMQ,oBAAoB/B,iBACxBiB,MACAa,eACA3B,wBACA;IAGF,MAAM6B,gBAAgB,GAAGpC,YAAYqB,MAAM,SAAS,CAAC;IAErD,MAAMgB,kBAAkB;QACtBpB,MAAMY;QACNS,kBAAkBf;QAClBY;QACAX;QACAY;QACAV;IACF;IAEAzC,cACEoC,MACAnC,kBAAkB,YAAYqD,OAAO,EAAE,UACvCX,KACAS;IAGF,MAAM1B,+BAA+BU,MAAMP;IAE3C,MAAM0B,aAAaxD,SACjB4C,KACA1C,kBAAkBsB,cAAcC,oBAAoB,OAAO;IAG7DyB,cAAcO,OAAO,CAAC,aAAa,GAAG;QACpCC,UAAU;QACVpB,SAAS;YACPqB,SAAS,CAAC,IAAI,EAAEH,WAAW,cAAc,CAAC;YAC1CI,KAAK;QACP;IACF;IACAV,cAAcO,OAAO,CAAC,MAAM,GAAG;QAC7BC,UAAU;QACVG,YAAY;QACZC,WAAW;YAAC;SAAa;QACzBxB,SAAS;YACPyB,UAAU;gBACR,CAAC,IAAI,EAAEP,WAAW,mBAAmB,CAAC;gBACtC,CAAC,IAAI,EAAEA,WAAW,sBAAsB,CAAC;aAC1C;YACDQ,UAAU;YACVJ,KAAK;QACP;IACF;IAEAxD,2BAA2BiC,MAAMM,oBAAoBO;IACrD,4EAA4E;IAC5E,kDAAkD;IAClD,MAAMe,MACJ3B,QAAQ4B,KAAK,KAAK,SAAS,MAAMvD,WAAW0B,MAAMC,QAAQ2B,GAAG,IAAIE;IAEnElD,qBACEoB,MACAM,oBACAT,4BACA+B,MAAM;QAAEA;IAAI,IAAI,CAAC;IAGnB,IAAI3B,QAAQ4B,KAAK,KAAK,QAAQ;QAC5B,MAAM5C,0BAA0Be,MAAM;YAAE4B;QAAI,GAAGnC;QAC/C,MAAMrB,iBAAiB4B,MAAM;YAC3B4B;YACAG,aAAazB;YACbJ;YACA8B,eAAevD,UAAUwB,QAAQL,IAAI;YACrCQ,WAAWD;YACX8B,aAAa1B;QACf;IACF;IAEAvC,kBAAkBgC,MAAMP,cAAc;QAAEwC,aAAa1B;IAAI;IAEzD,MAAM/B,gCAAgCwB,MAAM;QAACH;KAA2B;IAExE,MAAMxB,qBAAqB2B;IAC3B,OAAO,IACLzB,oBAAoByB,MAAMC,QAAQW,yBAAyB,EAAE;YAC3DsB,WAAW;gBAAC;gBAAc;aAAS;QACrC;AACJ,EAAE;AAEF,eAAenC,oBAAoB"}
+{"version":3,"sources":["../../../../../../packages/nx-plugin/src/py/dynamodb/generator.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { relative } from 'node:path';\nimport {\n  type GeneratorCallback,\n  generateFiles,\n  joinPathFragments,\n  OverwriteStrategy,\n  readProjectConfiguration,\n  type Tree,\n  updateProjectConfiguration,\n} from '@nx/devkit';\nimport { addPyDependencies } from '../../utils/add-dependencies.js';\nimport { addPythonReExport } from '../../utils/agent-connection/agent-connection.js';\nimport { resolveContainers } from '../../utils/containers.js';\nimport {\n  declareDependencies,\n  ownedElsewhere,\n} from '../../utils/declared-dependencies.js';\nimport {\n  DYNAMODB_LOCAL_IMAGE,\n  writeDynamoDBConfig,\n} from '../../utils/dynamodb-config.js';\nimport { addDynamoDBInfra } from '../../utils/dynamodb-constructs/dynamodb-constructs.js';\nimport { formatFilesInSubtree } from '../../utils/format.js';\nimport { resolveIac } from '../../utils/iac.js';\nimport { installDependencies } from '../../utils/install.js';\nimport { addGeneratorMetricsIfApplicable } from '../../utils/metrics.js';\nimport { kebabCase, toClassName } from '../../utils/names.js';\nimport { getNpmScope } from '../../utils/npm-scope.js';\nimport {\n  addGeneratorMetadata,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  projectExists,\n} from '../../utils/nx.js';\nimport { assignSharedPort } from '../../utils/port.js';\nimport {\n  SHARED_CONSTRUCTS_DEPENDENCIES,\n  sharedConstructsGenerator,\n} from '../../utils/shared-constructs.js';\nimport {\n  DYNAMODB_GENERATOR_IDS,\n  PACKAGES_DIR,\n  SHARED_SCRIPTS_DIR,\n} from '../../utils/shared-constructs-constants.js';\nimport {\n  SHARED_DYNAMODB_SCRIPTS_DEPENDENCIES,\n  sharedDynamoDBScriptsGenerator,\n} from '../../utils/shared-dynamodb-scripts.js';\nimport pyProjectGenerator, {\n  getPyProjectDetails,\n} from '../project/generator.js';\nimport type { PyDynamoDBGeneratorSchema } from './schema';\n\nexport const DEPENDENCIES = declareDependencies()({\n  ts: [\n    ...ownedElsewhere(SHARED_DYNAMODB_SCRIPTS_DEPENDENCIES),\n    ...ownedElsewhere(SHARED_CONSTRUCTS_DEPENDENCIES),\n  ],\n  py: [\n    { name: 'pynamodb' },\n    { name: 'boto3' },\n    { name: 'aws-lambda-powertools' },\n  ],\n});\n\nexport const PY_DYNAMODB_GENERATOR_INFO: NxGeneratorInfo = getGeneratorInfo(\n  import.meta.filename,\n);\n\nexport const pyDynamoDBGenerator = async (\n  tree: Tree,\n  options: PyDynamoDBGeneratorSchema,\n): Promise<GeneratorCallback> => {\n  const nameClassName = toClassName(options.name);\n  const localTableName = `${getNpmScope(tree)}-${kebabCase(options.tableName ?? options.name)}`;\n  const containerEngine = await resolveContainers(tree, 'inherit');\n  const { fullyQualifiedName, dir, normalizedModuleName } = getPyProjectDetails(\n    tree,\n    {\n      name: options.name,\n      directory: options.directory,\n      subDirectory: options.subDirectory,\n    },\n  );\n\n  if (!projectExists(tree, fullyQualifiedName)) {\n    await pyProjectGenerator(tree, {\n      name: options.name,\n      directory: options.directory,\n      subDirectory: options.subDirectory,\n      type: 'library',\n      preferInstallDependencies: false,\n    });\n  }\n\n  const projectConfig = readProjectConfiguration(tree, fullyQualifiedName);\n\n  const localDynamoDBPort = assignSharedPort(\n    tree,\n    projectConfig,\n    DYNAMODB_GENERATOR_IDS,\n    8000,\n  );\n\n  const containerName = `${getNpmScope(tree)}-dynamodb`;\n\n  const templateOptions = {\n    name: normalizedModuleName,\n    runtimeConfigKey: nameClassName,\n  };\n\n  // The entities and the client are the user's to author — the guide walks\n  // through adding entity modules under `entities/` — so a re-run leaves them\n  // alone.\n  generateFiles(\n    tree,\n    joinPathFragments(import.meta.dirname, 'files'),\n    dir,\n    templateOptions,\n    { overwriteStrategy: OverwriteStrategy.KeepExisting },\n  );\n\n  // The barrels re-export the generated base and example models. Adding to\n  // whatever is already there registers them without discarding the user's own\n  // entity exports on a re-run — which matters beyond the barrel itself, since\n  // PynamoDB only registers a model's discriminator once it is imported.\n  const entitiesInitPath = joinPathFragments(\n    dir,\n    normalizedModuleName,\n    'entities',\n    '__init__.py',\n  );\n  await addPythonReExport(tree, entitiesInitPath, '.base', 'BaseModel');\n  await addPythonReExport(tree, entitiesInitPath, '.example', 'ExampleModel');\n\n  const packageInitPath = joinPathFragments(\n    dir,\n    normalizedModuleName,\n    '__init__.py',\n  );\n  for (const importName of ['BaseModel', 'ExampleModel']) {\n    await addPythonReExport(tree, packageInitPath, '.entities', importName);\n  }\n\n  writeDynamoDBConfig(tree, dir, {\n    runtimeConfigKey: nameClassName,\n    localDev: {\n      port: localDynamoDBPort,\n      tableName: localTableName,\n      image: DYNAMODB_LOCAL_IMAGE,\n      containerName,\n      containerEngine,\n    },\n  });\n\n  await sharedDynamoDBScriptsGenerator(tree, DEPENDENCIES);\n\n  const scriptsDir = relative(\n    dir,\n    joinPathFragments(PACKAGES_DIR, SHARED_SCRIPTS_DIR, 'src', 'dynamodb'),\n  );\n\n  projectConfig.targets['pull-image'] = {\n    executor: 'nx:run-commands',\n    options: {\n      command: `tsx ${scriptsDir}/pull-image.ts`,\n      cwd: '{projectRoot}',\n    },\n  };\n  projectConfig.targets['dev'] = {\n    executor: 'nx:run-commands',\n    continuous: true,\n    dependsOn: ['pull-image'],\n    options: {\n      commands: [\n        `tsx ${scriptsDir}/start-container.ts`,\n        `tsx ${scriptsDir}/create-local-table.ts`,\n      ],\n      parallel: true,\n      cwd: '{projectRoot}',\n    },\n  };\n\n  updateProjectConfiguration(tree, fullyQualifiedName, projectConfig);\n  // Recorded so the version sync can tell a CDK project from a Terraform one;\n  // undefined when no infrastructure was generated.\n  const iac =\n    options.infra !== 'none' ? await resolveIac(tree, options.iac) : undefined;\n\n  addGeneratorMetadata(\n    tree,\n    fullyQualifiedName,\n    PY_DYNAMODB_GENERATOR_INFO,\n    iac ? { iac } : {},\n  );\n\n  if (options.infra !== 'none') {\n    await sharedConstructsGenerator(tree, { iac }, DEPENDENCIES);\n    await addDynamoDBInfra(tree, {\n      iac,\n      projectName: fullyQualifiedName,\n      nameClassName,\n      nameKebabCase: kebabCase(options.name),\n      tableName: localTableName,\n      projectRoot: dir,\n    });\n  }\n\n  addPyDependencies(tree, DEPENDENCIES, { projectRoot: dir });\n\n  await addGeneratorMetricsIfApplicable(tree, [PY_DYNAMODB_GENERATOR_INFO]);\n\n  await formatFilesInSubtree(tree);\n  return () =>\n    installDependencies(tree, options.preferInstallDependencies, {\n      languages: ['typescript', 'python'],\n    });\n};\n\nexport default pyDynamoDBGenerator;\n"],"names":["relative","generateFiles","joinPathFragments","OverwriteStrategy","readProjectConfiguration","updateProjectConfiguration","addPyDependencies","addPythonReExport","resolveContainers","declareDependencies","ownedElsewhere","DYNAMODB_LOCAL_IMAGE","writeDynamoDBConfig","addDynamoDBInfra","formatFilesInSubtree","resolveIac","installDependencies","addGeneratorMetricsIfApplicable","kebabCase","toClassName","getNpmScope","addGeneratorMetadata","getGeneratorInfo","projectExists","assignSharedPort","SHARED_CONSTRUCTS_DEPENDENCIES","sharedConstructsGenerator","DYNAMODB_GENERATOR_IDS","PACKAGES_DIR","SHARED_SCRIPTS_DIR","SHARED_DYNAMODB_SCRIPTS_DEPENDENCIES","sharedDynamoDBScriptsGenerator","pyProjectGenerator","getPyProjectDetails","DEPENDENCIES","ts","py","name","PY_DYNAMODB_GENERATOR_INFO","filename","pyDynamoDBGenerator","tree","options","nameClassName","localTableName","tableName","containerEngine","fullyQualifiedName","dir","normalizedModuleName","directory","subDirectory","type","preferInstallDependencies","projectConfig","localDynamoDBPort","containerName","templateOptions","runtimeConfigKey","dirname","overwriteStrategy","KeepExisting","entitiesInitPath","packageInitPath","importName","localDev","port","image","scriptsDir","targets","executor","command","cwd","continuous","dependsOn","commands","parallel","iac","infra","undefined","projectName","nameKebabCase","projectRoot","languages"],"mappings":"AAAA;;;CAGC,GAED,SAASA,QAAQ,QAAQ,YAAY;AACrC,SAEEC,aAAa,EACbC,iBAAiB,EACjBC,iBAAiB,EACjBC,wBAAwB,EAExBC,0BAA0B,QACrB,aAAa;AACpB,SAASC,iBAAiB,QAAQ,kCAAkC;AACpE,SAASC,iBAAiB,QAAQ,mDAAmD;AACrF,SAASC,iBAAiB,QAAQ,4BAA4B;AAC9D,SACEC,mBAAmB,EACnBC,cAAc,QACT,uCAAuC;AAC9C,SACEC,oBAAoB,EACpBC,mBAAmB,QACd,iCAAiC;AACxC,SAASC,gBAAgB,QAAQ,yDAAyD;AAC1F,SAASC,oBAAoB,QAAQ,wBAAwB;AAC7D,SAASC,UAAU,QAAQ,qBAAqB;AAChD,SAASC,mBAAmB,QAAQ,yBAAyB;AAC7D,SAASC,+BAA+B,QAAQ,yBAAyB;AACzE,SAASC,SAAS,EAAEC,WAAW,QAAQ,uBAAuB;AAC9D,SAASC,WAAW,QAAQ,2BAA2B;AACvD,SACEC,oBAAoB,EACpBC,gBAAgB,EAEhBC,aAAa,QACR,oBAAoB;AAC3B,SAASC,gBAAgB,QAAQ,sBAAsB;AACvD,SACEC,8BAA8B,EAC9BC,yBAAyB,QACpB,mCAAmC;AAC1C,SACEC,sBAAsB,EACtBC,YAAY,EACZC,kBAAkB,QACb,6CAA6C;AACpD,SACEC,oCAAoC,EACpCC,8BAA8B,QACzB,yCAAyC;AAChD,OAAOC,sBACLC,mBAAmB,QACd,0BAA0B;AAGjC,OAAO,MAAMC,eAAezB,sBAAsB;IAChD0B,IAAI;WACCzB,eAAeoB;WACfpB,eAAee;KACnB;IACDW,IAAI;QACF;YAAEC,MAAM;QAAW;QACnB;YAAEA,MAAM;QAAQ;QAChB;YAAEA,MAAM;QAAwB;KACjC;AACH,GAAG;AAEH,OAAO,MAAMC,6BAA8ChB,iBACzD,YAAYiB,QAAQ,EACpB;AAEF,OAAO,MAAMC,sBAAsB,OACjCC,MACAC;IAEA,MAAMC,gBAAgBxB,YAAYuB,QAAQL,IAAI;IAC9C,MAAMO,iBAAiB,GAAGxB,YAAYqB,MAAM,CAAC,EAAEvB,UAAUwB,QAAQG,SAAS,IAAIH,QAAQL,IAAI,GAAG;IAC7F,MAAMS,kBAAkB,MAAMtC,kBAAkBiC,MAAM;IACtD,MAAM,EAAEM,kBAAkB,EAAEC,GAAG,EAAEC,oBAAoB,EAAE,GAAGhB,oBACxDQ,MACA;QACEJ,MAAMK,QAAQL,IAAI;QAClBa,WAAWR,QAAQQ,SAAS;QAC5BC,cAAcT,QAAQS,YAAY;IACpC;IAGF,IAAI,CAAC5B,cAAckB,MAAMM,qBAAqB;QAC5C,MAAMf,mBAAmBS,MAAM;YAC7BJ,MAAMK,QAAQL,IAAI;YAClBa,WAAWR,QAAQQ,SAAS;YAC5BC,cAAcT,QAAQS,YAAY;YAClCC,MAAM;YACNC,2BAA2B;QAC7B;IACF;IAEA,MAAMC,gBAAgBlD,yBAAyBqC,MAAMM;IAErD,MAAMQ,oBAAoB/B,iBACxBiB,MACAa,eACA3B,wBACA;IAGF,MAAM6B,gBAAgB,GAAGpC,YAAYqB,MAAM,SAAS,CAAC;IAErD,MAAMgB,kBAAkB;QACtBpB,MAAMY;QACNS,kBAAkBf;IACpB;IAEA,yEAAyE;IACzE,4EAA4E;IAC5E,SAAS;IACT1C,cACEwC,MACAvC,kBAAkB,YAAYyD,OAAO,EAAE,UACvCX,KACAS,iBACA;QAAEG,mBAAmBzD,kBAAkB0D,YAAY;IAAC;IAGtD,yEAAyE;IACzE,6EAA6E;IAC7E,6EAA6E;IAC7E,uEAAuE;IACvE,MAAMC,mBAAmB5D,kBACvB8C,KACAC,sBACA,YACA;IAEF,MAAM1C,kBAAkBkC,MAAMqB,kBAAkB,SAAS;IACzD,MAAMvD,kBAAkBkC,MAAMqB,kBAAkB,YAAY;IAE5D,MAAMC,kBAAkB7D,kBACtB8C,KACAC,sBACA;IAEF,KAAK,MAAMe,cAAc;QAAC;QAAa;KAAe,CAAE;QACtD,MAAMzD,kBAAkBkC,MAAMsB,iBAAiB,aAAaC;IAC9D;IAEApD,oBAAoB6B,MAAMO,KAAK;QAC7BU,kBAAkBf;QAClBsB,UAAU;YACRC,MAAMX;YACNV,WAAWD;YACXuB,OAAOxD;YACP6C;YACAV;QACF;IACF;IAEA,MAAMf,+BAA+BU,MAAMP;IAE3C,MAAMkC,aAAapE,SACjBgD,KACA9C,kBAAkB0B,cAAcC,oBAAoB,OAAO;IAG7DyB,cAAce,OAAO,CAAC,aAAa,GAAG;QACpCC,UAAU;QACV5B,SAAS;YACP6B,SAAS,CAAC,IAAI,EAAEH,WAAW,cAAc,CAAC;YAC1CI,KAAK;QACP;IACF;IACAlB,cAAce,OAAO,CAAC,MAAM,GAAG;QAC7BC,UAAU;QACVG,YAAY;QACZC,WAAW;YAAC;SAAa;QACzBhC,SAAS;YACPiC,UAAU;gBACR,CAAC,IAAI,EAAEP,WAAW,mBAAmB,CAAC;gBACtC,CAAC,IAAI,EAAEA,WAAW,sBAAsB,CAAC;aAC1C;YACDQ,UAAU;YACVJ,KAAK;QACP;IACF;IAEAnE,2BAA2BoC,MAAMM,oBAAoBO;IACrD,4EAA4E;IAC5E,kDAAkD;IAClD,MAAMuB,MACJnC,QAAQoC,KAAK,KAAK,SAAS,MAAM/D,WAAW0B,MAAMC,QAAQmC,GAAG,IAAIE;IAEnE1D,qBACEoB,MACAM,oBACAT,4BACAuC,MAAM;QAAEA;IAAI,IAAI,CAAC;IAGnB,IAAInC,QAAQoC,KAAK,KAAK,QAAQ;QAC5B,MAAMpD,0BAA0Be,MAAM;YAAEoC;QAAI,GAAG3C;QAC/C,MAAMrB,iBAAiB4B,MAAM;YAC3BoC;YACAG,aAAajC;YACbJ;YACAsC,eAAe/D,UAAUwB,QAAQL,IAAI;YACrCQ,WAAWD;YACXsC,aAAalC;QACf;IACF;IAEA1C,kBAAkBmC,MAAMP,cAAc;QAAEgD,aAAalC;IAAI;IAEzD,MAAM/B,gCAAgCwB,MAAM;QAACH;KAA2B;IAExE,MAAMxB,qBAAqB2B;IAC3B,OAAO,IACLzB,oBAAoByB,MAAMC,QAAQW,yBAAyB,EAAE;YAC3D8B,WAAW;gBAAC;gBAAc;aAAS;QACrC;AACJ,EAAE;AAEF,eAAe3C,oBAAoB"}

package/src/py/lambda-function/__snapshots__/generator.spec.ts.snap

@@ -17,7 +17,8 @@ tracer: Tracer = Tracer()
 
 
 @tracer.capture_lambda_handler
-@metrics.log_metrics
+@logger.inject_lambda_context
+@metrics.log_metrics(capture_cold_start_metric=True)
 def lambda_handler(event: dict, context: LambdaContext):
     logger.info("Received event", extra={"event": event})
     metrics.add_metric(name="InvocationCount", unit=MetricUnit.Count, value=1)

package/src/py/lambda-function/files/handler/__lambdaFunctionSnakeCase__.py.template

@@ -16,7 +16,8 @@ metrics: Metrics = Metrics()
 tracer: Tracer = Tracer()
 
 @tracer.capture_lambda_handler
-@metrics.log_metrics
+@logger.inject_lambda_context
+@metrics.log_metrics(capture_cold_start_metric=True)
 <%_ if (event !== 'Any') { _%>
 @event_parser(model=<%= event %>)
 <%_ } _%>

package/src/py/project/schema.json

@@ -32,7 +32,9 @@
       "type": "string",
       "description": "Whether the project is an application or library",
       "default": "application",
-      "enum": ["application", "library"]
+      "enum": ["application", "library"],
+      "x-priority": "important",
+      "x-prompt": "What type of Python project would you like to create?"
     },
     "moduleName": {
       "type": "string",

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

@@ -170,9 +170,13 @@ def handler(event: dict[str, Any], _context: Any) -> dict[str, Any]:
 exports[`py#rdb generator > should generate MySQL dependencies and templates 4`] = `
 "FROM public.ecr.aws/lambda/python:3.14
 
-# Patch OS packages beyond the base image's pinned release so the image is
-# free of known HIGH/CRITICAL vulnerabilities
-RUN microdnf upgrade -y --releasever latest && microdnf clean all
+# Patch OS packages beyond the base image's pinned release, and drop pip, whose
+# vendored copies of msgpack and setuptools carry known HIGH vulnerabilities.
+# The handler's dependencies are already installed in the image, so nothing
+# installs packages at runtime.
+RUN microdnf upgrade -y --releasever latest && microdnf clean all \\
+    && rm -rf /var/lang/lib/python*/site-packages/pip \\
+    /var/lang/lib/python*/site-packages/pip-*.dist-info
 
 WORKDIR \${LAMBDA_TASK_ROOT}
 
@@ -188,9 +192,13 @@ CMD ["proj_db.migration_handler.handler"]
 exports[`py#rdb generator > should generate MySQL dependencies and templates 5`] = `
 "FROM public.ecr.aws/lambda/python:3.14
 
-# Patch OS packages beyond the base image's pinned release so the image is
-# free of known HIGH/CRITICAL vulnerabilities
-RUN microdnf upgrade -y --releasever latest && microdnf clean all
+# Patch OS packages beyond the base image's pinned release, and drop pip, whose
+# vendored copies of msgpack and setuptools carry known HIGH vulnerabilities.
+# The handler's dependencies are already installed in the image, so nothing
+# installs packages at runtime.
+RUN microdnf upgrade -y --releasever latest && microdnf clean all \\
+    && rm -rf /var/lang/lib/python*/site-packages/pip \\
+    /var/lang/lib/python*/site-packages/pip-*.dist-info
 
 WORKDIR \${LAMBDA_TASK_ROOT}
 
@@ -1382,9 +1390,13 @@ datefmt = %H:%M:%S
 exports[`py#rdb generator > should generate a PostgreSQL SQLModel project with Alembic 2`] = `
 "FROM public.ecr.aws/lambda/python:3.14
 
-# Patch OS packages beyond the base image's pinned release so the image is
-# free of known HIGH/CRITICAL vulnerabilities
-RUN microdnf upgrade -y --releasever latest && microdnf clean all
+# Patch OS packages beyond the base image's pinned release, and drop pip, whose
+# vendored copies of msgpack and setuptools carry known HIGH vulnerabilities.
+# The handler's dependencies are already installed in the image, so nothing
+# installs packages at runtime.
+RUN microdnf upgrade -y --releasever latest && microdnf clean all \\
+    && rm -rf /var/lang/lib/python*/site-packages/pip \\
+    /var/lang/lib/python*/site-packages/pip-*.dist-info
 
 WORKDIR \${LAMBDA_TASK_ROOT}
 
@@ -1400,9 +1412,13 @@ CMD ["proj_db.migration_handler.handler"]
 exports[`py#rdb generator > should generate a PostgreSQL SQLModel project with Alembic 3`] = `
 "FROM public.ecr.aws/lambda/python:3.14
 
-# Patch OS packages beyond the base image's pinned release so the image is
-# free of known HIGH/CRITICAL vulnerabilities
-RUN microdnf upgrade -y --releasever latest && microdnf clean all
+# Patch OS packages beyond the base image's pinned release, and drop pip, whose
+# vendored copies of msgpack and setuptools carry known HIGH vulnerabilities.
+# The handler's dependencies are already installed in the image, so nothing
+# installs packages at runtime.
+RUN microdnf upgrade -y --releasever latest && microdnf clean all \\
+    && rm -rf /var/lang/lib/python*/site-packages/pip \\
+    /var/lang/lib/python*/site-packages/pip-*.dist-info
 
 WORKDIR \${LAMBDA_TASK_ROOT}
 

package/src/py/rdb/files/Dockerfile.create-db-user.template

@@ -1,8 +1,12 @@
 FROM public.ecr.aws/lambda/python:3.14
 
-# Patch OS packages beyond the base image's pinned release so the image is
-# free of known HIGH/CRITICAL vulnerabilities
-RUN microdnf upgrade -y --releasever latest && microdnf clean all
+# Patch OS packages beyond the base image's pinned release, and drop pip, whose
+# vendored copies of msgpack and setuptools carry known HIGH vulnerabilities.
+# The handler's dependencies are already installed in the image, so nothing
+# installs packages at runtime.
+RUN microdnf upgrade -y --releasever latest && microdnf clean all \
+    && rm -rf /var/lang/lib/python*/site-packages/pip \
+    /var/lang/lib/python*/site-packages/pip-*.dist-info
 
 WORKDIR ${LAMBDA_TASK_ROOT}
 

package/src/py/rdb/files/Dockerfile.migration.template

@@ -1,8 +1,12 @@
 FROM public.ecr.aws/lambda/python:3.14
 
-# Patch OS packages beyond the base image's pinned release so the image is
-# free of known HIGH/CRITICAL vulnerabilities
-RUN microdnf upgrade -y --releasever latest && microdnf clean all
+# Patch OS packages beyond the base image's pinned release, and drop pip, whose
+# vendored copies of msgpack and setuptools carry known HIGH vulnerabilities.
+# The handler's dependencies are already installed in the image, so nothing
+# installs packages at runtime.
+RUN microdnf upgrade -y --releasever latest && microdnf clean all \
+    && rm -rf /var/lang/lib/python*/site-packages/pip \
+    /var/lang/lib/python*/site-packages/pip-*.dist-info
 
 WORKDIR ${LAMBDA_TASK_ROOT}
 

package/src/py/rdb/generator.js

@@ -290,34 +290,35 @@ export const pyRdbGenerator = async (tree, options)=>{
         }
     };
     if (options.infra !== 'none') {
-        if (iac === 'terraform') {
-            projectConfig.targets.docker = {
-                cache: IMAGE_BUILD_CACHE,
-                executor: 'nx:run-commands',
-                options: {
-                    commands: [
-                        `${containerEngine} build --platform linux/arm64 --provenance=false -t ${migrationDockerImageTag} ${migrationBundleDir}`,
-                        `${containerEngine} build --platform linux/arm64 --provenance=false -t ${createDbUserDockerImageTag} ${createDbUserBundleDir}`
-                    ],
-                    parallel: false
-                },
-                dependsOn: [
-                    'bundle-migration',
-                    'bundle-create-db-user'
-                ]
-            };
-            addArtifactDependencyToTargets(projectConfig, 'docker');
-            addDockerScanTarget(tree, {
-                project: projectConfig,
-                containerEngine,
-                trivyTargetName: 'trivy',
-                dockerTargetName: 'docker',
-                imageTags: [
-                    migrationDockerImageTag,
-                    createDbUserDockerImageTag
-                ]
-            }, DEPENDENCIES);
-        }
+        // Both images ship under either provider — Terraform pushes the locally
+        // built tags, CDK builds the same contexts as image assets — so the build
+        // and its scan are wired up for either.
+        projectConfig.targets.docker = {
+            cache: IMAGE_BUILD_CACHE,
+            executor: 'nx:run-commands',
+            options: {
+                commands: [
+                    `${containerEngine} build --platform linux/arm64 --provenance=false -t ${migrationDockerImageTag} ${migrationBundleDir}`,
+                    `${containerEngine} build --platform linux/arm64 --provenance=false -t ${createDbUserDockerImageTag} ${createDbUserBundleDir}`
+                ],
+                parallel: false
+            },
+            dependsOn: [
+                'bundle-migration',
+                'bundle-create-db-user'
+            ]
+        };
+        addArtifactDependencyToTargets(projectConfig, 'docker');
+        addDockerScanTarget(tree, {
+            project: projectConfig,
+            containerEngine,
+            trivyTargetName: 'trivy',
+            dockerTargetName: 'docker',
+            imageTags: [
+                migrationDockerImageTag,
+                createDbUserDockerImageTag
+            ]
+        }, DEPENDENCIES);
         addDependencyToTargetIfNotPresent(projectConfig, 'build', 'bundle-migration');
         addDependencyToTargetIfNotPresent(projectConfig, 'build', 'bundle-create-db-user');
     }