braintrust 0.0.3 → 0.0.5

data/lib/braintrust/trace/span_processor.rb

@@ -5,14 +5,16 @@ require "opentelemetry/sdk"
 module Braintrust
   module Trace
     # Custom span processor that adds Braintrust-specific attributes to spans
+    # and optionally filters spans based on custom filter functions.
     class SpanProcessor
       PARENT_ATTR_KEY = "braintrust.parent"
       ORG_ATTR_KEY = "braintrust.org"
       APP_URL_ATTR_KEY = "braintrust.app_url"
 
-      def initialize(wrapped_processor, state)
+      def initialize(wrapped_processor, state, filters = [])
         @wrapped = wrapped_processor
         @state = state
+        @filters = filters || []
       end
 
       def on_start(span, parent_context)
@@ -33,9 +35,10 @@ module Braintrust
         @wrapped.on_start(span, parent_context)
       end
 
-      # Called when a span ends
+      # Called when a span ends - apply filters before forwarding
       def on_finish(span)
-        @wrapped.on_finish(span)
+        # Only forward span if it passes filters
+        @wrapped.on_finish(span) if should_forward_span?(span)
       end
 
       # Shutdown the processor
@@ -73,6 +76,29 @@ module Braintrust
         # Return the parent attribute from the parent span
         parent_span.attributes&.[](PARENT_ATTR_KEY)
       end
+
+      # Determine if a span should be forwarded to the wrapped processor
+      # based on configured filters
+      def should_forward_span?(span)
+        # Always keep root spans (spans with no parent)
+        # Check if parent_span_id is the invalid/zero span ID
+        is_root = span.parent_span_id == OpenTelemetry::Trace::INVALID_SPAN_ID
+        return true if is_root
+
+        # If no filters, keep everything
+        return true if @filters.empty?
+
+        # Apply filters in order - first non-zero result wins
+        @filters.each do |filter|
+          result = filter.call(span)
+          return true if result > 0  # Keep span
+          return false if result < 0 # Drop span
+          # result == 0: no influence, continue to next filter
+        end
+
+        # All filters returned 0 (no influence), default to keep
+        true
+      end
     end
   end
 end

data/lib/braintrust/trace.rb

@@ -3,14 +3,33 @@
 require "opentelemetry/sdk"
 require "opentelemetry/exporter/otlp"
 require_relative "trace/span_processor"
+require_relative "trace/span_filter"
 require_relative "logger"
 
-# OpenAI integration is optional - automatically loaded if openai gem is available
+# OpenAI integrations - both ruby-openai and openai gems use require "openai"
+# so we detect which one actually loaded the code and require the appropriate integration
 begin
   require "openai"
-  require_relative "trace/contrib/openai"
+
+  # Check which OpenAI gem's code is actually loaded by inspecting $LOADED_FEATURES
+  # (both gems can be in Gem.loaded_specs, but only one's code can be loaded)
+  openai_load_path = $LOADED_FEATURES.find { |f| f.end_with?("/openai.rb") }
+
+  if openai_load_path&.include?("ruby-openai")
+    # alexrudall/ruby-openai gem (path contains "ruby-openai-X.Y.Z")
+    require_relative "trace/contrib/github.com/alexrudall/ruby-openai/ruby-openai"
+  elsif openai_load_path&.include?("/openai-")
+    # Official openai gem (path contains "openai-X.Y.Z")
+    require_relative "trace/contrib/openai"
+  elsif Gem.loaded_specs["ruby-openai"]
+    # Fallback: ruby-openai in loaded_specs (for unusual installation paths)
+    require_relative "trace/contrib/github.com/alexrudall/ruby-openai/ruby-openai"
+  elsif Gem.loaded_specs["openai"]
+    # Fallback: official openai in loaded_specs (for unusual installation paths)
+    require_relative "trace/contrib/openai"
+  end
 rescue LoadError
-  # OpenAI gem not installed - integration will not be available
+  # No OpenAI gem installed - integration will not be available
 end
 
 # Anthropic integration is optional - automatically loaded if anthropic gem is available
@@ -26,8 +45,9 @@ module Braintrust
     # Set up OpenTelemetry tracing with Braintrust
     # @param state [State] Braintrust state
     # @param tracer_provider [TracerProvider, nil] Optional tracer provider
+    # @param exporter [Exporter, nil] Optional exporter override (for testing)
     # @return [void]
-    def self.setup(state, tracer_provider = nil)
+    def self.setup(state, tracer_provider = nil, exporter: nil)
       if tracer_provider
         # Use the explicitly provided tracer provider
         # DO NOT set as global - user is managing it themselves
@@ -49,13 +69,17 @@ module Braintrust
       end
 
       # Enable Braintrust tracing (adds span processor)
-      enable(tracer_provider, state: state)
+      config = state.config
+      enable(tracer_provider, state: state, config: config, exporter: exporter)
     end
 
-    def self.enable(tracer_provider, state: nil, exporter: nil)
+    def self.enable(tracer_provider, state: nil, exporter: nil, config: nil)
       state ||= Braintrust.current_state
       raise Error, "No state available" unless state
 
