roseflow-openrouter 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 87a20a3c7e3eabe923827fb046a4d418901e2719d845244b3cd6488301044a89
+  data.tar.gz: b0b3cc0894bba52d00d95b75ec093a2c86cf1621d2bc9f7849de1a6c61b9b754
+SHA512:
+  metadata.gz: e393acceab954ba51cec59a2dd518928c9e9e3fcca4ec2f77475e1c4dc9b9caab8cd318b23f7d9026fbe8343e8cdf104d5f9f5d80a78e9505fe686444b6f6604
+  data.tar.gz: d026c4386e835b3841e47e8c14ad913d25413bf82d3b1c6506c5c596017001936bc13926e44fa905cb0b36c476053f5ce6c5070d47ef7adf8009cc6ac305d78b

data/.rspec

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

data/.standard.yml

@@ -0,0 +1,3 @@
+# For available configuration options, see:
+#   https://github.com/testdouble/standard
+ruby_version: 2.6

data/CHANGELOG.md

@@ -0,0 +1,5 @@
+## [Unreleased]
+
+## [0.1.0] - 2023-07-18
+
+- Initial release

data/CODE_OF_CONDUCT.md

@@ -0,0 +1,7 @@
+# Contributor Code of Conduct
+
+The Roseflow team is committed to fostering a welcoming community.
+
+**Our Code of Conduct can be found here**:
+
+https://roseflow.ai/conduct

