Skip to content

Commit ae74aa3

Browse files
authored
fix: Add openai dependency to sagemaker dependency group (#678)
It depends on OpenAI and we a got a report about the need to install it explicitly Co-authored-by: Mackenzie Zastrow <[email protected]>
1 parent fbd598a commit ae74aa3

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

pyproject.toml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,9 @@ writer = [
9292
sagemaker = [
9393
"boto3>=1.26.0,<2.0.0",
9494
"botocore>=1.29.0,<2.0.0",
95-
"boto3-stubs[sagemaker-runtime]>=1.26.0,<2.0.0"
95+
"boto3-stubs[sagemaker-runtime]>=1.26.0,<2.0.0",
96+
# uses OpenAI as part of the implementation
97+
"openai>=1.68.0,<2.0.0",
9698
]
9799

98100
a2a = [

0 commit comments

Comments
 (0)