llm_specs 0.1.1 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6920accbfa3bf50fabf644d681a5ceaf498828b72ea538a6e679822e7b7c488c
-  data.tar.gz: 115f7d7daa49ba5d455161dc196bd65931c089c452fd385c4143f4063c147e21
+  metadata.gz: e55746ce8150b1e3fc1554277363ad1b03828eee4a13b3199d461dc811547e64
+  data.tar.gz: 320e3b24f5b6373f2e76d98be22d35d8589d727cd48914f7f26c0f419aece373
 SHA512:
-  metadata.gz: 720806a631c89c2690c23ec4744c6ecfef396c1926a0c69787c2d2a3c946a9144f9f456bd9db00f715ec037c506d56fe9e781f6c49944e0ca8af3cfcbedd839e
-  data.tar.gz: 2a99e8ac6b1baf77a608e30f660da6675f27979c82784f2fc59949649199be4235e2e027ac7b898f509aa9c2254296e40619e7d363404a46db1c3b174fb9d59a
+  metadata.gz: d3ccbf4b868ebca48dcdf748a48a37e14cea5b0fabd7170ec26cb76f602eb182acff5225b3bc36f246ba0b15db5db31f8dae7fe771e19c62788990355cffc267
+  data.tar.gz: 01047dcd8b33b838da2bdeb67781416b58a5564031848aa02a4f78a165cbfce6e2ea929a5eb9deb2850a8e38412a58ceadc4c0f5705e44f803e2cdb37b7e7929

data/README.md

@@ -11,6 +11,35 @@ It provides a simple, efficient way to access model metadata with built-in cachi
 - Provides a queryable catalog of models via .models
 - Supports filtering and lookup by model attributes
 
+
+## Example Usage
+```ruby
+require 'llm_specs'
+
+models = LLMSpecs.models
+```
+
+### Find a model by ID
+```ruby
+model = LLMSpecs.models.find("claude-3-sonnet")
+```
+
+### Filter models
+```ruby
+# List all Anthropic models
+anthropic_models = LLMSpecs.models.where(provider: "anthropic")
+
+# List all Anthropic models by family "claude-3-7-sonnet"
+claude_models = LLMSpecs.models.where(provider: "anthropic", family: "claude-3-7-sonnet")
+
+# List all models that support streaming
+streaming_models = LLMSpecs.models.select(&:streaming?)
+
+# List all models that support audio output
+audio_models = LLMSpecs.models.select(&:audio_output?)
+
+```
+
 ##  Model Capabilities
 Each model is represented by an instance of `LLMSpecs::Model`, a value object that encapsulates:
 
@@ -27,36 +56,37 @@ Each model is represented by an instance of `LLMSpecs::Model`, a value object th
 
 - Pricing breakdowns via `input_pricing` and `output_pricing`
 
-You can easily check capabilities or pricing:
+You can easily check capabilities:
 ```ruby
-model.function_calling?  # => true/false
-model.input_pricing      # => $ per 1M input tokens (default)
-model.output_pricing(:image_tokens)
+model.supports?(:function_calling) # => true/false
+model.function_calling?  # => shortcut for above
+# with multiple arguments
+model.supports?(:function_calling, :batch) # => true/false
 ```
 
-## Example Usage
+or input or output modalities:
 ```ruby
-require 'llm_specs'
+models.supports_input?(:audio) # => true/false
+models.supports_input?(:audio, :text) # multiple argument
+models.audio_input? # shortcut
+models.image_input?
+models.text_input?
 
-models = LLMSpecs.models
+models.supports_output?(:audio) # => true/false
+models.supports_output?(:audio, :embeddings) # multiple argument
+models.audio_output? # shortcut
+models.image_output?
+models.text_output?
+models.embeddings_output?
 ```
 
-### Find a model by ID
+Pricing methods:
 ```ruby
-model = LLMSpecs.models.find("claude-3-sonnet")
+model.input_pricing  # => $ per 1M input tokens (default :text_tokens)
+model.input_pricing(:text_tokens, :batch)
+model.output_pricing(:embeddings)
 ```
 
-### Filter models
-```ruby
-anthropic_models = LLMSpecs.models.where(provider: "anthropic")
-
-claude_models = LLMSpecs.models.where(provider: "anthropic", family: "claude-3-7-sonnet")
-```
-
-### List all models that support streaming
-```ruby
-streaming_models = LLMSpecs.models.select(&:streaming?)
-```
 
 ## License
 

data/lib/llm_specs/model.rb

@@ -1,14 +1,30 @@
 # frozen_string_literal: true
 module LLMSpecs
   class Model < Data.define(:id, :name, :provider, :family, :context_window, :max_output_tokens, :modalities, :capabilities, :pricing)
-    def supports?(capability)
-      capabilities.include?(capability.to_s)
+    def supports?(*caps)
+      caps.all? { capabilities.include?(it.to_s) }
+    end
+    
+    def supports_input?(*modes)
+      modes.all? { input_modalities.include?(it.to_s) }
+    end
+    
+    def supports_output?(*modes)
+      modes.all? { output_modalities.include?(it.to_s) }
     end
     
     %i[function_calling structured_output batch reasoning citations streaming].each do |capability|
       define_method(:"#{capability}?") { supports?(capability) }
     end
     
+    %i[text image audio].each do |method|
+      define_method(:"#{method}_input?") { supports_input?(method) }
+    end
+    
+     %i[text image audio embeddings].each do |method|
+      define_method(:"#{method}_output?") { supports_output?(method) }
+    end
+    
     def input_modalities
       modalities[:input]
     end

data/lib/llm_specs/version.rb

@@ -1,4 +1,4 @@
 # frozen_string_literal: true
 module LLMSpecs
-  VERSION = "0.1.1"
+  VERSION = "0.1.3"
 end

data/lib/llm_specs.rb

@@ -9,10 +9,17 @@ require_relative "llm_specs/collection"
 require_relative "llm_specs/model"
 
 module LLMSpecs
-  API_URI    = "https://api.parsera.org/v1/llm-specs"
-  CACHE_PATH = "models.json"
+  API_URI = "https://api.parsera.org/v1/llm-specs"
   
-  def self.models
-    @models ||= Catalog.new(api_uri: API_URI, cache_path: CACHE_PATH).models
+  class << self
+    attr_writer :cache_path
+
+    def cache_path
+      @cache_path || "models.json"
+    end
+
+    def models
+      @models ||= Catalog.new(api_uri: API_URI, cache_path: cache_path).models
+    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llm_specs
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.3
 platform: ruby
 authors:
 - Max Power