Merge pull request #687 from kusaanko/fix-event-dispatch

Fix to check if events should be recorded in tracing
utilityai · Mar 3, 2025 · 6ab83db · 6ab83db
2 parents 209ad86 + 727419a
commit 6ab83db
Showing 1 changed file with 12 additions and 10 deletions.
diff --git a/llama-cpp-2/src/log.rs b/llama-cpp-2/src/log.rs
@@ -142,16 +142,18 @@ impl State {
         let (meta, fields) = meta_for_level(level);
 
         tracing::dispatcher::get_default(|dispatcher| {
-            dispatcher.event(&tracing::Event::new(
-                meta,
-                &meta.fields().value_set(&[
-                    (&fields.message, Some(&text as &dyn tracing::field::Value)),
-                    (
-                        &fields.target,
-                        module.as_ref().map(|s| s as &dyn tracing::field::Value),
-                    ),
-                ]),
-            ));
+            if dispatcher.enabled(meta) {
+                dispatcher.event(&tracing::Event::new(
+                    meta,
+                    &meta.fields().value_set(&[
+                        (&fields.message, Some(&text as &dyn tracing::field::Value)),
+                        (
+                            &fields.target,
+                            module.as_ref().map(|s| s as &dyn tracing::field::Value),
+                        ),
+                    ]),
+                ));
+            }
         });
     }