llm_specs 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6920accbfa3bf50fabf644d681a5ceaf498828b72ea538a6e679822e7b7c488c
-  data.tar.gz: 115f7d7daa49ba5d455161dc196bd65931c089c452fd385c4143f4063c147e21
+  metadata.gz: 3fb186c06a052eb3818811b07ff649ce0122313ac7be76036d48ed8810aaf825
+  data.tar.gz: 0ac219ca8e0619e96876bd113c09047a0b43d17f9d92ff555a9c50d3b194d782
 SHA512:
-  metadata.gz: 720806a631c89c2690c23ec4744c6ecfef396c1926a0c69787c2d2a3c946a9144f9f456bd9db00f715ec037c506d56fe9e781f6c49944e0ca8af3cfcbedd839e
-  data.tar.gz: 2a99e8ac6b1baf77a608e30f660da6675f27979c82784f2fc59949649199be4235e2e027ac7b898f509aa9c2254296e40619e7d363404a46db1c3b174fb9d59a
+  metadata.gz: eb8025aa56c42285c3bb97c325dba24ed5b81a6fafe0c8ded5b739b759df82d4c2d3b830215ba6a2741df0e0e3b5704f02139f4ad98d96172d3ac3fb89665223
+  data.tar.gz: 02542bf9b714b4319b53307c26d36d695094adb7eb8a70fa237a506afe98afe765ab19475e2396d892c4dbbcca8257706108691636a38d2a5ade427a200590a6

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.2"
 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.2
 platform: ruby
 authors:
 - Max Power