raix 2.0.3 → 2.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 44058aeb590e21ead9e7db9ebdc9a4ecffc9f4771b75663e137b22ecca4cd7d2
-  data.tar.gz: c59475844e68ef02379d85f7dde02540a58e7659d6b44a7fb83db7f491998635
+  metadata.gz: ec3b7449e51440e1d669e323dea0c371a3d18d7d4083090d61599ff1dbce3bbe
+  data.tar.gz: 6f2029218c2099ab1172d1d458be9ccb33836078d5abc9b04865cae0d497bc07
 SHA512:
-  metadata.gz: 76c43e2109fc3ec1be374c2b5df9e024ad09274e2ece3df976eb6f961d022123453bb995f6f0680921df5f224c439cbd19ec8eac8d5acc201d80193541396509
-  data.tar.gz: c0ee5f840e83718168668cf844a2cb3e7969f2cd5e199235b8532049660aab0cf7750abcf0e8748ffda7427fee73c7739fc37060c739efc64bcf7b04867c9da5
+  metadata.gz: 76e1d77e80023dbc634e4b2f72eaa269f05d539ea97470df1355e69e73d75a6cf4bd2bd7e627ab2cf01fcb381fd85c4b153d28005f32190ff1b8ae69c3bea275
+  data.tar.gz: 6fedf4bae5c406b72b83a1aacbc985849eb7c461237e7cc1e12d9ca73ea05a861912f13739bc92ba8ad992bf88f5a31fa530dc9468c5bdb4763f53caf3833843

data/CHANGELOG.md

@@ -1,5 +1,11 @@
 ## [Unreleased]
 
+## [2.0.4] - 2026-05-19
+
+### Fixed
+- `ruby_llm_request` now preserves the upstream provider's `id`, `model`, and `provider` fields, plus the full `usage` payload (including `prompt_tokens_details.cached_tokens` and `completion_tokens_details.reasoning_tokens`) on the OpenAI-compatible response hash. Previously the conversion dropped everything except `choices` and basic token counts, which broke callers that needed the generation id for authoritative cost lookups (e.g. OpenRouter's `/api/v1/generation` endpoint) or that wanted to verify prompt-cache hits via `cached_tokens`.
+- Added `require "active_support/core_ext/module/delegation"` so `Raix::ChatCompletion` loads cleanly without an external preload of ActiveSupport. The class uses `delegate :configuration, to: :class` but did not pull in the required core ext, so a bare `require "raix"` would raise `NoMethodError` for `delegate`.
+
 ## [2.0.3] - 2026-04-30
 
 ### Fixed

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    raix (2.0.3)
+    raix (2.0.4)
       activesupport (>= 6.0)
       faraday-retry (~> 2.0)
       ostruct

data/lib/raix/chat_completion.rb

@@ -4,6 +4,7 @@ require "active_support/concern"
 require "active_support/core_ext/object/blank"
 require "active_support/core_ext/string/filters"
 require "active_support/core_ext/hash/indifferent_access"
+require "active_support/core_ext/module/delegation"
 require "ruby_llm"
 
 module Raix
@@ -56,9 +57,7 @@ module Raix
     end
 
     # Instance level access to the class-level configuration.
-    def configuration
-      self.class.configuration
-    end
+    delegate :configuration, to: :class
 
     # This method performs chat completion based on the provided transcript and parameters.
     #
@@ -398,8 +397,34 @@ module Raix
         # Non-streaming mode - return OpenAI-compatible response format
         response_message = has_user_message ? chat.complete : chat.ask
 
-        # Convert RubyLLM response to OpenAI format for compatibility
+        # Pull through the raw provider payload when available. OpenRouter's
+        # `id` is the only handle we have to look up authoritative billing
+        # cost via /api/v1/generation, and callers that watch the response
+        # snapshot for `model` / cached-token counts shouldn't have to break
+        # out of the OpenAI-compatible shape to get them.
+        raw_body = response_message.raw.respond_to?(:body) ? response_message.raw.body : nil
+        raw_body = {} unless raw_body.is_a?(Hash)
+
+        usage_payload = {
+          "prompt_tokens" => response_message.input_tokens,
+          "completion_tokens" => response_message.output_tokens,
+          "total_tokens" => (response_message.input_tokens || 0) + (response_message.output_tokens || 0)
+        }
+
+        # Merge prompt_tokens_details / completion_tokens_details (cached tokens,
+        # reasoning tokens) when the provider supplied them.
+        if (upstream_usage = raw_body["usage"]).is_a?(Hash)
+          upstream_usage.each do |key, value|
+            next if usage_payload.key?(key)
+
+            usage_payload[key] = value
+          end
+        end
+
         {
+          "id" => raw_body["id"],
+          "model" => raw_body["model"] || response_message.model_id,
+          "provider" => raw_body["provider"],
           "choices" => [
             {
               "message" => {
@@ -410,11 +435,7 @@ module Raix
               "finish_reason" => response_message.tool_call? ? "tool_calls" : "stop"
             }
           ],
-          "usage" => {
-            "prompt_tokens" => response_message.input_tokens,
-            "completion_tokens" => response_message.output_tokens,
-            "total_tokens" => (response_message.input_tokens || 0) + (response_message.output_tokens || 0)
-          }
+          "usage" => usage_payload
         }
       end
     rescue StandardError => e

data/lib/raix/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Raix
-  VERSION = "2.0.3"
+  VERSION = "2.0.4"
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: raix
 version: !ruby/object:Gem::Version
-  version: 2.0.3
+  version: 2.0.4
 platform: ruby
 authors:
 - Obie Fernandez
 bindir: exe
 cert_chain: []
-date: 2026-04-30 00:00:00.000000000 Z
+date: 2026-05-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport