From 127b42a8020421af28ebd1b7634861c5078f3989 Mon Sep 17 00:00:00 2001
From: Mackenzie Zastrow <zastrowm@users.noreply.github.com>
Date: Fri, 15 Aug 2025 11:46:34 -0400
Subject: [PATCH] fix: Add openai dependency to sagemaker dependency group

It depends on OpenAI and we a got a report about the need to install it explicitly
---
 pyproject.toml | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 487b26691..6c0b6e3f7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -92,7 +92,9 @@ writer = [
 sagemaker = [
     "boto3>=1.26.0,<2.0.0",
     "botocore>=1.29.0,<2.0.0",
-    "boto3-stubs[sagemaker-runtime]>=1.26.0,<2.0.0"
+    "boto3-stubs[sagemaker-runtime]>=1.26.0,<2.0.0",
+    # uses OpenAI as part of the implementation
+    "openai>=1.68.0,<2.0.0",
 ]
 
 a2a = [