ruby_llm-agents 1.2.1 → 1.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b6f99247af4fcb9e76776ab2658bac4914d5b869423258bf3827583212447146
-  data.tar.gz: a6927cced434962ed4b923717d7fd582e983c3091ac8172554afb68671a2514d
+  metadata.gz: 5396493167b6da5bcc4fd878d502efdf7269246374ea30ef8d5ce25423f3868a
+  data.tar.gz: 2e80ab75249560be31be4e1a9aa8c78efd6b0d6e2c692f391962a20b7ea98a5b
 SHA512:
-  metadata.gz: c98ae91fde73edf152ac4efc5665fe5da30e17ec14ff9f2f9128102bf4665530f55196e705dc371f0dac2354af4ab7a2102be9c04b3b7ec2e8331b7b5445ce23
-  data.tar.gz: af36432d85c4ac5ee6219eba846411dd4ffaa0ba8d8c9998550ea3d3e757fd125fa8a2bea3cc5c98a0387db519cc03b45d1fd3f3d63de5facc2dc84338b02692
+  metadata.gz: 2943372fc8709cef913eba67f491431b96e96b5ba4cc43338eabbfaee291fa53434e55fc1db7b7d0ab19b7d05e54c0b228c0799d73a5570f9dd9a18881fb9b7a
+  data.tar.gz: b442321d5323899962a6fab0ec998fd5616339bd46428c4198291c611eb2741fc9c3f78080f90fdb055654b16ef491d903caddff5d95394c87c72f84e900fe06

data/lib/ruby_llm/agents/core/version.rb

@@ -4,6 +4,6 @@ module RubyLLM
   module Agents
     # Current version of the RubyLLM::Agents gem
     # @return [String] Semantic version string
-    VERSION = "1.2.1"
+    VERSION = "1.2.3"
   end
 end

data/lib/ruby_llm/agents/image/concerns/image_operation_execution.rb

@@ -29,10 +29,9 @@ module RubyLLM
 
         # Check budget before execution
         def check_budget!
-          BudgetTracker.check!(
-            agent_type: self.class.name,
-            tenant_id: @tenant_id,
-            execution_type: execution_type
+          BudgetTracker.check_budget!(
+            self.class.name,
+            tenant_id: @tenant_id
           )
         end
 

data/lib/ruby_llm/agents/image/pipeline/execution.rb

@@ -232,10 +232,9 @@ module RubyLLM
         end
 
         def check_budget!
-          BudgetTracker.check!(
-            agent_type: self.class.name,
-            tenant_id: @tenant_id,
-            execution_type: "image_pipeline"
+          BudgetTracker.check_budget!(
+            self.class.name,
+            tenant_id: @tenant_id
           )
         end
 

data/lib/ruby_llm/agents/pipeline/middleware/budget.rb

@@ -60,12 +60,11 @@ module RubyLLM
           # @param context [Context] The execution context
           # @raise [BudgetExceededError] If budget exceeded with hard enforcement
           def check_budget!(context)
-            BudgetTracker.check!(
-              agent_type: context.agent_class&.name,
-              tenant_id: context.tenant_id,
-              execution_type: context.agent_type&.to_s
+            BudgetTracker.check_budget!(
+              context.agent_class&.name,
+              tenant_id: context.tenant_id
             )
-          rescue BudgetExceededError
+          rescue RubyLLM::Agents::Reliability::BudgetExceededError
             # Re-raise budget errors
             raise
           rescue StandardError => e
@@ -80,10 +79,19 @@ module RubyLLM
             return unless context.total_cost&.positive?
 
             BudgetTracker.record_spend!(
-              tenant_id: context.tenant_id,
-              cost: context.total_cost,
-              tokens: context.total_tokens
+              context.agent_class&.name,
+              context.total_cost,
+              tenant_id: context.tenant_id
             )
+
+            # Also record tokens if available
+            if context.total_tokens&.positive?
+              BudgetTracker.record_tokens!(
+                context.agent_class&.name,
+                context.total_tokens,
+                tenant_id: context.tenant_id
+              )
+            end
           rescue StandardError => e
             # Log but don't fail on spend recording errors
             error("Failed to record spend: #{e.message}")

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm-agents
 version: !ruby/object:Gem::Version
-  version: 1.2.1
+  version: 1.2.3
 platform: ruby
 authors:
 - adham90