simforge 0.6.0 → 0.8.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0eaad388f5419c3933530ac93808928f219682b1e32d35fff040f691cbd4f410
-  data.tar.gz: a7922a5c2a3f063b5f6c13f43e5f6defed320c82d9b8d887a3cc12293ab03808
+  metadata.gz: 73a5e6deba272d1145d6403c0101d8f32a694f236ac4a976d02eb11b79b17c25
+  data.tar.gz: 89481b2c5f3e7a00946aba1e438c21df60b9c8138a04720b1a22701a35095c73
 SHA512:
-  metadata.gz: 32b1df24c8a1dd4aa8acb6b5ca6828736c1a3b5586982a169865acaaa753075b4e06009f6c2d0d6843e9ec29539cba0a28c4e0df83eaaeef65f31b98e1449637
-  data.tar.gz: 2abbabfde142faa7c8ca0adc6e7433787f16b6b60aaafd28d26bd23c0fc890be852bdae342312540fc97e081bea05d163af66ae1b043f393c05c1f63ddd94d44
+  metadata.gz: 2626a260759ff51764a1b6497b29ac1493acbc86ea9a0e347a39bd5b02987c5bc4012d7b5e7839887f634da72643370c75b6835a92da59c6b7768ae525fae6e2
+  data.tar.gz: 45b37048fc7d7338135f5b2d9cd8d6a4a111034587d055aeb9b58e9cd44ac562a1e00e19b9833293f68afc81ff5b47e184e31e410e5c7a795739ef2bdea17e5f

data/lib/simforge/client.rb

@@ -23,6 +23,8 @@ module Simforge
         @enabled = false
       end
       @http_client = HttpClient.new(api_key:, service_url: @service_url)
+      @pending_span_threads = {}
+      @pending_span_mutex = Mutex.new
     end
 
     # Execute a block inside a span context, sending trace data on completion.
@@ -42,9 +44,14 @@ module Simforge
         TraceState.create(trace_id)
       end
 
+      if is_root_span
+        @pending_span_mutex.synchronize { @pending_span_threads[trace_id] = [] }
+      end
+
       result = nil
       error = nil
       span_contexts = nil
+      span_prompt = nil
 
       begin
         SpanContext.with_span(trace_id:, span_id:) do
@@ -52,6 +59,7 @@ module Simforge
         ensure
           # Capture contexts before the span context is popped
           span_contexts = SpanContext.current&.dig(:contexts)
+          span_prompt = SpanContext.current&.dig(:prompt)
         end
       rescue => e
         error = e.message
@@ -61,7 +69,7 @@ module Simforge
         begin
           ended_at = Time.now.utc.strftime("%Y-%m-%dT%H:%M:%S.%3NZ")
 
-          send_span(
+          span_thread = send_span(
             trace_function_key:,
             trace_id:,
             span_id:,
@@ -70,6 +78,7 @@ module Simforge
             span_type:,
             function_name:,
             contexts: span_contexts,
+            prompt: span_prompt,
             args:,
             kwargs:,
             result:,
@@ -78,14 +87,21 @@ module Simforge
             ended_at:
           )
 
-          # For root spans, also send trace completion
           if is_root_span
+            pending = @pending_span_mutex.synchronize { @pending_span_threads.delete(trace_id) || [] }
+            pending << span_thread if span_thread
+            pending.each { |t| t.join(5) }
+
             send_trace_completion(
               trace_function_key:,
               trace_id:,
               started_at:,
               ended_at:
             )
+          else
+            @pending_span_mutex.synchronize do
+              @pending_span_threads[trace_id] << span_thread if span_thread && @pending_span_threads.key?(trace_id)
+            end
           end
         rescue Exception # rubocop:disable Lint/RescueException
           # Silently ignore — user's result/exception takes priority
@@ -141,7 +157,7 @@ module Simforge
     end
 
     def send_span(trace_function_key:, trace_id:, span_id:, parent_span_id:,
-      span_name:, span_type:, function_name:, contexts:, args:, kwargs:, result:, error:,
+      span_name:, span_type:, function_name:, contexts:, prompt:, args:, kwargs:, result:, error:,
       started_at:, ended_at:)
       # Human-readable JSON (input/output fields)
       human_inputs = Serialize.serialize_inputs(args, kwargs)
@@ -163,6 +179,7 @@ module Simforge
       span_data["output_serialized"] = marshalled_output if marshalled_output
       span_data["error"] = error if error
       span_data["contexts"] = contexts if contexts&.any?
+      span_data["prompt"] = prompt if prompt
 
       raw_span = {
         "id" => span_id,
@@ -179,7 +196,7 @@ module Simforge
         "sourceTraceId" => trace_id,
         "traceFunctionKey" => trace_function_key,
         "rawSpan" => raw_span
-      )
+      ) # Returns the background thread
     end
   end
 end

data/lib/simforge/http_client.rb

@@ -57,7 +57,8 @@ module Simforge
       raise last_error
     end
 
-    # Send an external span (fire-and-forget in background thread).
+    # Send an external span in a background thread.
+    # Returns the thread for callers that need to await completion.
     def send_external_span(payload)
       merged = payload.merge("sdkVersion" => VERSION)
 
@@ -92,16 +93,22 @@ module Simforge
 
   class << self
     # Run a block in a background thread with tracking.
+    # Returns the thread for callers that need to join on it.
     def _run_in_background(&block)
       thread = Thread.new do
         block.call
-      rescue
-        # Silently ignore failures in background spans
+      rescue => e
+        begin
+          warn "Simforge: Failed to send request: #{e.message}"
+        rescue
+          # Never crash the host app
+        end
       ensure
         @pending_threads_mutex.synchronize { @pending_threads.delete(Thread.current) }
       end
 
       @pending_threads_mutex.synchronize { @pending_threads << thread }
+      thread
     end
 
     # Wait for all pending background threads to complete.

data/lib/simforge/span_context.rb

@@ -25,6 +25,19 @@ module Simforge
     rescue
       # Silently ignore - never crash the host app
     end
+
+    # Set the prompt for this span.
+    # The prompt is stored in span_data.prompt. Calling multiple times
+    # overwrites the previous value.
+    #
+    # @param prompt [String] the prompt string to store
+    def set_prompt(prompt)
+      return unless prompt.is_a?(String)
+
+      @span_state[:prompt] = prompt
+    rescue
+      # Silently ignore - never crash the host app
+    end
   end
 
   # Handle to the current active trace, allowing trace-level context to be set.

data/lib/simforge/version.rb

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

data/lib/simforge.rb

@@ -19,6 +19,10 @@ module Simforge
     def add_context(_context)
       # No-op
     end
+
+    def set_prompt(_prompt)
+      # No-op
+    end
   end
 
   # No-op trace handle returned when outside a span context.

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: simforge
 version: !ruby/object:Gem::Version
-  version: 0.6.0
+  version: 0.8.0
 platform: ruby
 authors:
 - Harvest Team
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-02-13 00:00:00.000000000 Z
+date: 2026-02-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake