diff --git a/build_artifacts/v1/v1.6/v1.6.0/cpu.additional_packages_env.in b/build_artifacts/v1/v1.6/v1.6.0/cpu.additional_packages_env.in
index 93f4c0cc..924f89a8 100644
--- a/build_artifacts/v1/v1.6/v1.6.0/cpu.additional_packages_env.in
+++ b/build_artifacts/v1/v1.6/v1.6.0/cpu.additional_packages_env.in
@@ -1,2 +1,4 @@
 conda-forge::jupyter_server_fileid
-conda-forge::jupyter_ydoc
\ No newline at end of file
+conda-forge::jupyter_ydoc
+conda-forge::amazon_sagemaker_sql_editor[version='>=0.1.0,<2']
+conda-forge::amazon-sagemaker-sql-magic[version='>=0.1.0,<2']
\ No newline at end of file
diff --git a/build_artifacts/v1/v1.6/v1.6.0/gpu.additional_packages_env.in b/build_artifacts/v1/v1.6/v1.6.0/gpu.additional_packages_env.in
index 93f4c0cc..924f89a8 100644
--- a/build_artifacts/v1/v1.6/v1.6.0/gpu.additional_packages_env.in
+++ b/build_artifacts/v1/v1.6/v1.6.0/gpu.additional_packages_env.in
@@ -1,2 +1,4 @@
 conda-forge::jupyter_server_fileid
-conda-forge::jupyter_ydoc
\ No newline at end of file
+conda-forge::jupyter_ydoc
+conda-forge::amazon_sagemaker_sql_editor[version='>=0.1.0,<2']
+conda-forge::amazon-sagemaker-sql-magic[version='>=0.1.0,<2']
\ No newline at end of file
diff --git a/test/test_artifacts/v1/amazon-sagemaker-sql-magic.test.Dockerfile b/test/test_artifacts/v1/amazon-sagemaker-sql-magic.test.Dockerfile
new file mode 100644
index 00000000..44679eb8
--- /dev/null
+++ b/test/test_artifacts/v1/amazon-sagemaker-sql-magic.test.Dockerfile
@@ -0,0 +1,6 @@
+ARG COSMOS_IMAGE
+FROM $COSMOS_IMAGE
+
+ARG MAMBA_DOCKERFILE_ACTIVATE=1
+
+CMD ["python", "-c", "import amazon_sagemaker_sql_magic"]
diff --git a/test/test_artifacts/v1/amazon_sagemaker_sql_editor.test.Dockerfile b/test/test_artifacts/v1/amazon_sagemaker_sql_editor.test.Dockerfile
new file mode 100644
index 00000000..635358bf
--- /dev/null
+++ b/test/test_artifacts/v1/amazon_sagemaker_sql_editor.test.Dockerfile
@@ -0,0 +1,6 @@
+ARG COSMOS_IMAGE
+FROM $COSMOS_IMAGE
+
+ARG MAMBA_DOCKERFILE_ACTIVATE=1
+
+CMD ["python", "-c", "import amazon_sagemaker_sql_editor"]
diff --git a/test/test_dockerfile_based_harness.py b/test/test_dockerfile_based_harness.py
index 0f47d04b..9f881393 100644
--- a/test/test_dockerfile_based_harness.py
+++ b/test/test_dockerfile_based_harness.py
@@ -40,7 +40,9 @@
     ("jupyterlab-git.test.Dockerfile", ['jupyterlab-git']),
     ("serve.test.Dockerfile", ['serve-langchain']),
     ("jupyter-server-fileid.test.Dockerfile", ['jupyter_server_fileid']),
-    ("jupyter-ydoc.test.Dockerfile", ['jupyter_ydoc'])])
+    ("jupyter-ydoc.test.Dockerfile", ['jupyter_ydoc'])],
+    ("amazon-sagemaker-sql-magic.test.Dockerfile", ['amazon-sagemaker-sql-magic']),
+    ("amazon_sagemaker_sql_editor.test.Dockerfile", ['amazon_sagemaker_sql_editor']))
 def test_dockerfiles_for_cpu(dockerfile_path: str, required_packages: List[str],
                              local_image_version: str, use_gpu: bool):
     _validate_docker_images(dockerfile_path, required_packages, local_image_version, use_gpu, 'cpu')
@@ -72,7 +74,9 @@ def test_dockerfiles_for_cpu(dockerfile_path: str, required_packages: List[str],
     ("jupyterlab-git.test.Dockerfile", ['jupyterlab-git']),
     ("serve.test.Dockerfile", ['serve-langchain']),
     ("jupyter-server-fileid.test.Dockerfile", ['jupyter_server_fileid']),
-    ("jupyter-ydoc.test.Dockerfile", ['jupyter_ydoc'])])
+    ("jupyter-ydoc.test.Dockerfile", ['jupyter_ydoc'])],
+    ("amazon-sagemaker-sql-magic.test.Dockerfile", ['amazon-sagemaker-sql-magic']),
+    ("amazon_sagemaker_sql_editor.test.Dockerfile", ['amazon_sagemaker_sql_editor']))
 def test_dockerfiles_for_gpu(dockerfile_path: str, required_packages: List[str],
                              local_image_version: str, use_gpu: bool):
     _validate_docker_images(dockerfile_path, required_packages, local_image_version, use_gpu, 'gpu')