@aws/nx-plugin 1.0.0-rc.3 → 1.0.0-rc.31

package/src/py/agent/generator.js

@@ -1,35 +1,33 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.pyAgentGenerator = exports.PY_AGENT_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 versions_1 = require("../../utils/versions");
-const git_1 = require("../../utils/git");
-const object_1 = require("../../utils/object");
-const nx_1 = require("../../utils/nx");
-const metrics_1 = require("../../utils/metrics");
-const format_1 = require("../../utils/format");
-const names_1 = require("../../utils/names");
-const py_1 = require("../../utils/py");
-const agent_core_constructs_1 = require("../../utils/agent-core-constructs/agent-core-constructs");
-const agent_connection_1 = require("../../utils/agent-connection/agent-connection");
-const py_2 = require("../../utils/py");
-const bundle_1 = require("../../utils/bundle/bundle");
-const fs_1 = require("../../utils/fs");
-const npm_scope_1 = require("../../utils/npm-scope");
-const shared_constructs_1 = require("../../utils/shared-constructs");
-const nxlv_python_1 = require("../../utils/nxlv-python");
-const iac_1 = require("../../utils/iac");
-const containers_1 = require("../../utils/containers");
-const port_1 = require("../../utils/port");
-const paths_1 = require("../../utils/paths");
-exports.PY_AGENT_GENERATOR_INFO = (0, nx_1.getGeneratorInfo)(__filename);
-const pyAgentGenerator = async (tree, options) => {
-    const project = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.project);
-    const pyProjectPath = (0, devkit_1.joinPathFragments)(project.root, 'pyproject.toml');
+ */ import { addDependenciesToPackageJson, generateFiles, joinPathFragments, OverwriteStrategy, updateProjectConfiguration } from "@nx/devkit";
+import { ensureLicenseExceptions } from "../../license/config.js";
+import { AG_UI_LANGGRAPH_EXCEPTIONS } from "../../license/known-exceptions.js";
+import { addAgentChatScripts } from "../../utils/agent-chat/agent-chat.js";
+import { addPythonFrameworkBase, ensurePythonAgentConnectionProject, getPythonAgentConnectionModuleName, getPythonAgentConnectionProject } from "../../utils/agent-connection/agent-connection.js";
+import { addAgentInfra } from "../../utils/agent-core-constructs/agent-core-constructs.js";
+import { addPythonBundleTarget } from "../../utils/bundle/bundle.js";
+import { resolveContainers } from "../../utils/containers.js";
+import { formatFilesInSubtree } from "../../utils/format.js";
+import { FsCommands } from "../../utils/fs.js";
+import { updateGitIgnore } from "../../utils/git.js";
+import { resolveIac } from "../../utils/iac.js";
+import { installDependencies } from "../../utils/install.js";
+import { addGeneratorMetricsIfApplicable } from "../../utils/metrics.js";
+import { kebabCase, toClassName, toSnakeCase } from "../../utils/names.js";
+import { getNpmScope } from "../../utils/npm-scope.js";
+import { addComponentDevTarget, addComponentGeneratorMetadata, addDependencyToTargetIfNotPresent, getGeneratorInfo, normalizeTargetKeyOrder, readProjectConfigurationUnqualified } from "../../utils/nx.js";
+import { sortObjectKeys } from "../../utils/object.js";
+import { toProjectRelativePath } from "../../utils/paths.js";
+import { assignPort } from "../../utils/port.js";
+import { addDependenciesToDependencyGroupInPyProjectToml, addDependenciesToPyProjectToml, addWorkspaceDependencyToPyProject } from "../../utils/py.js";
+import { sharedConstructsGenerator } from "../../utils/shared-constructs.js";
+import { withVersions } from "../../utils/versions.js";
+export const PY_AGENT_GENERATOR_INFO = getGeneratorInfo(import.meta.filename);
+export const pyAgentGenerator = async (tree, options)=>{
+    const project = readProjectConfigurationUnqualified(tree, options.project);
+    const pyProjectPath = joinPathFragments(project.root, 'pyproject.toml');
     // Check if the project has a pyproject.toml file
     if (!pyProjectPath) {
         throw new Error(`Unsupported project ${options.project}. Expected a Python project (with a pyproject.toml)`);
@@ -40,13 +38,14 @@ const pyAgentGenerator = async (tree, options) => {
     // Module name is the last part of the source root,
     const sourceParts = project.sourceRoot.split('/');
     const moduleName = sourceParts[sourceParts.length - 1];
-    const name = (0, names_1.kebabCase)(options.name || `${(0, names_1.kebabCase)(project.name.split('.').pop())}-agent`);
+    const name = kebabCase(options.name || `${kebabCase(project.name.split('.').pop())}-agent`);
     const agentTargetPrefix = options.name ? name : 'agent';
-    const agentNameSnakeCase = (0, names_1.toSnakeCase)(options.name || 'agent');
-    const agentNameClassName = (0, names_1.toClassName)(name);
-    const targetSourceDir = (0, devkit_1.joinPathFragments)(project.sourceRoot, agentNameSnakeCase);
+    const agentNameSnakeCase = toSnakeCase(options.name || 'agent');
+    const agentNameClassName = toClassName(name);
+    const targetSourceDir = joinPathFragments(project.sourceRoot, agentNameSnakeCase);
     const infra = options.infra ?? 'agentcore';
     const protocol = options.protocol ?? 'http';
+    const framework = options.framework ?? 'strands';
     if (infra === 'none' && options.auth && options.auth !== 'iam') {
         console.warn('Warning: auth is ignored when no infrastructure is configured (no infrastructure is generated)');
     }
@@ -55,57 +54,112 @@ const pyAgentGenerator = async (tree, options) => {
     // point can import `session_id_context` and propagate the AgentCore
     // session ID to any downstream MCP / A2A clients a later connection
     // generator wires into this agent.
-    await (0, agent_connection_1.ensurePythonAgentConnectionProject)(tree);
-    const agentConnectionModuleName = (0, agent_connection_1.getPythonAgentConnectionModuleName)(tree);
-    const agentConnectionPackageName = (0, agent_connection_1.getPythonAgentConnectionPackageName)(tree);
-    (0, py_2.addWorkspaceDependencyToPyProject)(tree, project.root, agentConnectionPackageName);
+    await ensurePythonAgentConnectionProject(tree);
+    // The agent server imports the framework base helpers (session cache + model
+    // error logging) regardless of whether a connection client is wired in. The
+    // langchain framework has no base layer (its AG-UI foundation reuses only the
+    // framework-agnostic session context), so this is a no-op for langchain.
+    await addPythonFrameworkBase(tree, framework);
+    const agentConnectionModuleName = getPythonAgentConnectionModuleName(tree);
+    addWorkspaceDependencyToPyProject(tree, project, getPythonAgentConnectionProject(tree));
     const templateContext = {
         name,
         agentNameSnakeCase,
         agentNameClassName,
         moduleName,
         agentConnectionModuleName,
+        framework
     };
-    // Generate common files shared by both protocols
-    (0, devkit_1.generateFiles)(tree, (0, devkit_1.joinPathFragments)(__dirname, 'files', 'common'), targetSourceDir, templateContext, { overwriteStrategy: devkit_1.OverwriteStrategy.KeepExisting });
-    // Generate protocol-specific files
-    (0, devkit_1.generateFiles)(tree, (0, devkit_1.joinPathFragments)(__dirname, 'files', protocol.toLowerCase()), targetSourceDir, templateContext, { overwriteStrategy: devkit_1.OverwriteStrategy.KeepExisting });
-    (0, py_1.addDependenciesToPyProjectToml)(tree, project.root, [
+    // Generate common files shared by both protocols. The agent module (agent.py)
+    // is framework-specific (Strands yields a contextmanaged Agent; LangChain
+    // returns a compiled create_agent graph), so it comes from a per-framework
+    // dir. The package __init__.py is framework-agnostic (it stays in `common`).
+    generateFiles(tree, joinPathFragments(import.meta.dirname, 'files', framework === 'langchain' ? 'common-langchain' : 'common'), targetSourceDir, templateContext, {
+        overwriteStrategy: OverwriteStrategy.KeepExisting
+    });
+    if (framework === 'langchain') {
+        // common-langchain only carries agent.py; the empty package __init__.py is
+        // framework-agnostic, so emit it from the shared `common` dir.
+        generateFiles(tree, joinPathFragments(import.meta.dirname, 'files', 'common'), targetSourceDir, templateContext, {
+            overwriteStrategy: OverwriteStrategy.KeepExisting
+        });
+    }
+    // Generate protocol-specific files. Each protocol's server entry point is
+    // framework-specific (Strands yields a contextmanaged Agent; LangChain drives
+    // a compiled create_agent graph), so it comes from a per-framework dir
+    // `<protocol>-langchain` for LangChain, or `<protocol>` for Strands.
+    const protocolLower = protocol.toLowerCase();
+    const protocolTemplateDir = framework === 'langchain' ? `${protocolLower}-langchain` : protocolLower;
+    generateFiles(tree, joinPathFragments(import.meta.dirname, 'files', protocolTemplateDir), targetSourceDir, templateContext, {
+        overwriteStrategy: OverwriteStrategy.KeepExisting
+    });
+    // The HTTP server's init.py (the FastAPI app + JsonStreamingResponse) is
+    // framework-agnostic and the LangChain http main.py reuses it, so emit it
+    // from the shared `http` dir. Emitted after the langchain main.py above so
+    // KeepExisting preserves that main.py and only adds init.py (the shared dir's
+    // own Strands main.py is skipped).
+    if (protocolLower === 'http' && framework === 'langchain') {
+        generateFiles(tree, joinPathFragments(import.meta.dirname, 'files', 'http'), targetSourceDir, templateContext, {
+            overwriteStrategy: OverwriteStrategy.KeepExisting
+        });
+    }
+    addDependenciesToPyProjectToml(tree, project.root, [
         'aws-lambda-powertools',
         'aws-opentelemetry-distro',
         'bedrock-agentcore',
         'boto3',
         'fastapi',
         'mcp',
-        ...(protocol === 'a2a'
-            ? ['strands-agents[a2a]']
-            : protocol === 'ag-ui'
-                ? ['strands-agents', 'ag-ui-protocol', 'ag-ui-strands']
-                : ['strands-agents']),
-        'strands-agents-tools',
-        'uvicorn',
+        ...framework === 'langchain' ? // (langchain, langchain-aws, langgraph) plus the per-protocol server
+        // adapter: ag-ui-langgraph for AG-UI, a2a-sdk for A2A, nothing extra for
+        // HTTP (FastAPI is added above for every agent).
+        [
+            'langchain',
+            'langchain-aws',
+            'langgraph',
+            ...protocol === 'ag-ui' ? [
+                'ag-ui-protocol',
+                'ag-ui-langgraph'
+            ] : protocol === 'a2a' ? [
+                'a2a-sdk'
+            ] : []
+        ] : protocol === 'a2a' ? [
+            'strands-agents[a2a]',
+            'strands-agents-tools'
+        ] : protocol === 'ag-ui' ? [
+            'strands-agents',
+            'strands-agents-tools',
+            'ag-ui-protocol',
+            'ag-ui-strands'
+        ] : [
+            'strands-agents',
+            'strands-agents-tools'
+        ],
+        'uvicorn'
     ]);
-    (0, py_1.addDependenciesToDependencyGroupInPyProjectToml)(tree, project.root, 'dev', [
-        'fastapi[standard]',
+    addDependenciesToDependencyGroupInPyProjectToml(tree, project.root, 'dev', [
+        'fastapi[standard]'
     ]);
     if (infra === 'agentcore') {
-        const containers = await (0, containers_1.resolveContainers)(tree, 'inherit');
-        const dockerImageTag = `${(0, npm_scope_1.getNpmScope)(tree)}-${name}:latest`;
+        const containers = await resolveContainers(tree, 'inherit');
+        const dockerImageTag = `${getNpmScope(tree)}-${name}:latest`;
         // Add bundle target
-        const { bundleTargetName, bundleOutputDir } = (0, bundle_1.addPythonBundleTarget)(project, {
-            pythonPlatform: 'aarch64-manylinux_2_28',
+        const { bundleTargetName, bundleOutputDir } = addPythonBundleTarget(project, {
+            pythonPlatform: 'aarch64-manylinux_2_28'
         });
         // Add the Dockerfile
-        (0, devkit_1.generateFiles)(tree, (0, devkit_1.joinPathFragments)(__dirname, 'files', 'deploy'), targetSourceDir, {
+        generateFiles(tree, joinPathFragments(import.meta.dirname, 'files', 'deploy'), targetSourceDir, {
             agentNameSnakeCase,
             moduleName,
             bundleOutputDir,
-            protocol,
-        }, { overwriteStrategy: devkit_1.OverwriteStrategy.KeepExisting });
-        const dockerOutputDir = (0, devkit_1.joinPathFragments)('dist', project.root, 'docker', name);
+            protocol
+        }, {
+            overwriteStrategy: OverwriteStrategy.KeepExisting
+        });
+        const dockerOutputDir = joinPathFragments('dist', project.root, 'docker', name);
         const dockerTargetName = `${agentTargetPrefix}-docker`;
         // Add a docker target that prepares the docker context and builds the image
-        const fs = new fs_1.FsCommands(tree);
+        const fs = new FsCommands(tree);
         project.targets[dockerTargetName] = {
             cache: true,
             executor: 'nx:run-commands',
@@ -115,21 +169,25 @@ const pyAgentGenerator = async (tree, options) => {
                     fs.mkdir(dockerOutputDir),
                     fs.cp(bundleOutputDir, dockerOutputDir),
                     fs.cp(`${targetSourceDir}/Dockerfile`, `${dockerOutputDir}/Dockerfile`),
-                    `${containers} build --platform linux/arm64 -t ${dockerImageTag} ${dockerOutputDir}`,
+                    `${containers} build --platform linux/arm64 -t ${dockerImageTag} ${dockerOutputDir}`
                 ],
-                parallel: false,
+                parallel: false
             },
-            dependsOn: [bundleTargetName],
+            dependsOn: [
+                bundleTargetName
+            ]
         };
-        (0, nx_1.addDependencyToTargetIfNotPresent)(project, 'docker', dockerTargetName);
-        (0, nx_1.addDependencyToTargetIfNotPresent)(project, 'build', 'docker');
+        addDependencyToTargetIfNotPresent(project, 'docker', dockerTargetName);
+        addDependencyToTargetIfNotPresent(project, 'build', 'docker');
         // Add shared constructs
-        const iac = await (0, iac_1.resolveIac)(tree, options.iac);
-        await (0, shared_constructs_1.sharedConstructsGenerator)(tree, { iac });
+        const iac = await resolveIac(tree, options.iac);
+        await sharedConstructsGenerator(tree, {
+            iac
+        });
         // Add the construct to deploy the agent.
         // AG-UI uses HTTP as the AgentCore protocol type (AG-UI is HTTP-based with SSE over POST).
         const infraProtocol = protocol === 'ag-ui' ? 'http' : protocol;
-        await (0, agent_core_constructs_1.addAgentInfra)(tree, {
+        await addAgentInfra(tree, {
             agentNameKebabCase: name,
             agentNameClassName,
             dockerImageTag,
@@ -138,13 +196,18 @@ const pyAgentGenerator = async (tree, options) => {
             projectName: project.name,
             auth,
             serverProtocol: infraProtocol,
-            containers,
+            containers
         });
     }
     // A2A servers use port 9000 as per the Strands A2A SDK default and AgentCore A2A contract.
     // HTTP and AG-UI agents use port 8081+ to avoid conflict with VS Code server on 8080.
     const localDevPortStart = protocol === 'a2a' ? 9000 : 8081;
-    const localDevPort = (0, port_1.assignPort)(tree, project, localDevPortStart);
+    const localDevPort = assignPort(tree, project, localDevPortStart, {
+        component: {
+            info: PY_AGENT_GENERATOR_INFO,
+            name: agentTargetPrefix
+        }
+    });
     // All protocols use fastapi dev for hot reload:
     // - HTTP: FastAPI app directly defined in init.py
     // - A2A: A2AServer.to_fastapi_app() creates a FastAPI app in main.py
@@ -156,132 +219,169 @@ const pyAgentGenerator = async (tree, options) => {
     // we can run `agent-chat-cli` directly as a binary.
     const openApiTargetName = `${agentTargetPrefix}-openapi`;
     const clientGenTargetName = `${agentTargetPrefix}-generate-client`;
+    const scriptsDir = joinPathFragments(project.root, 'scripts', agentTargetPrefix);
     if (protocol === 'http') {
         // Emit the OpenAPI spec generator script (shared with react-connection)
-        (0, devkit_1.generateFiles)(tree, (0, devkit_1.joinPathFragments)(__dirname, 'react-connection', 'files', 'agent'), project.root, {
+        generateFiles(tree, joinPathFragments(import.meta.dirname, 'react-connection', 'files', 'agent'), project.root, {
             moduleName,
-            agentNameSnakeCase,
-        }, { overwriteStrategy: devkit_1.OverwriteStrategy.KeepExisting });
-        // Emit the chat CLI adapter script
-        const scriptsDir = (0, devkit_1.joinPathFragments)(project.root, 'scripts', agentTargetPrefix);
-        (0, devkit_1.generateFiles)(tree, (0, devkit_1.joinPathFragments)(__dirname, 'scripts', 'http'), scriptsDir, {
-            agentNameClassName,
-            agentTargetPrefix,
-            localDevPort,
-        }, { overwriteStrategy: devkit_1.OverwriteStrategy.KeepExisting });
+            agentNameSnakeCase
+        }, {
+            overwriteStrategy: OverwriteStrategy.KeepExisting
+        });
         // Ignore the generated client directory
-        (0, git_1.updateGitIgnore)(tree, project.root, (patterns) => [
-            ...patterns,
-            `scripts/${agentTargetPrefix}/generated/`,
-        ]);
+        updateGitIgnore(tree, project.root, (patterns)=>[
+                ...patterns,
+                `scripts/${agentTargetPrefix}/generated/`
+            ]);
     }
-    // Add TypeScript deps for the chat CLI. `agent-chat-cli` transitively
-    // bundles the protocol clients (@a2a-js/sdk, @ag-ui/client) and
-    // @clack/prompts, so nothing else is needed. HTTP also needs tsx to
-    // run the adapter script.
-    (0, devkit_1.addDependenciesToPackageJson)(tree, {}, (0, versions_1.withVersions)([
+    // Every protocol gets a standalone `chat.ts`. It connects to the local
+    // `dev` server by default, or to the deployed agent (with the
+    // appropriate auth) when `RUNTIME_CONFIG_APP_ID` is set.
+    addAgentChatScripts(tree, {
+        scriptsDir,
+        protocol,
+        language: 'py',
+        agentNameClassName,
+        auth
+    });
+    // TypeScript deps for the chat CLI, which runs standalone via tsx for every
+    // protocol and resolves the deployed agent from AppConfig. `agent-chat-cli`
+    // transitively bundles the protocol clients (@a2a-js/sdk, @ag-ui/client).
+    addDependenciesToPackageJson(tree, {}, withVersions([
         'agent-chat-cli',
-        ...(protocol === 'http'
-            ? ['tsx', '@types/node']
-            : []),
+        'tsx',
+        '@types/node',
+        '@aws-lambda-powertools/parameters',
+        '@aws-sdk/client-appconfigdata',
+        ...auth === 'iam' ? [
+            'aws4fetch',
+            '@aws-sdk/credential-providers'
+        ] : [],
+        ...protocol === 'a2a' ? [
+            '@a2a-js/sdk'
+        ] : []
     ]));
-    const chatUrl = protocol === 'ag-ui'
-        ? `http://localhost:${localDevPort}/invocations`
-        : `http://localhost:${localDevPort}`;
-    const chatCommand = protocol === 'http'
-        ? `tsx ./scripts/${agentTargetPrefix}/chat.ts`
-        : protocol === 'ag-ui'
-            ? `agent-chat-cli agui ${chatUrl}`
-            : `agent-chat-cli a2a ${chatUrl}`;
-    const httpOnlyTargets = protocol === 'http'
-        ? {
-            [openApiTargetName]: {
-                cache: true,
-                executor: 'nx:run-commands',
-                outputs: [
-                    `{workspaceRoot}/dist/{projectRoot}/openapi/${agentNameSnakeCase}`,
+    const chatUrl = protocol === 'ag-ui' ? `http://localhost:${localDevPort}/invocations` : `http://localhost:${localDevPort}`;
+    const chatCommand = `tsx ./scripts/${agentTargetPrefix}/chat.ts`;
+    const httpOnlyTargets = protocol === 'http' ? {
+        [openApiTargetName]: {
+            cache: true,
+            executor: 'nx:run-commands',
+            outputs: [
+                `{workspaceRoot}/dist/{projectRoot}/openapi/${agentNameSnakeCase}`
+            ],
+            options: {
+                commands: [
+                    `uv run python {projectRoot}/scripts/${agentNameSnakeCase}_openapi.py "dist/{projectRoot}/openapi/${agentNameSnakeCase}/openapi.json"`
+                ]
+            }
+        },
+        [clientGenTargetName]: {
+            cache: true,
+            executor: 'nx:run-commands',
+            dependsOn: [
+                openApiTargetName
+            ],
+            inputs: [
+                {
+                    dependentTasksOutputFiles: '**/*.json'
+                }
+            ],
+            outputs: [
+                `{projectRoot}/scripts/${agentTargetPrefix}/generated`
+            ],
+            options: {
+                commands: [
+                    `nx g @aws/nx-plugin:open-api#ts-client --openApiSpecPath="dist/${project.root}/openapi/${agentNameSnakeCase}/openapi.json" --outputPath="${project.root}/scripts/${agentTargetPrefix}/generated" --no-interactive`
+                ]
+            }
+        }
+    } : {};
+    const agentTargets = {
+        ...project.targets,
+        ...httpOnlyTargets,
+        [`${agentTargetPrefix}-serve`]: {
+            executor: 'nx:run-commands',
+            continuous: true,
+            options: {
+                commands: [
+                    serveCommand
                 ],
-                options: {
-                    commands: [
-                        `uv run python {projectRoot}/scripts/${agentNameSnakeCase}_openapi.py "dist/{projectRoot}/openapi/${agentNameSnakeCase}/openapi.json"`,
-                    ],
-                },
-            },
-            [clientGenTargetName]: {
-                cache: true,
-                executor: 'nx:run-commands',
-                inputs: [
-                    {
-                        dependentTasksOutputFiles: '**/*.json',
-                    },
+                cwd: '{projectRoot}',
+                env: {
+                    PORT: `${localDevPort}`
+                }
+            }
+        },
+        [`${agentTargetPrefix}-dev`]: {
+            executor: 'nx:run-commands',
+            continuous: true,
+            options: {
+                commands: [
+                    serveCommand
                 ],
-                outputs: [`{projectRoot}/scripts/${agentTargetPrefix}/generated`],
-                options: {
-                    commands: [
-                        `nx g @aws/nx-plugin:open-api#ts-client --openApiSpecPath="dist/${project.root}/openapi/${agentNameSnakeCase}/openapi.json" --outputPath="${project.root}/scripts/${agentTargetPrefix}/generated" --no-interactive`,
-                    ],
-                },
-                dependsOn: [openApiTargetName],
-            },
-        }
-        : {};
-    (0, devkit_1.updateProjectConfiguration)(tree, project.name, {
-        ...project,
-        targets: (0, object_1.sortObjectKeys)({
-            ...project.targets,
-            ...httpOnlyTargets,
-            [`${agentTargetPrefix}-serve`]: {
-                executor: 'nx:run-commands',
-                options: {
-                    commands: [serveCommand],
-                    cwd: '{projectRoot}',
-                    env: {
-                        PORT: `${localDevPort}`,
-                    },
-                },
-                continuous: true,
-            },
-            [`${agentTargetPrefix}-serve-local`]: {
-                executor: 'nx:run-commands',
-                options: {
-                    commands: [serveCommand],
-                    cwd: '{projectRoot}',
-                    env: {
-                        PORT: `${localDevPort}`,
-                        SERVE_LOCAL: 'true',
-                    },
-                },
-                continuous: true,
-            },
-            [`${agentTargetPrefix}-chat`]: {
-                executor: 'nx:run-commands',
-                options: {
-                    commands: [chatCommand],
-                    cwd: '{projectRoot}',
-                    env: {
-                        URL: chatUrl,
-                    },
-                },
-                dependsOn: [
-                    ...(protocol === 'http' ? [clientGenTargetName] : []),
-                    `${agentTargetPrefix}-serve-local`,
+                cwd: '{projectRoot}',
+                env: {
+                    PORT: `${localDevPort}`,
+                    LOCAL_DEV: 'true'
+                }
+            }
+        },
+        // HTTP chat imports the generated client, so ensure it's built first.
+        // No dev dependency — chat runs standalone against a local or
+        // deployed agent. normalizeTargetKeyOrder keeps the conditional dependsOn
+        // in Nx 23's serialization order so re-runs stay byte-identical.
+        [`${agentTargetPrefix}-chat`]: normalizeTargetKeyOrder({
+            executor: 'nx:run-commands',
+            options: {
+                commands: [
+                    chatCommand
                 ],
+                cwd: '{projectRoot}',
+                env: {
+                    URL: chatUrl
+                }
             },
-        }),
+            ...protocol === 'http' ? {
+                dependsOn: [
+                    clientGenTargetName
+                ]
+            } : {}
+        })
+    };
+    // Aggregate `<agent>-dev` under the project-level `dev` target.
+    addComponentDevTarget(agentTargets, `${agentTargetPrefix}-dev`);
+    updateProjectConfiguration(tree, project.name, {
+        ...project,
+        targets: sortObjectKeys(agentTargets)
     });
-    (0, nx_1.addComponentGeneratorMetadata)(tree, project.name, exports.PY_AGENT_GENERATOR_INFO, (0, paths_1.toProjectRelativePath)(project, targetSourceDir), agentTargetPrefix, {
+    addComponentGeneratorMetadata(tree, project.name, PY_AGENT_GENERATOR_INFO, toProjectRelativePath(project, targetSourceDir), agentTargetPrefix, {
         port: localDevPort,
         rc: agentNameClassName,
         auth,
         protocol,
+        // The mcp / gateway connection generators dispatch on this field to pick
+        // the Strands vs LangChain Layer-2 client + agent.py transform.
+        framework
     });
-    await (0, metrics_1.addGeneratorMetricsIfApplicable)(tree, [exports.PY_AGENT_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 addGeneratorMetricsIfApplicable(tree, [
+        PY_AGENT_GENERATOR_INFO
+    ]);
+    // langchain-core (pulled by every langchain agent regardless of protocol)
+    // brings jsonpatch/jsonpointer, whose wheels ship without resolvable SPDX
+    // license metadata, so register those exceptions so the workspace license
+    // check still passes.
+    if (framework === 'langchain') {
+        await ensureLicenseExceptions(tree, AG_UI_LANGGRAPH_EXCEPTIONS);
+    }
+    await formatFilesInSubtree(tree);
+    return ()=>installDependencies(tree, options.preferInstallDependencies, {
+            languages: [
+                'typescript',
+                'python'
+            ]
+        });
 };
-exports.pyAgentGenerator = pyAgentGenerator;
-exports.default = exports.pyAgentGenerator;
+export default pyAgentGenerator;
+
 //# sourceMappingURL=generator.js.map