@aws/nx-plugin 1.0.0-rc.71 → 1.0.0-rc.73

package/src/migrations/v1.0.0-rc.72/0001-py-agent-session-management-support/migration.js

@@ -0,0 +1,308 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */ import { readFileSync } from "node:fs";
+import { join } from "node:path";
+import { getProjects, joinPathFragments, visitNotIgnoredFiles } from "@nx/devkit";
+import { PY_AGENT_GENERATOR_INFO } from "../../../py/agent/generator.js";
+import { addPyDependencies } from "../../../utils/add-dependencies.js";
+import { addPythonReExport, getPythonAgentConnectionModuleName, getPythonAgentConnectionProjectDir } from "../../../utils/agent-connection/agent-connection.js";
+import { applyGritQL, captureGritQLVariable, matchGritQL } from "../../../utils/ast.js";
+import { declareDependencies } from "../../../utils/declared-dependencies.js";
+import { formatFilesInSubtree } from "../../../utils/format.js";
+import { kebabCase } from "../../../utils/names.js";
+import { getRelativePathToRootByDirectory, toProjectRelativePath } from "../../../utils/paths.js";
+/**
+ * Add session management support to py#agent.
+ *
+ * How to write a migration:
+ * - https://nx.dev/docs/kb/migration-generators
+ * - What `nextSteps` means: https://nx.dev/docs/reference/devkit/MigrationReturnObject
+ *
+ * Guardrails:
+ * - Pattern-match before writing: skip files that have diverged from the shape
+ *   your generators produce and report them via `nextSteps`, rather than
+ *   clobbering the user's changes.
+ * - Idempotent: re-running must be a no-op.
+ * - Format what you write: finish with `formatFilesInSubtree` so the files
+ *   this migration wrote are formatted correctly.
+ */ // Captures the agent-connection module a file imports `log_model_errors` (or
+// `session_id_context`) from, and gates every rewrite below on the import
+// actually being present. Generic over `$names` since a connection generator
+// may have merged its own import into the same line.
+const LOG_MODEL_ERRORS_IMPORT_CAPTURE_PATTERN = 'language python\n`from $mod import $names` where { $names <: contains `log_model_errors` }';
+const SESSION_ID_CONTEXT_IMPORT_CAPTURE_PATTERN = 'language python\n`from $mod import $names` where { $names <: contains `session_id_context` }';
+const DEPENDENCIES = declareDependencies()({
+    py: [
+        {
+            name: 'langgraph-checkpoint-sqlite'
+        },
+        {
+            name: 'aiosqlite'
+        }
+    ],
+    ts: []
+});
+const LANGCHAIN_AGENT_MATCH_PATTERN = 'language python\n`create_agent($args)` where { $args <: contains `system_prompt=$_`, $args <: contains `checkpointer=InMemorySaver()` }';
+const LANGCHAIN_CHECKPOINTER_PATTERN = 'language python\n`checkpointer=InMemorySaver()` => `checkpointer=get_checkpointer()`';
+const LANGCHAIN_IN_MEMORY_IMPORT_PATTERN = 'language python\n`from langgraph.checkpoint.memory import InMemorySaver` => .';
+const langChainSessionImportPattern = 'language python\n`from langchain.agents import create_agent` => raw`from langchain.agents import create_agent\n\nfrom .session import get_checkpointer`';
+// Every strands agent.py already hooks log_model_errors regardless of
+// protocol (hooks, unlike session_manager, carry over per-thread for AG-UI
+// without special wiring — see StrandsAgent's kwargs-forwarding), so
+// log_tool_errors is retrofitted the same way for every protocol.
+const LOG_TOOL_ERRORS_HOOKS_PATTERN = 'language python\n`hooks=[$hooks]` where { $hooks <: contains `log_model_errors`, $hooks <: not contains `log_tool_errors` }';
+const logToolErrorsImportPattern = (mod)=>`language python\n\`from contextlib import contextmanager\` => raw\`from contextlib import contextmanager\nfrom ${mod} import log_tool_errors\``;
+const LOG_TOOL_ERRORS_WRITE_PATTERN = 'language python\n`hooks=[log_model_errors]` => `hooks=[log_model_errors, log_tool_errors]`';
+// Existing agent-connection projects predate tool_errors_strands.py. This has
+// no EJS templating, so it's read verbatim from the generator's own template
+// rather than kept as a second, hand-copied source of truth.
+const TOOL_ERRORS_STRANDS_PY_CONTENT = readFileSync(join(import.meta.dirname, '../../../utils/agent-connection/files/py-core-strands/base/tool_errors_strands.py.template'), 'utf-8');
+/**
+ * Ensures the shared Python agent-connection project has tool_errors_strands.py
+ * and re-exports log_tool_errors, backfilling projects generated before this
+ * helper existed. Returns false (and leaves the tree untouched) if the
+ * agent-connection project itself can't be found.
+ */ async function ensureToolErrorsBase(tree) {
+    const projectDir = getPythonAgentConnectionProjectDir(tree);
+    if (!tree.exists(joinPathFragments(projectDir, 'project.json'))) {
+        return false;
+    }
+    const moduleDir = joinPathFragments(projectDir, getPythonAgentConnectionModuleName(tree));
+    const toolErrorsPath = joinPathFragments(moduleDir, 'core', 'tool_errors_strands.py');
+    if (!tree.exists(toolErrorsPath)) {
+        tree.write(toolErrorsPath, TOOL_ERRORS_STRANDS_PY_CONTENT);
+    }
+    await addPythonReExport(tree, joinPathFragments(moduleDir, '__init__.py'), '.core.tool_errors_strands', 'log_tool_errors');
+    return true;
+}
+// HTTP/A2A agents build one Agent per session (via with_session_id), so
+// wiring `session_manager` directly into the constructor is safe. Anchored
+// on a `tools=` keyword argument so this only matches an inline kwargs call
+// (as the generator produces), not e.g. `Agent(**agent_kwargs)`.
+const AGENT_PY_CONSTRUCTOR_MATCH_PATTERN = 'language python\n`Agent($args)` where { $args <: contains `tools=$_` }';
+const AGENT_PY_SESSION_MANAGER_PATTERN = 'language python\n`yield Agent($args)` => `yield Agent(session_manager=get_session_manager(), $args)` where { $args <: not contains `session_manager=$_` }';
+const AGENT_PY_SESSION_IMPORT_PATTERN = 'language python\n`from contextlib import contextmanager` => raw`from contextlib import contextmanager\n\nfrom .session import get_session_manager`';
+// AG-UI's StrandsAgent adapter needs a session_manager_provider (called once
+// per thread_id) rather than a plain session_manager.
+const AGUI_MAIN_IMPORT_PATTERN = 'language python\n`from ag_ui_strands import $names` => `from ag_ui_strands import StrandsAgentConfig, $names` where { $names <: contains `StrandsAgent`, $names <: not contains `StrandsAgentConfig` }';
+const AGUI_MAIN_CONSTRUCTOR_MATCH_PATTERN = 'language python\n`StrandsAgent($args)` where { $args <: contains `agent=$_` }';
+const AGUI_MAIN_CONSTRUCTOR_PATTERN = `${AGUI_MAIN_CONSTRUCTOR_MATCH_PATTERN} => \`StrandsAgent(config=StrandsAgentConfig(session_manager_provider=lambda _input_data: get_session_manager()), $args)\` where { $args <: not contains \`config=\` }`;
+const AGUI_MAIN_SESSION_IMPORT_PATTERN = 'language python\n`from .agent import get_agent` => raw`from .agent import get_agent\nfrom .session import get_session_manager` where { $program <: not contains `from .session import get_session_manager` }';
+// Existing agents predate session.py, so there's no prior storage to
+// preserve — default to in-memory, mirroring the ts#agent migration.
+const legacySessionManagerContent = (agentConnectionModule, localSessionsDir)=>`import os
+
+from strands.session import FileSessionManager, SessionManager
+
+from ${agentConnectionModule} import get_current_session_id
+
+
+def get_session_manager() -> SessionManager | None:
+    """Returns a SessionManager for persisting conversation state across
+    invocations. Local development always uses local file storage for
+    convenience, regardless of the configured session option. Without a
+    configured session option, conversation state is kept in memory only and
+    does not survive process restarts.
+    """
+    session_id = get_current_session_id()
+    if not session_id:
+        raise RuntimeError(
+            "No current session id — cannot resolve a SessionManager outside of a request scope."
+        )
+    if os.environ.get("LOCAL_DEV") == "true":
+        return FileSessionManager(session_id=session_id, storage_dir="${localSessionsDir}")
+    return None
+`;
+const legacyLangChainSessionContent = (localSessionsDir)=>`import os
+
+import aiosqlite
+from langgraph.checkpoint.base import BaseCheckpointSaver
+from langgraph.checkpoint.memory import InMemorySaver
+from langgraph.checkpoint.sqlite.aio import AsyncSqliteSaver
+
+
+def get_checkpointer() -> BaseCheckpointSaver:
+    """Returns the checkpointer used to persist conversation state."""
+    if os.environ.get("LOCAL_DEV") == "true":
+        os.makedirs("${localSessionsDir}", exist_ok=True)
+        conn = aiosqlite.connect(os.path.join("${localSessionsDir}", "checkpoints.sqlite"))
+        return AsyncSqliteSaver(conn)
+    return InMemorySaver()
+`;
+/** The Nx project owning `dirPath`, if any (the longest-matching project.root). */ const findOwningProject = (tree, dirPath)=>{
+    let best;
+    for (const project of getProjects(tree).values()){
+        if ((dirPath === project.root || dirPath.startsWith(`${project.root}/`)) && (!best || project.root.length > best.root.length)) {
+            best = project;
+        }
+    }
+    return best;
+};
+/** This agent's metadata, accepting both current and legacy component paths. */ const findAgentComponentMetadata = (project, dirRelativeToProjectRoot)=>project.metadata?.components?.find((component)=>component.generator === PY_AGENT_GENERATOR_INFO.id && (component.path === dirRelativeToProjectRoot || component.path === `${dirRelativeToProjectRoot}/agent.py`));
+/**
+ * This agent's real kebab-case name — the generator's own target-prefix
+ * (ComponentMetadata's `name` field), used directly rather than
+ * reconstructing it by kebab-casing the class name (`rc`). That round trip
+ * loses word boundaries for names with single-letter segments, e.g.
+ * `kebabCase("OldSHttp")` => "old-shttp", not the real "old-s-http".
+ *
+ * When no `--name` was given at generation time, the generator falls back
+ * to `${project}-agent` for the real name, but records a literal 'agent' as
+ * the target-prefix — indistinguishable from an agent explicitly named
+ * "agent" — so re-derive the fallback from the project name in that case,
+ * exactly as the generator itself does.
+ */ const agentTmpNameFor = (project, dir)=>{
+    const dirRelativeToProjectRoot = toProjectRelativePath(project, dir);
+    const targetPrefix = findAgentComponentMetadata(project, dirRelativeToProjectRoot)?.name;
+    if (!targetPrefix) return undefined;
+    if (targetPrefix !== 'agent') return targetPrefix;
+    return kebabCase(`${project.name.split('.').pop() ?? project.name}-agent`);
+};
+/** This agent's protocol + framework, from its ComponentMetadata entry. */ const agentComponentFor = (project, dir)=>{
+    const dirRelativeToProjectRoot = toProjectRelativePath(project, dir);
+    return findAgentComponentMetadata(project, dirRelativeToProjectRoot);
+};
+/** The relative path from `projectRoot` up to this agent's workspace-root-level local session storage. */ const localSessionsDirFor = (projectRoot, agentTmpName, framework = 'strands')=>joinPathFragments(getRelativePathToRootByDirectory(projectRoot), `tmp/agents/${framework}/${agentTmpName}`);
+export default async function migration(tree) {
+    const nextSteps = [];
+    const filePaths = [];
+    visitNotIgnoredFiles(tree, '', (filePath)=>filePaths.push(filePath));
+    for (const filePath of filePaths){
+        if (filePath.endsWith('/agent.py')) {
+            // log_tool_errors applies to every protocol (unlike session_manager),
+            // so retrofit it independently of whether the owning project can be
+            // resolved below. LOG_TOOL_ERRORS_HOOKS_PATTERN's own `where` clause
+            // (contains log_model_errors, not already containing log_tool_errors)
+            // doubles as the "still needs retrofitting" check.
+            if (await matchGritQL(tree, filePath, LOG_TOOL_ERRORS_HOOKS_PATTERN)) {
+                const mod = await captureGritQLVariable(tree, filePath, LOG_MODEL_ERRORS_IMPORT_CAPTURE_PATTERN, 'mod');
+                if (await ensureToolErrorsBase(tree) && mod) {
+                    await applyGritQL(tree, filePath, logToolErrorsImportPattern(mod));
+                    await applyGritQL(tree, filePath, LOG_TOOL_ERRORS_WRITE_PATTERN);
+                } else {
+                    nextSteps.push(`${filePath}: could not find the shared agent-connection project — manually add \`log_tool_errors\` to the hooks list (see the py#agent generator's template).`);
+                }
+            }
+            const dir = filePath.split('/').slice(0, -1).join('/');
+            const project = findOwningProject(tree, dir);
+            // Without a registered project there's no ComponentMetadata, so this
+            // agent's protocol/framework/agent-connection module can't be
+            // resolved — and a pre-migration AG-UI agent.py is textually identical
+            // to an HTTP/A2A one, so there's nothing left to safely guess from.
+            if (!project) {
+                if (await matchGritQL(tree, filePath, LOG_MODEL_ERRORS_IMPORT_CAPTURE_PATTERN)) {
+                    nextSteps.push(`${filePath}: could not determine the project root — manually verify whether this is an AG-UI or HTTP/A2A Strands agent and wire session_manager in accordingly (see the py#agent generator's template).`);
+                }
+                continue;
+            }
+            const component = agentComponentFor(project, dir);
+            if (component?.framework === 'langchain') {
+                const content = tree.read(filePath, 'utf-8') ?? '';
+                if (content.includes('checkpointer=get_checkpointer()')) {
+                    continue;
+                }
+                if (!await matchGritQL(tree, filePath, LANGCHAIN_AGENT_MATCH_PATTERN)) {
+                    nextSteps.push(`${filePath}: the create_agent call has diverged from the generated shape — manually create ${dir}/session.py and pass \`checkpointer=get_checkpointer()\` to \`create_agent\` (see the py#agent generator's template).`);
+                    continue;
+                }
+                const agentName = agentTmpNameFor(project, dir);
+                const sessionPath = `${dir}/session.py`;
+                if (!agentName) {
+                    nextSteps.push(`${filePath}: could not determine this agent's name from its ComponentMetadata — manually create ${sessionPath} and pass \`checkpointer=get_checkpointer()\` to \`create_agent\` (see the py#agent generator's template).`);
+                    continue;
+                }
+                if (!tree.exists(sessionPath)) {
+                    tree.write(sessionPath, legacyLangChainSessionContent(localSessionsDirFor(project.root, agentName, 'langchain')));
+                }
+                addPyDependencies(tree, DEPENDENCIES, {
+                    projectRoot: project.root
+                });
+                await applyGritQL(tree, filePath, langChainSessionImportPattern);
+                await applyGritQL(tree, filePath, LANGCHAIN_IN_MEMORY_IMPORT_PATTERN);
+                await applyGritQL(tree, filePath, LANGCHAIN_CHECKPOINTER_PATTERN);
+                const migratedContent = tree.read(filePath, 'utf-8') ?? '';
+                if (!migratedContent.includes('from .session import get_checkpointer') || !migratedContent.includes('checkpointer=get_checkpointer()')) {
+                    nextSteps.push(`${filePath}: session.py was created but the checkpointer could not be wired automatically — import \`get_checkpointer\` from it and pass \`checkpointer=get_checkpointer()\` to \`create_agent\`.`);
+                }
+                continue;
+            }
+            // AG-UI wires session_manager_provider on the StrandsAgent adapter in
+            // main.py instead — handled by the branch below.
+            if (component?.protocol === 'ag-ui') {
+                continue;
+            }
+            if (!component?.protocol) {
+                if (await matchGritQL(tree, filePath, LOG_MODEL_ERRORS_IMPORT_CAPTURE_PATTERN)) {
+                    nextSteps.push(`${filePath}: could not determine this agent's protocol/framework from its ComponentMetadata — manually verify whether it's AG-UI or HTTP/A2A Strands and wire session_manager in accordingly (see the py#agent generator's template).`);
+                }
+                continue;
+            }
+            // HTTP/A2A: wire session_manager into the Agent constructor and create
+            // the sibling session.py if it doesn't exist yet.
+            if (!await matchGritQL(tree, filePath, LOG_MODEL_ERRORS_IMPORT_CAPTURE_PATTERN)) {
+                continue;
+            }
+            const sessionPath = `${dir}/session.py`;
+            if (!await matchGritQL(tree, filePath, AGENT_PY_CONSTRUCTOR_MATCH_PATTERN)) {
+                nextSteps.push(`${filePath}: the Agent is not constructed with an inline \`Agent(...)\` call, so session_manager could not be wired in automatically. Manually add \`session_manager=get_session_manager()\` to its constructor (see the py#agent generator's template), creating ${sessionPath} first if it doesn't already exist.`);
+                continue;
+            }
+            if (!tree.exists(sessionPath)) {
+                const mod = await captureGritQLVariable(tree, filePath, LOG_MODEL_ERRORS_IMPORT_CAPTURE_PATTERN, 'mod');
+                const agentName = agentTmpNameFor(project, dir);
+                if (mod && agentName) {
+                    tree.write(sessionPath, legacySessionManagerContent(mod, localSessionsDirFor(project.root, agentName)));
+                } else {
+                    nextSteps.push(`${filePath}: could not determine the agent-connection module or this agent's name from its ComponentMetadata — manually create ${sessionPath} (see the py#agent generator's template) and wire \`session_manager=get_session_manager()\` into the Agent constructor.`);
+                    continue;
+                }
+            }
+            const agentContent = tree.read(filePath, 'utf-8') ?? '';
+            if (!agentContent.includes('from .session import get_session_manager')) {
+                await applyGritQL(tree, filePath, AGENT_PY_SESSION_IMPORT_PATTERN);
+                await applyGritQL(tree, filePath, AGENT_PY_SESSION_MANAGER_PATTERN);
+            }
+            if (!(tree.read(filePath, 'utf-8') ?? '').includes('get_session_manager')) {
+                nextSteps.push(`${filePath}: found ${sessionPath} but couldn't confirm the get_session_manager import — wire \`session_manager=get_session_manager()\` into the Agent constructor manually.`);
+                continue;
+            }
+            continue;
+        }
+        if (filePath.endsWith('/main.py') && (tree.read(filePath, 'utf-8') ?? '').includes('ag_ui_strands')) {
+            const alreadyWired = (tree.read(filePath, 'utf-8') ?? '').includes('StrandsAgentConfig');
+            if (!alreadyWired) {
+                const needsWiring = await matchGritQL(tree, filePath, AGUI_MAIN_CONSTRUCTOR_MATCH_PATTERN);
+                if (!needsWiring) {
+                    nextSteps.push(`${filePath}: the StrandsAgent constructor has diverged from the generated shape - left as-is. Manually add \`config=StrandsAgentConfig(session_manager_provider=lambda _input_data: get_session_manager())\` to it (see the py#agent generator's template).`);
+                    continue;
+                }
+                await applyGritQL(tree, filePath, AGUI_MAIN_IMPORT_PATTERN);
+                await applyGritQL(tree, filePath, AGUI_MAIN_CONSTRUCTOR_PATTERN);
+            }
+            const dir = filePath.split('/').slice(0, -1).join('/');
+            const sessionPath = `${dir}/session.py`;
+            if (!tree.exists(sessionPath)) {
+                const project = findOwningProject(tree, dir);
+                const mod = await captureGritQLVariable(tree, filePath, SESSION_ID_CONTEXT_IMPORT_CAPTURE_PATTERN, 'mod');
+                const agentName = project && agentTmpNameFor(project, dir);
+                if (project && mod && agentName) {
+                    tree.write(sessionPath, legacySessionManagerContent(mod, localSessionsDirFor(project.root, agentName)));
+                } else {
+                    nextSteps.push(`${filePath}: could not determine the project root, agent-connection module, or this agent's name from its ComponentMetadata — manually create ${sessionPath} (see the py#agent generator's template) and wire \`config=StrandsAgentConfig(session_manager_provider=lambda _input_data: get_session_manager())\` into the StrandsAgent constructor.`);
+                    continue;
+                }
+            }
+            await applyGritQL(tree, filePath, AGUI_MAIN_SESSION_IMPORT_PATTERN);
+            if (!(tree.read(filePath, 'utf-8') ?? '').includes('get_session_manager')) {
+                nextSteps.push(`${filePath}: found ${sessionPath} but couldn't confirm the get_session_manager import — wire it into the StrandsAgent constructor manually.`);
+            }
+        }
+    }
+    await formatFilesInSubtree(tree);
+    return {
+        nextSteps
+    };
+}
+
+//# sourceMappingURL=migration.js.map

