@aws/nx-plugin 1.0.0-rc.70 → 1.0.0-rc.72

package/src/py/agent/__snapshots__/generator.core.spec.ts.snap

@@ -0,0 +1,186 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`py#agent generator > should generate session.py returning InMemorySaver for the langchain framework > session.py (in-memory) 1`] = `
+"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 a LangGraph checkpointer for persisting conversation state across
+    invocations. Local development always uses a local SQLite database for
+    convenience, regardless of the configured session option.
+    """
+    if os.environ.get("LOCAL_DEV") == "true":
+        os.makedirs("../../tmp/agents/langchain/test-project-agent", exist_ok=True)
+        conn = aiosqlite.connect(
+            os.path.join(
+                "../../tmp/agents/langchain/test-project-agent", "checkpoints.sqlite"
+            )
+        )
+        return AsyncSqliteSaver(conn)
+
+    return InMemorySaver()
+"
+`;
+
+exports[`py#agent generator > should generate session.py returning None when session is in-memory > session.py (in-memory) 1`] = `
+"import os
+
+from proj_agent_connection import get_current_session_id
+from strands.session import FileSessionManager, SessionManager
+
+
+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="../../tmp/agents/strands/test-project-agent",
+        )
+
+    return None
+"
+`;
+
+exports[`py#agent generator > should generate session.py using DynamoDBSaver when session is dynamodb-s3 > session.py (dynamodb-s3) 1`] = `
+"import os
+
+import aiosqlite
+from langgraph.checkpoint.base import BaseCheckpointSaver
+from langgraph.checkpoint.sqlite.aio import AsyncSqliteSaver
+from langgraph_checkpoint_aws import DynamoDBSaver
+from proj_agent_connection.core.runtime_config import get_agentcore_runtime_config
+
+
+def get_checkpointer() -> BaseCheckpointSaver:
+    """Returns a LangGraph checkpointer for persisting conversation state across
+    invocations. Local development always uses a local SQLite database for
+    convenience, regardless of the configured session option.
+    """
+    if os.environ.get("LOCAL_DEV") == "true":
+        os.makedirs("../../tmp/agents/langchain/test-project-agent", exist_ok=True)
+        conn = aiosqlite.connect(
+            os.path.join(
+                "../../tmp/agents/langchain/test-project-agent", "checkpoints.sqlite"
+            )
+        )
+        return AsyncSqliteSaver(conn)
+
+    config = get_agentcore_runtime_config()
+    session_config = (
+        (config.get("agentRuntimes") or {})
+        .get("TestProjectAgent", {})
+        .get("session", {})
+    )
+    table_name = session_config.get("tableName")
+    bucket_name = session_config.get("bucketName")
+    if not table_name or not bucket_name:
+        raise RuntimeError(
+            "No DynamoDB checkpoint table or S3 offload bucket configured for this agent in runtime configuration."
+        )
+    return DynamoDBSaver(
+        table_name=table_name,
+        region_name=os.environ.get("AWS_REGION"),
+        s3_offload_config={"bucket_name": bucket_name},
+    )
+"
+`;
+
+exports[`py#agent generator > should generate session.py using S3 storage by default for the strands framework > session.py (s3) 1`] = `
+"import os
+
+from proj_agent_connection import get_current_session_id
+from proj_agent_connection.core.runtime_config import get_agentcore_runtime_config
+from strands.session import FileSessionManager, S3SessionManager, SessionManager
+
+
+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="../../tmp/agents/strands/test-project-agent",
+        )
+
+    config = get_agentcore_runtime_config()
+    bucket_name = (
+        (config.get("agentRuntimes") or {})
+        .get("TestProjectAgent", {})
+        .get("session", {})
+        .get("bucketName")
+    )
+    if not bucket_name:
+        raise RuntimeError(
+            "No S3 bucket configured for this agent's session in runtime configuration."
+        )
+    return S3SessionManager(session_id=session_id, bucket=bucket_name)
+"
+`;
+
+exports[`py#agent generator > should generate session.py using S3CheckpointSaver when session is s3 > session.py (s3) 1`] = `
+"import os
+
+import aiosqlite
+import boto3
+from langgraph.checkpoint.base import BaseCheckpointSaver
+from langgraph.checkpoint.sqlite.aio import AsyncSqliteSaver
+from proj_agent_connection.core.runtime_config import get_agentcore_runtime_config
+from proj_agent_connection.core.s3_checkpoint_saver_langchain import S3CheckpointSaver
+
+
+def get_checkpointer() -> BaseCheckpointSaver:
+    """Returns a LangGraph checkpointer for persisting conversation state across
+    invocations. Local development always uses a local SQLite database for
+    convenience, regardless of the configured session option.
+    """
+    if os.environ.get("LOCAL_DEV") == "true":
+        os.makedirs("../../tmp/agents/langchain/test-project-agent", exist_ok=True)
+        conn = aiosqlite.connect(
+            os.path.join(
+                "../../tmp/agents/langchain/test-project-agent", "checkpoints.sqlite"
+            )
+        )
+        return AsyncSqliteSaver(conn)
+
+    config = get_agentcore_runtime_config()
+    session_config = (
+        (config.get("agentRuntimes") or {})
+        .get("TestProjectAgent", {})
+        .get("session", {})
+    )
+    bucket_name = session_config.get("bucketName")
+    if not bucket_name:
+        raise RuntimeError(
+            "No S3 checkpoint bucket configured for this agent in runtime configuration."
+        )
+    return S3CheckpointSaver(
+        bucket_name=bucket_name,
+        prefix="checkpoints/",
+        s3_client=boto3.client("s3"),
+    )
+"
+`;

