nucliadb-telemetry 6.3.1.post3524__py3-none-any.whl → 6.3.1.post3531__py3-none-any.whl

nucliadb_telemetry/fastapi/metrics.py

@@ -28,7 +28,7 @@ from starlette.status import HTTP_500_INTERNAL_SERVER_ERROR
 from .utils import get_path_template
 
 try:
-    from starlette_prometheus.middleware import (
+    from starlette_prometheus.middleware import (  # type: ignore
         EXCEPTIONS,
         REQUESTS,
         REQUESTS_IN_PROGRESS,

nucliadb_telemetry/jetstream.py

@@ -19,13 +19,12 @@
 
 from datetime import datetime
 from functools import partial
-from typing import Any, Callable, Dict, List, Optional
+from typing import Any, Callable, Dict, List, Optional, Union
 
 import nats
 from nats.aio.client import Client
 from nats.aio.msg import Msg
 from nats.js.client import JetStreamContext
-from opentelemetry.context import attach
 from opentelemetry.propagate import extract, inject
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.semconv.trace import SpanAttributes
@@ -63,7 +62,7 @@ msg_sent_counter = metrics.Counter("nuclia_nats_msg_sent", labels={"subject": ""
 
 
 def start_span_message_receiver(tracer: Tracer, msg: Msg):
-    attributes = {
+    attributes: dict[str, Union[str, int]] = {
         SpanAttributes.MESSAGING_DESTINATION_KIND: "nats",
         SpanAttributes.MESSAGING_MESSAGE_PAYLOAD_SIZE_BYTES: len(msg.data),
         SpanAttributes.MESSAGING_MESSAGE_ID: msg.reply,
@@ -71,14 +70,14 @@ def start_span_message_receiver(tracer: Tracer, msg: Msg):
 
     # add some attributes from the metadata
     ctx = extract(msg.headers)
-    token = attach(ctx)
 
-    span = tracer.start_as_current_span(  # type: ignore
+    span = tracer.start_as_current_span(
         name=f"Received from {msg.subject}",
+        context=ctx,
         kind=SpanKind.SERVER,
         attributes=attributes,
     )
-    span._token = token
+
     return span
 
 
@@ -249,15 +248,13 @@ class NatsClientTelemetry:
 
         with start_span_message_publisher(tracer, subject) as span:
             try:
-                result = await self.nc.publish(subject, body, headers=headers, **kwargs)
+                await self.nc.publish(subject, body, headers=headers, **kwargs)
                 msg_sent_counter.inc({"subject": subject, "status": metrics.OK})
             except Exception as error:
                 set_span_exception(span, error)
                 msg_sent_counter.inc({"subject": subject, "status": metrics.ERROR})
                 raise error
 
-        return result
-
     async def request(
         self,
         subject: str,

{nucliadb_telemetry-6.3.1.post3524.dist-info → nucliadb_telemetry-6.3.1.post3531.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: nucliadb_telemetry
-Version: 6.3.1.post3524
+Version: 6.3.1.post3531
 Summary: NucliaDB Telemetry Library Python process
 Author-email: Nuclia <nucliadb@nuclia.com>
 License: AGPL

{nucliadb_telemetry-6.3.1.post3524.dist-info → nucliadb_telemetry-6.3.1.post3531.dist-info}/RECORD

@@ -7,7 +7,7 @@ nucliadb_telemetry/grpc.py,sha256=1CdueeJedxManSUQKwNOvqUuFiVhnMiZViXUJm81bIY,14
 nucliadb_telemetry/grpc_metrics.py,sha256=L2XMcJDxioNUYPX_IbKl_zvrG8uOFpF_a67c6yJq-Jw,5174
 nucliadb_telemetry/grpc_sentry.py,sha256=3eS_DVIX5MzvbJBkykf8JyZUf1b_b9w1KLhjoiW7oV0,2659
 nucliadb_telemetry/jaeger.py,sha256=HMQK8x_oaol5tCx3hCjaMG5Uzvsy_WWhKnD-FEZoYA4,7036
-nucliadb_telemetry/jetstream.py,sha256=9oMAAqtBrER7ISz8Jm2tbuVWCxQgfqLdHy3VraVFnIg,10057
+nucliadb_telemetry/jetstream.py,sha256=E2_SY5-8ljhZAY0n_q-qlvewUmi2kF2En3mcqyA26kY,9977
 nucliadb_telemetry/logs.py,sha256=ag1e7tKEoNFgB1FuAmyTxdIHtTFwOJlGxPl2s1bhUic,9311
 nucliadb_telemetry/metrics.py,sha256=jSGecUI7qNbPm3OEzo11v4Zo-lUhXhZrhkWPtX1eGf8,8022
 nucliadb_telemetry/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -16,7 +16,7 @@ nucliadb_telemetry/tracerprovider.py,sha256=ycVqSMo0NTa7EEv4QLC8qAt0NYKbOy8uC9k1
 nucliadb_telemetry/utils.py,sha256=lqbW5SjpGcfEhiJUScYeaNrtt5BFnOTaJn5dwnjCVMM,4765
 nucliadb_telemetry/fastapi/__init__.py,sha256=NRtQtyE1jZzTK5bZETYqb0d6bpjthTAt-UXiO8DAQWQ,3716
 nucliadb_telemetry/fastapi/context.py,sha256=qhl3rVDtqDfnsCHBAlv-eX8SUSlooMnsryBcV5FMZvU,1530
-nucliadb_telemetry/fastapi/metrics.py,sha256=15GDaTNslQ00AZEAZvQFC8OyVIKdKLjzhafnWbsfytw,3906
+nucliadb_telemetry/fastapi/metrics.py,sha256=fcnT5ShmU3oHci2FNJPoyYwWJiP60O0G3CEUAzUsC1w,3922
 nucliadb_telemetry/fastapi/tracing.py,sha256=U85I3QIFb27aj4v3Cl6l5evEgskP12LuEXW4pAsvGCc,15052
 nucliadb_telemetry/fastapi/utils.py,sha256=HtfhBSaYs_HWHx7SBp0pPAsoig5CLcHNFIFnzb-RXOE,2979
 nucliadb_telemetry/tests/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
@@ -30,7 +30,7 @@ nucliadb_telemetry/tests/grpc/helloworld_pb2.py,sha256=ht4dmi0pAy6qDrwcjkbtSf_hO
 nucliadb_telemetry/tests/grpc/helloworld_pb2.pyi,sha256=mBZCQE6z7riQw6fOegJgZ5lHmXqO23_zxL8iQMqSqms,1191
 nucliadb_telemetry/tests/grpc/helloworld_pb2_grpc.py,sha256=_jxUNxl4Fx-JztK9RO5R6osjNP2sVNVPAxLnmczEYOc,2677
 nucliadb_telemetry/tests/grpc/helloworld_pb2_grpc.pyi,sha256=Y6teCx-PhPU-rI6w5ItLBKaTb34FLpngPnuDVWtNve4,958
-nucliadb_telemetry-6.3.1.post3524.dist-info/METADATA,sha256=oFbESFZg31EEhSRJXAHBKeiWX0uGpH6AXwCPKdK7438,10640
-nucliadb_telemetry-6.3.1.post3524.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-nucliadb_telemetry-6.3.1.post3524.dist-info/top_level.txt,sha256=3qEHI_5ttqQIL2gkNYwSlKsFyBBiEzDiIy9UKISzOaQ,19
-nucliadb_telemetry-6.3.1.post3524.dist-info/RECORD,,
+nucliadb_telemetry-6.3.1.post3531.dist-info/METADATA,sha256=oJ9QQXrQH4HczD3HSSs2li-ykyUv9lH62kRAMUUnexE,10640
+nucliadb_telemetry-6.3.1.post3531.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
+nucliadb_telemetry-6.3.1.post3531.dist-info/top_level.txt,sha256=3qEHI_5ttqQIL2gkNYwSlKsFyBBiEzDiIy9UKISzOaQ,19
+nucliadb_telemetry-6.3.1.post3531.dist-info/RECORD,,