@aws/nx-plugin 1.0.0-rc.22 → 1.0.0-rc.24

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

@@ -1,43 +1,35 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.pyAgentA2aConnectionGenerator = exports.PY_AGENT_A2A_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 agent_connection_1 = require("../../../utils/agent-connection/agent-connection");
-const ast_1 = require("../../../utils/ast");
-const format_1 = require("../../../utils/format");
-const metrics_1 = require("../../../utils/metrics");
-const names_1 = require("../../../utils/names");
-const nx_1 = require("../../../utils/nx");
-const py_1 = require("../../../utils/py");
-/** Prefix a GritQL pattern with `language python` */
-const py = (pattern) => `language python\n${pattern}`;
-/** The module each framework imports the `tool` decorator from. */
-const TOOL_IMPORT_MODULE = {
+ */ import { generateFiles, installPackagesTask, joinPathFragments, OverwriteStrategy, updateProjectConfiguration } from "@nx/devkit";
+import { addPythonCoreClient, addPythonReExport, ensurePythonAgentConnectionProject, getPythonAgentConnectionModuleName, getPythonAgentConnectionProject, getPythonAgentConnectionProjectDir, PY_CLIENT_NAMING, resolveAgentFramework } from "../../../utils/agent-connection/agent-connection.js";
+import { addPythonDestructuredImport, applyGritQL, matchGritQL } from "../../../utils/ast.js";
+import { formatFilesInSubtree } from "../../../utils/format.js";
+import { addGeneratorMetricsIfApplicable } from "../../../utils/metrics.js";
+import { snakeCase } from "../../../utils/names.js";
+import { addDependencyToTargetIfNotPresent, getGeneratorInfo, readProjectConfigurationUnqualified } from "../../../utils/nx.js";
+import { addDependenciesToPyProjectToml, addWorkspaceDependencyToPyProject } from "../../../utils/py.js";
+/** Prefix a GritQL pattern with `language python` */ const py = (pattern)=>`language python\n${pattern}`;
+/** The module each framework imports the `tool` decorator from. */ const TOOL_IMPORT_MODULE = {
     strands: 'strands',
-    langchain: 'langchain_core.tools',
+    langchain: 'langchain_core.tools'
 };
-/** The GritQL pattern matching each framework's agent constructor. */
-const AGENT_CONSTRUCTOR = {
+/** The GritQL pattern matching each framework's agent constructor. */ const AGENT_CONSTRUCTOR = {
     strands: 'yield Agent($_)',
-    langchain: 'create_agent($_)',
+    langchain: 'create_agent($_)'
 };
 /**
  * The A2A transport dependency each framework's Layer-2 client needs. Strands
  * wraps strands' `A2AAgent` (strands-agents[a2a]); LangChain drives the a2a SDK
  * directly (a2a-sdk) and must not pull Strands in.
- */
-const A2A_TRANSPORT_DEP = {
+ */ const A2A_TRANSPORT_DEP = {
     strands: 'strands-agents[a2a]',
-    langchain: 'a2a-sdk',
+    langchain: 'a2a-sdk'
 };
-exports.PY_AGENT_A2A_CONNECTION_GENERATOR_INFO = (0, nx_1.getGeneratorInfo)(__filename);
-const pyAgentA2aConnectionGenerator = async (tree, options) => {
-    const sourceProject = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.sourceProject);
-    const targetProject = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.targetProject);
+export const PY_AGENT_A2A_CONNECTION_GENERATOR_INFO = getGeneratorInfo(import.meta.filename);
+export const pyAgentA2aConnectionGenerator = async (tree, options)=>{
+    const sourceProject = readProjectConfigurationUnqualified(tree, options.sourceProject);
+    const targetProject = readProjectConfigurationUnqualified(tree, options.targetProject);
     const agentComponent = options.sourceComponent;
     const targetAgentComponent = options.targetComponent;
     if (!agentComponent || !targetAgentComponent) {
@@ -46,13 +38,12 @@ const pyAgentA2aConnectionGenerator = async (tree, options) => {
     if ((targetAgentComponent.protocol ?? '').toLowerCase() !== 'a2a') {
         throw new Error(`Target agent '${targetAgentComponent.name}' uses the ${targetAgentComponent.protocol ?? 'http'} protocol — only A2A agents can be connected as tools.`);
     }
-    if (targetAgentComponent.auth &&
-        targetAgentComponent.auth.toLowerCase() !== 'iam') {
+    if (targetAgentComponent.auth && targetAgentComponent.auth.toLowerCase() !== 'iam') {
         throw new Error(`A2A agent connection currently only supports IAM authentication, but '${targetAgentComponent.name}' uses '${targetAgentComponent.auth}' authentication.`);
     }
     const targetAgentComponentName = targetAgentComponent.name ?? 'agent';
     const targetAgentClassName = targetAgentComponent.rc;
-    const targetAgentSnakeCase = (0, names_1.snakeCase)(targetAgentClassName);
+    const targetAgentSnakeCase = snakeCase(targetAgentClassName);
     const targetAgentPort = targetAgentComponent.port ?? 9000;
     // The source agent's framework drives everything that varies: the client
     // naming + app template (shared with the mcp/gateway generators via
@@ -61,41 +52,42 @@ const pyAgentA2aConnectionGenerator = async (tree, options) => {
     // (Strands' A2AAgent vs an a2a-sdk-only client) over the shared,
     // framework-agnostic A2A client config. Keyed on the framework, not a
     // boolean, so a third framework stays additive.
-    const framework = (0, agent_connection_1.resolveAgentFramework)(agentComponent.framework);
-    const naming = agent_connection_1.PY_CLIENT_NAMING[framework];
+    const framework = resolveAgentFramework(agentComponent.framework);
+    const naming = PY_CLIENT_NAMING[framework];
     // 1. Ensure the shared Python agent-connection project exists + has the
     //    A2A core client (framework Layer-2) and its shared SigV4 auth helper.
-    await (0, agent_connection_1.ensurePythonAgentConnectionProject)(tree);
-    await (0, agent_connection_1.addPythonCoreClient)(tree, 'a2a', framework);
-    const agentConnectionProjectDir = (0, agent_connection_1.getPythonAgentConnectionProjectDir)(tree);
-    const agentConnectionModuleName = (0, agent_connection_1.getPythonAgentConnectionModuleName)(tree);
-    const agentConnectionPackageName = (0, agent_connection_1.getPythonAgentConnectionPackageName)(tree);
+    await ensurePythonAgentConnectionProject(tree);
+    await addPythonCoreClient(tree, 'a2a', framework);
+    const agentConnectionProjectDir = getPythonAgentConnectionProjectDir(tree);
+    const agentConnectionModuleName = getPythonAgentConnectionModuleName(tree);
     // A2A client config + shared auth helper deps, plus the framework's A2A
     // transport dep (see A2A_TRANSPORT_DEP).
-    (0, py_1.addDependenciesToPyProjectToml)(tree, agentConnectionProjectDir, [
+    addDependenciesToPyProjectToml(tree, agentConnectionProjectDir, [
         'boto3',
         'httpx',
-        A2A_TRANSPORT_DEP[framework],
+        A2A_TRANSPORT_DEP[framework]
     ]);
     // 2. Generate the per-connection client into the shared agent-connection project
-    const appDir = (0, devkit_1.joinPathFragments)(agentConnectionProjectDir, agentConnectionModuleName, 'app');
-    (0, devkit_1.generateFiles)(tree, (0, devkit_1.joinPathFragments)(__dirname, 'files', naming.appTemplateSubdir), appDir, {
+    const appDir = joinPathFragments(agentConnectionProjectDir, agentConnectionModuleName, 'app');
+    generateFiles(tree, joinPathFragments(import.meta.dirname, 'files', naming.appTemplateSubdir), appDir, {
         targetAgentSnakeCase,
         targetAgentClassName,
         targetAgentPort,
-        agentConnectionModuleName,
-    }, { overwriteStrategy: devkit_1.OverwriteStrategy.KeepExisting });
+        agentConnectionModuleName
+    }, {
+        overwriteStrategy: OverwriteStrategy.KeepExisting
+    });
     // Ensure app/__init__.py exists
-    const appInitPath = (0, devkit_1.joinPathFragments)(appDir, '__init__.py');
+    const appInitPath = joinPathFragments(appDir, '__init__.py');
     if (!tree.exists(appInitPath)) {
         tree.write(appInitPath, '');
     }
     // Add re-export to module __init__.py
-    const moduleInitPath = (0, devkit_1.joinPathFragments)(agentConnectionProjectDir, agentConnectionModuleName, '__init__.py');
-    await (0, agent_connection_1.addPythonReExport)(tree, moduleInitPath, `.app.${targetAgentSnakeCase}_client_${naming.clientModuleSuffix}`, `${targetAgentClassName}Client${naming.clientClassSuffix}`);
+    const moduleInitPath = joinPathFragments(agentConnectionProjectDir, agentConnectionModuleName, '__init__.py');
+    await addPythonReExport(tree, moduleInitPath, `.app.${targetAgentSnakeCase}_client_${naming.clientModuleSuffix}`, `${targetAgentClassName}Client${naming.clientClassSuffix}`);
     // 3. Transform agent.py to add the A2A client import + wrap it as a tool
-    const agentSourceDir = (0, devkit_1.joinPathFragments)(sourceProject.root, agentComponent.path ?? 'src');
-    const agentFilePath = (0, devkit_1.joinPathFragments)(agentSourceDir, 'agent.py');
+    const agentSourceDir = joinPathFragments(sourceProject.root, agentComponent.path ?? 'src');
+    const agentFilePath = joinPathFragments(agentSourceDir, 'agent.py');
     if (tree.exists(agentFilePath)) {
         const clientClassName = `${targetAgentClassName}Client${naming.clientClassSuffix}`;
         const clientVarName = targetAgentSnakeCase;
@@ -105,45 +97,48 @@ const pyAgentA2aConnectionGenerator = async (tree, options) => {
         await addClientToolToGetAgent(tree, agentFilePath, clientClassName, clientVarName, toolName, targetAgentClassName, framework);
     }
     // 4. Add workspace dependency from agent project to agent-connection project
-    (0, py_1.addWorkspaceDependencyToPyProject)(tree, sourceProject.root, agentConnectionPackageName);
-    // 5. Set up serve-local target dependencies — chain onto the target agent
+    addWorkspaceDependencyToPyProject(tree, sourceProject, getPythonAgentConnectionProject(tree));
+    // 5. Set up dev target dependencies — chain onto the target agent
     const agentName = agentComponent.name ?? 'agent';
-    const serveLocalTargetName = `${agentName}-serve-local`;
-    const targetServeLocalTargetName = `${targetAgentComponentName}-serve-local`;
-    if (sourceProject.targets?.[serveLocalTargetName]) {
-        (0, nx_1.addDependencyToTargetIfNotPresent)(sourceProject, serveLocalTargetName, {
-            projects: [targetProject.name],
-            target: targetServeLocalTargetName,
+    const devTargetName = `${agentName}-dev`;
+    const targetDevTargetName = `${targetAgentComponentName}-dev`;
+    if (sourceProject.targets?.[devTargetName]) {
+        addDependencyToTargetIfNotPresent(sourceProject, devTargetName, {
+            projects: [
+                targetProject.name
+            ],
+            target: targetDevTargetName
         });
-        (0, devkit_1.updateProjectConfiguration)(tree, sourceProject.name, sourceProject);
+        updateProjectConfiguration(tree, sourceProject.name, sourceProject);
     }
-    await (0, metrics_1.addGeneratorMetricsIfApplicable)(tree, [
-        exports.PY_AGENT_A2A_CONNECTION_GENERATOR_INFO,
+    await addGeneratorMetricsIfApplicable(tree, [
+        PY_AGENT_A2A_CONNECTION_GENERATOR_INFO
     ]);
-    await (0, format_1.formatFilesInSubtree)(tree);
-    return () => {
-        (0, devkit_1.installPackagesTask)(tree);
+    await formatFilesInSubtree(tree);
+    return ()=>{
+        installPackagesTask(tree);
     };
 };
-exports.pyAgentA2aConnectionGenerator = pyAgentA2aConnectionGenerator;
 /**
  * Add imports to agent.py:
  *  - `tool` (from `strands`, or `langchain_core.tools` for LangChain agents)
  *  - the per-connection client class from the shared agent-connection module
- */
-const addImportToAgentFile = async (tree, filePath, agentConnectionModuleName, clientClassName, framework) => {
-    await (0, ast_1.addPythonDestructuredImport)(tree, filePath, ['tool'], TOOL_IMPORT_MODULE[framework]);
-    await (0, ast_1.addPythonDestructuredImport)(tree, filePath, [clientClassName], agentConnectionModuleName);
+ */ const addImportToAgentFile = async (tree, filePath, agentConnectionModuleName, clientClassName, framework)=>{
+    await addPythonDestructuredImport(tree, filePath, [
+        'tool'
+    ], TOOL_IMPORT_MODULE[framework]);
+    await addPythonDestructuredImport(tree, filePath, [
+        clientClassName
+    ], agentConnectionModuleName);
 };
 /**
  * Add the A2A tool to the Agent's tools array using GritQL.
  *
  * Scoped to the framework's agent constructor so unrelated `tools=` keyword
  * arguments elsewhere in the file are not touched.
- */
-const addToolToAgent = async (tree, filePath, toolName, framework) => {
+ */ const addToolToAgent = async (tree, filePath, toolName, framework)=>{
     const scope = AGENT_CONSTRUCTOR[framework];
-    await (0, ast_1.applyGritQL)(tree, filePath, py(`\`tools=$old\` where {
+    await applyGritQL(tree, filePath, py(`\`tools=$old\` where {
   $old <: within \`${scope}\`,
   $old <: not contains \`${toolName}\`,
   if ($old <: \`[]\`) {
@@ -160,9 +155,8 @@ const addToolToAgent = async (tree, filePath, toolName, framework) => {
  * Unlike MCP clients, A2A clients aren't context managers — we don't need
  * a `with` block around them. Creation + tool definition happen at the
  * top of the function body, anchored on the framework's agent constructor.
- */
-const addClientToolToGetAgent = async (tree, filePath, clientClassName, clientVarName, toolName, targetAgentClassName, framework) => {
-    if (await (0, ast_1.matchGritQL)(tree, filePath, py(`\`${clientClassName}.create\``))) {
+ */ const addClientToolToGetAgent = async (tree, filePath, clientClassName, clientVarName, toolName, targetAgentClassName, framework)=>{
+    if (await matchGritQL(tree, filePath, py(`\`${clientClassName}.create\``))) {
         return;
     }
     // Both frameworks' tools are @tool-decorated callables (the decorator's
@@ -179,7 +173,7 @@ const addClientToolToGetAgent = async (tree, filePath, clientClassName, clientVa
     // the framework's agent constructor. This works whether or not a prior MCP
     // block is present since we insert before any existing statements.
     const anchor = AGENT_CONSTRUCTOR[framework];
-    await (0, ast_1.applyGritQL)(tree, filePath, py(`\`def get_agent($params):
+    await applyGritQL(tree, filePath, py(`\`def get_agent($params):
     $body\` where {
   $body <: contains \`${anchor}\`,
   $body <: not contains \`${clientClassName}.create\`
@@ -187,5 +181,6 @@ const addClientToolToGetAgent = async (tree, filePath, clientClassName, clientVa
     ${toolBlock}
     $body\``));
 };
-exports.default = exports.pyAgentA2aConnectionGenerator;
+export default pyAgentA2aConnectionGenerator;
+
 //# sourceMappingURL=generator.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../../../../../packages/nx-plugin/src/py/agent/a2a-connection/generator.ts"],"names":[],"mappings":";;;AAAA;;;GAGG;AACH,uCAQoB;AACpB,uFAU0D;AAC1D,4CAI4B;AAC5B,kDAA6D;AAC7D,oDAAyE;AACzE,gDAAiD;AACjD,0CAK2B;AAC3B,0CAG2B;AAI3B,qDAAqD;AACrD,MAAM,EAAE,GAAG,CAAC,OAAe,EAAE,EAAE,CAAC,oBAAoB,OAAO,EAAE,CAAC;AAE9D,mEAAmE;AACnE,MAAM,kBAAkB,GAAmC;IACzD,OAAO,EAAE,SAAS;IAClB,SAAS,EAAE,sBAAsB;CAClC,CAAC;AAEF,sEAAsE;AACtE,MAAM,iBAAiB,GAAmC;IACxD,OAAO,EAAE,iBAAiB;IAC1B,SAAS,EAAE,kBAAkB;CAC9B,CAAC;AAEF;;;;GAIG;AACH,MAAM,iBAAiB,GAA0C;IAC/D,OAAO,EAAE,qBAAqB;IAC9B,SAAS,EAAE,SAAS;CACrB,CAAC;AAEW,QAAA,sCAAsC,GACjD,IAAA,qBAAgB,EAAC,UAAU,CAAC,CAAC;AAExB,MAAM,6BAA6B,GAAG,KAAK,EAChD,IAAU,EACV,OAA4C,EAChB,EAAE;IAC9B,MAAM,aAAa,GAAG,IAAA,wCAAmC,EACvD,IAAI,EACJ,OAAO,CAAC,aAAa,CACtB,CAAC;IACF,MAAM,aAAa,GAAG,IAAA,wCAAmC,EACvD,IAAI,EACJ,OAAO,CAAC,aAAa,CACtB,CAAC;IAEF,MAAM,cAAc,GAAG,OAAO,CAAC,eAAe,CAAC;IAC/C,MAAM,oBAAoB,GAAG,OAAO,CAAC,eAAe,CAAC;IAErD,IAAI,CAAC,cAAc,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC7C,MAAM,IAAI,KAAK,CACb,2FAA2F,CAC5F,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,oBAAoB,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,KAAK,KAAK,EAAE,CAAC;QAClE,MAAM,IAAI,KAAK,CACb,iBAAiB,oBAAoB,CAAC,IAAI,cAAc,oBAAoB,CAAC,QAAQ,IAAI,MAAM,wDAAwD,CACxJ,CAAC;IACJ,CAAC;IAED,IACE,oBAAoB,CAAC,IAAI;QACzB,oBAAoB,CAAC,IAAI,CAAC,WAAW,EAAE,KAAK,KAAK,EACjD,CAAC;QACD,MAAM,IAAI,KAAK,CACb,yEAAyE,oBAAoB,CAAC,IAAI,WAAW,oBAAoB,CAAC,IAAI,mBAAmB,CAC1J,CAAC;IACJ,CAAC;IAED,MAAM,wBAAwB,GAAG,oBAAoB,CAAC,IAAI,IAAI,OAAO,CAAC;IACtE,MAAM,oBAAoB,GAAG,oBAAoB,CAAC,EAAY,CAAC;IAC/D,MAAM,oBAAoB,GAAG,IAAA,iBAAS,EAAC,oBAAoB,CAAC,CAAC;IAC7D,MAAM,eAAe,GAAG,oBAAoB,CAAC,IAAI,IAAI,IAAI,CAAC;IAE1D,yEAAyE;IACzE,oEAAoE;IACpE,4EAA4E;IAC5E,4EAA4E;IAC5E,iEAAiE;IACjE,sEAAsE;IACtE,gDAAgD;IAChD,MAAM,SAAS,GAAG,IAAA,wCAAqB,EAAC,cAAc,CAAC,SAAS,CAAC,CAAC;IAClE,MAAM,MAAM,GAAG,mCAAgB,CAAC,SAAS,CAAC,CAAC;IAE3C,wEAAwE;IACxE,2EAA2E;IAC3E,MAAM,IAAA,qDAAkC,EAAC,IAAI,CAAC,CAAC;IAC/C,MAAM,IAAA,sCAAmB,EAAC,IAAI,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;IAElD,MAAM,yBAAyB,GAAG,IAAA,qDAAkC,EAAC,IAAI,CAAC,CAAC;IAC3E,MAAM,yBAAyB,GAAG,IAAA,qDAAkC,EAAC,IAAI,CAAC,CAAC;IAC3E,MAAM,0BAA0B,GAAG,IAAA,sDAAmC,EAAC,IAAI,CAAC,CAAC;IAE7E,wEAAwE;IACxE,yCAAyC;IACzC,IAAA,mCAA8B,EAAC,IAAI,EAAE,yBAAyB,EAAE;QAC9D,OAAO;QACP,OAAO;QACP,iBAAiB,CAAC,SAAS,CAAC;KAC7B,CAAC,CAAC;IAEH,iFAAiF;IACjF,MAAM,MAAM,GAAG,IAAA,0BAAiB,EAC9B,yBAAyB,EACzB,yBAAyB,EACzB,KAAK,CACN,CAAC;IACF,IAAA,sBAAa,EACX,IAAI,EACJ,IAAA,0BAAiB,EAAC,SAAS,EAAE,OAAO,EAAE,MAAM,CAAC,iBAAiB,CAAC,EAC/D,MAAM,EACN;QACE,oBAAoB;QACpB,oBAAoB;QACpB,eAAe;QACf,yBAAyB;KAC1B,EACD,EAAE,iBAAiB,EAAE,0BAAiB,CAAC,YAAY,EAAE,CACtD,CAAC;IAEF,gCAAgC;IAChC,MAAM,WAAW,GAAG,IAAA,0BAAiB,EAAC,MAAM,EAAE,aAAa,CAAC,CAAC;IAC7D,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC;QAC9B,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;IAC9B,CAAC;IAED,sCAAsC;IACtC,MAAM,cAAc,GAAG,IAAA,0BAAiB,EACtC,yBAAyB,EACzB,yBAAyB,EACzB,aAAa,CACd,CAAC;IACF,MAAM,IAAA,oCAAiB,EACrB,IAAI,EACJ,cAAc,EACd,QAAQ,oBAAoB,WAAW,MAAM,CAAC,kBAAkB,EAAE,EAClE,GAAG,oBAAoB,SAAS,MAAM,CAAC,iBAAiB,EAAE,CAC3D,CAAC;IAEF,yEAAyE;IACzE,MAAM,cAAc,GAAG,IAAA,0BAAiB,EACtC,aAAa,CAAC,IAAI,EAClB,cAAc,CAAC,IAAI,IAAI,KAAK,CAC7B,CAAC;IACF,MAAM,aAAa,GAAG,IAAA,0BAAiB,EAAC,cAAc,EAAE,UAAU,CAAC,CAAC;IAEpE,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;QAC/B,MAAM,eAAe,GAAG,GAAG,oBAAoB,SAAS,MAAM,CAAC,iBAAiB,EAAE,CAAC;QACnF,MAAM,aAAa,GAAG,oBAAoB,CAAC;QAC3C,MAAM,QAAQ,GAAG,OAAO,oBAAoB,EAAE,CAAC;QAE/C,MAAM,oBAAoB,CACxB,IAAI,EACJ,aAAa,EACb,yBAAyB,EACzB,eAAe,EACf,SAAS,CACV,CAAC;QACF,MAAM,cAAc,CAAC,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;QAC/D,MAAM,uBAAuB,CAC3B,IAAI,EACJ,aAAa,EACb,eAAe,EACf,aAAa,EACb,QAAQ,EACR,oBAAoB,EACpB,SAAS,CACV,CAAC;IACJ,CAAC;IAED,6EAA6E;IAC7E,IAAA,sCAAiC,EAC/B,IAAI,EACJ,aAAa,CAAC,IAAI,EAClB,0BAA0B,CAC3B,CAAC;IAEF,0EAA0E;IAC1E,MAAM,SAAS,GAAG,cAAc,CAAC,IAAI,IAAI,OAAO,CAAC;IACjD,MAAM,oBAAoB,GAAG,GAAG,SAAS,cAAc,CAAC;IACxD,MAAM,0BAA0B,GAAG,GAAG,wBAAwB,cAAc,CAAC;IAE7E,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC,oBAAoB,CAAC,EAAE,CAAC;QAClD,IAAA,sCAAiC,EAAC,aAAa,EAAE,oBAAoB,EAAE;YACrE,QAAQ,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC;YAC9B,MAAM,EAAE,0BAA0B;SACnC,CAAC,CAAC;QACH,IAAA,mCAA0B,EAAC,IAAI,EAAE,aAAa,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;IACtE,CAAC;IAED,MAAM,IAAA,yCAA+B,EAAC,IAAI,EAAE;QAC1C,8CAAsC;KACvC,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;AAtKW,QAAA,6BAA6B,iCAsKxC;AAEF;;;;GAIG;AACH,MAAM,oBAAoB,GAAG,KAAK,EAChC,IAAU,EACV,QAAgB,EAChB,yBAAiC,EACjC,eAAuB,EACvB,SAAyB,EACV,EAAE;IACjB,MAAM,IAAA,iCAA2B,EAC/B,IAAI,EACJ,QAAQ,EACR,CAAC,MAAM,CAAC,EACR,kBAAkB,CAAC,SAAS,CAAC,CAC9B,CAAC;IACF,MAAM,IAAA,iCAA2B,EAC/B,IAAI,EACJ,QAAQ,EACR,CAAC,eAAe,CAAC,EACjB,yBAAyB,CAC1B,CAAC;AACJ,CAAC,CAAC;AAEF;;;;;GAKG;AACH,MAAM,cAAc,GAAG,KAAK,EAC1B,IAAU,EACV,QAAgB,EAChB,QAAgB,EAChB,SAAyB,EACV,EAAE;IACjB,MAAM,KAAK,GAAG,iBAAiB,CAAC,SAAS,CAAC,CAAC;IAC3C,MAAM,IAAA,iBAAW,EACf,IAAI,EACJ,QAAQ,EACR,EAAE,CAAC;qBACc,KAAK;2BACC,QAAQ;;iBAElB,QAAQ;;iDAEwB,QAAQ;;EAEvD,CAAC,CACA,CAAC;AACJ,CAAC,CAAC;AAEF;;;;;;;GAOG;AACH,MAAM,uBAAuB,GAAG,KAAK,EACnC,IAAU,EACV,QAAgB,EAChB,eAAuB,EACvB,aAAqB,EACrB,QAAgB,EAChB,oBAA4B,EAC5B,SAAyB,EACV,EAAE;IACjB,IAAI,MAAM,IAAA,iBAAW,EAAC,IAAI,EAAE,QAAQ,EAAE,EAAE,CAAC,KAAK,eAAe,WAAW,CAAC,CAAC,EAAE,CAAC;QAC3E,OAAO;IACT,CAAC;IAED,wEAAwE;IACxE,sEAAsE;IACtE,4EAA4E;IAC5E,MAAM,SAAS,GAAG,GAAG,aAAa,MAAM,eAAe;;;UAG/C,QAAQ;+CAC6B,oBAAoB;qBAC9C,aAAa;CACjC,CAAC;IAEA,8EAA8E;IAC9E,2EAA2E;IAC3E,mEAAmE;IACnE,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,CAAC,CAAC;IAC5C,MAAM,IAAA,iBAAW,EACf,IAAI,EACJ,QAAQ,EACR,EAAE,CAAC;;wBAEiB,MAAM;4BACF,eAAe;;MAErC,SAAS;YACH,CAAC,CACV,CAAC;AACJ,CAAC,CAAC;AAEF,kBAAe,qCAA6B,CAAC"}
+{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/py/agent/a2a-connection/generator.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nimport {\n  type GeneratorCallback,\n  generateFiles,\n  installPackagesTask,\n  joinPathFragments,\n  OverwriteStrategy,\n  type Tree,\n  updateProjectConfiguration,\n} from '@nx/devkit';\nimport {\n  type AgentFramework,\n  addPythonCoreClient,\n  addPythonReExport,\n  ensurePythonAgentConnectionProject,\n  getPythonAgentConnectionModuleName,\n  getPythonAgentConnectionProject,\n  getPythonAgentConnectionProjectDir,\n  PY_CLIENT_NAMING,\n  resolveAgentFramework,\n} from '../../../utils/agent-connection/agent-connection';\nimport {\n  addPythonDestructuredImport,\n  applyGritQL,\n  matchGritQL,\n} from '../../../utils/ast';\nimport { formatFilesInSubtree } from '../../../utils/format';\nimport { addGeneratorMetricsIfApplicable } from '../../../utils/metrics';\nimport { snakeCase } from '../../../utils/names';\nimport {\n  addDependencyToTargetIfNotPresent,\n  getGeneratorInfo,\n  type NxGeneratorInfo,\n  readProjectConfigurationUnqualified,\n} from '../../../utils/nx';\nimport {\n  addDependenciesToPyProjectToml,\n  addWorkspaceDependencyToPyProject,\n} from '../../../utils/py';\nimport type { IPyDepVersion } from '../../../utils/versions';\nimport type { PyAgentA2aConnectionGeneratorSchema } from './schema';\n\n/** Prefix a GritQL pattern with `language python` */\nconst py = (pattern: string) => `language python\\n${pattern}`;\n\n/** The module each framework imports the `tool` decorator from. */\nconst TOOL_IMPORT_MODULE: Record<AgentFramework, string> = {\n  strands: 'strands',\n  langchain: 'langchain_core.tools',\n};\n\n/** The GritQL pattern matching each framework's agent constructor. */\nconst AGENT_CONSTRUCTOR: Record<AgentFramework, string> = {\n  strands: 'yield Agent($_)',\n  langchain: 'create_agent($_)',\n};\n\n/**\n * The A2A transport dependency each framework's Layer-2 client needs. Strands\n * wraps strands' `A2AAgent` (strands-agents[a2a]); LangChain drives the a2a SDK\n * directly (a2a-sdk) and must not pull Strands in.\n */\nconst A2A_TRANSPORT_DEP: Record<AgentFramework, IPyDepVersion> = {\n  strands: 'strands-agents[a2a]',\n  langchain: 'a2a-sdk',\n};\n\nexport const PY_AGENT_A2A_CONNECTION_GENERATOR_INFO: NxGeneratorInfo =\n  getGeneratorInfo(import.meta.filename);\n\nexport const pyAgentA2aConnectionGenerator = async (\n  tree: Tree,\n  options: PyAgentA2aConnectionGeneratorSchema,\n): Promise<GeneratorCallback> => {\n  const sourceProject = readProjectConfigurationUnqualified(\n    tree,\n    options.sourceProject,\n  );\n  const targetProject = readProjectConfigurationUnqualified(\n    tree,\n    options.targetProject,\n  );\n\n  const agentComponent = options.sourceComponent;\n  const targetAgentComponent = options.targetComponent;\n\n  if (!agentComponent || !targetAgentComponent) {\n    throw new Error(\n      'Both sourceComponent and targetComponent must be provided for py#agent -> a2a connections',\n    );\n  }\n\n  if ((targetAgentComponent.protocol ?? '').toLowerCase() !== 'a2a') {\n    throw new Error(\n      `Target agent '${targetAgentComponent.name}' uses the ${targetAgentComponent.protocol ?? 'http'} protocol — only A2A agents can be connected as tools.`,\n    );\n  }\n\n  if (\n    targetAgentComponent.auth &&\n    targetAgentComponent.auth.toLowerCase() !== 'iam'\n  ) {\n    throw new Error(\n      `A2A agent connection currently only supports IAM authentication, but '${targetAgentComponent.name}' uses '${targetAgentComponent.auth}' authentication.`,\n    );\n  }\n\n  const targetAgentComponentName = targetAgentComponent.name ?? 'agent';\n  const targetAgentClassName = targetAgentComponent.rc as string;\n  const targetAgentSnakeCase = snakeCase(targetAgentClassName);\n  const targetAgentPort = targetAgentComponent.port ?? 9000;\n\n  // The source agent's framework drives everything that varies: the client\n  // naming + app template (shared with the mcp/gateway generators via\n  // PY_CLIENT_NAMING), the A2A transport dep (see A2A_TRANSPORT_DEP), and the\n  // agent.py transform below. Each framework wraps its own A2A Layer-2 client\n  // (Strands' A2AAgent vs an a2a-sdk-only client) over the shared,\n  // framework-agnostic A2A client config. Keyed on the framework, not a\n  // boolean, so a third framework stays additive.\n  const framework = resolveAgentFramework(agentComponent.framework);\n  const naming = PY_CLIENT_NAMING[framework];\n\n  // 1. Ensure the shared Python agent-connection project exists + has the\n  //    A2A core client (framework Layer-2) and its shared SigV4 auth helper.\n  await ensurePythonAgentConnectionProject(tree);\n  await addPythonCoreClient(tree, 'a2a', framework);\n\n  const agentConnectionProjectDir = getPythonAgentConnectionProjectDir(tree);\n  const agentConnectionModuleName = getPythonAgentConnectionModuleName(tree);\n\n  // A2A client config + shared auth helper deps, plus the framework's A2A\n  // transport dep (see A2A_TRANSPORT_DEP).\n  addDependenciesToPyProjectToml(tree, agentConnectionProjectDir, [\n    'boto3',\n    'httpx',\n    A2A_TRANSPORT_DEP[framework],\n  ]);\n\n  // 2. Generate the per-connection client into the shared agent-connection project\n  const appDir = joinPathFragments(\n    agentConnectionProjectDir,\n    agentConnectionModuleName,\n    'app',\n  );\n  generateFiles(\n    tree,\n    joinPathFragments(import.meta.dirname, 'files', naming.appTemplateSubdir),\n    appDir,\n    {\n      targetAgentSnakeCase,\n      targetAgentClassName,\n      targetAgentPort,\n      agentConnectionModuleName,\n    },\n    { overwriteStrategy: OverwriteStrategy.KeepExisting },\n  );\n\n  // Ensure app/__init__.py exists\n  const appInitPath = joinPathFragments(appDir, '__init__.py');\n  if (!tree.exists(appInitPath)) {\n    tree.write(appInitPath, '');\n  }\n\n  // Add re-export to module __init__.py\n  const moduleInitPath = joinPathFragments(\n    agentConnectionProjectDir,\n    agentConnectionModuleName,\n    '__init__.py',\n  );\n  await addPythonReExport(\n    tree,\n    moduleInitPath,\n    `.app.${targetAgentSnakeCase}_client_${naming.clientModuleSuffix}`,\n    `${targetAgentClassName}Client${naming.clientClassSuffix}`,\n  );\n\n  // 3. Transform agent.py to add the A2A client import + wrap it as a tool\n  const agentSourceDir = joinPathFragments(\n    sourceProject.root,\n    agentComponent.path ?? 'src',\n  );\n  const agentFilePath = joinPathFragments(agentSourceDir, 'agent.py');\n\n  if (tree.exists(agentFilePath)) {\n    const clientClassName = `${targetAgentClassName}Client${naming.clientClassSuffix}`;\n    const clientVarName = targetAgentSnakeCase;\n    const toolName = `ask_${targetAgentSnakeCase}`;\n\n    await addImportToAgentFile(\n      tree,\n      agentFilePath,\n      agentConnectionModuleName,\n      clientClassName,\n      framework,\n    );\n    await addToolToAgent(tree, agentFilePath, toolName, framework);\n    await addClientToolToGetAgent(\n      tree,\n      agentFilePath,\n      clientClassName,\n      clientVarName,\n      toolName,\n      targetAgentClassName,\n      framework,\n    );\n  }\n\n  // 4. Add workspace dependency from agent project to agent-connection project\n  addWorkspaceDependencyToPyProject(\n    tree,\n    sourceProject,\n    getPythonAgentConnectionProject(tree),\n  );\n\n  // 5. Set up dev target dependencies — chain onto the target agent\n  const agentName = agentComponent.name ?? 'agent';\n  const devTargetName = `${agentName}-dev`;\n  const targetDevTargetName = `${targetAgentComponentName}-dev`;\n\n  if (sourceProject.targets?.[devTargetName]) {\n    addDependencyToTargetIfNotPresent(sourceProject, devTargetName, {\n      projects: [targetProject.name],\n      target: targetDevTargetName,\n    });\n    updateProjectConfiguration(tree, sourceProject.name, sourceProject);\n  }\n\n  await addGeneratorMetricsIfApplicable(tree, [\n    PY_AGENT_A2A_CONNECTION_GENERATOR_INFO,\n  ]);\n\n  await formatFilesInSubtree(tree);\n  return () => {\n    installPackagesTask(tree);\n  };\n};\n\n/**\n * Add imports to agent.py:\n *  - `tool` (from `strands`, or `langchain_core.tools` for LangChain agents)\n *  - the per-connection client class from the shared agent-connection module\n */\nconst addImportToAgentFile = async (\n  tree: Tree,\n  filePath: string,\n  agentConnectionModuleName: string,\n  clientClassName: string,\n  framework: AgentFramework,\n): Promise<void> => {\n  await addPythonDestructuredImport(\n    tree,\n    filePath,\n    ['tool'],\n    TOOL_IMPORT_MODULE[framework],\n  );\n  await addPythonDestructuredImport(\n    tree,\n    filePath,\n    [clientClassName],\n    agentConnectionModuleName,\n  );\n};\n\n/**\n * Add the A2A tool to the Agent's tools array using GritQL.\n *\n * Scoped to the framework's agent constructor so unrelated `tools=` keyword\n * arguments elsewhere in the file are not touched.\n */\nconst addToolToAgent = async (\n  tree: Tree,\n  filePath: string,\n  toolName: string,\n  framework: AgentFramework,\n): Promise<void> => {\n  const scope = AGENT_CONSTRUCTOR[framework];\n  await applyGritQL(\n    tree,\n    filePath,\n    py(`\\`tools=$old\\` where {\n  $old <: within \\`${scope}\\`,\n  $old <: not contains \\`${toolName}\\`,\n  if ($old <: \\`[]\\`) {\n    $old => \\`[${toolName}]\\`\n  } else {\n    $old <: \\`[$items]\\` where { $items += \\`, ${toolName}\\` }\n  }\n}`),\n  );\n};\n\n/**\n * Wrap the get_agent body so the remote A2A client is created and exposed\n * as a `@tool`-decorated closure before constructing the Agent.\n *\n * Unlike MCP clients, A2A clients aren't context managers — we don't need\n * a `with` block around them. Creation + tool definition happen at the\n * top of the function body, anchored on the framework's agent constructor.\n */\nconst addClientToolToGetAgent = async (\n  tree: Tree,\n  filePath: string,\n  clientClassName: string,\n  clientVarName: string,\n  toolName: string,\n  targetAgentClassName: string,\n  framework: AgentFramework,\n): Promise<void> => {\n  if (await matchGritQL(tree, filePath, py(`\\`${clientClassName}.create\\``))) {\n    return;\n  }\n\n  // Both frameworks' tools are @tool-decorated callables (the decorator's\n  // import differs, handled in addImportToAgentFile), defined inline in\n  // get_agent. The client is directly callable, returning the remote's reply.\n  const toolBlock = `${clientVarName} = ${clientClassName}.create()\n\n    @tool\n    def ${toolName}(prompt: str) -> str:\n        \"\"\"Delegate a question to the remote ${targetAgentClassName} A2A agent and return its reply.\"\"\"\n        return str(${clientVarName}(prompt))\n`;\n\n  // Prepend client creation + tool definition to the function body, anchored on\n  // the framework's agent constructor. This works whether or not a prior MCP\n  // block is present since we insert before any existing statements.\n  const anchor = AGENT_CONSTRUCTOR[framework];\n  await applyGritQL(\n    tree,\n    filePath,\n    py(`\\`def get_agent($params):\n    $body\\` where {\n  $body <: contains \\`${anchor}\\`,\n  $body <: not contains \\`${clientClassName}.create\\`\n} => \\`def get_agent($params):\n    ${toolBlock}\n    $body\\``),\n  );\n};\n\nexport default pyAgentA2aConnectionGenerator;\n"],"names":["generateFiles","installPackagesTask","joinPathFragments","OverwriteStrategy","updateProjectConfiguration","addPythonCoreClient","addPythonReExport","ensurePythonAgentConnectionProject","getPythonAgentConnectionModuleName","getPythonAgentConnectionProject","getPythonAgentConnectionProjectDir","PY_CLIENT_NAMING","resolveAgentFramework","addPythonDestructuredImport","applyGritQL","matchGritQL","formatFilesInSubtree","addGeneratorMetricsIfApplicable","snakeCase","addDependencyToTargetIfNotPresent","getGeneratorInfo","readProjectConfigurationUnqualified","addDependenciesToPyProjectToml","addWorkspaceDependencyToPyProject","py","pattern","TOOL_IMPORT_MODULE","strands","langchain","AGENT_CONSTRUCTOR","A2A_TRANSPORT_DEP","PY_AGENT_A2A_CONNECTION_GENERATOR_INFO","filename","pyAgentA2aConnectionGenerator","tree","options","sourceProject","targetProject","agentComponent","sourceComponent","targetAgentComponent","targetComponent","Error","protocol","toLowerCase","name","auth","targetAgentComponentName","targetAgentClassName","rc","targetAgentSnakeCase","targetAgentPort","port","framework","naming","agentConnectionProjectDir","agentConnectionModuleName","appDir","dirname","appTemplateSubdir","overwriteStrategy","KeepExisting","appInitPath","exists","write","moduleInitPath","clientModuleSuffix","clientClassSuffix","agentSourceDir","root","path","agentFilePath","clientClassName","clientVarName","toolName","addImportToAgentFile","addToolToAgent","addClientToolToGetAgent","agentName","devTargetName","targetDevTargetName","targets","projects","target","filePath","scope","toolBlock","anchor"],"mappings":"AAAA;;;CAGC,GACD,SAEEA,aAAa,EACbC,mBAAmB,EACnBC,iBAAiB,EACjBC,iBAAiB,EAEjBC,0BAA0B,QACrB,aAAa;AACpB,SAEEC,mBAAmB,EACnBC,iBAAiB,EACjBC,kCAAkC,EAClCC,kCAAkC,EAClCC,+BAA+B,EAC/BC,kCAAkC,EAClCC,gBAAgB,EAChBC,qBAAqB,QAChB,sDAAmD;AAC1D,SACEC,2BAA2B,EAC3BC,WAAW,EACXC,WAAW,QACN,wBAAqB;AAC5B,SAASC,oBAAoB,QAAQ,2BAAwB;AAC7D,SAASC,+BAA+B,QAAQ,4BAAyB;AACzE,SAASC,SAAS,QAAQ,0BAAuB;AACjD,SACEC,iCAAiC,EACjCC,gBAAgB,EAEhBC,mCAAmC,QAC9B,uBAAoB;AAC3B,SACEC,8BAA8B,EAC9BC,iCAAiC,QAC5B,uBAAoB;AAI3B,mDAAmD,GACnD,MAAMC,KAAK,CAACC,UAAoB,CAAC,iBAAiB,EAAEA,SAAS;AAE7D,iEAAiE,GACjE,MAAMC,qBAAqD;IACzDC,SAAS;IACTC,WAAW;AACb;AAEA,oEAAoE,GACpE,MAAMC,oBAAoD;IACxDF,SAAS;IACTC,WAAW;AACb;AAEA;;;;CAIC,GACD,MAAME,oBAA2D;IAC/DH,SAAS;IACTC,WAAW;AACb;AAEA,OAAO,MAAMG,yCACXX,iBAAiB,YAAYY,QAAQ,EAAE;AAEzC,OAAO,MAAMC,gCAAgC,OAC3CC,MACAC;IAEA,MAAMC,gBAAgBf,oCACpBa,MACAC,QAAQC,aAAa;IAEvB,MAAMC,gBAAgBhB,oCACpBa,MACAC,QAAQE,aAAa;IAGvB,MAAMC,iBAAiBH,QAAQI,eAAe;IAC9C,MAAMC,uBAAuBL,QAAQM,eAAe;IAEpD,IAAI,CAACH,kBAAkB,CAACE,sBAAsB;QAC5C,MAAM,IAAIE,MACR;IAEJ;IAEA,IAAI,AAACF,CAAAA,qBAAqBG,QAAQ,IAAI,EAAC,EAAGC,WAAW,OAAO,OAAO;QACjE,MAAM,IAAIF,MACR,CAAC,cAAc,EAAEF,qBAAqBK,IAAI,CAAC,WAAW,EAAEL,qBAAqBG,QAAQ,IAAI,OAAO,sDAAsD,CAAC;IAE3J;IAEA,IACEH,qBAAqBM,IAAI,IACzBN,qBAAqBM,IAAI,CAACF,WAAW,OAAO,OAC5C;QACA,MAAM,IAAIF,MACR,CAAC,sEAAsE,EAAEF,qBAAqBK,IAAI,CAAC,QAAQ,EAAEL,qBAAqBM,IAAI,CAAC,iBAAiB,CAAC;IAE7J;IAEA,MAAMC,2BAA2BP,qBAAqBK,IAAI,IAAI;IAC9D,MAAMG,uBAAuBR,qBAAqBS,EAAE;IACpD,MAAMC,uBAAuBhC,UAAU8B;IACvC,MAAMG,kBAAkBX,qBAAqBY,IAAI,IAAI;IAErD,yEAAyE;IACzE,oEAAoE;IACpE,4EAA4E;IAC5E,4EAA4E;IAC5E,iEAAiE;IACjE,sEAAsE;IACtE,gDAAgD;IAChD,MAAMC,YAAYzC,sBAAsB0B,eAAee,SAAS;IAChE,MAAMC,SAAS3C,gBAAgB,CAAC0C,UAAU;IAE1C,wEAAwE;IACxE,2EAA2E;IAC3E,MAAM9C,mCAAmC2B;IACzC,MAAM7B,oBAAoB6B,MAAM,OAAOmB;IAEvC,MAAME,4BAA4B7C,mCAAmCwB;IACrE,MAAMsB,4BAA4BhD,mCAAmC0B;IAErE,wEAAwE;IACxE,yCAAyC;IACzCZ,+BAA+BY,MAAMqB,2BAA2B;QAC9D;QACA;QACAzB,iBAAiB,CAACuB,UAAU;KAC7B;IAED,iFAAiF;IACjF,MAAMI,SAASvD,kBACbqD,2BACAC,2BACA;IAEFxD,cACEkC,MACAhC,kBAAkB,YAAYwD,OAAO,EAAE,SAASJ,OAAOK,iBAAiB,GACxEF,QACA;QACEP;QACAF;QACAG;QACAK;IACF,GACA;QAAEI,mBAAmBzD,kBAAkB0D,YAAY;IAAC;IAGtD,gCAAgC;IAChC,MAAMC,cAAc5D,kBAAkBuD,QAAQ;IAC9C,IAAI,CAACvB,KAAK6B,MAAM,CAACD,cAAc;QAC7B5B,KAAK8B,KAAK,CAACF,aAAa;IAC1B;IAEA,sCAAsC;IACtC,MAAMG,iBAAiB/D,kBACrBqD,2BACAC,2BACA;IAEF,MAAMlD,kBACJ4B,MACA+B,gBACA,CAAC,KAAK,EAAEf,qBAAqB,QAAQ,EAAEI,OAAOY,kBAAkB,EAAE,EAClE,GAAGlB,qBAAqB,MAAM,EAAEM,OAAOa,iBAAiB,EAAE;IAG5D,yEAAyE;IACzE,MAAMC,iBAAiBlE,kBACrBkC,cAAciC,IAAI,EAClB/B,eAAegC,IAAI,IAAI;IAEzB,MAAMC,gBAAgBrE,kBAAkBkE,gBAAgB;IAExD,IAAIlC,KAAK6B,MAAM,CAACQ,gBAAgB;QAC9B,MAAMC,kBAAkB,GAAGxB,qBAAqB,MAAM,EAAEM,OAAOa,iBAAiB,EAAE;QAClF,MAAMM,gBAAgBvB;QACtB,MAAMwB,WAAW,CAAC,IAAI,EAAExB,sBAAsB;QAE9C,MAAMyB,qBACJzC,MACAqC,eACAf,2BACAgB,iBACAnB;QAEF,MAAMuB,eAAe1C,MAAMqC,eAAeG,UAAUrB;QACpD,MAAMwB,wBACJ3C,MACAqC,eACAC,iBACAC,eACAC,UACA1B,sBACAK;IAEJ;IAEA,6EAA6E;IAC7E9B,kCACEW,MACAE,eACA3B,gCAAgCyB;IAGlC,kEAAkE;IAClE,MAAM4C,YAAYxC,eAAeO,IAAI,IAAI;IACzC,MAAMkC,gBAAgB,GAAGD,UAAU,IAAI,CAAC;IACxC,MAAME,sBAAsB,GAAGjC,yBAAyB,IAAI,CAAC;IAE7D,IAAIX,cAAc6C,OAAO,EAAE,CAACF,cAAc,EAAE;QAC1C5D,kCAAkCiB,eAAe2C,eAAe;YAC9DG,UAAU;gBAAC7C,cAAcQ,IAAI;aAAC;YAC9BsC,QAAQH;QACV;QACA5E,2BAA2B8B,MAAME,cAAcS,IAAI,EAAET;IACvD;IAEA,MAAMnB,gCAAgCiB,MAAM;QAC1CH;KACD;IAED,MAAMf,qBAAqBkB;IAC3B,OAAO;QACLjC,oBAAoBiC;IACtB;AACF,EAAE;AAEF;;;;CAIC,GACD,MAAMyC,uBAAuB,OAC3BzC,MACAkD,UACA5B,2BACAgB,iBACAnB;IAEA,MAAMxC,4BACJqB,MACAkD,UACA;QAAC;KAAO,EACR1D,kBAAkB,CAAC2B,UAAU;IAE/B,MAAMxC,4BACJqB,MACAkD,UACA;QAACZ;KAAgB,EACjBhB;AAEJ;AAEA;;;;;CAKC,GACD,MAAMoB,iBAAiB,OACrB1C,MACAkD,UACAV,UACArB;IAEA,MAAMgC,QAAQxD,iBAAiB,CAACwB,UAAU;IAC1C,MAAMvC,YACJoB,MACAkD,UACA5D,GAAG,CAAC;mBACW,EAAE6D,MAAM;yBACF,EAAEX,SAAS;;eAErB,EAAEA,SAAS;;+CAEqB,EAAEA,SAAS;;CAEzD,CAAC;AAEF;AAEA;;;;;;;CAOC,GACD,MAAMG,0BAA0B,OAC9B3C,MACAkD,UACAZ,iBACAC,eACAC,UACA1B,sBACAK;IAEA,IAAI,MAAMtC,YAAYmB,MAAMkD,UAAU5D,GAAG,CAAC,EAAE,EAAEgD,gBAAgB,SAAS,CAAC,IAAI;QAC1E;IACF;IAEA,wEAAwE;IACxE,sEAAsE;IACtE,4EAA4E;IAC5E,MAAMc,YAAY,GAAGb,cAAc,GAAG,EAAED,gBAAgB;;;QAGlD,EAAEE,SAAS;6CAC0B,EAAE1B,qBAAqB;mBACjD,EAAEyB,cAAc;AACnC,CAAC;IAEC,8EAA8E;IAC9E,2EAA2E;IAC3E,mEAAmE;IACnE,MAAMc,SAAS1D,iBAAiB,CAACwB,UAAU;IAC3C,MAAMvC,YACJoB,MACAkD,UACA5D,GAAG,CAAC;;sBAEc,EAAE+D,OAAO;0BACL,EAAEf,gBAAgB;;IAExC,EAAEc,UAAU;WACL,CAAC;AAEZ;AAEA,eAAerD,8BAA8B"}

package/src/py/agent/a2a-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/a2a-connection/schema.d.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/py/agent/a2a-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 PyAgentA2aConnectionGeneratorSchema {\n  sourceProject: string;\n  targetProject: string;\n  sourceComponent?: ComponentMetadata;\n  targetComponent?: ComponentMetadata;\n}\n"],"names":[],"mappings":"AAAA;;;CAGC,GAGD;;CAEC,GACD,WAKC"}

package/src/py/agent/files/ag-ui-langchain/main.py.template

@@ -6,6 +6,7 @@ from ag_ui.core import EventType, RunAgentInput, RunErrorEvent
 from ag_ui.encoder import EventEncoder
 from ag_ui_langgraph import LangGraphAgent
 from fastapi import FastAPI, Request
+from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse, StreamingResponse
 from <%- agentConnectionModuleName %> import get_current_session_id, session_id_context
 from starlette.middleware.base import BaseHTTPMiddleware
@@ -35,6 +36,15 @@ class _SessionIdMiddleware(BaseHTTPMiddleware):
 
 
 app = FastAPI(title="<%= agentNameClassName %>")
+# Allow browser-based AG-UI clients (e.g. a connected React website) to call
+# this agent cross-origin, including the CORS preflight.
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 app.add_middleware(_SessionIdMiddleware)
 
 

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

@@ -21,7 +21,7 @@ class AgentCoreGatewayMCPClientStrands:
 
     @staticmethod
     def without_auth(gateway_url: str) -> MCPClient:
-        """Plain-HTTP gateway client, for the local gateway started by serve-local."""
+        """Plain-HTTP gateway client, for the local gateway started by the dev target."""
         return MCPClient(AgentCoreGatewayMCPTransport.without_auth(gateway_url))
 "
 `;
@@ -53,7 +53,7 @@ class AgentCoreGatewayMCPTransport:
 
     @staticmethod
     def without_auth(gateway_url: str) -> TransportFactory:
-        """Plain-HTTP gateway transport, for the local gateway started by serve-local."""
+        """Plain-HTTP gateway transport, for the local gateway started by the dev target."""
         return no_auth_transport(gateway_url)
 "
 `;
@@ -63,7 +63,6 @@ exports[`py#agent#gateway-connection generator > should match snapshot for agent
 from typing import Any
 
 import boto3
-
 from mcp.client.streamable_http import streamablehttp_client
 
 from .auth.session import jwt_auth, plain_auth, sigv4_auth
@@ -105,29 +104,28 @@ def no_auth_transport(url: str) -> TransportFactory:
 exports[`py#agent#gateway-connection generator > should match snapshot for agent-connection core files > my_gateway_client_strands.py 1`] = `
 "import os
 
-from strands.tools.mcp.mcp_client import MCPClient
-
 from proj_agent_connection.core.agentcore_gateway_mcp_client_strands import (
     AgentCoreGatewayMCPClientStrands,
 )
 from proj_agent_connection.core.runtime_config import (
     get_agentcore_runtime_config,
 )
+from strands.tools.mcp.mcp_client import MCPClient
 
 
 class MyGatewayClientStrands:
     """Strands client for the MyGateway AgentCore Gateway.
 
     Deployed: a SigV4-signed MCP client pointing at the gateway URL.
-    Local (\`\`SERVE_LOCAL=true\`\`): a plain-HTTP client pointing at the local
-    gateway started by the gateway project's serve-local target. The local
+    Local (\`\`LOCAL_DEV=true\`\`): a plain-HTTP client pointing at the local
+    gateway started by the gateway project's dev target. The local
     gateway aggregates every attached MCP server but does not evaluate
     Cedar policies.
     """
 
     @staticmethod
     def create() -> MCPClient:
-        if os.environ.get("SERVE_LOCAL") == "true":
+        if os.environ.get("LOCAL_DEV") == "true":
             return AgentCoreGatewayMCPClientStrands.without_auth(
                 gateway_url="http://localhost:8100/mcp"
             )

package/src/py/agent/gateway-connection/files/agent-connection/app/__gatewaySnakeCase___client_strands.py.template

@@ -14,15 +14,15 @@ class <%- gatewayClassName %>ClientStrands:
     """Strands client for the <%- gatewayClassName %> AgentCore Gateway.
 
     Deployed: a SigV4-signed MCP client pointing at the gateway URL.
-    Local (``SERVE_LOCAL=true``): a plain-HTTP client pointing at the local
-    gateway started by the gateway project's serve-local target. The local
+    Local (``LOCAL_DEV=true``): a plain-HTTP client pointing at the local
+    gateway started by the gateway project's dev target. The local
     gateway aggregates every attached MCP server but does not evaluate
     Cedar policies.
     """
 
     @staticmethod
     def create() -> MCPClient:
-        if os.environ.get("SERVE_LOCAL") == "true":
+        if os.environ.get("LOCAL_DEV") == "true":
             return AgentCoreGatewayMCPClientStrands.without_auth(
                 gateway_url="http://localhost:<%- gatewayPort %>/mcp"
             )

package/src/py/agent/gateway-connection/files/langchain/agent-connection/app/__gatewaySnakeCase___client_langchain.py.template

@@ -14,14 +14,14 @@ class <%- gatewayClassName %>ClientLangChain:
     """Loads LangChain tools from the <%- gatewayClassName %> AgentCore Gateway.
 
     Deployed: SigV4-signed tools pointing at the gateway URL. Local
-    (``SERVE_LOCAL=true``): plain-HTTP tools pointing at the local gateway
-    started by the gateway project's serve-local target. The local gateway
+    (``LOCAL_DEV=true``): plain-HTTP tools pointing at the local gateway
+    started by the gateway project's dev target. The local gateway
     aggregates every attached MCP server but does not evaluate Cedar policies.
     """
 
     @staticmethod
     def create() -> list[BaseTool]:
-        if os.environ.get("SERVE_LOCAL") == "true":
+        if os.environ.get("LOCAL_DEV") == "true":
             return AgentCoreGatewayMCPClientLangChain.without_auth(
                 gateway_url="http://localhost:<%- gatewayPort %>/mcp"
             )

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

@@ -1,28 +1,24 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.pyAgentGatewayConnectionGenerator = exports.PY_AGENT_GATEWAY_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 generator_1 = require("../../../agentcore-gateway/generator");
-const agent_connection_1 = require("../../../utils/agent-connection/agent-connection");
-const ast_1 = require("../../../utils/ast");
-const format_1 = require("../../../utils/format");
-const metrics_1 = require("../../../utils/metrics");
-const names_1 = require("../../../utils/names");
-const nx_1 = require("../../../utils/nx");
-const py_1 = require("../../../utils/py");
-exports.PY_AGENT_GATEWAY_CONNECTION_GENERATOR_INFO = (0, nx_1.getGeneratorInfo)(__filename);
-const pyAgentGatewayConnectionGenerator = async (tree, options) => {
-    const sourceProject = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.sourceProject);
-    const targetProject = (0, nx_1.readProjectConfigurationUnqualified)(tree, options.targetProject);
+ */ import { generateFiles, installPackagesTask, joinPathFragments, OverwriteStrategy, updateProjectConfiguration } from "@nx/devkit";
+import { readAgentCoreGatewayMetadata } from "../../../agentcore-gateway/generator.js";
+import { addPythonCoreClient, addPythonReExport, ensurePythonAgentConnectionProject, getPythonAgentConnectionModuleName, getPythonAgentConnectionProject, getPythonAgentConnectionProjectDir, PY_MCP_FAMILY_CONNECTIONS, resolveAgentFramework } from "../../../utils/agent-connection/agent-connection.js";
+import { addPythonDestructuredImport } from "../../../utils/ast.js";
+import { formatFilesInSubtree } from "../../../utils/format.js";
+import { addGeneratorMetricsIfApplicable } from "../../../utils/metrics.js";
+import { snakeCase } from "../../../utils/names.js";
+import { addDependencyToTargetIfNotPresent, getGeneratorInfo, readProjectConfigurationUnqualified } from "../../../utils/nx.js";
+import { addDependenciesToPyProjectToml, addWorkspaceDependencyToPyProject } from "../../../utils/py.js";
+export const PY_AGENT_GATEWAY_CONNECTION_GENERATOR_INFO = getGeneratorInfo(import.meta.filename);
+export const pyAgentGatewayConnectionGenerator = async (tree, options)=>{
+    const sourceProject = readProjectConfigurationUnqualified(tree, options.sourceProject);
+    const targetProject = readProjectConfigurationUnqualified(tree, options.targetProject);
     const agentComponent = options.sourceComponent;
     if (!agentComponent) {
         throw new Error('sourceComponent must be provided for py#agent -> agentcore-gateway connections');
     }
-    const gateway = (0, generator_1.readAgentCoreGatewayMetadata)(targetProject);
+    const gateway = readAgentCoreGatewayMetadata(targetProject);
     if (gateway.protocol !== 'mcp') {
         throw new Error(`Gateway '${gateway.name}' has protocol='${gateway.protocol}'. Only MCP-protocol gateways are supported.`);
     }
@@ -33,81 +29,88 @@ const pyAgentGatewayConnectionGenerator = async (tree, options) => {
         throw new Error(`Agent '${agentComponent.name}' uses auth='${agentComponent.auth}'. Only IAM-authenticated agents can connect to IAM gateways.`);
     }
     const gatewayClassName = gateway.rc;
-    const gatewaySnakeCase = (0, names_1.snakeCase)(gatewayClassName);
-    const gatewayKebabCase = gatewaySnakeCase.replace(/_/g, '-');
-    const gatewayServeTargetName = `${gatewayKebabCase}-serve`;
-    const gatewayServeLocalTargetName = `${gatewayKebabCase}-serve-local`;
+    const gatewaySnakeCase = snakeCase(gatewayClassName);
+    // A gateway is its own standalone project, so it exposes plain serve / dev.
+    const gatewayServeTargetName = 'serve';
+    const gatewayDevTargetName = 'dev';
     // The source agent's framework selects the client shape, dependencies and the
     // agent.py transform — see PY_MCP_FAMILY_CONNECTIONS (keyed by framework, not a
     // boolean, so a third framework is an additive entry there). A Gateway exposes
     // its tools over MCP, so it reuses the same wiring as an MCP server connection.
-    const framework = (0, agent_connection_1.resolveAgentFramework)(agentComponent.framework);
-    const connection = agent_connection_1.PY_MCP_FAMILY_CONNECTIONS[framework];
-    await (0, agent_connection_1.ensurePythonAgentConnectionProject)(tree);
-    await (0, agent_connection_1.addPythonCoreClient)(tree, 'gateway', framework);
-    const agentConnectionProjectDir = (0, agent_connection_1.getPythonAgentConnectionProjectDir)(tree);
-    const agentConnectionModuleName = (0, agent_connection_1.getPythonAgentConnectionModuleName)(tree);
-    const agentConnectionPackageName = (0, agent_connection_1.getPythonAgentConnectionPackageName)(tree);
+    const framework = resolveAgentFramework(agentComponent.framework);
+    const connection = PY_MCP_FAMILY_CONNECTIONS[framework];
+    await ensurePythonAgentConnectionProject(tree);
+    await addPythonCoreClient(tree, 'gateway', framework);
+    const agentConnectionProjectDir = getPythonAgentConnectionProjectDir(tree);
+    const agentConnectionModuleName = getPythonAgentConnectionModuleName(tree);
     // Shared MCP transport + signed httpx auth deps, plus whatever extra deps the
     // framework's MCP client needs (e.g. langchain-mcp-adapters for LangChain).
-    (0, py_1.addDependenciesToPyProjectToml)(tree, agentConnectionProjectDir, [
+    addDependenciesToPyProjectToml(tree, agentConnectionProjectDir, [
         'boto3',
         'httpx',
         'mcp',
-        ...connection.deps,
+        ...connection.deps
     ]);
-    const appDir = (0, devkit_1.joinPathFragments)(agentConnectionProjectDir, agentConnectionModuleName, 'app');
+    const appDir = joinPathFragments(agentConnectionProjectDir, agentConnectionModuleName, 'app');
     // Local mode points at the gateway project's local gateway port.
-    (0, devkit_1.generateFiles)(tree, (0, devkit_1.joinPathFragments)(__dirname, 'files', connection.appTemplateSubdir), appDir, {
+    generateFiles(tree, joinPathFragments(import.meta.dirname, 'files', connection.appTemplateSubdir), appDir, {
         gatewaySnakeCase,
         gatewayClassName,
         agentConnectionModuleName,
-        gatewayPort: gateway.port,
-    }, { overwriteStrategy: devkit_1.OverwriteStrategy.KeepExisting });
-    const appInitPath = (0, devkit_1.joinPathFragments)(appDir, '__init__.py');
+        gatewayPort: gateway.port
+    }, {
+        overwriteStrategy: OverwriteStrategy.KeepExisting
+    });
+    const appInitPath = joinPathFragments(appDir, '__init__.py');
     if (!tree.exists(appInitPath)) {
         tree.write(appInitPath, '');
     }
-    const moduleInitPath = (0, devkit_1.joinPathFragments)(agentConnectionProjectDir, agentConnectionModuleName, '__init__.py');
-    await (0, agent_connection_1.addPythonReExport)(tree, moduleInitPath, `.app.${gatewaySnakeCase}_client_${connection.clientModuleSuffix}`, `${gatewayClassName}Client${connection.clientClassSuffix}`);
-    const agentSourceDir = (0, devkit_1.joinPathFragments)(sourceProject.root, agentComponent.path ?? 'src');
-    const agentFilePath = (0, devkit_1.joinPathFragments)(agentSourceDir, 'agent.py');
+    const moduleInitPath = joinPathFragments(agentConnectionProjectDir, agentConnectionModuleName, '__init__.py');
+    await addPythonReExport(tree, moduleInitPath, `.app.${gatewaySnakeCase}_client_${connection.clientModuleSuffix}`, `${gatewayClassName}Client${connection.clientClassSuffix}`);
+    const agentSourceDir = joinPathFragments(sourceProject.root, agentComponent.path ?? 'src');
+    const agentFilePath = joinPathFragments(agentSourceDir, 'agent.py');
     if (tree.exists(agentFilePath)) {
         const clientClassName = `${gatewayClassName}Client${connection.clientClassSuffix}`;
         const clientVarName = gatewaySnakeCase;
-        await (0, ast_1.addPythonDestructuredImport)(tree, agentFilePath, [clientClassName], agentConnectionModuleName);
+        await addPythonDestructuredImport(tree, agentFilePath, [
+            clientClassName
+        ], agentConnectionModuleName);
         await connection.wireClientIntoAgent(tree, agentFilePath, clientClassName, clientVarName);
     }
-    (0, py_1.addWorkspaceDependencyToPyProject)(tree, sourceProject.root, agentConnectionPackageName);
+    addWorkspaceDependencyToPyProject(tree, sourceProject, getPythonAgentConnectionProject(tree));
     const agentName = agentComponent.name ?? 'agent';
     const serveTargetName = `${agentName}-serve`;
-    const serveLocalTargetName = `${agentName}-serve-local`;
+    const devTargetName = `${agentName}-dev`;
     let projectConfigChanged = false;
     if (sourceProject.targets?.[serveTargetName]) {
-        (0, nx_1.addDependencyToTargetIfNotPresent)(sourceProject, serveTargetName, {
-            projects: [targetProject.name],
-            target: gatewayServeTargetName,
+        addDependencyToTargetIfNotPresent(sourceProject, serveTargetName, {
+            projects: [
+                targetProject.name
+            ],
+            target: gatewayServeTargetName
         });
         projectConfigChanged = true;
     }
-    if (sourceProject.targets?.[serveLocalTargetName]) {
-        (0, nx_1.addDependencyToTargetIfNotPresent)(sourceProject, serveLocalTargetName, {
-            projects: [targetProject.name],
-            target: gatewayServeLocalTargetName,
+    if (sourceProject.targets?.[devTargetName]) {
+        addDependencyToTargetIfNotPresent(sourceProject, devTargetName, {
+            projects: [
+                targetProject.name
+            ],
+            target: gatewayDevTargetName
         });
         projectConfigChanged = true;
     }
     if (projectConfigChanged) {
-        (0, devkit_1.updateProjectConfiguration)(tree, sourceProject.name, sourceProject);
+        updateProjectConfiguration(tree, sourceProject.name, sourceProject);
     }
-    await (0, metrics_1.addGeneratorMetricsIfApplicable)(tree, [
-        exports.PY_AGENT_GATEWAY_CONNECTION_GENERATOR_INFO,
+    await addGeneratorMetricsIfApplicable(tree, [
+        PY_AGENT_GATEWAY_CONNECTION_GENERATOR_INFO
     ]);
-    await (0, format_1.formatFilesInSubtree)(tree);
-    return () => {
-        (0, devkit_1.installPackagesTask)(tree);
+    await formatFilesInSubtree(tree);
+    return ()=>{
+        installPackagesTask(tree);
     };
 };
-exports.pyAgentGatewayConnectionGenerator = pyAgentGatewayConnectionGenerator;
-exports.default = exports.pyAgentGatewayConnectionGenerator;
+export default pyAgentGatewayConnectionGenerator;
+
 //# sourceMappingURL=generator.js.map