ask-tokens-rails 0.2.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 410d6416c7cc925ee0d07339c59ee071cb577b2bc9b32b41973ca73f1f5fecf5
-  data.tar.gz: 8ed2534684c2eb00471cb05eb0ea3de80b0c3b97950cb623da7fb1a7ad3b778e
+  metadata.gz: 9f36ea3ade0e11cbb7a52d00ae56f1168574ce7a559ce8995cc6ca43d6716de8
+  data.tar.gz: c5b9d81ebc12fe4a51080cd773d8728dad5eb75ffbd50f15329623083ba8dcf9
 SHA512:
-  metadata.gz: 6556aa40565dce96e88fafa73e644f5bdbe28a8bd231208bc3927d89478aa6127acc0affe51eccd6378d32b1294dc91b4dcfc5299485894686876652c8be6756
-  data.tar.gz: 9f80ef870f9a1c3b0a65ed1821b438067379a2d5b33d2a76b35e75932a0b2d33140c7b20a0f6390dbd01aa36c7b014d6c5f4fb991baf876d7d447d99b7048f40
+  metadata.gz: 48e3aaefc5ac428854b2c998fe9f4c66f9bd0b78483184ef17ac43494ddd52c77f0dc2939c4bdab114f14222bc208a5174536212c027b90417d3d77f8373e3a4
+  data.tar.gz: 29dff0d60146f50f546f20be6643e8c8bd938c671a9d74f238c84f030af048ca07558fd5154e63d44641111352496382f7db00ea379554d29ecf5e1baafb8bb5

data/lib/ask/tokens/rails/stores/active_record_store.rb

@@ -42,7 +42,8 @@ module Ask
             metadata: meta,
             expires_at: entry.expires_at,
             balance: entry.balance,
-            created_at: entry.created_at
+            created_at: entry.created_at,
+            **extract_columns_from_metadata(meta)
           )
           entry.with(id: txn.id)
         end
@@ -89,6 +90,18 @@ module Ask
             created_at: txn.created_at
           )
         end
+
+        # Extract well-known metadata keys into their matching model columns.
+        # Only writes keys that exist in metadata, so grants/debits without
+        # LLM context are unaffected. Symbol and string keys are both handled.
+        COLUMN_KEYS = %i[model_id provider input_tokens output_tokens cached_tokens llm_cost_usd multiplier].freeze
+
+        def extract_columns_from_metadata(meta)
+          COLUMN_KEYS.each_with_object({}) do |key, hash|
+            value = meta[key] || meta[key.to_s]
+            hash[key] = value if value
+          end
+        end
       end
     end
   end

data/lib/ask/tokens/rails/version.rb

@@ -3,7 +3,7 @@
 module Ask
   module Tokens
     module Rails
-      VERSION = "0.2.0"
+      VERSION = "0.2.1"
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ask-tokens-rails
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
 platform: ruby
 authors:
 - Kaka Ruto