lex-xai 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 12eb444e9136e46f0734f9a6e89dd976582746194f2fb19e9d911e75e810644e
-  data.tar.gz: 346af921176e50c420ddf9b685439453fdf3ba79a971401cbdcd03fef03ad80c
+  metadata.gz: f122be7885c3d5f79b22015679579e69bbf8ccbb841a4e5f903688e2ae2d1221
+  data.tar.gz: 7738c16fc492d2f975702b535d9b1d54ca370bbe8013a0d03946b7203070386b
 SHA512:
-  metadata.gz: fd63e18d3f814c311020ad7137aeddf3ef51b9697a5fae70d8cb553ee2cf550ce39d611da511356c5d39fcef6d39c29ab069d6140ab6bd4301e184f034c96a72
-  data.tar.gz: 56332345d7420caff6d052f3e7ddb1c94aa36910089350a24ef8396d8fe2b2e1769e9e6ff78df0d3e10c608808127ea588676f7179d677b117ae74bc0dc06969
+  metadata.gz: 6b39da9f13c8445ca28b69402162d9323fc280eb84770c96040a0979f4890eae53c7acc1017ed7f07c4db8e140a57b38c51abb065cee872b82423e5c59b77aec
+  data.tar.gz: 9aea0da6fc20892bad4346e9c72d27c25244987b10d650b79f273fc0db53b313f965b4f605b4e00960c11d34f893a77e0fc85b1cef28c4651c082d6c8d2b8fea

data/CHANGELOG.md

@@ -1,5 +1,10 @@
 # Changelog
 
+## [0.1.4] - 2026-03-31
+
+### Added
+- Standardized `usage:` key in `Runners::Chat#create` and `Runners::Embeddings#create` return hashes with `input_tokens`, `output_tokens`, `cache_read_tokens`, and `cache_write_tokens` fields for consumption by legion-llm's CostEstimator and metering
+
 ## [0.1.3] - 2026-03-30
 
 ### Changed

data/lib/legion/extensions/xai/runners/chat.rb

@@ -16,7 +16,16 @@ module Legion
             body[:temperature] = temperature if temperature
 
             response = client(api_key: api_key, **).post('/v1/chat/completions', body)
-            { result: response.body }
+            body = response.body
+            {
+              result: body,
+              usage:  {
+                input_tokens:       body.dig('usage', 'prompt_tokens') || 0,
+                output_tokens:      body.dig('usage', 'completion_tokens') || 0,
+                cache_read_tokens:  0,
+                cache_write_tokens: 0
+              }
+            }
           end
 
           include Legion::Extensions::Helpers::Lex if Legion::Extensions.const_defined?(:Helpers, false) &&

data/lib/legion/extensions/xai/runners/embeddings.rb

@@ -13,7 +13,16 @@ module Legion
             body = { model: model, input: input }
 
             response = client(api_key: api_key, **).post('/v1/embeddings', body)
-            { result: response.body }
+            body = response.body
+            {
+              result: body,
+              usage:  {
+                input_tokens:       body.dig('usage', 'prompt_tokens') || 0,
+                output_tokens:      body.dig('usage', 'completion_tokens') || 0,
+                cache_read_tokens:  0,
+                cache_write_tokens: 0
+              }
+            }
           end
 
           include Legion::Extensions::Helpers::Lex if Legion::Extensions.const_defined?(:Helpers, false) &&

data/lib/legion/extensions/xai/version.rb

@@ -3,7 +3,7 @@
 module Legion
   module Extensions
     module Xai
-      VERSION = '0.1.3'
+      VERSION = '0.1.4'
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lex-xai
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - Esity