bitfab 0.65.3 → 0.65.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1ccbb4e0e912ec35fe388c4cdb2225c4fa3c4f5e020e87c7d52f81a3358907da
-  data.tar.gz: 4e674dc9c1fb2e1c2ef6f5d5b1cf21146904a0d85989d7e41f25ce6c0b04c740
+  metadata.gz: a9c94aca223ea9794e110a16dae0e3e4f23ef45a2cf345c6bbaf396fc6b10b5f
+  data.tar.gz: 9c99a4266ec0109d155caeea5e1bb7792fb38c5e9d510e3552a2df60d192160a
 SHA512:
-  metadata.gz: a5f9344e7be229c578f2fb875de96aa11f49487b24775eec15955f32d18ab4552067842af6b4a8f12041f6140ee21bafbd3983630d0929e00a35236fb427e95f
-  data.tar.gz: 15bd45c45d9f6700edd7b96bbc792236441581763097307d13d4a14a76d0d8d733e5444c1586c9a24f6c990a8575e70851c37a80ff10b4f153bb84bb554e00c2
+  metadata.gz: fa01c2d028e06cc338df64527d011df63e0a01dcec25559b4816523c951aed2b55ab0034075c7c09babec22d927bb4e5f0654456e82fd76e5d41815d0851fa30
+  data.tar.gz: 6f865c897e2ac1975c1b2a64cfbdf5c933f24356db7294e97a8ce11712cbf1937b942701c5237cd62dd79a68ae422dfaf46ed6eba3545e6c8522bdf0070831af

data/lib/bitfab/client.rb

@@ -58,7 +58,8 @@ module Bitfab
       @resolved_api_key = nil
       @api_key_warned = false
       @http_client = HttpClient.new(api_key: -> { resolve_api_key }, service_url: @service_url, timeout:)
-      @http_client.configure_simulation_plan(enabled: simulation_plan, start: !api_key_resolved_lazily?)
+      @http_client.configure_simulation_plan(enabled: simulation_plan, start: !api_key_resolved_lazily?,
+        can_read: -> { simulation_plan_readable? })
       @datasets = Datasets.new(@http_client)
       @categories = Categories.new(@http_client)
       @organization_members = OrganizationMembers.new(@http_client)
@@ -607,7 +608,8 @@ module Bitfab
             end
           end
 
-          on_complete&.call(result: recorded_result, error: recorded_error, ended_at:)
+          on_complete&.call(result: recorded_result, error: recorded_error, ended_at:,
+            contexts: span_contexts, prompt: span_prompt)
 
           span_data = span_data.merge("replay_reusable" => replay_reusable)
           output_json = finalize.nil? && !result.is_a?(Enumerator) && SelectiveReplay.fingerprint(final_result)
@@ -852,6 +854,13 @@ module Bitfab
       !resolve_api_key.nil?
     end
 
+    def simulation_plan_readable?
+      recording = capture_configured? || ReplayContext.current || SeedContext.current
+      recording && !lookup_api_key.nil?
+    rescue
+      false
+    end
+
     def tracing_enabled?
       return false unless capture_configured?
 

data/lib/bitfab/http_client.rb

@@ -56,8 +56,8 @@ module Bitfab
       @close_condition = ConditionVariable.new
     end
 
-    def configure_simulation_plan(enabled: true, start: true)
-      @simulation_plan = SimulationPlan.new(self, enabled:)
+    def configure_simulation_plan(enabled: true, start: true, can_read: nil)
+      @simulation_plan = SimulationPlan.new(self, enabled:, can_read:)
       @simulation_plan.refresh if start
       @simulation_plan
     end

data/lib/bitfab/simulation_plan.rb

@@ -42,9 +42,10 @@ module Bitfab
       end
     end
 
-    def initialize(reader, enabled: true)
+    def initialize(reader, enabled: true, can_read: nil)
       @reader = reader
       @enabled = enabled
+      @can_read = can_read
       @pid = Process.pid
       @mutex = Mutex.new
       @condition = ConditionVariable.new
@@ -56,25 +57,26 @@ module Bitfab
       @stopped = false
       @first_read_finished = false
       @first_read_deadline = nil
+      @read_started = false
       self.class.register(self)
     end
 
     def disabled?
-      !@enabled || !ENV.fetch("BITFAB_DISABLE_SIM_PLAN", "").strip.empty?
+      switched_off? || (!@read_started && !readable?)
     end
 
     def refresh
       reset_after_fork
-      return if disabled? || @stopped
+      return if disabled?
 
