replicate-rails 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 187cc3b85a25e4c4affcf809fd9287f86eb3a59e7cd972fb4efa4ec05bf683e6
-  data.tar.gz: 9e11f3b84e40377fab28a1124cc5e2cfdcb076c0c229afc27e4b7d6315894a16
+  metadata.gz: d67d82a77422c0855515d26493290cb270a60a02c80be16417dba9a21cdf434d
+  data.tar.gz: 3a9f6b98554f066aee9a886c066547b9d0a22976c825cad22b758a43e30ac756
 SHA512:
-  metadata.gz: 93ff4864a34282eef64be50f62ee52ebbe4f62ec00a5662ad6a3805b4e4621fe8c29d735836aa211395a4a90828ed07fb677426381cbdac3c95c7a4f772fca17
-  data.tar.gz: b3f43be20e66a254406b7f0221ea949e38e0894a239821db5e4cab03f11063e983c645feed158fcf6f2c055f765ecd8577bb8ef7c20ace1d224b984deaeb3292
+  metadata.gz: e3766b6a8e3abf0d042389bf903901bbe955e24354da3e9d6a6057d70a8875493de676f5eac84b5fffcedd413d32cd029d8755ac1b690eb7a850f142374a7cb4
+  data.tar.gz: cc7fc5ce9377ac2c1c3677d96b02516d47f290ed7cc75019c1be3ae26fd5176397efe73d1f24768c8705a8bf99c2a2ac0ba8fbce6f4cc1c5a04fe7a6b5f6b3ae

data/Gemfile

@@ -1,12 +1,4 @@
 # frozen_string_literal: true
 
 source "https://rubygems.org"
-
-# Specify your gem's dependencies in replicate-rails.gemspec
 gemspec
-
-gem "rake", "~> 13.0"
-
-gem "minitest", "~> 5.0"
-
-gem "rubocop", "~> 1.21"

data/Gemfile.lock

@@ -1,25 +1,119 @@
 PATH
   remote: .
   specs:
-    replicate-rails (0.1.0)
+    replicate-rails (0.1.1)
+      activesupport (>= 3.1)
       replicate-ruby (~> 0.1)
 
 GEM
   remote: https://rubygems.org/
   specs:
+    actioncable (7.0.4)
+      actionpack (= 7.0.4)
+      activesupport (= 7.0.4)
+      nio4r (~> 2.0)
+      websocket-driver (>= 0.6.1)
+    actionmailbox (7.0.4)
+      actionpack (= 7.0.4)
+      activejob (= 7.0.4)
+      activerecord (= 7.0.4)
+      activestorage (= 7.0.4)
+      activesupport (= 7.0.4)
+      mail (>= 2.7.1)
+      net-imap
+      net-pop
+      net-smtp
+    actionmailer (7.0.4)
+      actionpack (= 7.0.4)
+      actionview (= 7.0.4)
+      activejob (= 7.0.4)
+      activesupport (= 7.0.4)
+      mail (~> 2.5, >= 2.5.4)
+      net-imap
+      net-pop
+      net-smtp
+      rails-dom-testing (~> 2.0)
+    actionpack (7.0.4)
+      actionview (= 7.0.4)
+      activesupport (= 7.0.4)
+      rack (~> 2.0, >= 2.2.0)
+      rack-test (>= 0.6.3)
+      rails-dom-testing (~> 2.0)
+      rails-html-sanitizer (~> 1.0, >= 1.2.0)
+    actiontext (7.0.4)
+      actionpack (= 7.0.4)
+      activerecord (= 7.0.4)
+      activestorage (= 7.0.4)
+      activesupport (= 7.0.4)
+      globalid (>= 0.6.0)
+      nokogiri (>= 1.8.5)
+    actionview (7.0.4)
+      activesupport (= 7.0.4)
+      builder (~> 3.1)
+      erubi (~> 1.4)
+      rails-dom-testing (~> 2.0)
+      rails-html-sanitizer (~> 1.1, >= 1.2.0)
+    activejob (7.0.4)
+      activesupport (= 7.0.4)
+      globalid (>= 0.3.6)
+    activemodel (7.0.4)
+      activesupport (= 7.0.4)
+    activerecord (7.0.4)
+      activemodel (= 7.0.4)
+      activesupport (= 7.0.4)
+    activestorage (7.0.4)
+      actionpack (= 7.0.4)
+      activejob (= 7.0.4)
+      activerecord (= 7.0.4)
+      activesupport (= 7.0.4)
+      marcel (~> 1.0)
+      mini_mime (>= 1.1.0)
+    activesupport (7.0.4)
+      concurrent-ruby (~> 1.0, >= 1.0.2)
+      i18n (>= 1.6, < 2)
+      minitest (>= 5.1)
+      tzinfo (~> 2.0)
     addressable (2.8.1)
       public_suffix (>= 2.0.2, < 6.0)
     ast (2.4.2)
