ruby-amazon-bedrock 0.1.1

Sign up to get free protection for your applications and to get access to all the features.
Files changed (89) hide show
  1. checksums.yaml +7 -0
  2. data/.circleci/config.yml +57 -0
  3. data/.gitignore +12 -0
  4. data/.rspec +3 -0
  5. data/.rubocop.yml +28 -0
  6. data/CHANGELOG.md +9 -0
  7. data/CODE_OF_CONDUCT.md +74 -0
  8. data/CONTRIBUTING.md +3 -0
  9. data/Gemfile +16 -0
  10. data/Gemfile.lock +109 -0
  11. data/LICENSE.txt +21 -0
  12. data/README.md +329 -0
  13. data/Rakefile +12 -0
  14. data/bin/console +12 -0
  15. data/bin/setup +8 -0
  16. data/lib/bedrock/version.rb +9 -0
  17. data/lib/bedrock.rb +10 -0
  18. data/lib/bedrock_runtime/client.rb +41 -0
  19. data/lib/bedrock_runtime/payload_builders/ai_21_labs/base.rb +55 -0
  20. data/lib/bedrock_runtime/payload_builders/ai_21_labs/j2_mid_v1.rb +22 -0
  21. data/lib/bedrock_runtime/payload_builders/ai_21_labs/j2_ultra_v1.rb +22 -0
  22. data/lib/bedrock_runtime/payload_builders/amazon/base.rb +51 -0
  23. data/lib/bedrock_runtime/payload_builders/amazon/titan_image_generator_v1.rb +45 -0
  24. data/lib/bedrock_runtime/payload_builders/amazon/titan_text_express_v1.rb +23 -0
  25. data/lib/bedrock_runtime/payload_builders/amazon/titan_text_lite_v1.rb +31 -0
  26. data/lib/bedrock_runtime/payload_builders/anthropic/base.rb +55 -0
  27. data/lib/bedrock_runtime/payload_builders/anthropic/claude_instant_v1.rb +21 -0
  28. data/lib/bedrock_runtime/payload_builders/anthropic/claude_v1.rb +23 -0
  29. data/lib/bedrock_runtime/payload_builders/anthropic/claude_v2.rb +21 -0
  30. data/lib/bedrock_runtime/payload_builders/base.rb +33 -0
  31. data/lib/bedrock_runtime/payload_builders/cohere/command_base.rb +46 -0
  32. data/lib/bedrock_runtime/payload_builders/cohere/command_light_text_v14.rb +23 -0
  33. data/lib/bedrock_runtime/payload_builders/cohere/command_text_v14.rb +21 -0
  34. data/lib/bedrock_runtime/payload_builders/cohere/embed_base.rb +41 -0
  35. data/lib/bedrock_runtime/payload_builders/cohere/embed_english_v3.rb +23 -0
  36. data/lib/bedrock_runtime/payload_builders/cohere/embed_multilingual_v3.rb +21 -0
  37. data/lib/bedrock_runtime/payload_builders/meta/base.rb +45 -0
  38. data/lib/bedrock_runtime/payload_builders/meta/llama213b_chat_v1.rb +23 -0
  39. data/lib/bedrock_runtime/payload_builders/meta/llama270b_chat_v1.rb +21 -0
  40. data/lib/bedrock_runtime/payload_builders/stability_ai/base.rb +48 -0
  41. data/lib/bedrock_runtime/payload_builders/stability_ai/stable_diffusion_xl_v0.rb +23 -0
  42. data/lib/bedrock_runtime/payload_builders/stability_ai/stable_diffusion_xl_v1.rb +23 -0
  43. data/lib/bedrock_runtime/payload_factory.rb +72 -0
  44. data/ruby-amazon-bedrock.gemspec +39 -0
  45. data/sig/ruby/amazon/bedrock.rbs +8 -0
  46. data/spec/bedrock_runtime/client_spec.rb +51 -0
  47. data/spec/bedrock_runtime/payload_builders/ai_21_labs/base_spec.rb +32 -0
  48. data/spec/bedrock_runtime/payload_builders/ai_21_labs/j2_mid_v1_spec.rb +33 -0
  49. data/spec/bedrock_runtime/payload_builders/ai_21_labs/j2_ultra_v1_spec.rb +33 -0
  50. data/spec/bedrock_runtime/payload_builders/amazon/base_spec.rb +31 -0
  51. data/spec/bedrock_runtime/payload_builders/amazon/titan_image_generator_v1_spec.rb +38 -0
  52. data/spec/bedrock_runtime/payload_builders/amazon/titan_text_express_v1_spec.rb +32 -0
  53. data/spec/bedrock_runtime/payload_builders/amazon/titan_text_lite_v1_spec.rb +32 -0
  54. data/spec/bedrock_runtime/payload_builders/anthropic/base_spec.rb +33 -0
  55. data/spec/bedrock_runtime/payload_builders/anthropic/claude_instant_v1_spec.rb +34 -0
  56. data/spec/bedrock_runtime/payload_builders/anthropic/claude_v1_spec.rb +34 -0
  57. data/spec/bedrock_runtime/payload_builders/anthropic/claude_v2_spec.rb +34 -0
  58. data/spec/bedrock_runtime/payload_builders/base_spec.rb +34 -0
  59. data/spec/bedrock_runtime/payload_builders/cohere/command_base_spec.rb +27 -0
  60. data/spec/bedrock_runtime/payload_builders/cohere/command_light_text_v14_spec.rb +28 -0
  61. data/spec/bedrock_runtime/payload_builders/cohere/command_text_v14_spec.rb +28 -0
  62. data/spec/bedrock_runtime/payload_builders/cohere/embed_base_spec.rb +26 -0
  63. data/spec/bedrock_runtime/payload_builders/cohere/embed_english_v3_spec.rb +27 -0
  64. data/spec/bedrock_runtime/payload_builders/cohere/embed_multilingual_v3_spec.rb +27 -0
  65. data/spec/bedrock_runtime/payload_builders/meta/base_spec.rb +28 -0
  66. data/spec/bedrock_runtime/payload_builders/meta/llama213b_chat_v1_spec.rb +29 -0
  67. data/spec/bedrock_runtime/payload_builders/meta/llama270b_chat_v1_spec.rb +29 -0
  68. data/spec/bedrock_runtime/payload_builders/stability_ai/base_spec.rb +30 -0
  69. data/spec/bedrock_runtime/payload_builders/stability_ai/stable_diffusion_xl_v0_spec.rb +31 -0
  70. data/spec/bedrock_runtime/payload_builders/stability_ai/stable_diffusion_xl_v1_spec.rb +31 -0
  71. data/spec/bedrock_runtime/payload_factory_spec.rb +56 -0
  72. data/spec/bedrock_spec.rb +7 -0
  73. data/spec/cassettes/models/amazon_titan-text-express-v1.yml +59 -0
  74. data/spec/cassettes/models/amazon_titan-text-lite-v1.yml +54 -0
  75. data/spec/cassettes/models/anthropic_claude-instant-v1.yml +62 -0
  76. data/spec/cassettes/models/anthropic_claude-v1.yml +61 -0
  77. data/spec/cassettes/models/anthropic_claude-v2.yml +59 -0
  78. data/spec/cassettes/models/cohere_command-light-text-v14.yml +54 -0
  79. data/spec/cassettes/models/cohere_command-text-v14.yml +55 -0
  80. data/spec/cassettes/models/cohere_embed-english-v3.yml +47 -0
  81. data/spec/cassettes/models/cohere_embed-multilingual-v3.yml +47 -0
  82. data/spec/cassettes/models/errors/bad_request.yml +45 -0
  83. data/spec/cassettes/models/errors/not_found.yml +45 -0
  84. data/spec/cassettes/models/meta_llama2-13b-chat-v1.yml +73 -0
  85. data/spec/cassettes/models/meta_llama2-70b-chat-v1.yml +55 -0
  86. data/spec/cassettes/models/stability_stable-diffusion-xl-v0.yml +51 -0
  87. data/spec/cassettes/models/stability_stable-diffusion-xl-v1.yml +45 -0
  88. data/spec/spec_helper.rb +29 -0
  89. metadata +153 -0
