diff --git a/package-lock.json b/package-lock.json
index f2b91ae3..a53c64c4 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -24,7 +24,7 @@
         "@aws-sdk/client-sts": "^3.893.0",
         "@aws-sdk/client-xray": "^3.1003.0",
         "@aws-sdk/credential-providers": "^3.893.0",
-        "@aws/agent-inspector": "0.1.0",
+        "@aws/agent-inspector": "0.2.1",
         "@commander-js/extra-typings": "^14.0.0",
         "@opentelemetry/api": "^1.9.0",
         "@opentelemetry/otlp-transformer": "^0.213.0",
@@ -2856,9 +2856,9 @@
       }
     },
     "node_modules/@aws/agent-inspector": {
-      "version": "0.1.0",
-      "resolved": "https://registry.npmjs.org/@aws/agent-inspector/-/agent-inspector-0.1.0.tgz",
-      "integrity": "sha512-ooYUUTh80oLSMr7VkJt0zCbF3mEgdeyZI46kPZapx4sy0TsP2glpObKZ3baqPI0agh6NeuUTNmAidusnVxeFrg==",
+      "version": "0.2.1",
+      "resolved": "https://registry.npmjs.org/@aws/agent-inspector/-/agent-inspector-0.2.1.tgz",
+      "integrity": "sha512-kyL6RBcTj1hYIchtrHDlDyeqm2viVYMBxhZKVn8wJn058YhI52GIDuUFlKD1avd57X+LJKlHr5VcKvBZp7Sg6A==",
       "license": "Apache-2.0",
       "dependencies": {
         "@ag-ui/core": "^0.0.52",
diff --git a/package.json b/package.json
index 1e17208d..2eb78d66 100644
--- a/package.json
+++ b/package.json
@@ -86,7 +86,7 @@
     "@aws-sdk/client-sts": "^3.893.0",
     "@aws-sdk/client-xray": "^3.1003.0",
     "@aws-sdk/credential-providers": "^3.893.0",
-    "@aws/agent-inspector": "0.1.0",
+    "@aws/agent-inspector": "0.2.1",
     "@commander-js/extra-typings": "^14.0.0",
     "@opentelemetry/api": "^1.9.0",
     "@opentelemetry/otlp-transformer": "^0.213.0",