Add more annotations to aws_xray_sdk.core.recorder (#6029)

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: ...