@@ -0,0 +1,29 @@
1
+ # frozen_string_literal: true
2
+
3
+ require 'bedrock'
4
+ require 'bedrock_runtime/client'
5
+ require 'bedrock_runtime/payload_factory'
6
+
7
+ require 'dotenv/load'
8
+ require 'simplecov'
9
+ require 'vcr'
10
+
11
+ SimpleCov.start
12
+
13
+ RSpec.configure do |config|
14
+ # Enable flags like --only-failures and --next-failure
15
+ config.example_status_persistence_file_path = '.rspec_status'
16
+
17
+ # Disable RSpec exposing methods globally on `Module` and `main`
18
+ config.disable_monkey_patching!
19
+
20
+ config.expect_with :rspec do |c|
21
+ c.syntax = :expect
22
+ end
23
+ end
24
+
25
+ VCR.configure do |config|
26
+ config.cassette_library_dir = 'spec/cassettes'
27
+ config.hook_into :webmock # or :faraday if you're using Faraday
28
+ config.configure_rspec_metadata!
29
+ end
metadata ADDED
@@ -0,0 +1,153 @@
1
+ --- !ruby/object:Gem::Specification
2
+ name: ruby-amazon-bedrock
3
+ version: !ruby/object:Gem::Version
4
+ version: 0.1.1
5
+ platform: ruby
6
+ authors:
7
+ - Alvaro Delgado
8
+ autorequire:
9
+ bindir: exe
10
+ cert_chain: []
11
+ date: 2023-12-13 00:00:00.000000000 Z
12
+ dependencies:
13
+ - !ruby/object:Gem::Dependency
14
+ name: aws-sdk-bedrockruntime
15
+ requirement: !ruby/object:Gem::Requirement
16
+ requirements:
17
+ - - "~>"
18
+ - !ruby/object:Gem::Version
19
+ version: '1.0'
20
+ type: :runtime
21
+ prerelease: false
22
+ version_requirements: !ruby/object:Gem::Requirement
23
+ requirements:
24
+ - - "~>"
25
+ - !ruby/object:Gem::Version
26
+ version: '1.0'
27
+ description: The ruby-amazon-bedrock gem offers Ruby developers an efficient and user-friendly
28
+ interface to Amazon Bedrock, a powerful library for AI-driven text and image
29
+ generation. This gem simplifies the process of connecting to Amazon Bedrock's
30
+ APIs, enabling developers to easily harness the capabilities of advanced machine learning
31
+ models for generating high-quality text and images.
32
+ email:
33
+ - hola@alvarodelgado.dev
34
+ executables: []
35
+ extensions: []
36
+ extra_rdoc_files: []
37
+ files:
38
+ - ".circleci/config.yml"
39
+ - ".gitignore"
40
+ - ".rspec"
41
+ - ".rubocop.yml"
42
+ - CHANGELOG.md
43
+ - CODE_OF_CONDUCT.md
44
+ - CONTRIBUTING.md
45
+ - Gemfile
46
+ - Gemfile.lock
47
+ - LICENSE.txt
48
+ - README.md
49
+ - Rakefile
50
+ - bin/console
51
+ - bin/setup
52
+ - lib/bedrock.rb
53
+ - lib/bedrock/version.rb
54
+ - lib/bedrock_runtime/client.rb
55
+ - lib/bedrock_runtime/payload_builders/ai_21_labs/base.rb
56
+ - lib/bedrock_runtime/payload_builders/ai_21_labs/j2_mid_v1.rb
57
+ - lib/bedrock_runtime/payload_builders/ai_21_labs/j2_ultra_v1.rb
58
+ - lib/bedrock_runtime/payload_builders/amazon/base.rb
59
+ - lib/bedrock_runtime/payload_builders/amazon/titan_image_generator_v1.rb
60
+ - lib/bedrock_runtime/payload_builders/amazon/titan_text_express_v1.rb
61
+ - lib/bedrock_runtime/payload_builders/amazon/titan_text_lite_v1.rb
62
+ - lib/bedrock_runtime/payload_builders/anthropic/base.rb
63
+ - lib/bedrock_runtime/payload_builders/anthropic/claude_instant_v1.rb
64
+ - lib/bedrock_runtime/payload_builders/anthropic/claude_v1.rb
65
+ - lib/bedrock_runtime/payload_builders/anthropic/claude_v2.rb
66
+ - lib/bedrock_runtime/payload_builders/base.rb
67
+ - lib/bedrock_runtime/payload_builders/cohere/command_base.rb
68
+ - lib/bedrock_runtime/payload_builders/cohere/command_light_text_v14.rb
69
+ - lib/bedrock_runtime/payload_builders/cohere/command_text_v14.rb
70
+ - lib/bedrock_runtime/payload_builders/cohere/embed_base.rb
71
+ - lib/bedrock_runtime/payload_builders/cohere/embed_english_v3.rb
72
+ - lib/bedrock_runtime/payload_builders/cohere/embed_multilingual_v3.rb
73
+ - lib/bedrock_runtime/payload_builders/meta/base.rb
74
+ - lib/bedrock_runtime/payload_builders/meta/llama213b_chat_v1.rb
75
+ - lib/bedrock_runtime/payload_builders/meta/llama270b_chat_v1.rb
76
+ - lib/bedrock_runtime/payload_builders/stability_ai/base.rb
77
+ - lib/bedrock_runtime/payload_builders/stability_ai/stable_diffusion_xl_v0.rb
78
+ - lib/bedrock_runtime/payload_builders/stability_ai/stable_diffusion_xl_v1.rb
79
+ - lib/bedrock_runtime/payload_factory.rb
80
+ - ruby-amazon-bedrock.gemspec
81
+ - sig/ruby/amazon/bedrock.rbs
82
+ - spec/bedrock_runtime/client_spec.rb
83
+ - spec/bedrock_runtime/payload_builders/ai_21_labs/base_spec.rb
84
+ - spec/bedrock_runtime/payload_builders/ai_21_labs/j2_mid_v1_spec.rb
85
+ - spec/bedrock_runtime/payload_builders/ai_21_labs/j2_ultra_v1_spec.rb
86
+ - spec/bedrock_runtime/payload_builders/amazon/base_spec.rb
87
+ - spec/bedrock_runtime/payload_builders/amazon/titan_image_generator_v1_spec.rb
88
+ - spec/bedrock_runtime/payload_builders/amazon/titan_text_express_v1_spec.rb
89
+ - spec/bedrock_runtime/payload_builders/amazon/titan_text_lite_v1_spec.rb
90
+ - spec/bedrock_runtime/payload_builders/anthropic/base_spec.rb
91
+ - spec/bedrock_runtime/payload_builders/anthropic/claude_instant_v1_spec.rb
92
+ - spec/bedrock_runtime/payload_builders/anthropic/claude_v1_spec.rb
93
+ - spec/bedrock_runtime/payload_builders/anthropic/claude_v2_spec.rb
94
+ - spec/bedrock_runtime/payload_builders/base_spec.rb
95
+ - spec/bedrock_runtime/payload_builders/cohere/command_base_spec.rb
96
+ - spec/bedrock_runtime/payload_builders/cohere/command_light_text_v14_spec.rb
97
+ - spec/bedrock_runtime/payload_builders/cohere/command_text_v14_spec.rb
98
+ - spec/bedrock_runtime/payload_builders/cohere/embed_base_spec.rb
99
+ - spec/bedrock_runtime/payload_builders/cohere/embed_english_v3_spec.rb
100
+ - spec/bedrock_runtime/payload_builders/cohere/embed_multilingual_v3_spec.rb
101
+ - spec/bedrock_runtime/payload_builders/meta/base_spec.rb
102
+ - spec/bedrock_runtime/payload_builders/meta/llama213b_chat_v1_spec.rb
103
+ - spec/bedrock_runtime/payload_builders/meta/llama270b_chat_v1_spec.rb
104
+ - spec/bedrock_runtime/payload_builders/stability_ai/base_spec.rb
105
+ - spec/bedrock_runtime/payload_builders/stability_ai/stable_diffusion_xl_v0_spec.rb
106
+ - spec/bedrock_runtime/payload_builders/stability_ai/stable_diffusion_xl_v1_spec.rb
107
+ - spec/bedrock_runtime/payload_factory_spec.rb
108
+ - spec/bedrock_spec.rb
109
+ - spec/cassettes/models/amazon_titan-text-express-v1.yml
110
+ - spec/cassettes/models/amazon_titan-text-lite-v1.yml
111
+ - spec/cassettes/models/anthropic_claude-instant-v1.yml
112
+ - spec/cassettes/models/anthropic_claude-v1.yml
113
+ - spec/cassettes/models/anthropic_claude-v2.yml
114
+ - spec/cassettes/models/cohere_command-light-text-v14.yml
115
+ - spec/cassettes/models/cohere_command-text-v14.yml
116
+ - spec/cassettes/models/cohere_embed-english-v3.yml
117
+ - spec/cassettes/models/cohere_embed-multilingual-v3.yml
118
+ - spec/cassettes/models/errors/bad_request.yml
119
+ - spec/cassettes/models/errors/not_found.yml
120
+ - spec/cassettes/models/meta_llama2-13b-chat-v1.yml
121
+ - spec/cassettes/models/meta_llama2-70b-chat-v1.yml
122
+ - spec/cassettes/models/stability_stable-diffusion-xl-v0.yml
123
+ - spec/cassettes/models/stability_stable-diffusion-xl-v1.yml
124
+ - spec/spec_helper.rb
125
+ homepage: https://github.com/AAlvAAro/ruby-amazon-bedrock
126
+ licenses:
127
+ - MIT
128
+ metadata:
129
+ homepage_uri: https://github.com/AAlvAAro/ruby-amazon-bedrock
130
+ source_code_uri: https://github.com/AAlvAAro/ruby-amazon-bedrock
131
+ changelog_uri: https://github.com/AAlvAAro/ruby-amazon-bedrock/blob/main/CHANGELOG.md
132
+ rubygems_mfa_required: 'true'
133
+ post_install_message:
134
+ rdoc_options: []
135
+ require_paths:
136
+ - lib
137
+ required_ruby_version: !ruby/object:Gem::Requirement
138
+ requirements:
139
+ - - ">="
140
+ - !ruby/object:Gem::Version
141
+ version: 2.6.0
142
+ required_rubygems_version: !ruby/object:Gem::Requirement
143
+ requirements:
144
+ - - ">="
145
+ - !ruby/object:Gem::Version
146
+ version: '0'
147
+ requirements: []
148
+ rubygems_version: 3.4.22
149
+ signing_key:
150
+ specification_version: 4
151
+ summary: Seamless Integration with Amazon Bedrock for AI-Powered Text and Image Generation
152
+ in Ruby.
153
+ test_files: []