openrouter_client 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1dfa50003cbe411f1e237d47c2872e053bbe649253abe52df64da017071bf791
-  data.tar.gz: 6279dcb4868bf470016c0df6cfe45fd3c5ebc19284cb7eb54d8d8d6c6ca78bb1
+  metadata.gz: f98e5959581fadc509a596c15ef862b8432d5355605e3c36c0fdefb0cb4dcf28
+  data.tar.gz: 04e7534567aa13c34c26a159a2df648a69cd13e1233f7c3e8ecb137f56935986
 SHA512:
-  metadata.gz: 20fa68b2374b7be8f14b0833fa794d3f2d52f0aea344a2707ab9d3330a6a8c43da36e2f7d27681af2a8a2a8ecfeb926bdb70b7f86819e904b38b76aa1c4f2156
-  data.tar.gz: ecc2aecf1224b2c5ddc1207780ebbc9014a07fd568969d2fda199cdfeaa46ea93ec8badbdacfca74cb6357c3f98ed171e144f584aef2782bfdf50c811ddab100
+  metadata.gz: d4d992b00d394aefb85553611fbce0e3d516f3ae93c35aeb243e171bf0f5493672fae79a8ee79666145a691a3aef14f68a40bda609a115bec058a48d86348e31
+  data.tar.gz: 164ee4e2f29f5991c872ad79cfeadd3496744f6c729d911eb3e42e7430a009a33b6fd8adaf8a6abe21aa87647d8fc8fe664a395a287fcbb523070c2341bc6d50

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    openrouter_client (0.1.1)
+    openrouter_client (0.1.2)
       faraday (>= 1)
 
 GEM

data/README.md

@@ -215,6 +215,16 @@ model = OpenRouter::Model.find_by!(id: "openai/gpt-4")
 puts model.name           # => "GPT-4"
 puts model.context_length # => 8192
 puts model.input_price    # => 0.00003 (per token)
+
+# Endpoints (available providers for this model)
+model.endpoints.each do |endpoint|
+  puts endpoint.provider_name      # => "OpenAI"
+  puts endpoint.context_length     # => 8192
+  puts endpoint.prompt_price       # => 0.00003
+  puts endpoint.completion_price   # => 0.00006
+  puts endpoint.available?         # => true
+  puts endpoint.supports?(:tools)  # => true
+end
 puts model.output_price   # => 0.00006 (per token)
 puts model.free?          # => false
 

data/lib/openrouter/endpoint.rb

@@ -0,0 +1,95 @@
+# frozen_string_literal: true
+
+module OpenRouter
+  # Represents a specific endpoint/provider for a model.
+  # Models can have multiple endpoints from different providers with different pricing and capabilities.
+  class Endpoint
+    # @return [String] Endpoint name
+    attr_reader :name
+    # @return [String] Model name
+    attr_reader :model_name
+    # @return [Integer] Context length in tokens
+    attr_reader :context_length
+    # @return [Hash] Pricing information
+    attr_reader :pricing
+    # @return [String] Provider name (e.g., "OpenAI", "Azure")
+    attr_reader :provider_name
+    # @return [String, nil] Tag
+    attr_reader :tag
+    # @return [Hash, nil] Quantization info
+    attr_reader :quantization
+    # @return [Integer, nil] Maximum completion tokens
+    attr_reader :max_completion_tokens
+    # @return [Integer, nil] Maximum prompt tokens
+    attr_reader :max_prompt_tokens
+    # @return [Array<String>] Supported parameters
+    attr_reader :supported_parameters
+    # @return [String, nil] Endpoint status
+    attr_reader :status
+    # @return [Float, nil] Uptime in the last 30 minutes
+    attr_reader :uptime_last_30m
+    # @return [Boolean] Whether the endpoint supports implicit caching
+    attr_reader :supports_implicit_caching
+
+    # @param attributes [Hash] Raw attributes from OpenRouter API
+    def initialize(attributes)
+      @name = attributes["name"]
+      @model_name = attributes["model_name"]
+      @context_length = attributes["context_length"]
+      @pricing = attributes["pricing"]
+      @provider_name = attributes["provider_name"]
+      @tag = attributes["tag"]
+      @quantization = attributes["quantization"]
+      @max_completion_tokens = attributes["max_completion_tokens"]
+      @max_prompt_tokens = attributes["max_prompt_tokens"]
+      @supported_parameters = attributes["supported_parameters"] || []
+      @status = attributes["status"]
+      @uptime_last_30m = attributes["uptime_last_30m"]
+      @supports_implicit_caching = attributes["supports_implicit_caching"] || false
+    end
+
+    # Get the prompt/input price per token.
+    # @return [Float, nil]
+    def prompt_price
+      @pricing&.dig("prompt")&.to_f
+    end
+
+    # Get the completion/output price per token.
+    # @return [Float, nil]
+    def completion_price
+      @pricing&.dig("completion")&.to_f
+    end
+
+    # Get the image price.
+    # @return [Float, nil]
+    def image_price
+      @pricing&.dig("image")&.to_f
+    end
+
+    # Get any discount percentage.
+    # @return [Float, nil]
+    def discount
+      @pricing&.dig("discount")&.to_f
+    end
+
+    # Check if this endpoint is free.
+    # @return [Boolean]
+    def free?
+      prompt_price&.zero? && completion_price&.zero?
+    end
+
+    # Check if this endpoint is currently available.
+    # Status "0" indicates the endpoint is healthy.
+    # @return [Boolean]
+    def available?
+      @status == "0"
+    end
+
+    # Check if this endpoint supports a specific parameter.
+    # @param parameter [String, Symbol] The parameter to check
+    # @return [Boolean]
+    def supports?(parameter)
+      @supported_parameters.include?(parameter.to_s)
+    end
+  end
+end

