Skip to content

Support for Amazon SageMaker AI endpoints as Model Provider #176

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 31 commits into
base: main
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
31 commits
Select commit Hold shift + click to select a range
c7de8d5
New SageMaker AI implementation
dgallitelli Jun 4, 2025
b128c3b
Update sagemaker.py
dgallitelli Jun 4, 2025
4ec3eaa
Fixed Usagemetadata class
dgallitelli Jun 4, 2025
aad3bbf
Merge branch 'strands-agents:main' into main
dgallitelli Jun 5, 2025
24fdf1c
Improved management of the streaming response
dgallitelli Jun 6, 2025
3dcd15f
Merge branch 'strands-agents:main' into main
dgallitelli Jun 6, 2025
00eb1e2
Merge remote-tracking branch 'refs/remotes/origin/main'
dgallitelli Jun 6, 2025
fba8e25
Updated test
dgallitelli Jun 6, 2025
6872f16
Fixed linter errors
dgallitelli Jun 6, 2025
c052493
Fixed tool calling loop
dgallitelli Jun 6, 2025
6bf8bdf
Merge branch 'strands-agents:main' into main
dgallitelli Jun 6, 2025
5ae1674
Update sagemaker.py
dgallitelli Jun 10, 2025
4894643
Merge branch 'strands-agents:main' into main
dgallitelli Jun 10, 2025
68ad410
Merge branch 'strands-agents:main' into main
dgallitelli Jun 11, 2025
1fb4c13
Updated implementation with stream control
dgallitelli Jun 11, 2025
0341df3
Merge remote-tracking branch 'refs/remotes/origin/main'
dgallitelli Jun 11, 2025
cba0a18
Removed leftover print
dgallitelli Jun 11, 2025
72bfa27
Added control over empty tools
dgallitelli Jun 11, 2025
216fc23
Merge branch 'strands-agents:main' into main
dgallitelli Jun 11, 2025
8dc3553
Merge branch 'main' of https://github.com/dgallitelli/strands-agents-…
dgallitelli Jul 7, 2025
e1a899f
Updated SageMaker implementation
dgallitelli Jul 7, 2025
7cf9b57
Update4d pyproject
dgallitelli Jul 7, 2025
05ab9eb
Updated SageMaker implementation
dgallitelli Jul 7, 2025
d4524ea
Added support for reasoning content
dgallitelli Jul 8, 2025
fffc4df
Fixed relative dependency and run tests successfully
dgallitelli Jul 8, 2025
3908e4e
Removed wrong chunk error type
dgallitelli Jul 8, 2025
bbccb4b
Raise exception
dgallitelli Jul 8, 2025
b728a53
Updated SageMaker implementation
dgallitelli Jul 9, 2025
883b4bb
Implementation complete
dgallitelli Jul 9, 2025
226b568
Updated implementation
dgallitelli Jul 11, 2025
c5dcdb6
Fixed run-lint issues
dgallitelli Jul 11, 2025
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 6 additions & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -89,13 +89,17 @@ a2a = [
"fastapi>=0.115.12",
"starlette>=0.46.2",
]
sagemaker = [
"boto3>=1.26.0,<2.0.0",
"botocore>=1.29.0,<2.0.0",
]

[tool.hatch.version]
# Tells Hatch to use your version control system (git) to determine the version.
source = "vcs"

[tool.hatch.envs.hatch-static-analysis]
features = ["anthropic", "litellm", "llamaapi", "ollama", "openai", "otel","mistral"]
features = ["anthropic", "litellm", "llamaapi", "ollama", "openai", "otel", "mistral", "sagemaker"]
dependencies = [
"mypy>=1.15.0,<2.0.0",
"ruff>=0.11.6,<0.12.0",
Expand All @@ -119,7 +123,7 @@ lint-fix = [
]

[tool.hatch.envs.hatch-test]
features = ["anthropic", "litellm", "llamaapi", "ollama", "openai", "otel","mistral"]
features = ["anthropic", "litellm", "llamaapi", "ollama", "openai", "otel", "mistral", "sagemaker"]
extra-dependencies = [
"moto>=5.1.0,<6.0.0",
"pytest>=8.0.0,<9.0.0",
Expand Down
Loading