diff --git a/poetry.lock b/poetry.lock
index b5f603bed..7da03d2ea 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -408,13 +408,13 @@ typing-extensions = ">=4.0.1"
 
 [[package]]
 name = "azure-ai-ml"
-version = "1.20.0"
+version = "1.21.1"
 description = "Microsoft Azure Machine Learning Client Library for Python"
 optional = false
 python-versions = ">=3.7"
 files = [
-    {file = "azure-ai-ml-1.20.0.tar.gz", hash = "sha256:6432a0da1b7250cb0db5a1c33202e0419935e19ea32d4c2b3220705f8f1d4101"},
-    {file = "azure_ai_ml-1.20.0-py3-none-any.whl", hash = "sha256:c7eb3c5ccf82a6ee94403c3e5060763decd38cf03ff2620a4a6577526e605104"},
+    {file = "azure_ai_ml-1.21.1-py3-none-any.whl", hash = "sha256:19ce58687d0946d781c0025a660f0622cf0a9af25327fb1c724e2bea91d254d9"},
+    {file = "azure_ai_ml-1.21.1.tar.gz", hash = "sha256:915436aa7e47e9d36b8d031114a6094ea8f9d473af6aa86c187729bd3d026bf1"},
 ]
 
 [package.dependencies]
@@ -7106,4 +7106,4 @@ test = ["big-O", "importlib-resources", "jaraco.functools", "jaraco.itertools",
 [metadata]
 lock-version = "2.0"
 python-versions = "^3.10"
-content-hash = "781249e7f5a66bf63f920b5b75093f48126f4410bfaabf69f55419440746b7f1"
+content-hash = "4ef29e19e2394e344ae987f258aaeab200ee1aa0a56f4a32f1284af81441c23c"
diff --git a/pyproject.toml b/pyproject.toml
index 9cf925511..74793f496 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -38,7 +38,7 @@ pillow = "10.4.0"
 azure-mgmt-cognitiveservices = "^13.5.0"
 jsonschema = "^4.23.0"
 semantic-kernel = {version = "1.3.0", python = "<3.13"}
-azure-ai-ml = "^1.20.0"
+azure-ai-ml = "^1.21.1"
 azure-cosmos = "^4.7.0"
 
 [tool.poetry.group.dev.dependencies]