+      # Get config from state if available
+      config ||= state.respond_to?(:config) ? state.config : nil
+
       # Create OTLP HTTP exporter unless override provided
       exporter ||= OpenTelemetry::Exporter::OTLP::Exporter.new(
         endpoint: "#{state.api_url}/otel/v1/traces",
@@ -64,11 +88,18 @@ module Braintrust
         }
       )
 
-      # Wrap in batch processor
-      batch_processor = OpenTelemetry::SDK::Trace::Export::BatchSpanProcessor.new(exporter)
+      # Use SimpleSpanProcessor for InMemorySpanExporter (testing), BatchSpanProcessor for production
+      span_processor = if exporter.is_a?(OpenTelemetry::SDK::Trace::Export::InMemorySpanExporter)
+        OpenTelemetry::SDK::Trace::Export::SimpleSpanProcessor.new(exporter)
+      else
+        OpenTelemetry::SDK::Trace::Export::BatchSpanProcessor.new(exporter)
+      end
+
+      # Build filters array from config
+      filters = build_filters(config)
 
-      # Wrap batch processor in our custom span processor to add Braintrust attributes
-      processor = SpanProcessor.new(batch_processor, state)
+      # Wrap span processor in our custom span processor to add Braintrust attributes and filters
+      processor = SpanProcessor.new(span_processor, state, filters)
 
       # Register with tracer provider
       tracer_provider.add_span_processor(processor)
@@ -83,6 +114,25 @@ module Braintrust
       self
     end
 
+    # Build filters array from config
+    # @param config [Config, nil] Configuration object
+    # @return [Array<Proc>] Array of filter functions
+    def self.build_filters(config)
+      filters = []
+
+      # Add custom filters first (they have priority)
+      if config&.span_filter_funcs&.any?
+        filters.concat(config.span_filter_funcs)
+      end
+
+      # Add AI filter if enabled
+      if config&.filter_ai_spans
+        filters << SpanFilter.method(:ai_filter)
+      end
+
+      filters
+    end
+
     # Generate a permalink URL for a span to view in the Braintrust UI
     # Returns an empty string if the permalink cannot be generated
     # @param span [OpenTelemetry::Trace::Span] The span to generate a permalink for

data/lib/braintrust/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Braintrust
-  VERSION = "0.0.3"
+  VERSION = "0.0.5"
 end

data/lib/braintrust.rb

@@ -37,8 +37,11 @@ module Braintrust
   # @param blocking_login [Boolean] Whether to block and login synchronously (default: false - async background login)
   # @param enable_tracing [Boolean] Whether to enable OpenTelemetry tracing (default: true)
   # @param tracer_provider [TracerProvider, nil] Optional tracer provider to use instead of creating one
+  # @param filter_ai_spans [Boolean, nil] Enable AI span filtering (overrides BRAINTRUST_OTEL_FILTER_AI_SPANS env var)
+  # @param span_filter_funcs [Array<Proc>, nil] Custom span filter functions
+  # @param exporter [Exporter, nil] Optional exporter override (for testing)
   # @return [State] the created state
-  def self.init(api_key: nil, org_name: nil, default_project: nil, app_url: nil, api_url: nil, set_global: true, blocking_login: false, enable_tracing: true, tracer_provider: nil)
+  def self.init(api_key: nil, org_name: nil, default_project: nil, app_url: nil, api_url: nil, set_global: true, blocking_login: false, enable_tracing: true, tracer_provider: nil, filter_ai_spans: nil, span_filter_funcs: nil, exporter: nil)
     state = State.from_env(
       api_key: api_key,
       org_name: org_name,
@@ -47,7 +50,10 @@ module Braintrust
       api_url: api_url,
       blocking_login: blocking_login,
       enable_tracing: enable_tracing,
-      tracer_provider: tracer_provider
+      tracer_provider: tracer_provider,
+      filter_ai_spans: filter_ai_spans,
+      span_filter_funcs: span_filter_funcs,
+      exporter: exporter
     )
 
     State.global = state if set_global

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: braintrust
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.5
 platform: ruby
 authors:
 - Braintrust
@@ -15,14 +15,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.0'
+        version: '1.3'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.0'
+        version: '1.3'
 - !ruby/object:Gem::Dependency
   name: opentelemetry-exporter-otlp
   requirement: !ruby/object:Gem::Requirement
@@ -202,8 +202,11 @@ files:
 - lib/braintrust/logger.rb
 - lib/braintrust/state.rb
 - lib/braintrust/trace.rb
+- lib/braintrust/trace/attachment.rb
 - lib/braintrust/trace/contrib/anthropic.rb
+- lib/braintrust/trace/contrib/github.com/alexrudall/ruby-openai/ruby-openai.rb
 - lib/braintrust/trace/contrib/openai.rb
+- lib/braintrust/trace/span_filter.rb
 - lib/braintrust/trace/span_processor.rb
 - lib/braintrust/version.rb
 homepage: https://github.com/braintrustdata/braintrust-sdk-ruby