Skip to content

Commit 3fe4b12

Browse files
tpaulshippytpaulshippy
authored
16: Add support for Claude through Bedrock (#50)
Resolves #16 for the Claude models. ## Purpose Support Claude through Bedrock. ## Approach Follow pattern of adding providers. Avoid additional dependencies by bringing in the signing and decoding pieces from aws-sdk-ruby directly. ## Testing Ran with test script locally that had the following environment variables configured: - AWS_ACCESS_KEY_ID - AWS_SECRET_ACCESS_KEY - AWS_SESSION_TOKEN - AWS_REGION Used `anthropic.claude-3-5-sonnet-20240620-v1:0` model. ## Screenshot ![image](https://github.com/user-attachments/assets/cfc94a89-473a-46f2-b548-effa67f1067f)
1 parent b32a435 commit 3fe4b12

37 files changed

+4335
-58
lines changed

‎LICENSES/APACHE-2.0.txt‎

Lines changed: 202 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,202 @@
1+
2+
Apache License
3+
Version 2.0, January 2004
4+
http://www.apache.org/licenses/
5+
6+
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
7+
8+
1. Definitions.
9+
10+
"License" shall mean the terms and conditions for use, reproduction,
11+
and distribution as defined by Sections 1 through 9 of this document.
12+
13+
"Licensor" shall mean the copyright owner or entity authorized by
14+
the copyright owner that is granting the License.
15+
16+
"Legal Entity" shall mean the union of the acting entity and all
17+
other entities that control, are controlled by, or are under common
18+
control with that entity. For the purposes of this definition,
19+
"control" means (i) the power, direct or indirect, to cause the
20+
direction or management of such entity, whether by contract or
21+
otherwise, or (ii) ownership of fifty percent (50%) or more of the
22+
outstanding shares, or (iii) beneficial ownership of such entity.
23+
24+
"You" (or "Your") shall mean an individual or Legal Entity
25+
exercising permissions granted by this License.
26+
27+
"Source" form shall mean the preferred form for making modifications,
28+
including but not limited to software source code, documentation
29+
source, and configuration files.
30+
31+
"Object" form shall mean any form resulting from mechanical
32+
transformation or translation of a Source form, including but
33+
not limited to compiled object code, generated documentation,
34+
and conversions to other media types.
35+
36+
"Work" shall mean the work of authorship, whether in Source or
37+
Object form, made available under the License, as indicated by a
38+
copyright notice that is included in or attached to the work
39+
(an example is provided in the Appendix below).
40+
41+
"Derivative Works" shall mean any work, whether in Source or Object
42+
form, that is based on (or derived from) the Work and for which the
43+
editorial revisions, annotations, elaborations, or other modifications
44+
represent, as a whole, an original work of authorship. For the purposes
45+
of this License, Derivative Works shall not include works that remain
46+
separable from, or merely link (or bind by name) to the interfaces of,
47+
the Work and Derivative Works thereof.
48+
49+
"Contribution" shall mean any work of authorship, including
50+
the original version of the Work and any modifications or additions
51+
to that Work or Derivative Works thereof, that is intentionally
52+
submitted to Licensor for inclusion in the Work by the copyright owner
53+
or by an individual or Legal Entity authorized to submit on behalf of
54+
the copyright owner. For the purposes of this definition, "submitted"
55+
means any form of electronic, verbal, or written communication sent
56+
to the Licensor or its representatives, including but not limited to
57+
communication on electronic mailing lists, source code control systems,
58+
and issue tracking systems that are managed by, or on behalf of, the
59+
Licensor for the purpose of discussing and improving the Work, but
60+
excluding communication that is conspicuously marked or otherwise
61+
designated in writing by the copyright owner as "Not a Contribution."
62+
63+
"Contributor" shall mean Licensor and any individual or Legal Entity
64+
on behalf of whom a Contribution has been received by Licensor and
65+
subsequently incorporated within the Work.
66+
67+
2. Grant of Copyright License. Subject to the terms and conditions of
68+
this License, each Contributor hereby grants to You a perpetual,
69+
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
70+
copyright license to reproduce, prepare Derivative Works of,
71+
publicly display, publicly perform, sublicense, and distribute the
72+
Work and such Derivative Works in Source or Object form.
73+
74+
3. Grant of Patent License. Subject to the terms and conditions of
75+
this License, each Contributor hereby grants to You a perpetual,
76+
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
77+
(except as stated in this section) patent license to make, have made,
78+
use, offer to sell, sell, import, and otherwise transfer the Work,
79+
where such license applies only to those patent claims licensable
80+
by such Contributor that are necessarily infringed by their
81+
Contribution(s) alone or by combination of their Contribution(s)
82+
with the Work to which such Contribution(s) was submitted. If You
83+
institute patent litigation against any entity (including a
84+
cross-claim or counterclaim in a lawsuit) alleging that the Work
85+
or a Contribution incorporated within the Work constitutes direct
86+
or contributory patent infringement, then any patent licenses
87+
granted to You under this License for that Work shall terminate
88+
as of the date such litigation is filed.
89+
90+
4. Redistribution. You may reproduce and distribute copies of the
91+
Work or Derivative Works thereof in any medium, with or without
92+
modifications, and in Source or Object form, provided that You
93+
meet the following conditions:
94+
95+
(a) You must give any other recipients of the Work or
96+
Derivative Works a copy of this License; and
97+
98+
(b) You must cause any modified files to carry prominent notices
99+
stating that You changed the files; and
100+
101+
(c) You must retain, in the Source form of any Derivative Works
102+
that You distribute, all copyright, patent, trademark, and
103+
attribution notices from the Source form of the Work,
104+
excluding those notices that do not pertain to any part of
105+
the Derivative Works; and
106+
107+
(d) If the Work includes a "NOTICE" text file as part of its
108+
distribution, then any Derivative Works that You distribute must
109+
include a readable copy of the attribution notices contained
110+
within such NOTICE file, excluding those notices that do not
111+
pertain to any part of the Derivative Works, in at least one
112+
of the following places: within a NOTICE text file distributed
113+
as part of the Derivative Works; within the Source form or
114+
documentation, if provided along with the Derivative Works; or,
115+
within a display generated by the Derivative Works, if and
116+
wherever such third-party notices normally appear. The contents
117+
of the NOTICE file are for informational purposes only and
118+
do not modify the License. You may add Your own attribution
119+
notices within Derivative Works that You distribute, alongside
120+
or as an addendum to the NOTICE text from the Work, provided
121+
that such additional attribution notices cannot be construed
122+
as modifying the License.
123+
124+
You may add Your own copyright statement to Your modifications and
125+
may provide additional or different license terms and conditions
126+
for use, reproduction, or distribution of Your modifications, or
127+
for any such Derivative Works as a whole, provided Your use,
128+
reproduction, and distribution of the Work otherwise complies with
129+
the conditions stated in this License.
130+
131+
5. Submission of Contributions. Unless You explicitly state otherwise,
132+
any Contribution intentionally submitted for inclusion in the Work
133+
by You to the Licensor shall be under the terms and conditions of
134+
this License, without any additional terms or conditions.
135+
Notwithstanding the above, nothing herein shall supersede or modify
136+
the terms of any separate license agreement you may have executed
137+
with Licensor regarding such Contributions.
138+
139+
6. Trademarks. This License does not grant permission to use the trade
140+
names, trademarks, service marks, or product names of the Licensor,
141+
except as required for reasonable and customary use in describing the
142+
origin of the Work and reproducing the content of the NOTICE file.
143+
144+
7. Disclaimer of Warranty. Unless required by applicable law or
145+
agreed to in writing, Licensor provides the Work (and each
146+
Contributor provides its Contributions) on an "AS IS" BASIS,
147+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
148+
implied, including, without limitation, any warranties or conditions
149+
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
150+
PARTICULAR PURPOSE. You are solely responsible for determining the
151+
appropriateness of using or redistributing the Work and assume any
152+
risks associated with Your exercise of permissions under this License.
153+
154+
8. Limitation of Liability. In no event and under no legal theory,
155+
whether in tort (including negligence), contract, or otherwise,
156+
unless required by applicable law (such as deliberate and grossly
157+
negligent acts) or agreed to in writing, shall any Contributor be
158+
liable to You for damages, including any direct, indirect, special,
159+
incidental, or consequential damages of any character arising as a
160+
result of this License or out of the use or inability to use the
161+
Work (including but not limited to damages for loss of goodwill,
162+
work stoppage, computer failure or malfunction, or any and all
163+
other commercial damages or losses), even if such Contributor
164+
has been advised of the possibility of such damages.
165+
166+
9. Accepting Warranty or Additional Liability. While redistributing
167+
the Work or Derivative Works thereof, You may choose to offer,
168+
and charge a fee for, acceptance of support, warranty, indemnity,
169+
or other liability obligations and/or rights consistent with this
170+
License. However, in accepting such obligations, You may act only
171+
on Your own behalf and on Your sole responsibility, not on behalf
172+
of any other Contributor, and only if You agree to indemnify,
173+
defend, and hold each Contributor harmless for any liability
174+
incurred by, or claims asserted against, such Contributor by reason
175+
of your accepting any such warranty or additional liability.
176+
177+
END OF TERMS AND CONDITIONS
178+
179+
APPENDIX: How to apply the Apache License to your work.
180+
181+
To apply the Apache License to your work, attach the following
182+
boilerplate notice, with the fields enclosed by brackets "[]"
183+
replaced with your own identifying information. (Don't include
184+
the brackets!) The text should be enclosed in the appropriate
185+
comment syntax for the file format. We also recommend that a
186+
file or class name and description of purpose be included on the
187+
same "printed page" as the copyright notice for easier
188+
identification within third-party archives.
189+
190+
Copyright [yyyy] [name of copyright owner]
191+
192+
Licensed under the Apache License, Version 2.0 (the "License");
193+
you may not use this file except in compliance with the License.
194+
You may obtain a copy of the License at
195+
196+
http://www.apache.org/licenses/LICENSE-2.0
197+
198+
Unless required by applicable law or agreed to in writing, software
199+
distributed under the License is distributed on an "AS IS" BASIS,
200+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
201+
See the License for the specific language governing permissions and
202+
limitations under the License.

‎NOTICE‎

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
1+
RubyLLM Project
2+
Copyright 2025 Carmine Paolino. All Rights Reserved.
3+
Licensed under the MIT License.
4+
5+
This project includes software developed by third parties, licensed under
6+
terms compatible with the MIT License. The required notices and license
7+
texts are included below and/or in the LICENSES/ directory.
8+
9+
--------------------------------------------------------------------------
10+
11+
This project incorporates code from aws-sdk-ruby.
12+
The original NOTICE file for aws-sdk-ruby contains the following:
13+
14+
--- BEGIN aws-sdk-ruby NOTICE ---
15+
AWS SDK for Ruby
16+
Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
17+
--- END aws-sdk-ruby NOTICE ---
18+
19+
The aws-sdk-ruby components used in this project are licensed under the
20+
Apache License, Version 2.0 (the "License"). You may obtain a copy of
21+
the License at:
22+
23+
LICENSES/APACHE-2.0.txt
24+
25+
Unless required by applicable law or agreed to in writing, software
26+
distributed under the License is distributed on an "AS IS" BASIS,
27+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
28+
See the License for the specific language governing permissions and
29+
limitations under the License.
30+
31+
--------------------------------------------------------------------------

‎README.md‎

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -99,10 +99,14 @@ Configure with your API keys:
9999

100100
```ruby
101101
RubyLLM.configure do |config|
102-
config.openai_api_key = ENV['OPENAI_API_KEY']
103-
config.anthropic_api_key = ENV['ANTHROPIC_API_KEY']
104-
config.gemini_api_key = ENV['GEMINI_API_KEY']
105-
config.deepseek_api_key = ENV['DEEPSEEK_API_KEY']
102+
config.openai_api_key = ENV.fetch('OPENAI_API_KEY', nil)
103+
config.anthropic_api_key = ENV.fetch('ANTHROPIC_API_KEY', nil)
104+
config.gemini_api_key = ENV.fetch('GEMINI_API_KEY', nil)
105+
config.deepseek_api_key = ENV.fetch('DEEPSEEK_API_KEY', nil)
106+
config.bedrock_api_key = ENV.fetch('AWS_ACCESS_KEY_ID', nil)
107+
config.bedrock_secret_key = ENV.fetch('AWS_SECRET_ACCESS_KEY', nil)
108+
config.bedrock_region = ENV.fetch('AWS_REGION', nil)
109+
config.bedrock_session_token = ENV.fetch('AWS_SESSION_TOKEN', nil)
106110
end
107111
```
108112

‎bin/console‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,10 @@ RubyLLM.configure do |config|
1212
config.anthropic_api_key = ENV.fetch('ANTHROPIC_API_KEY', nil)
1313
config.gemini_api_key = ENV.fetch('GEMINI_API_KEY', nil)
1414
config.deepseek_api_key = ENV.fetch('DEEPSEEK_API_KEY', nil)
15+
config.bedrock_api_key = ENV.fetch('AWS_ACCESS_KEY_ID', nil)
16+
config.bedrock_secret_key = ENV.fetch('AWS_SECRET_ACCESS_KEY', nil)
17+
config.bedrock_region = ENV.fetch('AWS_REGION', nil)
18+
config.bedrock_session_token = ENV.fetch('AWS_SESSION_TOKEN', nil)
1519
end
1620

1721
IRB.start(__FILE__)

‎docs/guides/getting-started.md‎

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,10 +26,14 @@ require 'ruby_llm'
2626

2727
RubyLLM.configure do |config|
2828
# Add the API keys you have available
29-
config.openai_api_key = ENV['OPENAI_API_KEY']
30-
config.anthropic_api_key = ENV['ANTHROPIC_API_KEY']
31-
config.gemini_api_key = ENV['GEMINI_API_KEY']
32-
config.deepseek_api_key = ENV['DEEPSEEK_API_KEY']
29+
config.openai_api_key = ENV.fetch('OPENAI_API_KEY', nil)
30+
config.anthropic_api_key = ENV.fetch('ANTHROPIC_API_KEY', nil)
31+
config.gemini_api_key = ENV.fetch('GEMINI_API_KEY', nil)
32+
config.deepseek_api_key = ENV.fetch('DEEPSEEK_API_KEY', nil)
33+
config.bedrock_api_key = ENV.fetch('AWS_ACCESS_KEY_ID', nil)
34+
config.bedrock_secret_key = ENV.fetch('AWS_SECRET_ACCESS_KEY', nil)
35+
config.bedrock_region = ENV.fetch('AWS_REGION', nil)
36+
config.bedrock_session_token = ENV.fetch('AWS_SESSION_TOKEN', nil)
3337
end
3438
```
3539

‎docs/guides/rails.md‎

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -92,10 +92,14 @@ In an initializer (e.g., `config/initializers/ruby_llm.rb`):
9292

9393
```ruby
9494
RubyLLM.configure do |config|
95-
config.openai_api_key = ENV['OPENAI_API_KEY']
96-
config.anthropic_api_key = ENV['ANTHROPIC_API_KEY']
97-
config.gemini_api_key = ENV['GEMINI_API_KEY']
98-
config.deepseek_api_key = ENV['DEEPSEEK_API_KEY']
95+
config.openai_api_key = ENV.fetch('OPENAI_API_KEY', nil)
96+
config.anthropic_api_key = ENV.fetch('ANTHROPIC_API_KEY', nil)
97+
config.gemini_api_key = ENV.fetch('GEMINI_API_KEY', nil)
98+
config.deepseek_api_key = ENV.fetch('DEEPSEEK_API_KEY', nil)
99+
config.bedrock_api_key = ENV.fetch('AWS_ACCESS_KEY_ID', nil)
100+
config.bedrock_secret_key = ENV.fetch('AWS_SECRET_ACCESS_KEY', nil)
101+
config.bedrock_region = ENV.fetch('AWS_REGION', nil)
102+
config.bedrock_session_token = ENV.fetch('AWS_SESSION_TOKEN', nil)
99103
end
100104
```
101105

‎docs/installation.md‎

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -51,10 +51,14 @@ require 'ruby_llm'
5151

5252
RubyLLM.configure do |config|
5353
# Required: At least one API key
54-
config.openai_api_key = ENV['OPENAI_API_KEY']
55-
config.anthropic_api_key = ENV['ANTHROPIC_API_KEY']
56-
config.gemini_api_key = ENV['GEMINI_API_KEY']
57-
config.deepseek_api_key = ENV['DEEPSEEK_API_KEY']
54+
config.openai_api_key = ENV.fetch('OPENAI_API_KEY', nil)
55+
config.anthropic_api_key = ENV.fetch('ANTHROPIC_API_KEY', nil)
56+
config.gemini_api_key = ENV.fetch('GEMINI_API_KEY', nil)
57+
config.deepseek_api_key = ENV.fetch('DEEPSEEK_API_KEY', nil)
58+
config.bedrock_api_key = ENV.fetch('AWS_ACCESS_KEY_ID', nil)
59+
config.bedrock_secret_key = ENV.fetch('AWS_SECRET_ACCESS_KEY', nil)
60+
config.bedrock_region = ENV.fetch('AWS_REGION', nil)
61+
config.bedrock_session_token = ENV.fetch('AWS_SESSION_TOKEN', nil)
5862

5963
# Optional: Set default models
6064
config.default_model = 'gpt-4o-mini' # Default chat model

‎lib/ruby_llm.rb‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,8 @@
1515
'llm' => 'LLM',
1616
'openai' => 'OpenAI',
1717
'api' => 'API',
18-
'deepseek' => 'DeepSeek'
18+
'deepseek' => 'DeepSeek',
19+
'bedrock' => 'Bedrock'
1920
)
2021
loader.ignore("#{__dir__}/tasks")
2122
loader.ignore("#{__dir__}/ruby_llm/railtie")
@@ -71,6 +72,7 @@ def logger
7172
RubyLLM::Provider.register :anthropic, RubyLLM::Providers::Anthropic
7273
RubyLLM::Provider.register :gemini, RubyLLM::Providers::Gemini
7374
RubyLLM::Provider.register :deepseek, RubyLLM::Providers::DeepSeek
75+
RubyLLM::Provider.register :bedrock, RubyLLM::Providers::Bedrock
7476

7577
if defined?(Rails::Railtie)
7678
require 'ruby_llm/railtie'

‎lib/ruby_llm/aliases.json‎

Lines changed: 22 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,24 +1,39 @@
11
{
22
"claude-3-5-sonnet": {
3-
"anthropic": "claude-3-5-sonnet-20241022"
3+
"anthropic": "claude-3-5-sonnet-20241022",
4+
"bedrock": "anthropic.claude-3-5-sonnet-20241022-v2:0"
45
},
56
"claude-3-5-haiku": {
6-
"anthropic": "claude-3-5-haiku-20241022"
7+
"anthropic": "claude-3-5-haiku-20241022",
8+
"bedrock": "anthropic.claude-3-5-haiku-20241022-v1:0"
79
},
810
"claude-3-7-sonnet": {
9-
"anthropic": "claude-3-7-sonnet-20250219"
11+
"anthropic": "claude-3-7-sonnet-20250219",
12+
"bedrock": "us.anthropic.claude-3-7-sonnet-20250219-v1:0"
1013
},
1114
"claude-3-opus": {
12-
"anthropic": "claude-3-opus-20240229"
15+
"anthropic": "claude-3-opus-20240229",
16+
"bedrock": "anthropic.claude-3-opus-20240229-v1:0"
1317
},
1418
"claude-3-sonnet": {
15-
"anthropic": "claude-3-sonnet-20240229"
19+
"anthropic": "claude-3-sonnet-20240229",
20+
"bedrock": "anthropic.claude-3-sonnet-20240229-v1:0"
1621
},
1722
"claude-3-haiku": {
18-
"anthropic": "claude-3-haiku-20240307"
23+
"anthropic": "claude-3-haiku-20240307",
24+
"bedrock": "anthropic.claude-3-haiku-20240307-v1:0"
1925
},
2026
"claude-3": {
21-
"anthropic": "claude-3-sonnet-20240229"
27+
"anthropic": "claude-3-sonnet-20240229",
28+
"bedrock": "anthropic.claude-3-sonnet-20240229-v1:0"
29+
},
30+
"claude-2": {
31+
"anthropic": "claude-2.0",
32+
"bedrock": "anthropic.claude-2.0"
33+
},
34+
"claude-2-1": {
35+
"anthropic": "claude-2.1",
36+
"bedrock": "anthropic.claude-2.1"
2237
},
2338
"gpt-4o": {
2439
"openai": "gpt-4o-2024-11-20"

‎lib/ruby_llm/configuration.rb‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,10 @@ class Configuration
1414
:anthropic_api_key,
1515
:gemini_api_key,
1616
:deepseek_api_key,
17+
:bedrock_api_key,
18+
:bedrock_secret_key,
19+
:bedrock_region,
20+
:bedrock_session_token,
1721
:default_model,
1822
:default_embedding_model,
1923
:default_image_model,

0 commit comments

Comments
 (0)