diff --git a/agency_swarm/threads/thread.py b/agency_swarm/threads/thread.py
index e1434e43..b28e1ae1 100644
--- a/agency_swarm/threads/thread.py
+++ b/agency_swarm/threads/thread.py
@@ -78,6 +78,7 @@ def init_thread(self):
                     **example,
                 )
 
+    @get_tracker().get_observe_decorator()
     def get_completion_stream(
         self,
         message: Union[str, List[dict], None],
@@ -101,6 +102,7 @@ def get_completion_stream(
             response_format=response_format,
         )
 
+    @get_tracker().get_observe_decorator()
     def get_completion(
         self,
         message: Union[str, List[dict], None],
diff --git a/tests/demos/demo_observability.py b/tests/demos/demo_observability.py
index 1867b0f1..63418bd6 100644
--- a/tests/demos/demo_observability.py
+++ b/tests/demos/demo_observability.py
@@ -5,8 +5,8 @@
 from agency_swarm import Agency, Agent
 from agency_swarm.util.oai import _get_openai_module, set_tracker
 
-TRACKER = "sqlite"
-# TRACKER = "langfuse"
+# TRACKER = "sqlite"
+TRACKER = "langfuse"
 
 if TRACKER == "sqlite":
     # Test SQLite configuration