diff --git a/providers/common/ai/provider.yaml b/providers/common/ai/provider.yaml
index f7f8ef3f2ea06..abb5ca41d3d7d 100644
--- a/providers/common/ai/provider.yaml
+++ b/providers/common/ai/provider.yaml
@@ -454,6 +454,7 @@ operators:
 toolsets:
   - integration-name: Common AI
     python-modules:
+      - airflow.providers.common.ai.toolsets.aws
       - airflow.providers.common.ai.toolsets.hook
       - airflow.providers.common.ai.toolsets.sql
       - airflow.providers.common.ai.toolsets.datafusion