lex-foundry 0.1.3 → 0.1.4

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (7) hide show
  1. checksums.yaml +4 -4
  2. data/CHANGELOG.md +5 -0
  3. data/lib/legion/extensions/foundry/runners/connections.rb +18 -6
  4. data/lib/legion/extensions/foundry/runners/deployments.rb +18 -6
  5. data/lib/legion/extensions/foundry/runners/models.rb +15 -2
  6. data/lib/legion/extensions/foundry/version.rb +1 -1
  7. metadata +1 -1
checksums.yaml CHANGED
@@ -1,7 +1,7 @@
1
1
  ---
2
2
  SHA256:
3
- metadata.gz: ca48a900f575df9bc27d4fae5ece1d41c371870dee4ea10e6a8b2b9899e02fee
4
- data.tar.gz: 481f632e9fe917799aab24132a5a2acbc56ec74ff30f54208ad76f177df3bdec
3
+ metadata.gz: 66e9d1e3d64d09ca76c4bc919fc4aece39c9bceb70308e6cb84339267348f486
4
+ data.tar.gz: 18613233a290dd61cf4d1e90a2b1011139e1b0918f2c8e8a1d545e0491c2d316
5
5
  SHA512:
6
- metadata.gz: 7b784251f2beed3bf26623e92ce3f86bd4f997fdf4c116b78eece35fa40a4f12a1ee14b6e25ccd78b2b12d1eada1e4e2ff6aaa4939cff2c7ed5828290e34c81f
7
- data.tar.gz: a0db5eb0162e03fc1df59b3f7f5b42208883de2a4896d07457d59993e7219b38dbb15ee9c4290146ae7e9f8b36c4133f34f85d06e2a953c9c0607aa13a7fe388
6
+ metadata.gz: 816d38b816d3aded715b26f80212aeb69c1ebe8eee25c03db7d359a44e46ee1d400035f5bc2165361ffe56b1233be23b74ebe256416ef46c48abd194975a1050
7
+ data.tar.gz: 2c4863f6325177e788db78062bfbb7a08a91ba36ec1d458eaba21c8b7c75039da5c774e4960b8222d3a66016c6cea853b74601b4a03deee64dd2468eca0455d1
data/CHANGELOG.md CHANGED
@@ -1,5 +1,10 @@
1
1
  # Changelog
2
2
 
3
+ ## [0.1.4] - 2026-03-31
4
+
5
+ ### Added
6
+ - Standardized `usage:` key in all runner return hashes with `input_tokens`, `output_tokens`, `cache_read_tokens`, and `cache_write_tokens` fields parsed from Anthropic Foundry API responses (defaults to zero when not present)
7
+
3
8
  ## [0.1.3] - 2026-03-30
4
9
 
5
10
  ### Changed
data/lib/legion/extensions/foundry/runners/connections.rb CHANGED
@@ -14,7 +14,8 @@ module Legion
14
14
  path = arm_connections_path(subscription_id, resource_group, workspace)
15
15
  response = management_client(token: token, endpoint: endpoint)
16
16
  .get("#{path}?api-version=#{api_version}")
17
- { connections: response.body }
17
+ body = response.body
18
+ { connections: body, usage: extract_usage(body) }
18
19
  end
19
20
 