+    builder (3.2.4)
+    byebug (11.1.3)
     coderay (1.1.3)
+    concurrent-ruby (1.1.10)
+    crass (1.0.6)
+    erubi (1.11.0)
     faraday (2.6.0)
       faraday-net_http (>= 2.0, < 3.1)
       ruby2_keywords (>= 0.0.4)
     faraday-net_http (3.0.1)
     faraday-retry (2.0.0)
       faraday (~> 2.0)
+    globalid (1.0.0)
+      activesupport (>= 5.0)
+    i18n (1.12.0)
+      concurrent-ruby (~> 1.0)
     json (2.6.2)
+    loofah (2.19.0)
+      crass (~> 1.0.2)
+      nokogiri (>= 1.5.9)
+    mail (2.7.1)
+      mini_mime (>= 0.1.1)
+    marcel (1.0.2)
     method_source (1.0.0)
+    mini_mime (1.1.2)
     minitest (5.16.3)
+    mocha (1.15.0)
+    net-imap (0.3.1)
+      net-protocol
+    net-pop (0.1.2)
+      net-protocol
+    net-protocol (0.1.3)
+      timeout
+    net-smtp (0.3.2)
+      net-protocol
+    nio4r (2.5.8)
+    nokogiri (1.13.8-x86_64-darwin)
+      racc (~> 1.4)
     parallel (1.22.1)
     parser (3.1.2.1)
       ast (~> 2.4.1)
@@ -27,10 +121,40 @@ GEM
       coderay (~> 1.1)
       method_source (~> 1.0)
     public_suffix (5.0.0)
+    racc (1.6.0)
+    rack (2.2.4)
+    rack-test (2.0.2)
+      rack (>= 1.3)
+    rails (7.0.4)
+      actioncable (= 7.0.4)
+      actionmailbox (= 7.0.4)
+      actionmailer (= 7.0.4)
+      actionpack (= 7.0.4)
+      actiontext (= 7.0.4)
+      actionview (= 7.0.4)
+      activejob (= 7.0.4)
+      activemodel (= 7.0.4)
+      activerecord (= 7.0.4)
+      activestorage (= 7.0.4)
+      activesupport (= 7.0.4)
+      bundler (>= 1.15.0)
+      railties (= 7.0.4)
+    rails-dom-testing (2.0.3)
+      activesupport (>= 4.2.0)
+      nokogiri (>= 1.6)
+    rails-html-sanitizer (1.4.3)
+      loofah (~> 2.3)
+    railties (7.0.4)
+      actionpack (= 7.0.4)
+      activesupport (= 7.0.4)
+      method_source
+      rake (>= 12.2)
+      thor (~> 1.0)
+      zeitwerk (~> 2.5)
     rainbow (3.1.1)
     rake (13.0.6)
     regexp_parser (2.6.0)
-    replicate-ruby (0.1.1)
+    replicate-ruby (0.1.2)
       addressable
       faraday (>= 2.0)
       faraday-retry
@@ -49,14 +173,25 @@ GEM
       parser (>= 3.1.1.0)
     ruby-progressbar (1.11.0)
     ruby2_keywords (0.0.5)
+    thor (1.2.1)
+    timeout (0.3.0)
+    tzinfo (2.0.5)
+      concurrent-ruby (~> 1.0)
     unicode-display_width (2.3.0)
+    websocket-driver (0.7.5)
+      websocket-extensions (>= 0.1.0)
+    websocket-extensions (0.1.5)
+    zeitwerk (2.6.1)
 
 PLATFORMS
   x86_64-darwin-21
 
 DEPENDENCIES
+  byebug
   minitest (~> 5.0)
+  mocha
   pry
+  rails (>= 6)
   rake (~> 13.0)
   replicate-rails!
   rubocop (~> 1.21)