package/src/py/agent/__snapshots__/generator.frameworks.spec.ts.snap

@@ -577,7 +577,8 @@ exports[`py#agent generator > langchain framework > should match snapshot for la
 from langchain.agents import create_agent
 from langchain_aws import ChatBedrockConverse
 from langchain_core.tools import tool
-from langgraph.checkpoint.memory import InMemorySaver
+
+from .session import get_checkpointer
 
 REGION = os.environ.get("AWS_REGION", "us-east-1")
 MODEL_ID = os.environ.get("MODEL_ID", "global.anthropic.claude-haiku-4-5-20251001-v1:0")
@@ -591,7 +592,6 @@ def subtract(a: int, b: int) -> int:
 
 def get_agent():
     model = ChatBedrockConverse(model=MODEL_ID, region_name=REGION)
-    # Swap InMemorySaver for a durable checkpointer in a multi-replica deployment.
     return create_agent(
         model=model,
         tools=[subtract],
@@ -600,7 +600,7 @@ You are a mathematical wizard.
 Use your tools for mathematical tasks.
 Refer to tools as your 'spellbook'.
 """,
-        checkpointer=InMemorySaver(),
+        checkpointer=get_checkpointer(),
     )
 "
 `;
@@ -721,6 +721,8 @@ dependencies = [
   "langchain==1.3.14",
   "langchain-aws==1.7.0",
   "langgraph==1.2.10",
+  "langgraph-checkpoint-sqlite==3.1.1",
+  "aiosqlite==0.22.1",
   "ag-ui-protocol==0.1.19",
   "ag-ui-langgraph==0.0.42",
   "uvicorn==0.52.1"
@@ -736,3 +738,45 @@ dev-dependencies = [ ]
 workspace = true
 "
 `;
+
+exports[`py#agent generator > langchain framework > should match snapshot for langchain generated files > langchain-session.py 1`] = `
+"import os
+
+import aiosqlite
+import boto3
+from langgraph.checkpoint.base import BaseCheckpointSaver
+from langgraph.checkpoint.sqlite.aio import AsyncSqliteSaver
+from proj_agent_connection.core.runtime_config import get_agentcore_runtime_config
+from proj_agent_connection.core.s3_checkpoint_saver_langchain import S3CheckpointSaver
+
+
+def get_checkpointer() -> BaseCheckpointSaver:
+    """Returns a LangGraph checkpointer for persisting conversation state across
+    invocations. Local development always uses a local SQLite database for
+    convenience, regardless of the configured session option.
+    """
+    if os.environ.get("LOCAL_DEV") == "true":
+        os.makedirs("../../tmp/agents/langchain/snapshot-agent", exist_ok=True)
+        conn = aiosqlite.connect(
+            os.path.join(
+                "../../tmp/agents/langchain/snapshot-agent", "checkpoints.sqlite"
+            )
+        )
+        return AsyncSqliteSaver(conn)
+
+    config = get_agentcore_runtime_config()
+    session_config = (
+        (config.get("agentRuntimes") or {}).get("SnapshotAgent", {}).get("session", {})
+    )
+    bucket_name = session_config.get("bucketName")
+    if not bucket_name:
+        raise RuntimeError(
+            "No S3 checkpoint bucket configured for this agent in runtime configuration."
+        )
+    return S3CheckpointSaver(
+        bucket_name=bucket_name,
+        prefix="checkpoints/",
+        s3_client=boto3.client("s3"),
+    )
+"
+`;