20
21
  def get(name:, token:, endpoint:, subscription_id:, resource_group:, workspace:,
@@ -22,16 +23,18 @@ module Legion
22
23
  path = arm_connections_path(subscription_id, resource_group, workspace)
23
24
  response = management_client(token: token, endpoint: endpoint)
24
25
  .get("#{path}/#{name}?api-version=#{api_version}")
25
- { connection: response.body }
26
+ body = response.body
27
+ { connection: body, usage: extract_usage(body) }
26
28
  end
27
29
 
28
30
  def create(name:, type:, target:, token:, endpoint:, subscription_id:, resource_group:, workspace:,
29
31
  api_version: '2024-10-01-preview', **)
30
32
  path = arm_connections_path(subscription_id, resource_group, workspace)
31
- body = { properties: { category: type, target: target } }
33
+ request_body = { properties: { category: type, target: target } }
32
34
  response = management_client(token: token, endpoint: endpoint)
33
- .put("#{path}/#{name}?api-version=#{api_version}", body)
34
- { connection: response.body }
35
+ .put("#{path}/#{name}?api-version=#{api_version}", request_body)
36
+ body = response.body
37
+ { connection: body, usage: extract_usage(body) }
35
38
  end
36
39
 
37
40
  def delete(name:, token:, endpoint:, subscription_id:, resource_group:, workspace:,
@@ -39,7 +42,7 @@ module Legion
39
42
  path = arm_connections_path(subscription_id, resource_group, workspace)
40
43
  management_client(token: token, endpoint: endpoint)
41
44
  .delete("#{path}/#{name}?api-version=#{api_version}")
42
- { deleted: true }
45
+ { deleted: true, usage: extract_usage(nil) }
43
46
  end
44
47
  include Legion::Extensions::Helpers::Lex if Legion::Extensions.const_defined?(:Helpers, false) &&
45
48
  Legion::Extensions::Helpers.const_defined?(:Lex, false)
@@ -50,6 +53,15 @@ module Legion
50
53
  "/subscriptions/#{subscription_id}/resourceGroups/#{resource_group}" \
51
54
  "/providers/Microsoft.MachineLearningServices/workspaces/#{workspace}/connections"
52
55
  end
56
+
57
+ def extract_usage(body)
58
+ {
59
+ input_tokens: body&.dig('usage', 'input_tokens') || 0,
60
+ output_tokens: body&.dig('usage', 'output_tokens') || 0,
61
+ cache_read_tokens: body&.dig('usage', 'cache_read_input_tokens') || 0,
62
+ cache_write_tokens: body&.dig('usage', 'cache_creation_input_tokens') || 0
63
+ }
64
+ end
53
65
  end
54
66
  end
55
67
  end
data/lib/legion/extensions/foundry/runners/deployments.rb CHANGED
@@ -14,7 +14,8 @@ module Legion
14
14
  path = arm_deployments_path(subscription_id, resource_group, workspace)
15
15
  response = management_client(token: token, endpoint: endpoint)
16
16
  .get("#{path}?api-version=#{api_version}")
17
- { deployments: response.body }
17
+ body = response.body
18
+ { deployments: body, usage: extract_usage(body) }
18
19
  end
19
20
 
20
21
  def get(name:, token:, endpoint:, subscription_id:, resource_group:, workspace:,
@@ -22,16 +23,18 @@ module Legion
22
23
  path = arm_deployments_path(subscription_id, resource_group, workspace)
23
24
  response = management_client(token: token, endpoint: endpoint)
24
25
  .get("#{path}/#{name}?api-version=#{api_version}")
25
- { deployment: response.body }
26
+ body = response.body
27
+ { deployment: body, usage: extract_usage(body) }
26
28
  end
27
29
 
28
30
  def create(name:, model:, token:, endpoint:, subscription_id:, resource_group:, workspace:,
29
31
  sku: 'Standard', api_version: '2024-10-01-preview', **)
30
32
  path = arm_deployments_path(subscription_id, resource_group, workspace)
31
- body = { properties: { model: model }, sku: { name: sku } }
33
+ request_body = { properties: { model: model }, sku: { name: sku } }
32
34
  response = management_client(token: token, endpoint: endpoint)
33
- .put("#{path}/#{name}?api-version=#{api_version}", body)
34
- { deployment: response.body }
35
+ .put("#{path}/#{name}?api-version=#{api_version}", request_body)
36
+ body = response.body
37
+ { deployment: body, usage: extract_usage(body) }
35
38
  end
36
39
 
37
40
  def delete(name:, token:, endpoint:, subscription_id:, resource_group:, workspace:,
@@ -39,7 +42,7 @@ module Legion
39
42
  path = arm_deployments_path(subscription_id, resource_group, workspace)
40
43
  management_client(token: token, endpoint: endpoint)
41
44
  .delete("#{path}/#{name}?api-version=#{api_version}")
42
- { deleted: true }
45
+ { deleted: true, usage: extract_usage(nil) }
43
46
  end
44
47
  include Legion::Extensions::Helpers::Lex if Legion::Extensions.const_defined?(:Helpers, false) &&
45
48
  Legion::Extensions::Helpers.const_defined?(:Lex, false)
@@ -51,6 +54,15 @@ module Legion
51
54
  "/providers/Microsoft.MachineLearningServices/workspaces/#{workspace}" \
52
55
  '/onlineEndpoints/default/deployments'
53
56
  end
57
+
58
+ def extract_usage(body)
59
+ {
60
+ input_tokens: body&.dig('usage', 'input_tokens') || 0,
61
+ output_tokens: body&.dig('usage', 'output_tokens') || 0,
62
+ cache_read_tokens: body&.dig('usage', 'cache_read_input_tokens') || 0,
63
+ cache_write_tokens: body&.dig('usage', 'cache_creation_input_tokens') || 0
64
+ }
65
+ end
54
66
  end
55
67
  end
56
68
  end
data/lib/legion/extensions/foundry/runners/models.rb CHANGED
@@ -12,17 +12,30 @@ module Legion
12
12
  def list(token:, endpoint:, api_version: '2024-10-01-preview', **)
13
13
  response = workspace_client(token: token, endpoint: endpoint)
14
14
  .get("/models?api-version=#{api_version}")
15
- { models: response.body }
15
+ body = response.body
16
+ { models: body, usage: extract_usage(body) }
16
17
  end
17
18
 
18
19
  def get(model_id:, token:, endpoint:, api_version: '2024-10-01-preview', **)
19
20
  response = workspace_client(token: token, endpoint: endpoint)
20
21
  .get("/models/#{model_id}?api-version=#{api_version}")
21
- { model: response.body }
22
+ body = response.body
23
+ { model: body, usage: extract_usage(body) }
22
24
  end
23
25
 
24
26
  include Legion::Extensions::Helpers::Lex if Legion::Extensions.const_defined?(:Helpers, false) &&
25
27
  Legion::Extensions::Helpers.const_defined?(:Lex, false)
28
+
29
+ private
30
+
31
+ def extract_usage(body)
32
+ {
33
+ input_tokens: body&.dig('usage', 'input_tokens') || 0,
34
+ output_tokens: body&.dig('usage', 'output_tokens') || 0,
35
+ cache_read_tokens: body&.dig('usage', 'cache_read_input_tokens') || 0,
36
+ cache_write_tokens: body&.dig('usage', 'cache_creation_input_tokens') || 0
37
+ }
38
+ end
26
39
  end
27
40
  end
28
41
  end
data/lib/legion/extensions/foundry/version.rb CHANGED
@@ -3,7 +3,7 @@
3
3
  module Legion
4
4
  module Extensions
5
5
  module Foundry
6
- VERSION = '0.1.3'
6
+ VERSION = '0.1.4'
7
7
  end
8
8
  end
9
9
  end
metadata CHANGED
@@ -1,7 +1,7 @@
1
1
  --- !ruby/object:Gem::Specification
2
2
  name: lex-foundry
3
3
  version: !ruby/object:Gem::Version
4
- version: 0.1.3
4
+ version: 0.1.4
5
5
  platform: ruby
6
6
  authors:
7
7
  - Esity