llm_logs 0.1.4 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 87a8d5b11b7782e2dbda613ce9ef2c0717f2984574953d1070e7df443366d9d4
-  data.tar.gz: b18d3628a4658e2288744f680df716850a99bac16afaf4de07af199513034d46
+  metadata.gz: e59977f8fa67dd219ec626f20f5320c325b0eab9349c7b7b69ba75c200353513
+  data.tar.gz: cc78c8b7ed85a03e2b2b85ebee10107c51089dc1fe012de14e2fdd3de93cb126
 SHA512:
-  metadata.gz: c5ddca4dfee273333daf57dbd8b7e96959183547fece1e3c08b1de9e4f556bc8bdf1c2001ecd0110eab703580a559d4a75f4d8472fe6df06ef7a2b5bb757ec59
-  data.tar.gz: 1a740b579f7d7c84950edd7b3a70a33a843f8bf8e32a4e5eef902de80110b3a161eef1e2e65ec556b353c9ab7d7ed77c07f84c883d0b62a1e30c168866ca9919
+  metadata.gz: f97806146008910196d8072f6c7c051fc67362705c6a28ed6486afe465bc0d58865e0f590efee24b41be57ef2fc853eac942741b64b940d705212a096ac5577e
+  data.tar.gz: 166a50f08ab90b2568a2cd8bc6b0942ab1bcd75e97839e82d9ceb8f6b91e3a8e5aec7f23cc318af444b96f3b2eafea31f2c8ea10f4f1059678761bd356b34b27

data/app/helpers/llm_logs/formatting_helper.rb

@@ -16,8 +16,16 @@ module LlmLogs
     end
 
     def render_markdown(text)
+      # Escape HTML metacharacters before Markdown so prompt delimiters such as
+      # <user_request> render as visible text instead of being parsed as unknown
+      # HTML tags and then stripped by `sanitize`. `&` is escaped first so the
+      # `&` in `&lt;`/`&gt;` is not double-escaped. Markdown syntax (#, *, -, `,
+      # tables) is untouched, so headings/bold/lists/code still render. Caveat:
+      # <, >, & that appear inside fenced or inline code are shown as entities.
+      escaped = text.to_s.gsub("&", "&amp;").gsub("<", "&lt;").gsub(">", "&gt;")
+
       html = Kramdown::Document.new(
-        text.to_s,
+        escaped,
         input: "GFM",
         hard_wrap: true,
         syntax_highlighter: nil

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.6"
 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.6
 platform: ruby
 authors:
 - Anton