simforge 0.5.0 → 0.5.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 662f1ce8942083e856907a38568a40b252c4d6360f043fc25da58900a46e36e7
-  data.tar.gz: 9d74cf518171472d22b49c6b8fb3544e258607e5284acebe0f6b46e71213e0cd
+  metadata.gz: 712b471d28942f55c0ac0a78d05507f1060396de60c584173d3824f1e8c6d9b7
+  data.tar.gz: efa9dfc4e4173382d853327ee4edeafa54114482e04d97c86a3933465ee24285
 SHA512:
-  metadata.gz: 556e17aa0f010ecb394f7a5937c3c02a43ef0d76bffe00082899074a19f26c555bec5c4d47dfe1afa98c34f7093beae5dd4b4a4d2d9d00a68d0155aadd979aea
-  data.tar.gz: 14cd3fbc8c4ea4096036ce312e6eac67ab7b57a634c10faba0962fa3dd10368928ebc3c25c1236b5ff6019a8db36094b0a4e4a8eeb737da2c76fc42e5525cd62
+  metadata.gz: 1ca7cb706679154120fc89cac714b7d2eaa45017389f33f726afee7a77a0a0d1203dd7e14da32c8aa042d828381654bb87fe42fcb75c276edf61c52a27ba83e8
+  data.tar.gz: b9d9f5e70bf640447f046f9e272054c5f27f2aba849f42df20c6424c7a11c923c3403f89b23d0eb6b9eb50d53c39d0292b99213a5c4e2b745c7580948613a7d9

data/lib/simforge/traceable.rb

@@ -26,6 +26,39 @@ module Simforge
       base.extend(ClassMethods)
     end
 
+    # Wrap an existing method on an external class with span tracing.
+    # Use this to trace third-party library calls without modifying their source.
+    #
+    # @example
+    #   Simforge::Traceable.wrap(OpenAI::Client, :chat,
+    #     trace_function_key: "openai", name: "Chat", type: "llm")
+    #
+    # @param klass [Class, Module] the class to wrap
+    # @param method_name [Symbol] the method to wrap
+    # @param trace_function_key [String] the trace function key
+    # @param name [String, nil] explicit span name (defaults to method name)
+    # @param type [String] span type: llm, agent, function, guardrail, handoff, custom
+    def self.wrap(klass, method_name, trace_function_key:, name: nil, type: "custom")
+      span_name = name || method_name.to_s
+      method_name_str = method_name.to_s
+
+      wrapper = Module.new do
+        define_method(method_name) do |*args, **kwargs, &block|
+          Simforge.client.send(:execute_span,
+            trace_function_key:,
+            span_name:,
+            span_type: type,
+            function_name: method_name_str,
+            args:,
+            kwargs:) do
+            super(*args, **kwargs, &block)
+          end
+        end
+      end
+
+      klass.prepend(wrapper)
+    end
+
     module ClassMethods
       # Set the trace function key for this class.
       # All spans declared in this class will be grouped under this key.

data/lib/simforge/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Simforge
-  VERSION = "0.5.0"
+  VERSION = "0.5.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: simforge
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.5.1
 platform: ruby
 authors:
 - Harvest Team