data/lib/openrouter/model.rb

@@ -18,7 +18,7 @@ module OpenRouter
     attr_reader :pricing
     # @return [Integer, nil] Top provider information
     attr_reader :top_provider
-    # @return [String, nil] Model architecture
+    # @return [Hash, nil] Model architecture
     attr_reader :architecture
     # @return [Array<String>, nil] Supported parameters
     attr_reader :supported_parameters
@@ -26,6 +26,8 @@ module OpenRouter
     attr_reader :per_request_limits
     # @return [String, nil] Created timestamp
     attr_reader :created
+    # @return [Array<OpenRouter::Endpoint>, nil] Available endpoints for this model
+    attr_reader :endpoints
 
     # @param attributes [Hash] Raw attributes from OpenRouter API
     # @param client [OpenRouter::Client] HTTP client
@@ -68,26 +70,32 @@ module OpenRouter
 
     class << self
       # Find a specific model by ID. Raises NotFoundError if not found.
-      # @param id [String] The model identifier
+      # @param id [String] The model identifier (e.g., "openai/gpt-4")
       # @param client [OpenRouter::Client] HTTP client
       # @return [OpenRouter::Model]
       # @raise [OpenRouter::NotFoundError] if model not found
       def find(id, client: OpenRouter.client)
-        model = find_by(id: id, client: client)
-        raise NotFoundError, "Model not found: #{id}" unless model
+        author, slug = parse_model_id(id)
+        path = "#{MODELS_PATH}/#{author}/#{slug}/endpoints"
+
+        response = client.get(path)
+        return new(response["data"], client: client) if response && response["data"]
 
-        model
+        raise NotFoundError, "Model not found: #{id}"
+      rescue NotFoundError
+        raise
+      rescue OpenRouter::Error
+        raise NotFoundError, "Model not found: #{id}"
       end
 
       # Find a specific model by ID. Returns nil if not found.
-      # @param id [String] The model identifier
+      # @param id [String] The model identifier (e.g., "openai/gpt-4")
       # @param client [OpenRouter::Client] HTTP client
       # @return [OpenRouter::Model, nil]
       def find_by(id:, client: OpenRouter.client)
-        response = client.get(MODELS_PATH)
-        models = Array(response && response["data"])
-        entry = models.find { |model| model["id"] == id }
-        entry ? new(entry, client: client) : nil
+        find(id, client: client)
+      rescue NotFoundError
+        nil
       end
 
       # Find a specific model by ID. Raises NotFoundError if not found.
@@ -144,6 +152,19 @@ module OpenRouter
       def free(client: OpenRouter.client)
         all(client: client).select(&:free?)
       end
+
+      private
+
+      # Parse a model ID into author and slug components.
+      # @param id [String] The model identifier (e.g., "openai/gpt-4")
+      # @return [Array<String>] [author, slug]
+      # @raise [ArgumentError] if the ID format is invalid
+      def parse_model_id(id)
+        parts = id.to_s.split("/", 2)
+        raise ArgumentError, "Invalid model ID format: #{id}. Expected 'author/slug'" if parts.length != 2
+
+        parts
+      end
     end
 
     private
@@ -159,6 +180,13 @@ module OpenRouter
       @supported_parameters = attributes["supported_parameters"]
       @per_request_limits = attributes["per_request_limits"]
       @created = attributes["created"]
