diff --git a/stubs/aws-xray-sdk/aws_xray_sdk/core/recorder.pyi b/stubs/aws-xray-sdk/aws_xray_sdk/core/recorder.pyi index 410cd1ce9..f1c47af48 100644 --- a/stubs/aws-xray-sdk/aws_xray_sdk/core/recorder.pyi +++ b/stubs/aws-xray-sdk/aws_xray_sdk/core/recorder.pyi @@ -50,24 +50,24 @@ class AWSXRayRecorder: sampler: LocalSampler | DefaultSampler | None = ..., stream_sql: bool | None = ..., ) -> None: ... - def in_segment(self, name: str | None = ..., **segment_kwargs): ... - def in_subsegment(self, name: str | None = ..., **subsegment_kwargs): ... + def in_segment(self, name: str | None = ..., **segment_kwargs) -> SegmentContextManager: ... + def in_subsegment(self, name: str | None = ..., **subsegment_kwargs) -> SubsegmentContextManager: ... def begin_segment( self, name: str | None = ..., traceid: str | None = ..., parent_id: str | None = ..., sampling: bool | None = ... - ): ... + ) -> Segment | DummySegment: ... def end_segment(self, end_time: time.struct_time | None = ...) -> None: ... - def current_segment(self): ... - def begin_subsegment(self, name: str, namespace: str = ...): ... - def current_subsegment(self): ... + def current_segment(self) -> Segment: ... + def begin_subsegment(self, name: str, namespace: str = ...) -> DummySubsegment | Subsegment | None: ... + def current_subsegment(self) -> Subsegment | DummySubsegment | None: ... def end_subsegment(self, end_time: time.struct_time | None = ...) -> None: ... def put_annotation(self, key: str, value: Any) -> None: ... def put_metadata(self, key: str, value: Any, namespace: str = ...) -> None: ... - def is_sampled(self): ... - def get_trace_entity(self): ... - def set_trace_entity(self, trace_entity: Context) -> None: ... + def is_sampled(self) -> bool: ... + def get_trace_entity(self) -> Segment | Subsegment | DummySegment | DummySubsegment: ... + def set_trace_entity(self, trace_entity: Segment | Subsegment | DummySegment | DummySubsegment) -> None: ... def clear_trace_entities(self) -> None: ... def stream_subsegments(self) -> None: ... - def capture(self, name: str | None = ...): ... + def capture(self, name: str | None = ...) -> SubsegmentContextManager: ... def record_subsegment( self, wrapped: Callable[..., Any], @@ -77,48 +77,48 @@ class AWSXRayRecorder: name: str, namespace: str, meta_processor: Callable[..., Any], - ): ... + ) -> Any: ... @property - def enabled(self): ... + def enabled(self) -> bool: ... @enabled.setter def enabled(self, value: bool) -> None: ... @property - def sampling(self): ... + def sampling(self) -> bool: ... @sampling.setter def sampling(self, value: bool) -> None: ... @property - def sampler(self): ... + def sampler(self) -> LocalSampler | DefaultSampler: ... @sampler.setter def sampler(self, value: LocalSampler | DefaultSampler) -> None: ... @property - def service(self): ... + def service(self) -> str: ... @service.setter def service(self, value: str) -> None: ... @property - def dynamic_naming(self): ... + def dynamic_naming(self) -> Any | DefaultDynamicNaming: ... @dynamic_naming.setter - def dynamic_naming(self, value) -> None: ... + def dynamic_naming(self, value: Any | DefaultDynamicNaming) -> None: ... @property - def context(self): ... + def context(self) -> Context: ... @context.setter def context(self, cxt: Context) -> None: ... @property - def emitter(self): ... + def emitter(self) -> UDPEmitter: ... @emitter.setter def emitter(self, value: UDPEmitter) -> None: ... @property - def streaming(self): ... + def streaming(self) -> DefaultStreaming: ... @streaming.setter def streaming(self, value: DefaultStreaming) -> None: ... @property - def streaming_threshold(self): ... + def streaming_threshold(self) -> int: ... @streaming_threshold.setter def streaming_threshold(self, value: int) -> None: ... @property - def max_trace_back(self): ... + def max_trace_back(self) -> int: ... @max_trace_back.setter def max_trace_back(self, value: int) -> None: ... @property - def stream_sql(self): ... + def stream_sql(self) -> bool: ... @stream_sql.setter def stream_sql(self, value: bool) -> None: ...