From 3cafbf5b7a2c2e155be5dd1fd0faf15f2b601fde Mon Sep 17 00:00:00 2001
From: Trivikram Kamat <16024985+trivikr@users.noreply.github.com>
Date: Mon, 4 Nov 2024 20:02:16 -0800
Subject: [PATCH] Fix package name of AugmentedAIRuntime and EKSAuth clients
 (#962)

---
 .changeset/mean-sheep-refuse.md                              | 5 +++++
 .../v2-to-v3/__fixtures__/new-client/global-import.output.js | 4 ++--
 src/transforms/v2-to-v3/config/CLIENT_PACKAGE_NAMES_MAP.ts   | 3 ++-
 3 files changed, 9 insertions(+), 3 deletions(-)
 create mode 100644 .changeset/mean-sheep-refuse.md

diff --git a/.changeset/mean-sheep-refuse.md b/.changeset/mean-sheep-refuse.md
new file mode 100644
index 000000000..bb43218b5
--- /dev/null
+++ b/.changeset/mean-sheep-refuse.md
@@ -0,0 +1,5 @@
+---
+"aws-sdk-js-codemod": patch
+---
+
+Fix package name of AugmentedAIRuntime and EKSAuth clients
diff --git a/src/transforms/v2-to-v3/__fixtures__/new-client/global-import.output.js b/src/transforms/v2-to-v3/__fixtures__/new-client/global-import.output.js
index 3884e1658..c3735978c 100644
--- a/src/transforms/v2-to-v3/__fixtures__/new-client/global-import.output.js
+++ b/src/transforms/v2-to-v3/__fixtures__/new-client/global-import.output.js
@@ -126,7 +126,7 @@ import { ECRPUBLIC } from "@aws-sdk/client-ecr-public";
 import { ECS } from "@aws-sdk/client-ecs";
 import { EFS } from "@aws-sdk/client-efs";
 import { EKS } from "@aws-sdk/client-eks";
-import { EKSAuth } from "@aws-sdk/client-eksauth";
+import { EKSAuth } from "@aws-sdk/client-eks-auth";
 import { ElasticBeanstalk } from "@aws-sdk/client-elastic-beanstalk";
 import { ElasticInference } from "@aws-sdk/client-elastic-inference";
 import { ElasticLoadBalancing } from "@aws-sdk/client-elastic-load-balancing";
@@ -314,8 +314,8 @@ import { RUM } from "@aws-sdk/client-rum";
 import { S3 } from "@aws-sdk/client-s3";
 import { S3Control } from "@aws-sdk/client-s3-control";
 import { S3Outposts } from "@aws-sdk/client-s3outposts";
-import { SageMakerA2IRuntime } from "@aws-sdk/client-sage-maker-a2iruntime";
 import { SageMaker } from "@aws-sdk/client-sagemaker";
+import { SageMakerA2IRuntime } from "@aws-sdk/client-sagemaker-a2i-runtime";
 import { SagemakerEdge } from "@aws-sdk/client-sagemaker-edge";
 import { SageMakerFeatureStoreRuntime } from "@aws-sdk/client-sagemaker-featurestore-runtime";
 import { SageMakerGeospatial } from "@aws-sdk/client-sagemaker-geospatial";
diff --git a/src/transforms/v2-to-v3/config/CLIENT_PACKAGE_NAMES_MAP.ts b/src/transforms/v2-to-v3/config/CLIENT_PACKAGE_NAMES_MAP.ts
index 0484f7c3f..2b7d087ce 100644
--- a/src/transforms/v2-to-v3/config/CLIENT_PACKAGE_NAMES_MAP.ts
+++ b/src/transforms/v2-to-v3/config/CLIENT_PACKAGE_NAMES_MAP.ts
@@ -45,7 +45,7 @@ export const CLIENT_PACKAGE_NAMES_MAP: Record<string, string> = {
   AppTest: "client-apptest",
   ApplicationCostProfiler: "client-applicationcostprofiler",
   ARCZonalShift: "client-arc-zonal-shift",
-  AugmentedAIRuntime: "client-sage-maker-a2iruntime",
+  AugmentedAIRuntime: "client-sagemaker-a2i-runtime",
   AuditManager: "client-auditmanager",
   BackupStorage: "client-backupstorage",
   CUR: "client-cost-and-usage-report-service",
@@ -69,6 +69,7 @@ export const CLIENT_PACKAGE_NAMES_MAP: Record<string, string> = {
   ECRPUBLIC: "client-ecr-public",
   ELB: "client-elastic-load-balancing",
   ELBv2: "client-elastic-load-balancing-v2",
+  EKSAuth: "client-eks-auth",
   ElastiCache: "client-elasticache",
   EntityResolution: "client-entityresolution",
   EMRcontainers: "client-emr-containers",