ruby-amazon-bedrock 0.2.2 → 0.2.4

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (40) hide show
  1. checksums.yaml +4 -4
  2. data/.circleci/config.yml +0 -2
  3. data/.env.sample +5 -0
  4. data/.gitignore +2 -2
  5. data/CHANGELOG.md +12 -0
  6. data/Gemfile +1 -0
  7. data/Gemfile.lock +55 -41
  8. data/README.md +19 -11
  9. data/lib/amazon_bedrock/version.rb +1 -1
  10. data/lib/amazon_bedrock.rb +2 -1
  11. data/lib/bedrock_runtime/client.rb +13 -8
  12. data/lib/bedrock_runtime/payload_builders/ai_21_labs/base.rb +2 -2
  13. data/lib/bedrock_runtime/payload_builders/amazon/base.rb +1 -1
  14. data/lib/bedrock_runtime/payload_builders/meta/llama370b_instruct_v1.rb +21 -0
  15. data/lib/bedrock_runtime/payload_builders/meta/llama38b_instruct_v1.rb +21 -0
  16. data/lib/bedrock_runtime/payload_factory.rb +4 -0
  17. data/lib/bedrock_runtime/response_builders/ai_21_labs.rb +32 -0
  18. data/lib/bedrock_runtime/response_builders/{image.rb → amazon_image.rb} +42 -7
  19. data/lib/bedrock_runtime/response_builders/amazon_text.rb +32 -0
  20. data/lib/bedrock_runtime/response_builders/anthropic.rb +32 -0
  21. data/lib/bedrock_runtime/response_builders/cohere_command.rb +32 -0
  22. data/lib/bedrock_runtime/response_builders/cohere_embed.rb +32 -0
  23. data/lib/bedrock_runtime/response_builders/{text.rb → meta.rb} +11 -5
  24. data/lib/bedrock_runtime/response_builders/stability_ai.rb +85 -0
  25. data/lib/bedrock_runtime/response_factory.rb +44 -16
  26. data/ruby-amazon-bedrock.gemspec +2 -0
  27. data/spec/bedrock_runtime/client_spec.rb +4 -1
  28. data/spec/bedrock_runtime/payload_builders/ai_21_labs/base_spec.rb +1 -1
  29. data/spec/bedrock_runtime/payload_builders/ai_21_labs/j2_mid_v1_spec.rb +1 -1
  30. data/spec/bedrock_runtime/payload_builders/ai_21_labs/j2_ultra_v1_spec.rb +1 -1
  31. data/spec/bedrock_runtime/payload_builders/meta/llama370b_instruct_v1_spec.rb +26 -0
  32. data/spec/bedrock_runtime/payload_builders/meta/llama38b_instruct_v1_spec.rb +26 -0
  33. data/spec/bedrock_runtime/payload_factory_spec.rb +3 -1
  34. data/spec/bedrock_runtime/response_builders/{text_spec.rb → ai_21_labs_spec.rb} +3 -4
  35. data/spec/bedrock_runtime/response_builders/{image_spec.rb → stability_ai_spec.rb} +2 -2
  36. data/spec/bedrock_runtime/response_factory_spec.rb +74 -14
  37. data/spec/cassettes/models/meta_llama3-70b-instruct-v1_0.yml +73 -0
  38. data/spec/cassettes/models/meta_llama3-8b-instruct-v1_0.yml +71 -0
  39. data/spec/support/helpers.rb +1 -3
  40. metadata +48 -7
@@ -1,7 +1,5 @@
1
1
  # frozen_string_literal: true
2
2
 
3
- require 'pry'
4
-
5
3
  RSpec.shared_examples 'a payload builder' do
6
4
  it 'validates the payload structure' do
7
5
  payload_builder = described_class.new(prompt, options)
@@ -28,7 +26,7 @@ RSpec.shared_context 'a121 labs parameters' do
28
26
  let(:body) do
