lex-azure-ai 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 18fefac48c3aa974940b59aee94a3f8bd9a36ffcdf0434d12e7b6914a4e65066
-  data.tar.gz: '0209e8bce28f5988d3b65f7dbcc34c1f95ade99f7f4c0577e613fe3bd146d173'
+  metadata.gz: d62cce58c74c7629813ee041e1170b89b8bf840290e4780df44c498842e74575
+  data.tar.gz: d7ec97c677ba8c83c3e13fcfacf5e5ac9f3fb34eae518b65d57f21d37e1314e6
 SHA512:
-  metadata.gz: abf756e30f1d28a10ee439ae0e6ed5ff57482dae2343576e673970cf5a7e4cf47378aae2127c8f0b96f846bf2e116cc7e06f03fa0ddfe620fb0ea49b5e1ca219
-  data.tar.gz: eec52ba23401836ea655aebfa873d1277b0a05ef77933ce4014017d3a54303268dd16f2b0b9d062f9ed3f88ba7723a4778c8b5285e3ebd75691ea5d1cfbd23b6
+  metadata.gz: 2658586dd660d0a5dfce414f12932ca82582a7d142ad4605ef7eaf649aaf30368ebb3fd31a48dc6459f3120539622c34c82fd017a68d961e15075a23ae896998
+  data.tar.gz: dde8723318bf4c19c9c7396d6f2df1f6e25abcbf59f9f24051f1f0fdd1a3f95cc26153955aba57bbf16db34cca825a2a7d0f1c65a556532a70565003b0ab7daf

data/CHANGELOG.md

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

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

@@ -17,7 +17,15 @@ module Legion
 
             path = "/openai/deployments/#{deployment}/chat/completions?api-version=#{api_version}"
             response = client(api_key: api_key, endpoint: endpoint, api_version: api_version).post(path, body)
-            { result: response.body }
+            {
+              result: response.body,
+              usage:  {
+                input_tokens:       response.body.dig('usage', 'prompt_tokens') || 0,
+                output_tokens:      response.body.dig('usage', 'completion_tokens') || 0,
+                cache_read_tokens:  response.body.dig('usage', 'prompt_tokens_details', 'cached_tokens') || 0,
+                cache_write_tokens: 0
+              }
+            }
           end
 
           include Legion::Extensions::Helpers::Lex if Legion::Extensions.const_defined?(:Helpers, false) &&

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

@@ -13,7 +13,15 @@ module Legion
             body = { input: input }
             path = "/openai/deployments/#{deployment}/embeddings?api-version=#{api_version}"
             response = client(api_key: api_key, endpoint: endpoint, api_version: api_version).post(path, body)
-            { result: response.body }
+            {
+              result: response.body,
+              usage:  {
+                input_tokens:       response.body.dig('usage', 'prompt_tokens') || 0,
+                output_tokens:      response.body.dig('usage', 'completion_tokens') || 0,
+                cache_read_tokens:  response.body.dig('usage', 'prompt_tokens_details', 'cached_tokens') || 0,
+                cache_write_tokens: 0
+              }
+            }
           end
 
           include Legion::Extensions::Helpers::Lex if Legion::Extensions.const_defined?(:Helpers, false) &&

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

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

metadata

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