llm_logs 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 87a8d5b11b7782e2dbda613ce9ef2c0717f2984574953d1070e7df443366d9d4
-  data.tar.gz: b18d3628a4658e2288744f680df716850a99bac16afaf4de07af199513034d46
+  metadata.gz: c9f1ecb908532e887783dec76409b9d1064e414c2560864b984dde135757e507
+  data.tar.gz: e2a6567f55ef52cb4defaed521506502640c2a57e1df33709c8a598910ef767c
 SHA512:
-  metadata.gz: c5ddca4dfee273333daf57dbd8b7e96959183547fece1e3c08b1de9e4f556bc8bdf1c2001ecd0110eab703580a559d4a75f4d8472fe6df06ef7a2b5bb757ec59
-  data.tar.gz: 1a740b579f7d7c84950edd7b3a70a33a843f8bf8e32a4e5eef902de80110b3a161eef1e2e65ec556b353c9ab7d7ed77c07f84c883d0b62a1e30c168866ca9919
+  metadata.gz: 60677029d50f6bb17c6a3e580643bbe2615a8886b2f9e5c40678e737175e41d6ab56d39990abccf7ce8a17354e4d101d7724273e987f70c0b65fdb48c1037317
+  data.tar.gz: 46fdf2fb90bfaf0ea4b85044c2f12baf38decb36d5d7f96e540d99cc4afecdf7a8b6a43322c3f45320111ecdc990aaf21a899d14623a7443aafff327a82e4dd5

data/app/models/llm_logs/span.rb

@@ -15,8 +15,8 @@ module LlmLogs
         duration_ms: (Time.current - started_at) * 1000
       )
 
-      # Restore parent span as current
-      Thread.current[:llm_logs_span] = parent_span
+      # Restore parent span as current (Fiber[] so child fibers inherit it)
+      LlmLogs::Tracer.current_span = parent_span
     end
 
     def record_response(message)

data/lib/llm_logs/tracer.rb

@@ -1,11 +1,20 @@
 module LlmLogs
   module Tracer
+    # Trace/span context is kept in Fiber-local storage (Fiber[]), which child
+    # fibers inherit on creation. This matters for fiber-based schedulers such as
+    # socketry/async: work driven inside a child fiber still sees the active
+    # trace. The legacy Thread.current[:key] store is *not* inherited by child
+    # fibers, which caused spans to auto-create orphan traces instead of nesting.
     def self.current_trace
-      Thread.current[:llm_logs_trace]
+      Fiber[:llm_logs_trace]
     end
 
     def self.current_span
-      Thread.current[:llm_logs_span]
+      Fiber[:llm_logs_span]
+    end
+
+    def self.current_span=(span)
+      Fiber[:llm_logs_span] = span
     end
 
     def self.start_trace(name, metadata: {})
@@ -16,10 +25,10 @@ module LlmLogs
         started_at: Time.current
       )
 
-      previous_trace = Thread.current[:llm_logs_trace]
-      previous_span = Thread.current[:llm_logs_span]
-      Thread.current[:llm_logs_trace] = trace
-      Thread.current[:llm_logs_span] = nil
+      previous_trace = Fiber[:llm_logs_trace]
+      previous_span = Fiber[:llm_logs_span]
+      Fiber[:llm_logs_trace] = trace
+      Fiber[:llm_logs_span] = nil
 
       begin
         yield trace
@@ -28,8 +37,8 @@ module LlmLogs
         raise
       ensure
         trace.complete! if trace.status == "running"
-        Thread.current[:llm_logs_trace] = previous_trace
-        Thread.current[:llm_logs_span] = previous_span
+        Fiber[:llm_logs_trace] = previous_trace
+        Fiber[:llm_logs_span] = previous_span
       end
     end
 
@@ -50,7 +59,7 @@ module LlmLogs
         started_at: Time.current
       )
 
-      Thread.current[:llm_logs_span] = span
+      Fiber[:llm_logs_span] = span
       span
     end
 
@@ -60,7 +69,7 @@ module LlmLogs
         status: "running",
         started_at: Time.current
       )
-      Thread.current[:llm_logs_trace] = trace
+      Fiber[:llm_logs_trace] = trace
       trace
     end
   end

data/lib/llm_logs/version.rb

@@ -1,3 +1,3 @@
 module LlmLogs
-  VERSION = "0.1.4"
+  VERSION = "0.1.5"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llm_logs
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - Anton