package/src/migrations/v1.0.0-rc.72/0001-py-agent-session-management-support/migration.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../../../../packages/nx-plugin/src/migrations/v1.0.0-rc.72/0001-py-agent-session-management-support/migration.ts"],"sourcesContent":["/**\n * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.\n * SPDX-License-Identifier: Apache-2.0\n */\nimport { readFileSync } from 'node:fs';\nimport { join } from 'node:path';\nimport {\n  getProjects,\n  joinPathFragments,\n  type MigrationReturnObject,\n  type ProjectConfiguration,\n  type Tree,\n  visitNotIgnoredFiles,\n} from '@nx/devkit';\nimport { PY_AGENT_GENERATOR_INFO } from '../../../py/agent/generator';\nimport { addPyDependencies } from '../../../utils/add-dependencies';\nimport {\n  addPythonReExport,\n  getPythonAgentConnectionModuleName,\n  getPythonAgentConnectionProjectDir,\n} from '../../../utils/agent-connection/agent-connection';\nimport {\n  applyGritQL,\n  captureGritQLVariable,\n  matchGritQL,\n} from '../../../utils/ast';\nimport { declareDependencies } from '../../../utils/declared-dependencies';\nimport { formatFilesInSubtree } from '../../../utils/format';\nimport { kebabCase } from '../../../utils/names';\nimport type { ComponentMetadata } from '../../../utils/nx';\nimport {\n  getRelativePathToRootByDirectory,\n  toProjectRelativePath,\n} from '../../../utils/paths';\n\n/**\n * Add session management support to py#agent.\n *\n * How to write a migration:\n * - https://nx.dev/docs/kb/migration-generators\n * - What `nextSteps` means: https://nx.dev/docs/reference/devkit/MigrationReturnObject\n *\n * Guardrails:\n * - Pattern-match before writing: skip files that have diverged from the shape\n *   your generators produce and report them via `nextSteps`, rather than\n *   clobbering the user's changes.\n * - Idempotent: re-running must be a no-op.\n * - Format what you write: finish with `formatFilesInSubtree` so the files\n *   this migration wrote are formatted correctly.\n */\n\n// Captures the agent-connection module a file imports `log_model_errors` (or\n// `session_id_context`) from, and gates every rewrite below on the import\n// actually being present. Generic over `$names` since a connection generator\n// may have merged its own import into the same line.\nconst LOG_MODEL_ERRORS_IMPORT_CAPTURE_PATTERN =\n  'language python\\n`from $mod import $names` where { $names <: contains `log_model_errors` }';\nconst SESSION_ID_CONTEXT_IMPORT_CAPTURE_PATTERN =\n  'language python\\n`from $mod import $names` where { $names <: contains `session_id_context` }';\n\nconst DEPENDENCIES = declareDependencies()({\n  py: [{ name: 'langgraph-checkpoint-sqlite' }, { name: 'aiosqlite' }],\n  ts: [],\n});\n\nconst LANGCHAIN_AGENT_MATCH_PATTERN =\n  'language python\\n`create_agent($args)` where { $args <: contains `system_prompt=$_`, $args <: contains `checkpointer=InMemorySaver()` }';\nconst LANGCHAIN_CHECKPOINTER_PATTERN =\n  'language python\\n`checkpointer=InMemorySaver()` => `checkpointer=get_checkpointer()`';\nconst LANGCHAIN_IN_MEMORY_IMPORT_PATTERN =\n  'language python\\n`from langgraph.checkpoint.memory import InMemorySaver` => .';\nconst langChainSessionImportPattern =\n  'language python\\n`from langchain.agents import create_agent` => raw`from langchain.agents import create_agent\\n\\nfrom .session import get_checkpointer`';\n\n// Every strands agent.py already hooks log_model_errors regardless of\n// protocol (hooks, unlike session_manager, carry over per-thread for AG-UI\n// without special wiring — see StrandsAgent's kwargs-forwarding), so\n// log_tool_errors is retrofitted the same way for every protocol.\nconst LOG_TOOL_ERRORS_HOOKS_PATTERN =\n  'language python\\n`hooks=[$hooks]` where { $hooks <: contains `log_model_errors`, $hooks <: not contains `log_tool_errors` }';\nconst logToolErrorsImportPattern = (mod: string) =>\n  `language python\\n\\`from contextlib import contextmanager\\` => raw\\`from contextlib import contextmanager\\nfrom ${mod} import log_tool_errors\\``;\nconst LOG_TOOL_ERRORS_WRITE_PATTERN =\n  'language python\\n`hooks=[log_model_errors]` => `hooks=[log_model_errors, log_tool_errors]`';\n\n// Existing agent-connection projects predate tool_errors_strands.py. This has\n// no EJS templating, so it's read verbatim from the generator's own template\n// rather than kept as a second, hand-copied source of truth.\nconst TOOL_ERRORS_STRANDS_PY_CONTENT = readFileSync(\n  join(\n    import.meta.dirname,\n    '../../../utils/agent-connection/files/py-core-strands/base/tool_errors_strands.py.template',\n  ),\n  'utf-8',\n);\n\n/**\n * Ensures the shared Python agent-connection project has tool_errors_strands.py\n * and re-exports log_tool_errors, backfilling projects generated before this\n * helper existed. Returns false (and leaves the tree untouched) if the\n * agent-connection project itself can't be found.\n */\nasync function ensureToolErrorsBase(tree: Tree): Promise<boolean> {\n  const projectDir = getPythonAgentConnectionProjectDir(tree);\n  if (!tree.exists(joinPathFragments(projectDir, 'project.json'))) {\n    return false;\n  }\n  const moduleDir = joinPathFragments(\n    projectDir,\n    getPythonAgentConnectionModuleName(tree),\n  );\n  const toolErrorsPath = joinPathFragments(\n    moduleDir,\n    'core',\n    'tool_errors_strands.py',\n  );\n  if (!tree.exists(toolErrorsPath)) {\n    tree.write(toolErrorsPath, TOOL_ERRORS_STRANDS_PY_CONTENT);\n  }\n  await addPythonReExport(\n    tree,\n    joinPathFragments(moduleDir, '__init__.py'),\n    '.core.tool_errors_strands',\n    'log_tool_errors',\n  );\n  return true;\n}\n\n// HTTP/A2A agents build one Agent per session (via with_session_id), so\n// wiring `session_manager` directly into the constructor is safe. Anchored\n// on a `tools=` keyword argument so this only matches an inline kwargs call\n// (as the generator produces), not e.g. `Agent(**agent_kwargs)`.\nconst AGENT_PY_CONSTRUCTOR_MATCH_PATTERN =\n  'language python\\n`Agent($args)` where { $args <: contains `tools=$_` }';\nconst AGENT_PY_SESSION_MANAGER_PATTERN =\n  'language python\\n`yield Agent($args)` => `yield Agent(session_manager=get_session_manager(), $args)` where { $args <: not contains `session_manager=$_` }';\nconst AGENT_PY_SESSION_IMPORT_PATTERN =\n  'language python\\n`from contextlib import contextmanager` => raw`from contextlib import contextmanager\\n\\nfrom .session import get_session_manager`';\n\n// AG-UI's StrandsAgent adapter needs a session_manager_provider (called once\n// per thread_id) rather than a plain session_manager.\nconst AGUI_MAIN_IMPORT_PATTERN =\n  'language python\\n`from ag_ui_strands import $names` => `from ag_ui_strands import StrandsAgentConfig, $names` where { $names <: contains `StrandsAgent`, $names <: not contains `StrandsAgentConfig` }';\nconst AGUI_MAIN_CONSTRUCTOR_MATCH_PATTERN =\n  'language python\\n`StrandsAgent($args)` where { $args <: contains `agent=$_` }';\nconst AGUI_MAIN_CONSTRUCTOR_PATTERN = `${AGUI_MAIN_CONSTRUCTOR_MATCH_PATTERN} => \\`StrandsAgent(config=StrandsAgentConfig(session_manager_provider=lambda _input_data: get_session_manager()), $args)\\` where { $args <: not contains \\`config=\\` }`;\nconst AGUI_MAIN_SESSION_IMPORT_PATTERN =\n  'language python\\n`from .agent import get_agent` => raw`from .agent import get_agent\\nfrom .session import get_session_manager` where { $program <: not contains `from .session import get_session_manager` }';\n\n// Existing agents predate session.py, so there's no prior storage to\n// preserve — default to in-memory, mirroring the ts#agent migration.\nconst legacySessionManagerContent = (\n  agentConnectionModule: string,\n  localSessionsDir: string,\n): string =>\n  `import os\n\nfrom strands.session import FileSessionManager, SessionManager\n\nfrom ${agentConnectionModule} import get_current_session_id\n\n\ndef get_session_manager() -> SessionManager | None:\n    \"\"\"Returns a SessionManager for persisting conversation state across\n    invocations. Local development always uses local file storage for\n    convenience, regardless of the configured session option. Without a\n    configured session option, conversation state is kept in memory only and\n    does not survive process restarts.\n    \"\"\"\n    session_id = get_current_session_id()\n    if not session_id:\n        raise RuntimeError(\n            \"No current session id — cannot resolve a SessionManager outside of a request scope.\"\n        )\n    if os.environ.get(\"LOCAL_DEV\") == \"true\":\n        return FileSessionManager(session_id=session_id, storage_dir=\"${localSessionsDir}\")\n    return None\n`;\n\nconst legacyLangChainSessionContent = (localSessionsDir: string): string =>\n  `import os\n\nimport aiosqlite\nfrom langgraph.checkpoint.base import BaseCheckpointSaver\nfrom langgraph.checkpoint.memory import InMemorySaver\nfrom langgraph.checkpoint.sqlite.aio import AsyncSqliteSaver\n\n\ndef get_checkpointer() -> BaseCheckpointSaver:\n    \"\"\"Returns the checkpointer used to persist conversation state.\"\"\"\n    if os.environ.get(\"LOCAL_DEV\") == \"true\":\n        os.makedirs(\"${localSessionsDir}\", exist_ok=True)\n        conn = aiosqlite.connect(os.path.join(\"${localSessionsDir}\", \"checkpoints.sqlite\"))\n        return AsyncSqliteSaver(conn)\n    return InMemorySaver()\n`;\n\n/** The Nx project owning `dirPath`, if any (the longest-matching project.root). */\nconst findOwningProject = (\n  tree: Tree,\n  dirPath: string,\n): ProjectConfiguration | undefined => {\n  let best: ProjectConfiguration | undefined;\n  for (const project of getProjects(tree).values()) {\n    if (\n      (dirPath === project.root || dirPath.startsWith(`${project.root}/`)) &&\n      (!best || project.root.length > best.root.length)\n    ) {\n      best = project;\n    }\n  }\n  return best;\n};\n\n/** This agent's metadata, accepting both current and legacy component paths. */\nconst findAgentComponentMetadata = (\n  project: ProjectConfiguration,\n  dirRelativeToProjectRoot: string,\n): ComponentMetadata | undefined =>\n  (project.metadata as { components?: ComponentMetadata[] })?.components?.find(\n    (component) =>\n      component.generator === PY_AGENT_GENERATOR_INFO.id &&\n      (component.path === dirRelativeToProjectRoot ||\n        component.path === `${dirRelativeToProjectRoot}/agent.py`),\n  );\n\n/**\n * This agent's real kebab-case name — the generator's own target-prefix\n * (ComponentMetadata's `name` field), used directly rather than\n * reconstructing it by kebab-casing the class name (`rc`). That round trip\n * loses word boundaries for names with single-letter segments, e.g.\n * `kebabCase(\"OldSHttp\")` => \"old-shttp\", not the real \"old-s-http\".\n *\n * When no `--name` was given at generation time, the generator falls back\n * to `${project}-agent` for the real name, but records a literal 'agent' as\n * the target-prefix — indistinguishable from an agent explicitly named\n * \"agent\" — so re-derive the fallback from the project name in that case,\n * exactly as the generator itself does.\n */\nconst agentTmpNameFor = (\n  project: ProjectConfiguration,\n  dir: string,\n): string | undefined => {\n  const dirRelativeToProjectRoot = toProjectRelativePath(project, dir);\n  const targetPrefix = findAgentComponentMetadata(\n    project,\n    dirRelativeToProjectRoot,\n  )?.name;\n  if (!targetPrefix) return undefined;\n  if (targetPrefix !== 'agent') return targetPrefix;\n  return kebabCase(`${project.name.split('.').pop() ?? project.name}-agent`);\n};\n\n/** This agent's protocol + framework, from its ComponentMetadata entry. */\nconst agentComponentFor = (\n  project: ProjectConfiguration,\n  dir: string,\n): ComponentMetadata | undefined => {\n  const dirRelativeToProjectRoot = toProjectRelativePath(project, dir);\n  return findAgentComponentMetadata(project, dirRelativeToProjectRoot);\n};\n\n/** The relative path from `projectRoot` up to this agent's workspace-root-level local session storage. */\nconst localSessionsDirFor = (\n  projectRoot: string,\n  agentTmpName: string,\n  framework: 'strands' | 'langchain' = 'strands',\n): string =>\n  joinPathFragments(\n    getRelativePathToRootByDirectory(projectRoot),\n    `tmp/agents/${framework}/${agentTmpName}`,\n  );\n\nexport default async function migration(\n  tree: Tree,\n): Promise<MigrationReturnObject> {\n  const nextSteps: string[] = [];\n\n  const filePaths: string[] = [];\n  visitNotIgnoredFiles(tree, '', (filePath) => filePaths.push(filePath));\n\n  for (const filePath of filePaths) {\n    if (filePath.endsWith('/agent.py')) {\n      // log_tool_errors applies to every protocol (unlike session_manager),\n      // so retrofit it independently of whether the owning project can be\n      // resolved below. LOG_TOOL_ERRORS_HOOKS_PATTERN's own `where` clause\n      // (contains log_model_errors, not already containing log_tool_errors)\n      // doubles as the \"still needs retrofitting\" check.\n      if (await matchGritQL(tree, filePath, LOG_TOOL_ERRORS_HOOKS_PATTERN)) {\n        const mod = await captureGritQLVariable(\n          tree,\n          filePath,\n          LOG_MODEL_ERRORS_IMPORT_CAPTURE_PATTERN,\n          'mod',\n        );\n        if ((await ensureToolErrorsBase(tree)) && mod) {\n          await applyGritQL(tree, filePath, logToolErrorsImportPattern(mod));\n          await applyGritQL(tree, filePath, LOG_TOOL_ERRORS_WRITE_PATTERN);\n        } else {\n          nextSteps.push(\n            `${filePath}: could not find the shared agent-connection project — manually add \\`log_tool_errors\\` to the hooks list (see the py#agent generator's template).`,\n          );\n        }\n      }\n\n      const dir = filePath.split('/').slice(0, -1).join('/');\n      const project = findOwningProject(tree, dir);\n\n      // Without a registered project there's no ComponentMetadata, so this\n      // agent's protocol/framework/agent-connection module can't be\n      // resolved — and a pre-migration AG-UI agent.py is textually identical\n      // to an HTTP/A2A one, so there's nothing left to safely guess from.\n      if (!project) {\n        if (\n          await matchGritQL(\n            tree,\n            filePath,\n            LOG_MODEL_ERRORS_IMPORT_CAPTURE_PATTERN,\n          )\n        ) {\n          nextSteps.push(\n            `${filePath}: could not determine the project root — manually verify whether this is an AG-UI or HTTP/A2A Strands agent and wire session_manager in accordingly (see the py#agent generator's template).`,\n          );\n        }\n        continue;\n      }\n\n      const component = agentComponentFor(project, dir);\n\n      if (component?.framework === 'langchain') {\n        const content = tree.read(filePath, 'utf-8') ?? '';\n        if (content.includes('checkpointer=get_checkpointer()')) {\n          continue;\n        }\n        if (\n          !(await matchGritQL(tree, filePath, LANGCHAIN_AGENT_MATCH_PATTERN))\n        ) {\n          nextSteps.push(\n            `${filePath}: the create_agent call has diverged from the generated shape — manually create ${dir}/session.py and pass \\`checkpointer=get_checkpointer()\\` to \\`create_agent\\` (see the py#agent generator's template).`,\n          );\n          continue;\n        }\n\n        const agentName = agentTmpNameFor(project, dir);\n        const sessionPath = `${dir}/session.py`;\n        if (!agentName) {\n          nextSteps.push(\n            `${filePath}: could not determine this agent's name from its ComponentMetadata — manually create ${sessionPath} and pass \\`checkpointer=get_checkpointer()\\` to \\`create_agent\\` (see the py#agent generator's template).`,\n          );\n          continue;\n        }\n\n        if (!tree.exists(sessionPath)) {\n          tree.write(\n            sessionPath,\n            legacyLangChainSessionContent(\n              localSessionsDirFor(project.root, agentName, 'langchain'),\n            ),\n          );\n        }\n        addPyDependencies(tree, DEPENDENCIES, { projectRoot: project.root });\n        await applyGritQL(tree, filePath, langChainSessionImportPattern);\n        await applyGritQL(tree, filePath, LANGCHAIN_IN_MEMORY_IMPORT_PATTERN);\n        await applyGritQL(tree, filePath, LANGCHAIN_CHECKPOINTER_PATTERN);\n        const migratedContent = tree.read(filePath, 'utf-8') ?? '';\n        if (\n          !migratedContent.includes('from .session import get_checkpointer') ||\n          !migratedContent.includes('checkpointer=get_checkpointer()')\n        ) {\n          nextSteps.push(\n            `${filePath}: session.py was created but the checkpointer could not be wired automatically — import \\`get_checkpointer\\` from it and pass \\`checkpointer=get_checkpointer()\\` to \\`create_agent\\`.`,\n          );\n        }\n        continue;\n      }\n\n      // AG-UI wires session_manager_provider on the StrandsAgent adapter in\n      // main.py instead — handled by the branch below.\n      if (component?.protocol === 'ag-ui') {\n        continue;\n      }\n\n      if (!component?.protocol) {\n        if (\n          await matchGritQL(\n            tree,\n            filePath,\n            LOG_MODEL_ERRORS_IMPORT_CAPTURE_PATTERN,\n          )\n        ) {\n          nextSteps.push(\n            `${filePath}: could not determine this agent's protocol/framework from its ComponentMetadata — manually verify whether it's AG-UI or HTTP/A2A Strands and wire session_manager in accordingly (see the py#agent generator's template).`,\n          );\n        }\n        continue;\n      }\n\n      // HTTP/A2A: wire session_manager into the Agent constructor and create\n      // the sibling session.py if it doesn't exist yet.\n      if (\n        !(await matchGritQL(\n          tree,\n          filePath,\n          LOG_MODEL_ERRORS_IMPORT_CAPTURE_PATTERN,\n        ))\n      ) {\n        continue;\n      }\n\n      const sessionPath = `${dir}/session.py`;\n\n      if (\n        !(await matchGritQL(tree, filePath, AGENT_PY_CONSTRUCTOR_MATCH_PATTERN))\n      ) {\n        nextSteps.push(\n          `${filePath}: the Agent is not constructed with an inline \\`Agent(...)\\` call, so session_manager could not be wired in automatically. Manually add \\`session_manager=get_session_manager()\\` to its constructor (see the py#agent generator's template), creating ${sessionPath} first if it doesn't already exist.`,\n        );\n        continue;\n      }\n\n      if (!tree.exists(sessionPath)) {\n        const mod = await captureGritQLVariable(\n          tree,\n          filePath,\n          LOG_MODEL_ERRORS_IMPORT_CAPTURE_PATTERN,\n          'mod',\n        );\n        const agentName = agentTmpNameFor(project, dir);\n\n        if (mod && agentName) {\n          tree.write(\n            sessionPath,\n            legacySessionManagerContent(\n              mod,\n              localSessionsDirFor(project.root, agentName),\n            ),\n          );\n        } else {\n          nextSteps.push(\n            `${filePath}: could not determine the agent-connection module or this agent's name from its ComponentMetadata — manually create ${sessionPath} (see the py#agent generator's template) and wire \\`session_manager=get_session_manager()\\` into the Agent constructor.`,\n          );\n          continue;\n        }\n      }\n\n      const agentContent = tree.read(filePath, 'utf-8') ?? '';\n      if (!agentContent.includes('from .session import get_session_manager')) {\n        await applyGritQL(tree, filePath, AGENT_PY_SESSION_IMPORT_PATTERN);\n        await applyGritQL(tree, filePath, AGENT_PY_SESSION_MANAGER_PATTERN);\n      }\n\n      if (\n        !(tree.read(filePath, 'utf-8') ?? '').includes('get_session_manager')\n      ) {\n        nextSteps.push(\n          `${filePath}: found ${sessionPath} but couldn't confirm the get_session_manager import — wire \\`session_manager=get_session_manager()\\` into the Agent constructor manually.`,\n        );\n        continue;\n      }\n\n      continue;\n    }\n\n    if (\n      filePath.endsWith('/main.py') &&\n      (tree.read(filePath, 'utf-8') ?? '').includes('ag_ui_strands')\n    ) {\n      const alreadyWired = (tree.read(filePath, 'utf-8') ?? '').includes(\n        'StrandsAgentConfig',\n      );\n\n      if (!alreadyWired) {\n        const needsWiring = await matchGritQL(\n          tree,\n          filePath,\n          AGUI_MAIN_CONSTRUCTOR_MATCH_PATTERN,\n        );\n        if (!needsWiring) {\n          nextSteps.push(\n            `${filePath}: the StrandsAgent constructor has diverged from the generated shape - left as-is. Manually add \\`config=StrandsAgentConfig(session_manager_provider=lambda _input_data: get_session_manager())\\` to it (see the py#agent generator's template).`,\n          );\n          continue;\n        }\n\n        await applyGritQL(tree, filePath, AGUI_MAIN_IMPORT_PATTERN);\n        await applyGritQL(tree, filePath, AGUI_MAIN_CONSTRUCTOR_PATTERN);\n      }\n\n      const dir = filePath.split('/').slice(0, -1).join('/');\n      const sessionPath = `${dir}/session.py`;\n\n      if (!tree.exists(sessionPath)) {\n        const project = findOwningProject(tree, dir);\n        const mod = await captureGritQLVariable(\n          tree,\n          filePath,\n          SESSION_ID_CONTEXT_IMPORT_CAPTURE_PATTERN,\n          'mod',\n        );\n        const agentName = project && agentTmpNameFor(project, dir);\n\n        if (project && mod && agentName) {\n          tree.write(\n            sessionPath,\n            legacySessionManagerContent(\n              mod,\n              localSessionsDirFor(project.root, agentName),\n            ),\n          );\n        } else {\n          nextSteps.push(\n            `${filePath}: could not determine the project root, agent-connection module, or this agent's name from its ComponentMetadata — manually create ${sessionPath} (see the py#agent generator's template) and wire \\`config=StrandsAgentConfig(session_manager_provider=lambda _input_data: get_session_manager())\\` into the StrandsAgent constructor.`,\n          );\n          continue;\n        }\n      }\n\n      await applyGritQL(tree, filePath, AGUI_MAIN_SESSION_IMPORT_PATTERN);\n\n      if (\n        !(tree.read(filePath, 'utf-8') ?? '').includes('get_session_manager')\n      ) {\n        nextSteps.push(\n          `${filePath}: found ${sessionPath} but couldn't confirm the get_session_manager import — wire it into the StrandsAgent constructor manually.`,\n        );\n      }\n    }\n  }\n\n  await formatFilesInSubtree(tree);\n\n  return { nextSteps };\n}\n"],"names":["readFileSync","join","getProjects","joinPathFragments","visitNotIgnoredFiles","PY_AGENT_GENERATOR_INFO","addPyDependencies","addPythonReExport","getPythonAgentConnectionModuleName","getPythonAgentConnectionProjectDir","applyGritQL","captureGritQLVariable","matchGritQL","declareDependencies","formatFilesInSubtree","kebabCase","getRelativePathToRootByDirectory","toProjectRelativePath","LOG_MODEL_ERRORS_IMPORT_CAPTURE_PATTERN","SESSION_ID_CONTEXT_IMPORT_CAPTURE_PATTERN","DEPENDENCIES","py","name","ts","LANGCHAIN_AGENT_MATCH_PATTERN","LANGCHAIN_CHECKPOINTER_PATTERN","LANGCHAIN_IN_MEMORY_IMPORT_PATTERN","langChainSessionImportPattern","LOG_TOOL_ERRORS_HOOKS_PATTERN","logToolErrorsImportPattern","mod","LOG_TOOL_ERRORS_WRITE_PATTERN","TOOL_ERRORS_STRANDS_PY_CONTENT","dirname","ensureToolErrorsBase","tree","projectDir","exists","moduleDir","toolErrorsPath","write","AGENT_PY_CONSTRUCTOR_MATCH_PATTERN","AGENT_PY_SESSION_MANAGER_PATTERN","AGENT_PY_SESSION_IMPORT_PATTERN","AGUI_MAIN_IMPORT_PATTERN","AGUI_MAIN_CONSTRUCTOR_MATCH_PATTERN","AGUI_MAIN_CONSTRUCTOR_PATTERN","AGUI_MAIN_SESSION_IMPORT_PATTERN","legacySessionManagerContent","agentConnectionModule","localSessionsDir","legacyLangChainSessionContent","findOwningProject","dirPath","best","project","values","root","startsWith","length","findAgentComponentMetadata","dirRelativeToProjectRoot","metadata","components","find","component","generator","id","path","agentTmpNameFor","dir","targetPrefix","undefined","split","pop","agentComponentFor","localSessionsDirFor","projectRoot","agentTmpName","framework","migration","nextSteps","filePaths","filePath","push","endsWith","slice","content","read","includes","agentName","sessionPath","migratedContent","protocol","agentContent","alreadyWired","needsWiring"],"mappings":"AAAA;;;CAGC,GACD,SAASA,YAAY,QAAQ,UAAU;AACvC,SAASC,IAAI,QAAQ,YAAY;AACjC,SACEC,WAAW,EACXC,iBAAiB,EAIjBC,oBAAoB,QACf,aAAa;AACpB,SAASC,uBAAuB,QAAQ,iCAA8B;AACtE,SAASC,iBAAiB,QAAQ,qCAAkC;AACpE,SACEC,iBAAiB,EACjBC,kCAAkC,EAClCC,kCAAkC,QAC7B,sDAAmD;AAC1D,SACEC,WAAW,EACXC,qBAAqB,EACrBC,WAAW,QACN,wBAAqB;AAC5B,SAASC,mBAAmB,QAAQ,0CAAuC;AAC3E,SAASC,oBAAoB,QAAQ,2BAAwB;AAC7D,SAASC,SAAS,QAAQ,0BAAuB;AAEjD,SACEC,gCAAgC,EAChCC,qBAAqB,QAChB,0BAAuB;AAE9B;;;;;;;;;;;;;;CAcC,GAED,6EAA6E;AAC7E,0EAA0E;AAC1E,6EAA6E;AAC7E,qDAAqD;AACrD,MAAMC,0CACJ;AACF,MAAMC,4CACJ;AAEF,MAAMC,eAAeP,sBAAsB;IACzCQ,IAAI;QAAC;YAAEC,MAAM;QAA8B;QAAG;YAAEA,MAAM;QAAY;KAAE;IACpEC,IAAI,EAAE;AACR;AAEA,MAAMC,gCACJ;AACF,MAAMC,iCACJ;AACF,MAAMC,qCACJ;AACF,MAAMC,gCACJ;AAEF,sEAAsE;AACtE,2EAA2E;AAC3E,qEAAqE;AACrE,kEAAkE;AAClE,MAAMC,gCACJ;AACF,MAAMC,6BAA6B,CAACC,MAClC,CAAC,+GAA+G,EAAEA,IAAI,yBAAyB,CAAC;AAClJ,MAAMC,gCACJ;AAEF,8EAA8E;AAC9E,6EAA6E;AAC7E,6DAA6D;AAC7D,MAAMC,iCAAiChC,aACrCC,KACE,YAAYgC,OAAO,EACnB,+FAEF;AAGF;;;;;CAKC,GACD,eAAeC,qBAAqBC,IAAU;IAC5C,MAAMC,aAAa3B,mCAAmC0B;IACtD,IAAI,CAACA,KAAKE,MAAM,CAAClC,kBAAkBiC,YAAY,kBAAkB;QAC/D,OAAO;IACT;IACA,MAAME,YAAYnC,kBAChBiC,YACA5B,mCAAmC2B;IAErC,MAAMI,iBAAiBpC,kBACrBmC,WACA,QACA;IAEF,IAAI,CAACH,KAAKE,MAAM,CAACE,iBAAiB;QAChCJ,KAAKK,KAAK,CAACD,gBAAgBP;IAC7B;IACA,MAAMzB,kBACJ4B,MACAhC,kBAAkBmC,WAAW,gBAC7B,6BACA;IAEF,OAAO;AACT;AAEA,wEAAwE;AACxE,2EAA2E;AAC3E,4EAA4E;AAC5E,iEAAiE;AACjE,MAAMG,qCACJ;AACF,MAAMC,mCACJ;AACF,MAAMC,kCACJ;AAEF,6EAA6E;AAC7E,sDAAsD;AACtD,MAAMC,2BACJ;AACF,MAAMC,sCACJ;AACF,MAAMC,gCAAgC,GAAGD,oCAAoC,sKAAsK,CAAC;AACpP,MAAME,mCACJ;AAEF,qEAAqE;AACrE,qEAAqE;AACrE,MAAMC,8BAA8B,CAClCC,uBACAC,mBAEA,CAAC;;;;KAIE,EAAED,sBAAsB;;;;;;;;;;;;;;;;sEAgByC,EAAEC,iBAAiB;;AAEzF,CAAC;AAED,MAAMC,gCAAgC,CAACD,mBACrC,CAAC;;;;;;;;;;;qBAWkB,EAAEA,iBAAiB;+CACO,EAAEA,iBAAiB;;;AAGlE,CAAC;AAED,iFAAiF,GACjF,MAAME,oBAAoB,CACxBjB,MACAkB;IAEA,IAAIC;IACJ,KAAK,MAAMC,WAAWrD,YAAYiC,MAAMqB,MAAM,GAAI;QAChD,IACE,AAACH,CAAAA,YAAYE,QAAQE,IAAI,IAAIJ,QAAQK,UAAU,CAAC,GAAGH,QAAQE,IAAI,CAAC,CAAC,CAAC,CAAA,KACjE,CAAA,CAACH,QAAQC,QAAQE,IAAI,CAACE,MAAM,GAAGL,KAAKG,IAAI,CAACE,MAAM,AAAD,GAC/C;YACAL,OAAOC;QACT;IACF;IACA,OAAOD;AACT;AAEA,8EAA8E,GAC9E,MAAMM,6BAA6B,CACjCL,SACAM,2BAECN,QAAQO,QAAQ,EAA2CC,YAAYC,KACtE,CAACC,YACCA,UAAUC,SAAS,KAAK7D,wBAAwB8D,EAAE,IACjDF,CAAAA,UAAUG,IAAI,KAAKP,4BAClBI,UAAUG,IAAI,KAAK,GAAGP,yBAAyB,SAAS,CAAC,AAAD;AAGhE;;;;;;;;;;;;CAYC,GACD,MAAMQ,kBAAkB,CACtBd,SACAe;IAEA,MAAMT,2BAA2B5C,sBAAsBsC,SAASe;IAChE,MAAMC,eAAeX,2BACnBL,SACAM,2BACCvC;IACH,IAAI,CAACiD,cAAc,OAAOC;IAC1B,IAAID,iBAAiB,SAAS,OAAOA;IACrC,OAAOxD,UAAU,GAAGwC,QAAQjC,IAAI,CAACmD,KAAK,CAAC,KAAKC,GAAG,MAAMnB,QAAQjC,IAAI,CAAC,MAAM,CAAC;AAC3E;AAEA,yEAAyE,GACzE,MAAMqD,oBAAoB,CACxBpB,SACAe;IAEA,MAAMT,2BAA2B5C,sBAAsBsC,SAASe;IAChE,OAAOV,2BAA2BL,SAASM;AAC7C;AAEA,wGAAwG,GACxG,MAAMe,sBAAsB,CAC1BC,aACAC,cACAC,YAAqC,SAAS,GAE9C5E,kBACEa,iCAAiC6D,cACjC,CAAC,WAAW,EAAEE,UAAU,CAAC,EAAED,cAAc;AAG7C,eAAe,eAAeE,UAC5B7C,IAAU;IAEV,MAAM8C,YAAsB,EAAE;IAE9B,MAAMC,YAAsB,EAAE;IAC9B9E,qBAAqB+B,MAAM,IAAI,CAACgD,WAAaD,UAAUE,IAAI,CAACD;IAE5D,KAAK,MAAMA,YAAYD,UAAW;QAChC,IAAIC,SAASE,QAAQ,CAAC,cAAc;YAClC,sEAAsE;YACtE,oEAAoE;YACpE,qEAAqE;YACrE,sEAAsE;YACtE,mDAAmD;YACnD,IAAI,MAAMzE,YAAYuB,MAAMgD,UAAUvD,gCAAgC;gBACpE,MAAME,MAAM,MAAMnB,sBAChBwB,MACAgD,UACAjE,yCACA;gBAEF,IAAI,AAAC,MAAMgB,qBAAqBC,SAAUL,KAAK;oBAC7C,MAAMpB,YAAYyB,MAAMgD,UAAUtD,2BAA2BC;oBAC7D,MAAMpB,YAAYyB,MAAMgD,UAAUpD;gBACpC,OAAO;oBACLkD,UAAUG,IAAI,CACZ,GAAGD,SAAS,kJAAkJ,CAAC;gBAEnK;YACF;YAEA,MAAMb,MAAMa,SAASV,KAAK,CAAC,KAAKa,KAAK,CAAC,GAAG,CAAC,GAAGrF,IAAI,CAAC;YAClD,MAAMsD,UAAUH,kBAAkBjB,MAAMmC;YAExC,qEAAqE;YACrE,8DAA8D;YAC9D,uEAAuE;YACvE,oEAAoE;YACpE,IAAI,CAACf,SAAS;gBACZ,IACE,MAAM3C,YACJuB,MACAgD,UACAjE,0CAEF;oBACA+D,UAAUG,IAAI,CACZ,GAAGD,SAAS,4LAA4L,CAAC;gBAE7M;gBACA;YACF;YAEA,MAAMlB,YAAYU,kBAAkBpB,SAASe;YAE7C,IAAIL,WAAWc,cAAc,aAAa;gBACxC,MAAMQ,UAAUpD,KAAKqD,IAAI,CAACL,UAAU,YAAY;gBAChD,IAAII,QAAQE,QAAQ,CAAC,oCAAoC;oBACvD;gBACF;gBACA,IACE,CAAE,MAAM7E,YAAYuB,MAAMgD,UAAU3D,gCACpC;oBACAyD,UAAUG,IAAI,CACZ,GAAGD,SAAS,gFAAgF,EAAEb,IAAI,qHAAqH,CAAC;oBAE1N;gBACF;gBAEA,MAAMoB,YAAYrB,gBAAgBd,SAASe;gBAC3C,MAAMqB,cAAc,GAAGrB,IAAI,WAAW,CAAC;gBACvC,IAAI,CAACoB,WAAW;oBACdT,UAAUG,IAAI,CACZ,GAAGD,SAAS,qFAAqF,EAAEQ,YAAY,0GAA0G,CAAC;oBAE5N;gBACF;gBAEA,IAAI,CAACxD,KAAKE,MAAM,CAACsD,cAAc;oBAC7BxD,KAAKK,KAAK,CACRmD,aACAxC,8BACEyB,oBAAoBrB,QAAQE,IAAI,EAAEiC,WAAW;gBAGnD;gBACApF,kBAAkB6B,MAAMf,cAAc;oBAAEyD,aAAatB,QAAQE,IAAI;gBAAC;gBAClE,MAAM/C,YAAYyB,MAAMgD,UAAUxD;gBAClC,MAAMjB,YAAYyB,MAAMgD,UAAUzD;gBAClC,MAAMhB,YAAYyB,MAAMgD,UAAU1D;gBAClC,MAAMmE,kBAAkBzD,KAAKqD,IAAI,CAACL,UAAU,YAAY;gBACxD,IACE,CAACS,gBAAgBH,QAAQ,CAAC,4CAC1B,CAACG,gBAAgBH,QAAQ,CAAC,oCAC1B;oBACAR,UAAUG,IAAI,CACZ,GAAGD,SAAS,sLAAsL,CAAC;gBAEvM;gBACA;YACF;YAEA,sEAAsE;YACtE,iDAAiD;YACjD,IAAIlB,WAAW4B,aAAa,SAAS;gBACnC;YACF;YAEA,IAAI,CAAC5B,WAAW4B,UAAU;gBACxB,IACE,MAAMjF,YACJuB,MACAgD,UACAjE,0CAEF;oBACA+D,UAAUG,IAAI,CACZ,GAAGD,SAAS,0NAA0N,CAAC;gBAE3O;gBACA;YACF;YAEA,uEAAuE;YACvE,kDAAkD;YAClD,IACE,CAAE,MAAMvE,YACNuB,MACAgD,UACAjE,0CAEF;gBACA;YACF;YAEA,MAAMyE,cAAc,GAAGrB,IAAI,WAAW,CAAC;YAEvC,IACE,CAAE,MAAM1D,YAAYuB,MAAMgD,UAAU1C,qCACpC;gBACAwC,UAAUG,IAAI,CACZ,GAAGD,SAAS,uPAAuP,EAAEQ,YAAY,mCAAmC,CAAC;gBAEvT;YACF;YAEA,IAAI,CAACxD,KAAKE,MAAM,CAACsD,cAAc;gBAC7B,MAAM7D,MAAM,MAAMnB,sBAChBwB,MACAgD,UACAjE,yCACA;gBAEF,MAAMwE,YAAYrB,gBAAgBd,SAASe;gBAE3C,IAAIxC,OAAO4D,WAAW;oBACpBvD,KAAKK,KAAK,CACRmD,aACA3C,4BACElB,KACA8C,oBAAoBrB,QAAQE,IAAI,EAAEiC;gBAGxC,OAAO;oBACLT,UAAUG,IAAI,CACZ,GAAGD,SAAS,oHAAoH,EAAEQ,YAAY,uHAAuH,CAAC;oBAExQ;gBACF;YACF;YAEA,MAAMG,eAAe3D,KAAKqD,IAAI,CAACL,UAAU,YAAY;YACrD,IAAI,CAACW,aAAaL,QAAQ,CAAC,6CAA6C;gBACtE,MAAM/E,YAAYyB,MAAMgD,UAAUxC;gBAClC,MAAMjC,YAAYyB,MAAMgD,UAAUzC;YACpC;YAEA,IACE,CAAC,AAACP,CAAAA,KAAKqD,IAAI,CAACL,UAAU,YAAY,EAAC,EAAGM,QAAQ,CAAC,wBAC/C;gBACAR,UAAUG,IAAI,CACZ,GAAGD,SAAS,QAAQ,EAAEQ,YAAY,0IAA0I,CAAC;gBAE/K;YACF;YAEA;QACF;QAEA,IACER,SAASE,QAAQ,CAAC,eAClB,AAAClD,CAAAA,KAAKqD,IAAI,CAACL,UAAU,YAAY,EAAC,EAAGM,QAAQ,CAAC,kBAC9C;YACA,MAAMM,eAAe,AAAC5D,CAAAA,KAAKqD,IAAI,CAACL,UAAU,YAAY,EAAC,EAAGM,QAAQ,CAChE;YAGF,IAAI,CAACM,cAAc;gBACjB,MAAMC,cAAc,MAAMpF,YACxBuB,MACAgD,UACAtC;gBAEF,IAAI,CAACmD,aAAa;oBAChBf,UAAUG,IAAI,CACZ,GAAGD,SAAS,gPAAgP,CAAC;oBAE/P;gBACF;gBAEA,MAAMzE,YAAYyB,MAAMgD,UAAUvC;gBAClC,MAAMlC,YAAYyB,MAAMgD,UAAUrC;YACpC;YAEA,MAAMwB,MAAMa,SAASV,KAAK,CAAC,KAAKa,KAAK,CAAC,GAAG,CAAC,GAAGrF,IAAI,CAAC;YAClD,MAAM0F,cAAc,GAAGrB,IAAI,WAAW,CAAC;YAEvC,IAAI,CAACnC,KAAKE,MAAM,CAACsD,cAAc;gBAC7B,MAAMpC,UAAUH,kBAAkBjB,MAAMmC;gBACxC,MAAMxC,MAAM,MAAMnB,sBAChBwB,MACAgD,UACAhE,2CACA;gBAEF,MAAMuE,YAAYnC,WAAWc,gBAAgBd,SAASe;gBAEtD,IAAIf,WAAWzB,OAAO4D,WAAW;oBAC/BvD,KAAKK,KAAK,CACRmD,aACA3C,4BACElB,KACA8C,oBAAoBrB,QAAQE,IAAI,EAAEiC;gBAGxC,OAAO;oBACLT,UAAUG,IAAI,CACZ,GAAGD,SAAS,mIAAmI,EAAEQ,YAAY,sLAAsL,CAAC;oBAEtV;gBACF;YACF;YAEA,MAAMjF,YAAYyB,MAAMgD,UAAUpC;YAElC,IACE,CAAC,AAACZ,CAAAA,KAAKqD,IAAI,CAACL,UAAU,YAAY,EAAC,EAAGM,QAAQ,CAAC,wBAC/C;gBACAR,UAAUG,IAAI,CACZ,GAAGD,SAAS,QAAQ,EAAEQ,YAAY,0GAA0G,CAAC;YAEjJ;QACF;IACF;IAEA,MAAM7E,qBAAqBqB;IAE3B,OAAO;QAAE8C;IAAU;AACrB"}

