Update utils/tracing.py

utils/tracing.py

@@ -1,30 +1,38 @@
-import os
-import json
-from typing import Optional
-
-class Tracer:
-    def __init__(self, client=None, trace_url: Optional[str] = None):
-        self.client = client
-        self.trace_url = trace_url
-
-    @classmethod
-    def from_env(cls):
-        try:
-            from langfuse import Langfuse
-            pk = os.getenv("LANGFUSE_PUBLIC_KEY")
-            sk = os.getenv("LANGFUSE_SECRET_KEY")
-            host = os.getenv("LANGFUSE_HOST", "https://cloud.langfuse.com")
-            if pk and sk:
-                client = Langfuse(public_key=pk, secret_key=sk, host=host)
-                session = client.trace("tabular-agentic-xai")
-                return cls(client=session, trace_url=session.get_url() if hasattr(session, "get_url") else None)
-        except Exception:
-            pass
-        return cls()
-
-    def trace_event(self, name: str, payload: dict):
-        if self.client:
-            try:
-                self.client.event(name=name, input=json.dumps(payload))
-            except Exception:
-                pass
+import os
+import json
+
+class Tracer:
+    def __init__(self):
+        self.enabled = False
+        self.client = None
+        self.trace = None
+        try:
+            from langfuse import Langfuse
+            pk = os.getenv("LANGFUSE_PUBLIC_KEY")
+            sk = os.getenv("LANGFUSE_SECRET_KEY")
+            host = os.getenv("LANGFUSE_HOST", "https://cloud.langfuse.com")
+            if pk and sk:
+                self.client = Langfuse(public_key=pk, secret_key=sk, host=host)
+                self.trace = self.client.trace(name="tabular-agentic-xai")
+                self.enabled = True
+        except Exception:
+            pass
+
+    @classmethod
+    def from_env(cls):
+        return cls()
+
+    @property
+    def trace_url(self):
+        try:
+            return self.trace.get_url() if (self.enabled and hasattr(self.trace, "get_url")) else None
+        except Exception:
+            return None
+
+    def trace_event(self, name: str, payload: dict):
+        if not self.enabled or not self.trace:
+            return
+        try:
+            self.trace.event(name=name, input=json.dumps(payload))
+        except Exception:
+            pass