llm_specs 0.1.7 → 0.1.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 94bc17db460eba0ae57d26432140f80e6b9d966f9f6d5b00bafabf1390269f16
-  data.tar.gz: 180a85f7aab4e81acb09a50c18de4f446e0067d2a559e0ff778a16f404c807ff
+  metadata.gz: 81de36354efcb2d8856d761c6e773ab49e7056d22360695f768bd346ed044958
+  data.tar.gz: f5ad326858cb9e75cea2ff30edc38ddccf5838fb2660f819a69bfbe82d232129
 SHA512:
-  metadata.gz: 3b84d7ecddcea54b697f20d898bfcc8a3ba18efa2195381a18347919bb55a1a7c22da048fa45abc8086a8b94f400c4aadd6eb41be58fb8a8eff8586b99ea91b3
-  data.tar.gz: 898b8bb26b823e0034b6470ac6257b7b34ded0f1b5aa743183f06fc5a2fc0d6d11e5851dce8a922e396e1ba3029c5529bd67278bffff31a7860d416b13aa3368
+  metadata.gz: 2d11e01888a0e88d5fb129fba566992f12ba6ee9a173fa3584786406bf8927e2c48ada1e4843d1685006096ef525c5c61721c0b4d5a50ede9aed73b21ebb9841
+  data.tar.gz: 7855d013340c075965cfa83adaaa4d656710f77ebe8cd24927cfb10600b06da10da029d347b883fda58954bbaa1e3ed0265cee22b80e1292edb63200a9e2f991

data/README.md

@@ -16,7 +16,12 @@ It provides a simple, efficient way to access model metadata with built-in cachi
 ```ruby
 require 'llm_specs'
 
-models = LLMSpecs.models
+models = LLMSpecs.models # all models
+
+models = LLMSpecs.gemini # provider specific models
+models = LLMSpecs.anthropic
+models = LLMSpecs.openai
+models = LLMSpecs.deepseek
 ```
 
 ### Find a model by ID
@@ -28,9 +33,13 @@ model = LLMSpecs.models.find("claude-3-sonnet")
 ```ruby
 # List all Anthropic models
 anthropic_models = LLMSpecs.models.where(provider: "anthropic")
+# or 
+anthropic_models = LLMSpecs.anthropic
 
 # List all Anthropic models by family "claude-3-7-sonnet"
 claude_models = LLMSpecs.models.where(provider: "anthropic", family: "claude-3-7-sonnet")
+#or 
+claude_models = LLMSpecs.anthropic.where(family: "claude-3-7-sonnet")
 
 # List all models that support streaming
 streaming_models = LLMSpecs.models.select(&:streaming?)

data/lib/llm_specs/errors.rb

@@ -2,7 +2,7 @@
 module LLMSpecs
   class ModelNotFound < StandardError
     def initialize(id)
-      super("Couldn't find model with id='#{id}'")
+      super "Couldn't find model with id='#{id}'"
     end
   end
 end

data/lib/llm_specs/model.rb

@@ -1,28 +1,28 @@
 # frozen_string_literal: true
 module LLMSpecs
   class Model < Data.define(:id, :name, :provider, :family, :context_window, :max_output_tokens, :modalities, :capabilities, :pricing)
-    def supports?(*caps)
-      caps.all? { capabilities.include?(it.to_s) }
+    def supports?(*args)
+      args.all? { capabilities.include? it.to_s }
     end
     
-    def supports_input?(*modes)
-      modes.all? { input_modalities.include?(it.to_s) }
+    def supports_input?(*args)
+      args.all? { input_modalities.include? it.to_s }
     end
     
-    def supports_output?(*modes)
-      modes.all? { output_modalities.include?(it.to_s) }
+    def supports_output?(*args)
+      args.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) }
+      define_method(:"#{capability}?") { supports? capability }
     end
     
-    %i[text image audio].each do |method|
-      define_method(:"#{method}_input?") { supports_input?(method) }
+    %i[text image audio].each do |mode|
+      define_method(:"#{mode}_input?") { supports_input? mode }
     end
     
-     %i[text image audio embeddings].each do |method|
-      define_method(:"#{method}_output?") { supports_output?(method) }
+     %i[text image audio embeddings].each do |mode|
+      define_method(:"#{mode}_output?") { supports_output? mode }
     end
     
     def input_modalities

data/lib/llm_specs/version.rb

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

data/lib/llm_specs.rb

@@ -20,5 +20,9 @@ module LLMSpecs
     def models
       @models ||= Catalog.new(api_uri: API_URI, cache_path: cache_path).models
     end
+    
+    %i[anthropic deepseek gemini openai].each do |prov|
+      define_method(prov) { models.where(provider: prov) }
+    end
   end
 end

metadata

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