Class swarmauri_standard.tracing.SimpleTraceContext.SimpleTraceContext

swarmauri_standard.tracing.SimpleTraceContext.SimpleTraceContext

SimpleTraceContext(trace_id, name, initial_attributes=None)

Bases: ITraceContext

Source code in swarmauri_standard/tracing/SimpleTraceContext.py

def __init__(
    self,
    trace_id: str,
    name: str,
    initial_attributes: Optional[Dict[str, Any]] = None,
):
    self.trace_id = trace_id
    self.name = name
    self.attributes = initial_attributes if initial_attributes else {}
    self.start_time = datetime.now()
    self.end_time = None

trace_id instance-attribute

trace_id = trace_id

name instance-attribute

name = name

attributes instance-attribute

attributes = (
    initial_attributes if initial_attributes else {}
)

start_time instance-attribute

start_time = now()

end_time instance-attribute

end_time = None

get_trace_id

get_trace_id()

Source code in swarmauri_standard/tracing/SimpleTraceContext.py

def get_trace_id(self) -> str:
    return self.trace_id

add_attribute

add_attribute(key, value)

Source code in swarmauri_standard/tracing/SimpleTraceContext.py

def add_attribute(self, key: str, value: Any):
    self.attributes[key] = value

close

close()

Source code in swarmauri_standard/tracing/SimpleTraceContext.py

def close(self):
    self.end_time = datetime.now()