tokenr-ruby 0.1.0 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ce5be4fe9dded47d914a3e7e348b86b807d90a53b75d2e875c7a63e24d52344b
-  data.tar.gz: 8b36455f2b65965afd9b304e88c467117b0e3d14abbdf3e3e05449ec34092a69
+  metadata.gz: 15ea6965561601d93515ffdf3448f286976ad610bcb2374f225d30404fa6d008
+  data.tar.gz: d4e501b932db24625348ed27c82003e702dcf4e05ec32664e7bd1255eb13ac43
 SHA512:
-  metadata.gz: aeec3f6ee5c391a14f503c666989cd14bb62e1693dcd320a2b695a9f0fbed01c73107909f60a88b4191ce6421308cc375b1ed295e7095e4a8f3f8b3829bcf158
-  data.tar.gz: bc7c5d3133f07e146020eb3946a96956563003a068746296537259ba2361e769aab4bb8b39efeb92f601c6c0278c43155a86599846119770c09feba88e1bcf60
+  metadata.gz: 255f3b2269c15a41ad3e29c3f302dd79a2498e4098351afa8827db98cc8dde58d75b0acc625e899af24d68ce2bd892a2128e8471f8ac599b6a8aad2fe5555e2f
+  data.tar.gz: 9f46b51a1e802659dd855474f4fa1c67cbfc8a320560d16130c6ba6e87bb8c8ae84f62f635b8035c5454fbbcbaa2e0d83e8e44d9c4fa8393f9956a0e13588349

data/lib/tokenr/client.rb

@@ -2,6 +2,7 @@
 
 require "net/http"
 require "json"
+require "openssl"
 require "uri"
 
 module Tokenr
@@ -66,9 +67,17 @@ module Tokenr
     def request(method, path, body = nil, params = {})
       uri  = build_uri(path, params)
       http = Net::HTTP.new(uri.host, uri.port)
-      http.use_ssl       = uri.scheme == "https"
-      http.open_timeout  = 5
-      http.read_timeout  = 30
+      http.use_ssl      = uri.scheme == "https"
+      http.open_timeout = 5
+      http.read_timeout = 30
+      if http.use_ssl?
+        # Use VERIFY_PEER with an explicit cert store that has no CRL check flags.
+        # This keeps certificate chain validation while avoiding CRL fetch failures
+        # on servers whose CA certificates have CRL distribution points that are
+        # unreachable or return errors.
+        http.verify_mode = OpenSSL::SSL::VERIFY_PEER
+        http.cert_store  = OpenSSL::X509::Store.new.tap(&:set_default_paths)
+      end
 
       req = build_request(method, uri, body)
       handle_response(http.request(req))

data/lib/tokenr/integrations/anthropic.rb

@@ -39,7 +39,7 @@ module Tokenr
 
         def messages(model:, messages:, **params)
           start    = Process.clock_gettime(Process::CLOCK_MONOTONIC)
-          response = client.messages(model: model, messages: messages, **params)
+          response = client.messages.create(model: model, messages: messages, **params)
           latency  = ((Process.clock_gettime(Process::CLOCK_MONOTONIC) - start) * 1000).round
 
           track_response(model, response, latency)

data/lib/tokenr/integrations/openai.rb

@@ -34,33 +34,39 @@ module Tokenr
         end
 
         def chat(parameters:)
-          timed { client.chat(parameters: parameters) } do |response, latency_ms|
-            track_chat_response(parameters[:model], response, latency_ms)
-          end
+          start    = Process.clock_gettime(Process::CLOCK_MONOTONIC)
+          response = client.chat(parameters: parameters)
+          latency  = ((Process.clock_gettime(Process::CLOCK_MONOTONIC) - start) * 1000).round
+          track_chat_response(parameters[:model], response, latency)
+          response
         end
 
         def completions(parameters:)
-          timed { client.completions(parameters: parameters) } do |response, latency_ms|
-            track_chat_response(parameters[:model], response, latency_ms)
-          end
+          start    = Process.clock_gettime(Process::CLOCK_MONOTONIC)
+          response = client.completions(parameters: parameters)
+          latency  = ((Process.clock_gettime(Process::CLOCK_MONOTONIC) - start) * 1000).round
+          track_chat_response(parameters[:model], response, latency)
+          response
         end
 
         def embeddings(parameters:)
-          timed { client.embeddings(parameters: parameters) } do |response, latency_ms|
-            usage = response.dig("usage")
-            next unless usage
-
+          start    = Process.clock_gettime(Process::CLOCK_MONOTONIC)
+          response = client.embeddings(parameters: parameters)
+          latency  = ((Process.clock_gettime(Process::CLOCK_MONOTONIC) - start) * 1000).round
+          usage    = response.dig("usage")
+          if usage
             Tokenr.track(
-              model:        parameters[:model],
-              provider:     "openai",
-              input_tokens: usage["prompt_tokens"] || 0,
+              model:         parameters[:model],
+              provider:      "openai",
+              input_tokens:  usage["prompt_tokens"] || 0,
               output_tokens: 0,
-              latency_ms:   latency_ms,
-              agent_id:     agent_id,
-              feature_name: feature_name,
-              tags:         tags
+              latency_ms:    latency,
+              agent_id:      agent_id,
+              feature_name:  feature_name,
+              tags:          tags
             )
           end
+          response
         end
 
         # Pass any other methods straight through to the underlying client.
@@ -74,14 +80,6 @@ module Tokenr
 
         private
 
-        def timed
-          start    = Process.clock_gettime(Process::CLOCK_MONOTONIC)
-          response = yield
-          latency  = ((Process.clock_gettime(Process::CLOCK_MONOTONIC) - start) * 1000).round
-          yield response, latency if block_given?
-          response
-        end
-
         def track_chat_response(model, response, latency_ms)
           usage = response.dig("usage")
           return unless usage

data/lib/tokenr/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Tokenr
-  VERSION = "0.1.0"
+  VERSION = "0.1.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tokenr-ruby
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.3
 platform: ruby
 authors:
 - Tokenr
@@ -50,7 +50,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.9
+rubygems_version: 4.0.6
 specification_version: 4
 summary: Automatic LLM cost tracking for OpenAI, Anthropic, and more
 test_files: []