Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions quotientai/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.

Expand All @@ -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):
"""
Expand Down Expand Up @@ -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
Expand Down