data/README.md

@@ -1,8 +1,6 @@
-# ReplicateRails
+# Replicate Rails Gem
 
-Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/replicate/rails`. To experiment with that code, run `bin/console` for an interactive prompt.
-
-TODO: Delete this and the text above, and describe your gem
+This gem bundles the [replicate-ruby](https://github.com/dreamingtulpa/replicate-ruby) gem and adds webhook support for replicate.com.
 
 ## Installation
 
@@ -12,17 +10,39 @@ Add this line to your application's Gemfile:
 gem 'replicate-rails'
 ```
 
-And then execute:
+## Usage
 
-    $ bundle install
+Setup an initializer and webhook adapter class:
 
-Or install it yourself as:
+```ruby
+# config/initializers/replicate.rb
+Replicate.client.api_token = "your-api-token"
+
+ReplicateRails.configure do |config|
+  config.webhook_adapter = ReplicateWebhook.new
+end
+
+class ReplicateWebhook
+  def call(prediction)
+    # do your thing
+  end
+end
+```
 
-    $ gem install replicate-rails
+and mount the route:
 
-## Usage
+```ruby
+# config/routes.rb
+mount ReplicateRails::Engine => "/replicate/webhook"
+```
+
+Now you can run predictions as follows:
 
-TODO: Write usage instructions here
+```
+model = Replicate.client.retrieve_model("stability-ai/stable-diffusion")
+version = model.latest_version
+version.predict(prompt: "a beautiful sunset", "https://yourdomain.tld/replicate/webhook")
+```
 
 ## Development
 

data/app/controllers/replicate_rails/webhook_controller.rb

@@ -0,0 +1,20 @@
+module ReplicateRails
+  class WebhookController < ActionController::Base
+    if Rails.application.config.action_controller.default_protect_from_forgery
+      skip_before_action :verify_authenticity_token
+    end
+
+    def event
+      ReplicateRails.config.webhook_adapter.call(prediction)
+      head :ok
+    rescue ReplicateRails::WebhookError
+      head :unprocessable_entity
+    end
+
+    private
+
+    def prediction
+      Replicate::Record::Prediction.new(Replicate.client, JSON.parse(request.body.read))
+    end
+  end
+end

data/config/routes.rb

@@ -0,0 +1,3 @@
+ReplicateRails::Engine.routes.draw do
+  root to: 'webhook#event', via: :post
+end

data/lib/replicate_rails/configuration.rb

@@ -0,0 +1,5 @@
+module ReplicateRails
+  class Configuration
+    attr_accessor :webhook_adapter
+  end
+end

data/lib/replicate_rails/engine.rb

@@ -0,0 +1,5 @@
+module ReplicateRails
+  class Engine < ::Rails::Engine
+    isolate_namespace ReplicateRails
+  end
+end

data/lib/replicate_rails/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ReplicateRails
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/lib/replicate_rails.rb

@@ -1,8 +1,22 @@
 # frozen_string_literal: true
 
-require_relative "replicate_rails/version"
+require "replicate"
+require "replicate_rails/version"
+require "replicate_rails/engine" if defined?(Rails)
+
+require "replicate_rails/configuration"
 
 module ReplicateRails
   class Error < StandardError; end
-  # Your code goes here...
+  class WebhookError < Error; end
+
+  class << self
+    def config
+      @config ||= Configuration.new
+    end
+
+    def configure
+      yield(config)
+    end
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: replicate-rails
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Dreaming Tulpa
@@ -10,6 +10,20 @@ bindir: exe
 cert_chain: []
 date: 2022-10-14 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '3.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '3.1'
 - !ruby/object:Gem::Dependency
   name: replicate-ruby
   requirement: !ruby/object:Gem::Requirement
@@ -38,6 +52,90 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: byebug
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '6'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.21'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.21'
 description:
 email:
 - hey@dreamingtulpa.com
@@ -51,7 +149,11 @@ files:
 - Gemfile.lock
 - README.md
 - Rakefile
+- app/controllers/replicate_rails/webhook_controller.rb
+- config/routes.rb
 - lib/replicate_rails.rb
+- lib/replicate_rails/configuration.rb
+- lib/replicate_rails/engine.rb
 - lib/replicate_rails/version.rb
 - sig/replicate_rails.rb
 homepage: https://github.com/dreamingtulpa/replicate-rails