-      @mutex.synchronize { start_reader(0) if @thread.nil? && now >= @refresh_after }
+      refresh_for_record
     end
 
     def send_span(payload, declared_node: false, discard: nil, &submit)
-      refresh
+      refresh_for_record
       data = payload.dig("rawSpan", "span_data")
       key = payload["rootTraceFunctionKey"] || payload["traceFunctionKey"]
-      if disabled? || key.nil? || !data.is_a?(Hash) || framework?(payload)
+      if switched_off? || key.nil? || !data.is_a?(Hash) || framework?(payload)
         submit.call(payload)
         return
       end
@@ -84,11 +86,11 @@ module Bitfab
     end
 
     def send_trace(payload, &submit)
-      refresh
+      refresh_for_record
       entry = [payload, nil, submit]
       ready = @mutex.synchronize do
         id = trace_id(payload)
-        if !disabled? && !@stopped && id && (@draining.include?(id) || @held.any? { |held| held[1] && trace_id(held[0]) == id })
+        if !switched_off? && !@stopped && id && (@draining.include?(id) || @held.any? { |held| held[1] && trace_id(held[0]) == id })
           hold(entry)
         else
           entry
@@ -172,6 +174,21 @@ module Bitfab
 
     private
 
+    def refresh_for_record
+      reset_after_fork
+      return if switched_off? || @stopped
+
+      @mutex.synchronize { start_reader(0) if @thread.nil? && now >= @refresh_after }
+    end
+
+    def switched_off?
+      !@enabled || !ENV.fetch("BITFAB_DISABLE_SIM_PLAN", "").strip.empty?
+    end
+
+    def readable?
+      @can_read.nil? || @can_read.call == true
+    end
+
     def now
       Process.clock_gettime(Process::CLOCK_MONOTONIC)
     end
@@ -194,7 +211,7 @@ module Bitfab
 
     def hold(entry)
       @held << entry
-      start_reader([@refresh_after - now, 0].max) if @thread.nil? && !@stopped && !disabled?
+      start_reader([@refresh_after - now, 0].max) if @thread.nil? && !@stopped && !switched_off?
       @held.shift if @held.length > MAX_HELD
     end
 
@@ -210,7 +227,7 @@ module Bitfab
     end
 
     def prepare(payload, key)
-      return payload if key.nil? || disabled?
+      return payload if key.nil? || switched_off?
       return apply(payload, key) if @mutex.synchronize { @content_off }
 
       without_unread_content(payload)
@@ -232,6 +249,7 @@ module Bitfab
     end
 
     def start_reader(delay)
+      @read_started = true
       @first_read_deadline ||= now + delay + READ_TIMEOUT
       @thread = Thread.new { read_until_loaded(delay) }
       @thread.name = "bitfab-sim-plan"
@@ -270,7 +288,7 @@ module Bitfab
 
     def read_until_loaded(delay)
       pause(delay) if delay.positive?
-      parsed = (disabled? || @stopped) ? nil : read_once
+      parsed = (switched_off? || @stopped) ? nil : read_once
       entries = @mutex.synchronize do
         @first_read_finished = true
         @content_off = parsed if parsed

data/lib/bitfab/subtree.rb

@@ -53,6 +53,8 @@ module Bitfab
       :experiment_id,
       :span_data,
       :declared_node,
+      :contexts,
+      :prompt,
       keyword_init: true
     )
 
@@ -150,7 +152,7 @@ module Bitfab
           span_id: @owner.current_parent_span_id, surface: :trace, &body)
       end
 
-      def complete(result:, error:, ended_at:)
+      def complete(result:, error:, ended_at:, contexts: nil, prompt: nil)
         return if @completed
 
         @completed = true
@@ -158,6 +160,8 @@ module Bitfab
           next if !entry || session.equal?(@owner)
 
           entry.error = error
+          entry.contexts = contexts
+          entry.prompt = prompt
           session.emit_configured(entry, result:)
         end
       end
@@ -741,8 +745,8 @@ module Bitfab
           declared_node: entry.declared_node == true,
           span_data: entry.span_data || {},
           function_name: entry.name,
-          contexts: nil,
-          prompt: nil,
+          contexts: entry.contexts,
+          prompt: entry.prompt,
           args: entry.args,
           kwargs: entry.kwargs,
           result:,

data/lib/bitfab/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Bitfab
-  VERSION = "0.65.3"
+  VERSION = "0.65.5"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: bitfab
 version: !ruby/object:Gem::Version
-  version: 0.65.3
+  version: 0.65.5
 platform: ruby
 authors:
 - Harvest Team