replicate-api 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 6a977c31bb41a70a5734b042c1a3147d10b5d53976a0c6ad5f8425da6fabe74b
+  data.tar.gz: ea9f526e0b3afd6f221edb0a937aca33a8ad21fd7cb2be1c640659764789ee10
+SHA512:
+  metadata.gz: 5389b2593291fd89e28a5f6e6beb9fbf877c03f2cdc3fd2e28972615c03c5222e778b97d51c81a63ce8628aefaab966b95df86460dd88251316483a1ff93899c
+  data.tar.gz: cf6c9e95bcc21b96f2676f9a8707a9823240fbafefab1f01ba9b5602833246882c2b1b8c5488b76a13adc81fbe188b7a1d0ef12a919829876d0328170ebfbec5

data/.gem/credentials

@@ -0,0 +1 @@
+:github: Bearer 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

data/.rspec

@@ -0,0 +1,3 @@
+--format documentation
+--color
+--require spec_helper

data/Gemfile

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+source "https://rubygems.org"
+
+# Specify your gem's dependencies in replicate-api.gemspec
+gemspec
+
+gem "rake", "~> 13.0"
+
+gem "rspec", "~> 3.0"
+
+gem 'pry', '~> 0.13.1'
+

data/Gemfile.lock

@@ -0,0 +1,46 @@
+PATH
+  remote: .
+  specs:
+    replicate-api (0.1.0)
+      faraday (~> 2.7.1)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    coderay (1.1.3)
+    diff-lcs (1.5.0)
+    faraday (2.7.1)
+      faraday-net_http (>= 2.0, < 3.1)
+      ruby2_keywords (>= 0.0.4)
+    faraday-net_http (3.0.2)
+    method_source (1.0.0)
+    pry (0.13.1)
+      coderay (~> 1.1)
+      method_source (~> 1.0)
+    rake (13.0.6)
+    rspec (3.12.0)
+      rspec-core (~> 3.12.0)
+      rspec-expectations (~> 3.12.0)
+      rspec-mocks (~> 3.12.0)
+    rspec-core (3.12.0)
+      rspec-support (~> 3.12.0)
+    rspec-expectations (3.12.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.12.0)
+    rspec-mocks (3.12.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.12.0)
+    rspec-support (3.12.0)
+    ruby2_keywords (0.0.5)
+
+PLATFORMS
+  arm64-darwin-21
+
+DEPENDENCIES
+  pry (~> 0.13.1)
+  rake (~> 13.0)
+  replicate-api!
+  rspec (~> 3.0)
+
+BUNDLED WITH
+   2.3.7

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2022 Olivia Li
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,39 @@
+# Replicate Ruby client
+
+This is a Ruby client for Replicate. It lets you run models with your Ruby code. 
+
+Grab your token from replicate.com/account to use this library. 
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'replicate-api'
+```
+
+And then execute:
+
+    $ bundle install
+
+Or install it yourself as:
+
+    $ gem install replicate-api
+
+## Usage
+
+You can run a model and get its output:
+
+```
+replicate = Replicate::Client.new(api_key: "#{YOUR API KEY HERE}")
+model = replicate.models("stability-ai/stable-diffusion")
+model.predict(prompt="a 19th century portrait of a wombat gentleman")
+```
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/olivia-li/replicate-ruby.
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).

data/Rakefile

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task default: :spec

data/lib/replicate/client.rb

@@ -0,0 +1,57 @@
+require 'faraday'
+require 'pry'
+
+
+module Replicate
+  REPLICATE_API_BASE_URL = "https://api.replicate.com/v1"
+
+  # Client for the Replicate API
+  class Client
+    attr_reader :token, :conn
+
+    def initialize(token:, &block)
+      @token = token
+      if block_given?
+        conn &block
+      else
+        conn
+      end
+
+    end
+
+    def conn(&block)
+      @conn ||= Faraday.new(url: REPLICATE_API_BASE_URL) do |f|
+        yield f if block_given?
+        f.request :json
+        f.response :json, content_type: /\bjson$/
+        f.headers = headers()
+      end
+    end
+
+    def models(path)
+      return Replicate::Model.new(client: self, path: path)
+    end
+  
+    def headers
+      return {
+        "Authorization": "Token #{@token}",
+        "User-Agent": "replicate-ruby@#{Replicate::VERSION}",
+      }
+    end
+
+    def requests(method:, path:, **kwargs)
+      if ["GET", "OPTIONS"].include? method
+        kwargs = kwargs.merge({allow_redirects: true})
+      end 
+      if ["HEAD"].include? method
+        kwargs = kwargs.merge({allow_redirects: false})
+      end
+      response = @conn.send(method.downcase, path, kwargs)
+      if 400 <= response.status && response.status < 600
+        raise ("HTTP error: #{response.status}, #{response.reason_phrase}")
+      end
+      return response
+    end
+
+  end
+end

data/lib/replicate/model.rb

@@ -0,0 +1,32 @@
+module Replicate
+  class Model
+    attr_reader :version_id, :path
+
+    def initialize(client:, path:) 
+      @client = client
+      @path = path
+      response = client.requests(method: "GET", path: "models/#{path}")
+      @version_id = response.body["latest_version"]["id"]
+    end
+
+    def predict(prompt:)
+      body = {version: @version_id, input: {prompt: prompt}}
+      response = @client.requests(method: "POST", path: "predictions", **body)
+      poll(prediction_id: response.body["id"])
+    end
+
+    def poll(prediction_id:)
+      response = @client.requests(method: "GET", path: "predictions/#{prediction_id}")
+      puts(response.body["status"])
+      if !["succeeded", "failed", "canceled"].include?(response.body["status"])
+        sleep(2)
+        poll(prediction_id: prediction_id)
+      elsif response.body["status"] == "succeeded" 
+        return response.body["output"][0]
+      else
+        raise response.body["error"]
+      end
+    end
+
+  end
+end

data/lib/replicate/version.rb

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

data/lib/replicate.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+require_relative "replicate/version"
+
+module Replicate
+  autoload :Client, "replicate/client"
+  autoload :Error, "replicate/error"
+  autoload :Model, "replicate/model"
+  # Your code goes here...
+end

data/sig/replicate/ruby.rbs

@@ -0,0 +1,6 @@
+module Replicate
+  module Ruby
+    VERSION: String
+    # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+  end
+end

metadata

@@ -0,0 +1,71 @@
+--- !ruby/object:Gem::Specification
+name: replicate-api
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Olivia Li
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2022-11-27 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.7.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.7.1
+description:
+email:
+- oliviali.ouyang@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gem/credentials"
+- ".rspec"
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/replicate.rb
+- lib/replicate/client.rb
+- lib/replicate/model.rb
+- lib/replicate/version.rb
+- sig/replicate/ruby.rbs
+homepage: https://github.com/Olivia-li/replicate-ruby.git
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/Olivia-li/replicate-ruby.git
+  source_code_uri: https://github.com/Olivia-li/replicate-ruby.git
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.6.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.2.3
+signing_key:
+specification_version: 4
+summary: Ruby API wrapper for Replicate
+test_files: []