+      @endpoints = build_endpoints(attributes["endpoints"])
+    end
+
+    def build_endpoints(endpoints_data)
+      return nil unless endpoints_data
+
+      endpoints_data.map { |attrs| OpenRouter::Endpoint.new(attrs) }
     end
   end
 end

data/lib/openrouter/version.rb

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

data/lib/openrouter.rb

@@ -113,6 +113,7 @@ end
 require_relative "openrouter/client"
 require_relative "openrouter/completion"
 require_relative "openrouter/stream"
+require_relative "openrouter/endpoint"
 require_relative "openrouter/model"
 require_relative "openrouter/generation"
 require_relative "openrouter/credit"

data/rbi/openrouter/endpoint.rbi

@@ -0,0 +1,69 @@
+# typed: strict
+# frozen_string_literal: true
+
+module OpenRouter
+  class Endpoint
+    sig { returns(String) }
+    def name; end
+
+    sig { returns(String) }
+    def model_name; end
+
+    sig { returns(Integer) }
+    def context_length; end
+
+    sig { returns(T.nilable(T::Hash[T.untyped, T.untyped])) }
+    def pricing; end
+
+    sig { returns(String) }
+    def provider_name; end
+
+    sig { returns(T.nilable(String)) }
+    def tag; end
+
+    sig { returns(T.nilable(T::Hash[T.untyped, T.untyped])) }
+    def quantization; end
+
+    sig { returns(T.nilable(Integer)) }
+    def max_completion_tokens; end
+
+    sig { returns(T.nilable(Integer)) }
+    def max_prompt_tokens; end
+
+    sig { returns(T::Array[String]) }
+    def supported_parameters; end
+
+    sig { returns(T.nilable(String)) }
+    def status; end
+
+    sig { returns(T.nilable(Float)) }
+    def uptime_last_30m; end
+
+    sig { returns(T::Boolean) }
+    def supports_implicit_caching; end
+
+    sig { params(attributes: T.untyped).void }
+    def initialize(attributes); end
+
+    sig { returns(T.nilable(Float)) }
+    def prompt_price; end
+
+    sig { returns(T.nilable(Float)) }
+    def completion_price; end
+
+    sig { returns(T.nilable(Float)) }
+    def image_price; end
+
+    sig { returns(T.nilable(Float)) }
+    def discount; end
+
+    sig { returns(T::Boolean) }
+    def free?; end
+
+    sig { returns(T::Boolean) }
+    def available?; end
+
+    sig { params(parameter: T.any(String, Symbol)).returns(T::Boolean) }
+    def supports?(parameter); end
+  end
+end

data/rbi/openrouter/model.rbi

@@ -35,6 +35,9 @@ module OpenRouter
     sig { returns(T.nilable(String)) }
     def created; end
 
+    sig { returns(T.nilable(T::Array[OpenRouter::Endpoint])) }
+    def endpoints; end
+
     sig { params(attributes: T.untyped, client: OpenRouter::Client).void }
     def initialize(attributes, client: OpenRouter.client); end
 
@@ -82,11 +85,19 @@ module OpenRouter
 
       sig { params(client: OpenRouter::Client).returns(T::Array[OpenRouter::Model]) }
       def free(client: OpenRouter.client); end
+
+      private
+
+      sig { params(id: String).returns([String, String]) }
+      def parse_model_id(id); end
     end
 
     private
 
     sig { params(attributes: T.untyped).void }
     def reset_attributes(attributes); end
+
+    sig { params(endpoints_data: T.nilable(T::Array[T::Hash[T.untyped, T.untyped]])).returns(T.nilable(T::Array[OpenRouter::Endpoint])) }
+    def build_endpoints(endpoints_data); end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: openrouter_client
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Dylan Player
@@ -44,6 +44,7 @@ files:
 - lib/openrouter/client.rb
 - lib/openrouter/completion.rb
 - lib/openrouter/credit.rb
+- lib/openrouter/endpoint.rb
 - lib/openrouter/generation.rb
 - lib/openrouter/model.rb
 - lib/openrouter/stream.rb
@@ -53,6 +54,7 @@ files:
 - rbi/openrouter/client.rbi
 - rbi/openrouter/completion.rbi
 - rbi/openrouter/credit.rbi
+- rbi/openrouter/endpoint.rbi
 - rbi/openrouter/generation.rbi
 - rbi/openrouter/model.rbi
 - rbi/openrouter/openrouter.rbi