diff --git a/quotientai/client.py b/quotientai/client.py index 0248044..0171bba 100644 --- a/quotientai/client.py +++ b/quotientai/client.py @@ -493,7 +493,7 @@ def init( return self - def trace(self): + def trace(self, name: Optional[str] = None): """ Decorator to trace function calls for Quotient. @@ -515,7 +515,7 @@ async def my_async_function(): return lambda func: func # Call the tracing resource without parameters since it's now configured - return self.tracing_resource.trace() + return self.tracing_resource.trace(name) def force_flush(self): """ @@ -720,9 +720,9 @@ def log( ) return log_id - def trace(self): + def trace(self, name: Optional[str] = None): """Direct access to the tracer's trace decorator.""" - return self.tracer.trace() + return self.tracer.trace(name) def poll_for_detection( self, log_id: str, timeout: int = 300, poll_interval: float = 2.0