package/src/preset/__snapshots__/generator.spec.ts.snap

@@ -252,7 +252,7 @@ export default {
 
 exports[`preset generator > should run successfully > biome.json 1`] = `
 "{
-  "$schema": "https://biomejs.dev/schemas/2.5.7/schema.json",
+  "$schema": "https://biomejs.dev/schemas/2.5.8/schema.json",
   "root": true,
   "formatter": {
     "enabled": true,
@@ -418,7 +418,7 @@ catalog:
   '@nx/js': 23.1.1
   '@nx/workspace': 23.1.1
   typescript: ~6.0.3
-  '@biomejs/biome': 2.5.7
+  '@biomejs/biome': 2.5.8
   '@aws/nx-plugin': ^0.0.0
   '@aws/nx-plugin-mcp': ^0.0.0
 "

package/src/preset/generator.d.ts

@@ -17,7 +17,7 @@ export declare const DEPENDENCIES: import("../utils/declared-dependencies").Depe
     readonly name: "typescript";
 }, {
     readonly name: "@biomejs/biome";
-}], readonly import("../utils/declared-dependencies").DeclaredPyDependency<"a2a-sdk" | "ag-ui-langgraph" | "ag-ui-protocol" | "ag-ui-strands" | "aws-lambda-powertools" | "aws-lambda-powertools[tracer]" | "aws-lambda-powertools[parser]" | "aws-opentelemetry-distro" | "bedrock-agentcore" | "boto3" | "checkov" | "fastapi" | "fastapi[standard]" | "httpx" | "langchain" | "langchain-aws" | "langchain-mcp-adapters" | "langgraph" | "mcp" | "pip-check-updates" | "pip-licenses" | "ruff" | "strands-agents" | "strands-agents[a2a]" | "strands-agents-tools" | "ty" | "pynamodb" | "uvicorn" | "sqlmodel" | "alembic" | "aiomysql" | "asyncpg" | "greenlet", Record<string, never>>[]>;
+}], readonly import("../utils/declared-dependencies").DeclaredPyDependency<"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" | "ruff" | "strands-agents" | "strands-agents[a2a]" | "strands-agents-tools" | "ty" | "pynamodb" | "uvicorn" | "sqlmodel" | "alembic" | "aiomysql" | "asyncpg" | "greenlet", Record<string, never>>[]>;
 export declare const PRESET_GENERATOR_INFO: NxGeneratorInfo;
 /**
  * Determines if the current user is an Amazon employee based on their git configuration.