29
27
  {
30
28
  prompt: prompt,
31
- maxTokenCount: 100,
29
+ maxTokens: 100,
32
30
  temperature: 0.5,
33
31
  topP: 2,
34
32
  stopSequences: %w[stop1 stop2],
metadata CHANGED
@@ -1,14 +1,14 @@
1
1
  --- !ruby/object:Gem::Specification
2
2
  name: ruby-amazon-bedrock
3
3
  version: !ruby/object:Gem::Version
4
- version: 0.2.2
4
+ version: 0.2.4
5
5
  platform: ruby
6
6
  authors:
7
7
  - Alvaro Delgado
8
8
  autorequire:
9
9
  bindir: exe
10
10
  cert_chain: []
11
- date: 2024-01-11 00:00:00.000000000 Z
11
+ date: 2024-09-17 00:00:00.000000000 Z
12
12
  dependencies:
13
13
  - !ruby/object:Gem::Dependency
14
14
  name: aws-sdk-bedrockruntime
@@ -24,6 +24,34 @@ dependencies:
24
24
  - - "~>"
25
25
  - !ruby/object:Gem::Version
26
26
  version: '1.0'
27
+ - !ruby/object:Gem::Dependency
28
+ name: aws-sdk-s3
29
+ requirement: !ruby/object:Gem::Requirement
30
+ requirements:
31
+ - - "~>"
32
+ - !ruby/object:Gem::Version
33
+ version: '1.0'
34
+ type: :runtime
35
+ prerelease: false
36
+ version_requirements: !ruby/object:Gem::Requirement
37
+ requirements:
38
+ - - "~>"
39
+ - !ruby/object:Gem::Version
40
+ version: '1.0'
41
+ - !ruby/object:Gem::Dependency
42
+ name: base64
43
+ requirement: !ruby/object:Gem::Requirement
44
+ requirements:
45
+ - - "~>"
46
+ - !ruby/object:Gem::Version
47
+ version: '0.2'
48
+ type: :runtime
49
+ prerelease: false
50
+ version_requirements: !ruby/object:Gem::Requirement
51
+ requirements:
52
+ - - "~>"
53
+ - !ruby/object:Gem::Version
54
+ version: '0.2'
27
55
  description: The ruby-amazon-bedrock gem offers Ruby developers an efficient and user-friendly
28
56
  interface to Amazon Bedrock, a powerful library for AI-driven text and image
29
57
  generation. This gem simplifies the process of connecting to Amazon Bedrock's
@@ -36,6 +64,7 @@ extensions: []
36
64
  extra_rdoc_files: []
37
65
  files:
38
66
  - ".circleci/config.yml"
67
+ - ".env.sample"
39
68
  - ".gitignore"
40
69
  - ".rspec"
41
70
  - ".rubocop.yml"
@@ -76,12 +105,20 @@ files:
76
105
  - lib/bedrock_runtime/payload_builders/meta/base.rb
77
106
  - lib/bedrock_runtime/payload_builders/meta/llama213b_chat_v1.rb
78
107
  - lib/bedrock_runtime/payload_builders/meta/llama270b_chat_v1.rb
108
+ - lib/bedrock_runtime/payload_builders/meta/llama370b_instruct_v1.rb
109
+ - lib/bedrock_runtime/payload_builders/meta/llama38b_instruct_v1.rb
79
110
  - lib/bedrock_runtime/payload_builders/stability_ai/base.rb
80
111
  - lib/bedrock_runtime/payload_builders/stability_ai/stable_diffusion_xl_v0.rb
81
112
  - lib/bedrock_runtime/payload_builders/stability_ai/stable_diffusion_xl_v1.rb
82
113
  - lib/bedrock_runtime/payload_factory.rb
83
- - lib/bedrock_runtime/response_builders/image.rb
84
- - lib/bedrock_runtime/response_builders/text.rb
114
+ - lib/bedrock_runtime/response_builders/ai_21_labs.rb
115
+ - lib/bedrock_runtime/response_builders/amazon_image.rb
116
+ - lib/bedrock_runtime/response_builders/amazon_text.rb
117
+ - lib/bedrock_runtime/response_builders/anthropic.rb
118
+ - lib/bedrock_runtime/response_builders/cohere_command.rb
119
+ - lib/bedrock_runtime/response_builders/cohere_embed.rb
120
+ - lib/bedrock_runtime/response_builders/meta.rb
121
+ - lib/bedrock_runtime/response_builders/stability_ai.rb
85
122
  - lib/bedrock_runtime/response_factory.rb
86
123
  - ruby-amazon-bedrock.gemspec
87
124
  - sig/ruby_amazon_bedrock.rbs
@@ -108,12 +145,14 @@ files:
108
145
  - spec/bedrock_runtime/payload_builders/meta/base_spec.rb
109
146
  - spec/bedrock_runtime/payload_builders/meta/llama213b_chat_v1_spec.rb
110
147
  - spec/bedrock_runtime/payload_builders/meta/llama270b_chat_v1_spec.rb
148
+ - spec/bedrock_runtime/payload_builders/meta/llama370b_instruct_v1_spec.rb
149
+ - spec/bedrock_runtime/payload_builders/meta/llama38b_instruct_v1_spec.rb
111
150
  - spec/bedrock_runtime/payload_builders/stability_ai/base_spec.rb
112
151
  - spec/bedrock_runtime/payload_builders/stability_ai/stable_diffusion_xl_v0_spec.rb
113
152
  - spec/bedrock_runtime/payload_builders/stability_ai/stable_diffusion_xl_v1_spec.rb
114
153
  - spec/bedrock_runtime/payload_factory_spec.rb
115
- - spec/bedrock_runtime/response_builders/image_spec.rb
116
- - spec/bedrock_runtime/response_builders/text_spec.rb
154
+ - spec/bedrock_runtime/response_builders/ai_21_labs_spec.rb
155
+ - spec/bedrock_runtime/response_builders/stability_ai_spec.rb
117
156
  - spec/bedrock_runtime/response_factory_spec.rb
118
157
  - spec/cassettes/models/amazon_titan-text-express-v1.yml
119
158
  - spec/cassettes/models/amazon_titan-text-lite-v1.yml
@@ -128,6 +167,8 @@ files:
128
167
  - spec/cassettes/models/errors/not_found.yml
129
168
  - spec/cassettes/models/meta_llama2-13b-chat-v1.yml
130
169
  - spec/cassettes/models/meta_llama2-70b-chat-v1.yml
170
+ - spec/cassettes/models/meta_llama3-70b-instruct-v1_0.yml
171
+ - spec/cassettes/models/meta_llama3-8b-instruct-v1_0.yml
131
172
  - spec/cassettes/models/stability_stable-diffusion-xl-v0.yml
132
173
  - spec/cassettes/models/stability_stable-diffusion-xl-v1.yml
133
174
  - spec/fixtures/sample_image.jpg
@@ -156,7 +197,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
156
197
  - !ruby/object:Gem::Version
157
198
  version: '0'
158
199
  requirements: []
159
- rubygems_version: 3.4.22
200
+ rubygems_version: 3.2.3
160
201
  signing_key:
161
202
  specification_version: 4
162
203
  summary: Seamless Integration with Amazon Bedrock for AI-Powered Text and Image Generation