OpenRouter support

Author: crmne

.overcommit.yml

@@ -18,6 +18,11 @@ PreCommit:
     exclude:
      - '**/db/structure.sql' # Ignore trailing whitespace in generated files
 
+  RakeTarget:
+    enabled: true
+    targets: ['models:update', 'models:docs']
+    on_warn: fail
+
 PostCheckout:
   ALL: # Special hook name that customizes all hooks of this type
     quiet: true # Change all post-checkout hooks to only display output on failure

bin/console

@@ -16,6 +16,7 @@ RubyLLM.configure do |config|
   config.bedrock_secret_key = ENV.fetch('AWS_SECRET_ACCESS_KEY', nil)
   config.bedrock_region = ENV.fetch('AWS_REGION', nil)
   config.bedrock_session_token = ENV.fetch('AWS_SESSION_TOKEN', nil)
+  config.openrouter_api_key = ENV.fetch('OPENROUTER_API_KEY', nil)
 end
 
 IRB.start(__FILE__)

docs/guides/available-models.md

@@ -16,7 +16,8 @@
   'openai' => 'OpenAI',
   'api' => 'API',
   'deepseek' => 'DeepSeek',
-  'bedrock' => 'Bedrock'
+  'bedrock' => 'Bedrock',
+  'openrouter' => 'OpenRouter'
 )
 loader.ignore("#{__dir__}/tasks")
 loader.ignore("#{__dir__}/ruby_llm/railtie")
@@ -79,6 +80,7 @@ def logger
 RubyLLM::Provider.register :gemini, RubyLLM::Providers::Gemini
 RubyLLM::Provider.register :deepseek, RubyLLM::Providers::DeepSeek
 RubyLLM::Provider.register :bedrock, RubyLLM::Providers::Bedrock
+RubyLLM::Provider.register :openrouter, RubyLLM::Providers::OpenRouter
 
 if defined?(Rails::Railtie)
   require 'ruby_llm/railtie'

lib/ruby_llm.rb

@@ -1,15 +1,18 @@
 {
   "claude-3-5-sonnet": {
     "anthropic": "claude-3-5-sonnet-20241022",
-    "bedrock": "anthropic.claude-3-5-sonnet-20241022-v2:0"
+    "bedrock": "anthropic.claude-3-5-sonnet-20241022-v2:0",
+    "openrouter": "anthropic/claude-3.5-sonnet"
   },
   "claude-3-5-haiku": {
     "anthropic": "claude-3-5-haiku-20241022",
-    "bedrock": "anthropic.claude-3-5-haiku-20241022-v1:0"
+    "bedrock": "anthropic.claude-3-5-haiku-20241022-v1:0",
+    "openrouter": "anthropic/claude-3.5-haiku-20241022"
   },
   "claude-3-7-sonnet": {
     "anthropic": "claude-3-7-sonnet-20250219",
-    "bedrock": "us.anthropic.claude-3-7-sonnet-20250219-v1:0"
+    "bedrock": "us.anthropic.claude-3-7-sonnet-20250219-v1:0",
+    "openrouter": "us.anthropic.claude-3-7-sonnet-20250219-v1:0"
   },
   "claude-3-opus": {
     "anthropic": "claude-3-opus-20240229",
@@ -25,14 +28,53 @@
   },
   "claude-3": {
     "anthropic": "claude-3-sonnet-20240229",
-    "bedrock": "anthropic.claude-3-sonnet-20240229-v1:0"
+    "bedrock": "anthropic.claude-3-sonnet-20240229-v1:0",
+    "openrouter": "anthropic/claude-3-sonnet"
   },
   "claude-2": {
     "anthropic": "claude-2.0",
-    "bedrock": "anthropic.claude-2.0"
+    "bedrock": "anthropic.claude-2.0",
+    "openrouter": "anthropic/claude-2"
   },
   "claude-2-1": {
     "anthropic": "claude-2.1",
-    "bedrock": "anthropic.claude-2.1"
+    "bedrock": "anthropic.claude-2.1",
+    "openrouter": "anthropic/claude-2.1"
+  },
+  "gpt-4o": {
+    "openai": "gpt-4o",
+    "openrouter": "openai/gpt-4o"
+  },
+  "gpt-4o-mini": {
+    "openai": "gpt-4o-mini",
+    "openrouter": "openai/gpt-4o-mini"
+  },
+  "gpt-4-turbo": {
+    "openai": "gpt-4-turbo",
+    "openrouter": "openai/gpt-4-turbo"
+  },
+  "gemini-1.5-flash": {
+    "gemini": "gemini-1.5-flash",
+    "openrouter": "google/gemini-flash-1.5"
+  },
+  "gemini-1.5-flash-8b": {
+    "gemini": "gemini-1.5-flash-8b",
+    "openrouter": "google/gemini-flash-1.5-8b"
+  },
+  "gemini-1.5-pro": {
+    "gemini": "gemini-1.5-pro",
+    "openrouter": "google/gemini-pro-1.5"
+  },
+  "gemini-2.0-flash": {
+    "gemini": "gemini-2.0-flash",
+    "openrouter": "google/gemini-2.0-flash-001"
+  },
+  "o1": {
+    "openai": "o1",
+    "openrouter": "openai/o1"
+  },
+  "o3-mini": {
+    "openai": "o3-mini",
+    "openrouter": "openai/o3-mini"
   }
 }

lib/ruby_llm/aliases.json

@@ -20,6 +20,7 @@ class Configuration
                   :bedrock_secret_key,
                   :bedrock_region,
                   :bedrock_session_token,
+                  :openrouter_api_key,
                   # Default models
                   :default_model,
                   :default_embedding_model,

lib/ruby_llm/configuration.rb

@@ -19,6 +19,7 @@ def initialize(provider, config)
     end
 
     def post(url, payload)
+      RubyLLM.logger.debug "POST #{url} with payload: #{payload.inspect}"
       body = payload.is_a?(Hash) ? JSON.generate(payload, ascii_only: false) : payload
       @connection.post url, body do |req|
         req.headers.merge! @provider.headers(@config) if @provider.respond_to?(:headers)