data/Gemfile

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+source "https://rubygems.org"
+
+# Specify your gem's dependencies in roseflow-openrouter.gemspec
+gemspec
+
+gem "rake", "~> 13.0"
+gem "rspec", "~> 3.0"
+gem "standard", "~> 1.3"

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2023 Lauri Jutila
+
+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,47 @@
+# Roseflow OpenRouter integration
+
+This gem adds [OpenRouter](https://openrouter.ai) support and integration for [Roseflow](https://github.com/roseflow-ai/roseflow), a framework for interacting with AI in Ruby.
+
+## Prerequisites
+
+To use this gem effectively, you need the [core Roseflow gem](https://github.com/roseflow-ai/roseflow).
+
+## Installation
+
+Install the gem and add to the application's Gemfile by executing:
+
+    $ bundle add roseflow-openrouter
+
+If bundler is not being used to manage dependencies, install the gem by executing:
+
+    $ gem install roseflow-openrouter
+
+## Usage
+
+See full documentation how to configure and use Roseflow with OpenRouter at [docs.roseflow.ai](https://docs.roseflow.ai/openrouter).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/roseflow-openrouter.
+
+## Community
+
+### Discord
+
+Join us in our [Discord](https://discord.gg/roseflow).
+
+### Twitter
+
+Connect with the core team on Twitter.
+
+<a href="https://twitter.com/ljuti" target="_blank">
+  <img alt="Twitter Follow" src="https://img.shields.io/twitter/follow/ljuti?logo=twitter&style=social">
+</a>
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
+
+## Code of Conduct
+
+Everyone interacting in the Roseflow OpenRouter project's codebases, issue trackers, chat rooms and mailing lists is expected to follow the [code of conduct](https://github.com/roseflow-ai/roseflow-openrouter/blob/main/CODE_OF_CONDUCT.md).

data/Rakefile

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+require "standard/rake"
+
+task default: %i[spec standard]

data/config/openrouter.yml

@@ -0,0 +1,14 @@
+default: &default
+  referer: <YOUR REFERER URL>
+
+development:
+  <<: *default
+  api_key: <YOUR OPENROUTER API KEY FOR DEVELOPMENT>
+
+test:
+  <<: *default
+  api_key: <YOUR OPENROUTER API KEY FOR TEST>
+
+production:
+  <<: *default
+  api_key: <YOUR OPENROUTER API KEY FOR PRODUCTION>

data/lib/roseflow/open_router/client.rb

@@ -0,0 +1,92 @@
+# frozen_string_literal: true
+
+require "faraday"
+require "faraday/retry"
+require "roseflow/open_router/config"
+require "roseflow/open_router/model"
+
+FARADAY_RETRY_OPTIONS = {
+  max: 3,
+  interval: 0.05,
+  interval_randomness: 0.5,
+  backoff_factor: 2,
+}
+
+module Roseflow
+  module OpenRouter
+    class Client
+      def initialize(config = Config.new, provider = nil)
+        @config = config
+        @provider = provider
+      end
+
+      def models
+        response = connection.get("/api/v1/models")
+        body = JSON.parse(response.body)
+        body.fetch("data", []).map do |model|
+          OpenRouter::Model.new(model, self)
+        end
+      end
+
+      def completion(model:, prompt:, **options)
+        connection.post("/api/v1/chat/completions") do |request|
+          request.body = options.merge({
+            model: model.name,
+            messages: prompt,
+          })
+        end
+      end
+
+      def streaming_completion(model:, prompt:, **options, &block)
+        streamed = []
+        raw_ = !!options.delete(:raw)
+        options.delete(:streaming)
+
+        connection.post("/api/v1/chat/completions") do |request|
+          request.body = options.merge({
+            model: model.name,
+            messages: prompt,
+            stream: true,
+          })
+
+          request.options.on_data = Proc.new do |chunk|
+            if block_given?
+              yield streaming_chunk(chunk) unless raw_
+              yield chunk if raw_
+            end
+
+            streamed << chunk unless block_given?
+          end
+        end
+
+        streamed unless block_given?
+      end
+
+      private
+
+      attr_reader :config, :provider
+
+      def connection
+        @connection ||= Faraday.new(
+          url: Config::OPENROUTER_API_URL,
+          headers: {
+            "HTTP-Referer": config.referer,
+          },
+        ) do |faraday|
+          faraday.request :authorization, "Bearer", -> { config.api_key }
+          faraday.request :json
+          faraday.request :retry, FARADAY_RETRY_OPTIONS
+          faraday.adapter Faraday.default_adapter
+        end
+      end
+
+      def streaming_chunk(chunk)
+        return chunk unless chunk.match(/{.*}/)
+
+        chunk.scan(/{.*}/).map do |json|
+          JSON.parse(json).dig("choices", 0, "delta", "content")
+        end.join("")
+      end
+    end
+  end
+end

data/lib/roseflow/open_router/config.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+require "anyway_config"
+
+module Roseflow
+  module OpenRouter
+    class Config < Anyway::Config
+      config_name :openrouter
+
+      attr_config :api_key
+      attr_config :referer
+
+      required :api_key
+      required :referer
+
+      OPENROUTER_API_URL = "https://openrouter.ai/api/v1".freeze
+    end
+  end
+end

data/lib/roseflow/open_router/model.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+require "dry/struct"
+require "active_support/core_ext/module/delegation"
+
+module Types
+  include Dry.Types()
+end
+
+module Roseflow
+  module OpenRouter
+    class Model
+      attr_reader :name, :id, :context_length
+
+      def initialize(model, provider)
+        @model_ = model
+        @provider_ = provider
+        assign_attributes
+      end
+
+      def max_tokens
+        @context_length
+      end
+
+      private
+
+      def assign_attributes
+        @name = @model_.fetch("id")
+        @id = @model_.fetch("id")
+        @context_length = @model_.fetch("context_length")
+      end
+    end
+  end
+end

data/lib/roseflow/open_router/model_repository.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module Roseflow
+  module OpenRouter
+    class ModelRepository
+      attr_reader :models
+
+      delegate :each, :all, :first, :last, to: :models
+
+      def initialize(provider)
+        @provider = provider
+        @models = provider.client.models
+      end
+
+      def find(name)
+        @models.select { |model| model.name == name }.first
+      end
+    end
+  end
+end

data/lib/roseflow/open_router/provider.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+module Roseflow
+  module OpenRouter
+    class Provider
+      def initialize(config = Roseflow::OpenRouter::Config.new)
+        @config = config
+      end
+
+      def client
+        @client ||= Client.new(config, self)
+      end
+
+      def models
+        @models ||= ModelRepository.new(self)
+      end
+
+      def completion(model:, prompt:, **options)
+        streaming = options.fetch(:streaming, false)
+
+        if streaming
+          client.streaming_completion(model: model, prompt: prompt, **options)
+        else
+          client.completion(model: model, prompt: prompt, **options)
+        end
+      end
+
+      private
+
+      attr_reader :config
+    end
+  end
+end

data/lib/roseflow/open_router/version.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module Roseflow
+  module OpenRouter
+    def self.gem_version
+      Gem::Version.new VERSION::STRING
+    end
+
+    module VERSION
+      MAJOR = 0
+      MINOR = 1
+      PATCH = 0
+      PRE = nil
+
+      STRING = [MAJOR, MINOR, PATCH, PRE].compact.join(".")
+    end
+  end
+end

data/lib/roseflow/open_router.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+require_relative "open_router/version"
+require_relative "open_router/client"
+require_relative "open_router/config"
+require_relative "open_router/model_repository"
+require_relative "open_router/model"
+require_relative "open_router/provider"
+
+module Roseflow
+  module OpenRouter
+    class Error < StandardError; end
+  end
+end

data/roseflow-openrouter.gemspec

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+require_relative "lib/roseflow/open_router/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "roseflow-openrouter"
+  spec.version = Roseflow::OpenRouter.gem_version
+  spec.authors = ["Lauri Jutila"]
+  spec.email = ["ljuti@users.noreply.github.com"]
+
+  spec.summary = "Roseflow meets OpenRouter"
+  spec.description = "OpenRouter integration and support for Roseflow"
+  spec.homepage = "https://github.com/roseflow-ai/roseflow-openrouter"
+  spec.license = "MIT"
+  spec.required_ruby_version = ">= 3.2.0"
+
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = "https://github.com/roseflow-ai/roseflow-openrouter"
+  spec.metadata["changelog_uri"] = "https://github.com/roseflow-ai/roseflow-openrouter/blob/master/CHANGELOG.md"
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files = Dir.chdir(__dir__) do
+    `git ls-files -z`.split("\x0").reject do |f|
+      (File.expand_path(f) == __FILE__) || f.start_with?(*%w[bin/ test/ spec/ features/ .git .circleci appveyor])
+    end
+  end
+  spec.bindir = "exe"
+  spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "activesupport"
+  spec.add_dependency "anyway_config", "~> 2.0"
+  spec.add_dependency "dry-struct"
+  spec.add_dependency "faraday"
+  spec.add_dependency "faraday-retry"
+
+  spec.add_development_dependency "awesome_print"
+  spec.add_development_dependency "pry"
+  spec.add_development_dependency "roseflow"
+  spec.add_development_dependency "webmock"
+  spec.add_development_dependency "vcr"
+end

data/sig/roseflow/openrouter.rbs

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

metadata

@@ -0,0 +1,204 @@
+--- !ruby/object:Gem::Specification
+name: roseflow-openrouter
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Lauri Jutila
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2023-07-19 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: anyway_config
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: dry-struct
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: faraday-retry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: awesome_print
+  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: pry
+  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: roseflow
+  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: webmock
+  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: vcr
+  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'
+description: OpenRouter integration and support for Roseflow
+email:
+- ljuti@users.noreply.github.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".rspec"
+- ".standard.yml"
+- CHANGELOG.md
+- CODE_OF_CONDUCT.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- config/openrouter.yml
+- lib/roseflow/open_router.rb
+- lib/roseflow/open_router/client.rb
+- lib/roseflow/open_router/config.rb
+- lib/roseflow/open_router/model.rb
+- lib/roseflow/open_router/model_repository.rb
+- lib/roseflow/open_router/provider.rb
+- lib/roseflow/open_router/version.rb
+- roseflow-openrouter.gemspec
+- sig/roseflow/openrouter.rbs
+homepage: https://github.com/roseflow-ai/roseflow-openrouter
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/roseflow-ai/roseflow-openrouter
+  source_code_uri: https://github.com/roseflow-ai/roseflow-openrouter
+  changelog_uri: https://github.com/roseflow-ai/roseflow-openrouter/blob/master/CHANGELOG.md
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.2.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.4.1
+signing_key:
+specification_version: 4
+summary: Roseflow meets OpenRouter
+test_files: []