cloudflare-ai 0.6.0 → 0.7.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8f1d5e2776ba81382b1cb02355ac23348bee4a661d141c7c144136f1b9547eed
-  data.tar.gz: b651a6faed4d18a9b00cfed3f2217bc3cdc47ff9a2f9371cf50ab9c1f6bfab6f
+  metadata.gz: 2f16a1fd5c112fb69440d4d89b1f27d167e854c537ca952b91e1f3f673c0efb5
+  data.tar.gz: 59f999e994f5bd21071ee2f125b5e138ba353feb857912b1df472f35acc05006
 SHA512:
-  metadata.gz: 2fe5ec654b60e662113f79d2e66adf0483c17c0b2ffd2ea483c8f3b45c0d2ae747605b831e5cf6eed9ea8cd7ae614e6992199cf3fcad41870cb8ca8e269cb048
-  data.tar.gz: 3f4eac6cea6d8cdd00a00f71c3eca95ccfb61b2de8b2f45a4dfa46a7d81b45978eb5c92883569c8fd4398874005d0bad456096baa01cb7c82e22587b9700dc2a
+  metadata.gz: 1d258b49887ca664f616aad4dbc74afc4bb8d49d0452933c2ee907306af56cf81da0bf084c260ddbe4db9d96f522812cc5abfb771c3d94757f12342334b16fd6
+  data.tar.gz: 133fcfd21cedbdb324604ec0e9bd4aacb0750b9f486df7bc217998e4dd117f4e6d482572ba1feee8b629e3362857cb8b8f28d2f1e3c6f1d89acddd18b73752e9

data/README.md

@@ -24,7 +24,7 @@ It's still early days, and here are my immediate priorities:
   * [x] [Text Classification](https://developers.cloudflare.com/workers-ai/models/text-classification/)
   * [x] [Translation](https://developers.cloudflare.com/workers-ai/models/translation/)
   * [x] [Image Classification](https://developers.cloudflare.com/workers-ai/models/image-classification/)
-  * [ ] [Text-to-Image](https://developers.cloudflare.com/workers-ai/models/text-to-image/)
+  * [x] [Text-to-Image](https://developers.cloudflare.com/workers-ai/models/text-to-image/)
   * [ ] [Automatic Speech Recognition](https://developers.cloudflare.com/workers-ai/models/speech-recognition/)
 
 # Table of Contents
@@ -164,7 +164,16 @@ p result.result # => {"result":[{"label":"TABBY","score":0.6159140467643738},{"l
 ```
 
 #### Result object
-All invocations of the `classify` methods return a `Cloudflare::AI::Results::TextClassification`.
+All invocations of the `classify` method returns a `Cloudflare::AI::Results::TextClassification`.
+
+### Text to Image
+```ruby
+result = client.draw(prompt: "robot with blue eyes")
+p result.result # => File:0x0000000110deed68 (png tempfile)
+```
+
+#### Result object
+All invocations of the `draw` method returns a `Cloudflare::AI::Results::TextToImage`.
 
 ### Translation
 ```ruby

data/lib/cloudflare/ai/client.rb

@@ -41,6 +41,17 @@ class Cloudflare::AI::Client
     post_streamable_request(url, payload, &block)
   end
 
+  def draw(prompt:, num_steps: 20, model_name: Cloudflare::AI::Models.text_to_image.first)
+    url = service_url_for(account_id: account_id, model_name: model_name)
+    payload = {prompt: prompt, num_steps: num_steps}.to_json
+
+    result = connection.post(url, payload).body
+    binary_data = result.split(",").map(&:to_i).pack("C*")
+    Cloudflare::AI::Results::TextToImage.new(
+      Tempfile.new(["cloudflare-ai", ".png"], binmode: true).tap { |result| result.write(binary_data) }
+    )
+  end
+
   def embed(text:, model_name: Cloudflare::AI::Models.text_embedding.first)
     url = service_url_for(account_id: account_id, model_name: model_name)
     payload = {text: text}.to_json

data/lib/cloudflare/ai/result.rb

@@ -30,7 +30,7 @@ class Cloudflare::AI::Result
     result_data.to_json
   end
 
-  private
+  protected
 
   attr_reader :result_data
 

data/lib/cloudflare/ai/results/text_to_image.rb

@@ -0,0 +1,5 @@
+class Cloudflare::AI::Results::TextToImage < Cloudflare::AI::Result
+  def initialize(file)
+    @result_data = {result: file, success: true, errors: [], messages: []}
+  end
+end

data/lib/cloudflare/ai/version.rb

@@ -2,6 +2,6 @@
 
 module Cloudflare
   module AI
-    VERSION = "0.6.0"
+    VERSION = "0.7.0"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: cloudflare-ai
 version: !ruby/object:Gem::Version
-  version: 0.6.0
+  version: 0.7.0
 platform: ruby
 authors:
 - Ajay Krishnan
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-01-24 00:00:00.000000000 Z
+date: 2024-01-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activemodel
@@ -117,6 +117,7 @@ files:
 - lib/cloudflare/ai/results/text_classification.rb
 - lib/cloudflare/ai/results/text_embedding.rb
 - lib/cloudflare/ai/results/text_generation.rb
+- lib/cloudflare/ai/results/text_to_image.rb
 - lib/cloudflare/ai/results/translation.rb
 - lib/cloudflare/ai/version.rb
 homepage: https://rubygems.org/gems/cloudflare-ai