open_router_enhanced 2.1.0 → 2.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1f08650282b1232abfa779330c8f8ec9aad7bf6a984fd31e1a74ce1f3cad6962
-  data.tar.gz: e41ce6204abde21a34dd6c2976801666b8f4b9339cf81c845861f3837c15097c
+  metadata.gz: 9f4936ab4591643976d7cf5f63b92fec61fef1fda2900ce6ed022ce35671bd3a
+  data.tar.gz: 1ac1e6ea82c9eacc1bc28e4b99490507bcf1f335446f3ab1cbab2a2ca8d749b4
 SHA512:
-  metadata.gz: 1d15f1e25db721734f0f9d2e43d01f6546af170096e65253c1c8438c3362484d6bad28f36669e2b8c34856ca297bb07dc1ffb18c7dc4442e2fe8c93ad79474b1
-  data.tar.gz: 2db9c96b66dcb0eae4b03f463db956192b38ef7444c879f7fe72a767e31c378d1ed702ee38a4c3b76e187945e6d942ac5dcbf7e7733fe8621315bee729e74940
+  metadata.gz: ef3a651dee0fec1b4d0508e88578c78ed1d259b5dc7c5fb5917d7af0690a2e69c12732f6fc5a0b423340da7afddbc4de80f08f8a27e616c9d7c15a1051508a39
+  data.tar.gz: 017ec8a29aeb205e396265409c8f9d53f3f7280d025543e89b2c95510d6b21fe78f230461dcc5291a433c1b60f850425ba754dcfc1ab32e98517b1f522688692

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    open_router_enhanced (2.0.1)
+    open_router_enhanced (2.2.1)
       activesupport (>= 6.0, < 9.0)
       dotenv (>= 2.0, < 4.0)
       faraday (>= 1.0, < 3.0)

data/Rakefile

@@ -40,7 +40,6 @@ namespace :models do
     puts " done. #{count} models cached."
   end
 
-
   desc "Display summary of available models"
   task :summary do
     require_relative "lib/open_router"

data/lib/open_router/callbacks.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+module OpenRouter
+  # Mixin providing event callback registration and dispatch for Client.
+  module Callbacks
+    # Register a callback for a specific event
+    #
+    # @param event [Symbol] The event to register for (:before_request, :after_response, :on_tool_call, :on_error, :on_stream_chunk, :on_healing)
+    # @param block [Proc] The callback to execute
+    # @return [self] Returns self for method chaining
+    #
+    # @example
+    #   client.on(:after_response) do |response|
+    #     puts "Used #{response.total_tokens} tokens"
+    #   end
+    def on(event, &block)
+      raise ArgumentError, "Invalid event: #{event}. Valid events are: #{@callbacks.keys.join(", ")}" unless @callbacks.key?(event)
+
+      @callbacks[event] << block
+      self
+    end
+
+    # Remove all callbacks for a specific event
+    #
+    # @param event [Symbol] The event to clear callbacks for
+    # @return [self] Returns self for method chaining
+    def clear_callbacks(event = nil)
+      if event
+        @callbacks[event] = [] if @callbacks.key?(event)
+      else
+        @callbacks.each_key { |key| @callbacks[key] = [] }
+      end
+      self
+    end
+
+    # Trigger callbacks for a specific event
+    #
+    # @param event [Symbol] The event to trigger
+    # @param data [Object] Data to pass to the callbacks
+    def trigger_callbacks(event, data = nil)
+      return unless @callbacks[event]
+
+      @callbacks[event].each do |callback|
+        callback.call(data)
+      rescue StandardError => e
+        warn "[OpenRouter] Callback error for #{event}: #{e.message}"
+      end
+    end
+  end
+end