llm_specs 0.1.3 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e55746ce8150b1e3fc1554277363ad1b03828eee4a13b3199d461dc811547e64
-  data.tar.gz: 320e3b24f5b6373f2e76d98be22d35d8589d727cd48914f7f26c0f419aece373
+  metadata.gz: 79037b669b30677efe9d924a7b8083f426f943f54d66ea975de9e3b9757bc85f
+  data.tar.gz: d778a3d458ef032b79cf451a6c1ccfd28ba8aac22472613c7535153bed8f4498
 SHA512:
-  metadata.gz: d3ccbf4b868ebca48dcdf748a48a37e14cea5b0fabd7170ec26cb76f602eb182acff5225b3bc36f246ba0b15db5db31f8dae7fe771e19c62788990355cffc267
-  data.tar.gz: 01047dcd8b33b838da2bdeb67781416b58a5564031848aa02a4f78a165cbfce6e2ea929a5eb9deb2850a8e38412a58ceadc4c0f5705e44f803e2cdb37b7e7929
+  metadata.gz: ef501f8bdc9550691ce68b124d047115bb66c3c14b0ee73a4bd49405d05be00ed03d021f9afb25f51c26a11378c2220be1dc57fd16d8ce423a19e396571b7a50
+  data.tar.gz: 791ea85d4793569859f2389fbbfd0ecf847a365f2e1004805fff5410b60cda2234fa2588fbf7fc54a17e4708dbd642aa1dfc5c36b9bacd636d777573745bd165

data/README.md

@@ -66,27 +66,34 @@ model.supports?(:function_calling, :batch) # => true/false
 
 or input or output modalities:
 ```ruby
-models.supports_input?(:audio) # => true/false
-models.supports_input?(:audio, :text) # multiple argument
-models.audio_input? # shortcut
-models.image_input?
-models.text_input?
-
-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?
+model.supports_input?(:audio) # => true/false
+model.supports_input?(:audio, :text) # multiple argument
+model.audio_input? # shortcut
+model.image_input?
+model.text_input?
+
+model.supports_output?(:audio) # => true/false
+model.supports_output?(:audio, :embeddings) # multiple argument
+model.audio_output? # shortcut
+model.image_output?
+model.text_output?
+model.embeddings_output?
 ```
 
 Pricing methods:
 ```ruby
-model.input_pricing  # => $ per 1M input tokens (default :text_tokens)
+model.input_pricing  # 1.1 => $ per 1M input tokens (default :text_tokens)
+model.output_pricing
+
 model.input_pricing(:text_tokens, :batch)
 model.output_pricing(:embeddings)
 ```
 
+## Cache configuration
+
+```ruby
+LLMSpecs.cache_path = Rails.root.join("tmp", "cache", "llm-specs-cache.json")
+```
 
 ## License
 

data/lib/llm_specs/cache.rb

@@ -8,9 +8,7 @@ module LLMSpecs
     
     def fetch
       return read if valid?
-      data = yield
-      write(data)
-      data
+      yield.tap { write(it) }
     end
     
     def valid?

data/lib/llm_specs/catalog.rb

@@ -14,10 +14,14 @@ module LLMSpecs
     
     def fetch
       @cache.fetch do
-        response = Net::HTTP.get_response(@uri)
-        raise FetchError.new(response) unless response.is_a? Net::HTTPSuccess
-        JSON.parse(response.body, symbolize_names: true)
+        parse Net::HTTP.get_response(@uri).tap(&:value) # raises Net::HTTPError if not 2xx 
       end
     end
+    
+    def parse(response)
+      JSON.parse(response.body, symbolize_names: true)
+    end
   end
 end
+
+

data/lib/llm_specs/errors.rb

@@ -5,12 +5,4 @@ module LLMSpecs
       super("Couldn't find model with id='#{id}'")
     end
   end
-  
-  class FetchError < StandardError
-    attr_reader :response
-    def initialize(response)
-      @response = response
-      super("HTTP fetch failed (status: #{response.status})")
-    end
-  end
 end

data/lib/llm_specs/version.rb

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

data/lib/llm_specs.rb

@@ -15,7 +15,7 @@ module LLMSpecs
     attr_writer :cache_path
 
     def cache_path
-      @cache_path || "models.json"
+      @cache_path || "llm-specs.json"
     end
 
     def models

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llm_specs
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.5
 platform: ruby
 authors:
 - Max Power
@@ -43,7 +43,6 @@ files:
 - lib/llm_specs/errors.rb
 - lib/llm_specs/model.rb
 - lib/llm_specs/version.rb
-- models.json
 - sig/llm_specs.rbs
 homepage: https://github.com/max-power/llm_specs
 licenses: