gen-ai 0.2.2 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2b25d130c100818a73d652eaf9b585a4d024f6134a66de3f20eb5dd6e3d2212a
-  data.tar.gz: 7d971d676915f71edf74fec57cd7cb59b2d2c6607d7ed818d05d07b19f8cdecc
+  metadata.gz: e6fbef446e1383e14734c390b0e342433991858e95a083e0aaf9a31f844f4503
+  data.tar.gz: e8bda40ea6377b4ada7175bcabae430681ac6d707c75cce745e4a079ce17dba9
 SHA512:
-  metadata.gz: 98554f5f32925f6ea70ea78e8c462d3d24949fb265fed69ae5f73fc722206c455e81bec75a879aa21bf841b372518f6ffce5497ec3c82bf60260b4d4c8ed2525
-  data.tar.gz: 905b2aa2b483711de399aa166dc5c83f9e7a482a1649f77281c995adbcf0f3dc9847b3ecbc0bdc21028a0f79f4ffb770c0a09dbad6a02d2a191d206dfeb43712
+  metadata.gz: ade0a401f9c6cb8fdb1c1de1fd5e31b52a79ffc69af2267644f4ffe6f2bf8892d3f92bcf1a2e2aebfd1ea0790dab1dd3a8c5b42fa59e870bc0cbab13766ce1b9
+  data.tar.gz: 39826a4c7f854b2a0ffd24010306e612b2cad1465e60719ba518f5b2ee43a17338c268684e615b376388f280dd20740f750fc6c8e5b49dfbc822c1f2873ba9f6

data/README.md

@@ -20,13 +20,20 @@ If bundler is not being used to manage dependencies, install the gem by executin
 
 ## Usage
 
+Require it in you code:
+```ruby
+require 'gen_ai'
+```
+
+
+
 ### Feature support
 ✅ - Supported  |  ❌ - Not supported  |  🛠️ - Work in progress
 
 Language models capabilities
 
 | Provider         | Embedding | Completion | Conversation | Sentiment | Summarization |
-| ---------------- | --------- | :--------: | :----------: | :-------: | :-----------: |
+| ---------------- | :-------: | :--------: | :----------: | :-------: | :-----------: |
 | **OpenAI**       | ✅        | ✅         | ✅            | 🛠️        | 🛠️             |
 | **Google Palm2** | ✅        | ✅         | ✅            | 🛠️        | 🛠️             |
 
@@ -34,7 +41,7 @@ Language models capabilities
 Image generation model capabilities
 
 | Provider         | Generate  | Variations | Edit         | Upscale   |
-| ---------------- | --------- | :--------: | :----------: | :-------: |
+| ---------------- | :-------: | :--------: | :----------: | :-------: |
 | **OpenAI**       | ✅        | ✅         | ✅            | ❌        |
 | **StabilityAI**  | ✅        | ❌         | ✅            | ✅        |
 

data/lib/gen/ai.rb

@@ -0,0 +1,3 @@
+# frozen_string_literal: true
+
+require_relative '../gen_ai'

data/lib/gen_ai/image/open_ai.rb

@@ -4,6 +4,7 @@ module GenAI
   class Image
     class OpenAI < Base
       DEFAULT_SIZE = '256x256'
+      DEFAULT_MODEL = 'dall-e-3'
       RESPONSE_FORMAT = 'b64_json'
 
       def initialize(token:, options: {})
@@ -19,7 +20,7 @@ module GenAI
 
         build_result(
           raw: response,
-          model: 'dall-e',
+          model: parameters[:model],
           parsed: response['data'].map { |datum| datum[RESPONSE_FORMAT] }
         )
       end
@@ -54,6 +55,7 @@ module GenAI
         {
           prompt: prompt,
           size: options.delete(:size) || DEFAULT_SIZE,
+          model: options.delete(:model) || DEFAULT_MODEL,
           response_format: options.delete(:response_format) || RESPONSE_FORMAT
         }.merge(options)
       end

data/lib/gen_ai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module GenAI
-  VERSION = '0.2.2'
+  VERSION = '0.2.3'
 end

data/lib/gen_ai.rb

@@ -8,6 +8,7 @@ loader.inflector.inflect(
   'open_ai' => 'OpenAI',
   'stability_ai' => 'StabilityAI'
 )
+loader.ignore("#{__dir__}/gen")
 loader.setup
 
 module GenAI

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: gen-ai
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.2.3
 platform: ruby
 authors:
 - Alex Chaplinsky
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-10-29 00:00:00.000000000 Z
+date: 2023-11-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -96,6 +96,7 @@ files:
 - README.md
 - Rakefile
 - gen-ai.gemspec
+- lib/gen/ai.rb
 - lib/gen_ai.rb
 - lib/gen_ai/api/client.rb
 - lib/gen_ai/base.rb