ioa-observe-sdk 1.0.23__py3-none-any.whl → 1.0.24__py3-none-any.whl

ioa_observe/sdk/instrumentations/a2a.py

@@ -89,6 +89,60 @@ class A2AInstrumentor(BaseInstrumentor):
 
         A2AClient.send_message = instrumented_send_message
 
+        # Instrument broadcast_message
+        if hasattr(A2AClient, "broadcast_message"):
+            original_broadcast_message = A2AClient.broadcast_message
+
+            @functools.wraps(original_broadcast_message)
+            async def instrumented_broadcast_message(self, request, *args, **kwargs):
+                # Put context into A2A message metadata instead of HTTP headers
+                with _global_tracer.start_as_current_span("a2a.broadcast_message"):
+                    traceparent = get_current_traceparent()
+                    session_id = None
+                    if traceparent:
+                        session_id = kv_store.get(f"execution.{traceparent}")
+                        if session_id:
+                            kv_store.set(f"execution.{traceparent}", session_id)
+
+                    # Ensure metadata dict exists
+                    try:
+                        md = getattr(request.params, "metadata", None)
+                    except AttributeError:
+                        md = None
+                    metadata = md if isinstance(md, dict) else {}
+
+                    observe_meta = dict(metadata.get("observe", {}))
+
+                    # Inject W3C trace context + baggage into observe_meta
+                    TraceContextTextMapPropagator().inject(carrier=observe_meta)
+                    W3CBaggagePropagator().inject(carrier=observe_meta)
+
+                    if traceparent:
+                        observe_meta["traceparent"] = traceparent
+                    if session_id:
+                        observe_meta["session_id"] = session_id
+                        baggage.set_baggage(f"execution.{traceparent}", session_id)
+
+                    metadata["observe"] = observe_meta
+
+                    # Write back metadata (pydantic models are mutable by default in v2)
+                    try:
+                        request.params.metadata = metadata
+                    except Exception:
+                        # Fallback
+                        request = request.model_copy(
+                            update={
+                                "params": request.params.model_copy(
+                                    update={"metadata": metadata}
+                                )
+                            }
+                        )
+
+                # Call through without transport-specific kwargs
+                return await original_broadcast_message(self, request, *args, **kwargs)
+
+            A2AClient.broadcast_message = instrumented_broadcast_message
+
         # Instrument server handler
         from a2a.server.request_handlers import DefaultRequestHandler
 
@@ -156,6 +210,12 @@ class A2AInstrumentor(BaseInstrumentor):
 
         A2AClient.send_message = A2AClient.send_message.__wrapped__
 
+        # Uninstrument `broadcast_message`
+        if hasattr(A2AClient, "broadcast_message") and hasattr(
+            A2AClient.broadcast_message, "__wrapped__"
+        ):
+            A2AClient.broadcast_message = A2AClient.broadcast_message.__wrapped__
+
         # Uninstrument server handler
         from a2a.server.request_handlers import DefaultRequestHandler
 

ioa_observe/sdk/instrumentations/slim.py

@@ -478,19 +478,19 @@ class SLIMInstrumentor(BaseInstrumentor):
             slim_bindings.Slim.listen_for_session = instrumented_listen_for_session
 
     def _instrument_session_methods(self, slim_bindings):
-        # check if slim_bindings >= v0.6.0 is installed by looking for Session class
-        if not hasattr(slim_bindings, "Session"):
-            return
-
-        # In v0.6.0+, we need to instrument session classes dynamically
         # Try to find session-related classes in the slim_bindings module
         session_classes = []
 
-        # Look for common session class names
-        for attr_name in ["Session", "P2PSession", "GroupSession"]:
-            if hasattr(slim_bindings, attr_name):
-                session_class = getattr(slim_bindings, attr_name)
-                session_classes.append((attr_name, session_class))
+        # Check for v0.6.0+ Session classes
+        if hasattr(slim_bindings, "Session"):
+            for attr_name in ["Session", "P2PSession", "GroupSession"]:
+                if hasattr(slim_bindings, attr_name):
+                    session_class = getattr(slim_bindings, attr_name)
+                    session_classes.append((attr_name, session_class))
+
+        # Check for older PySession class (pre-v0.6.0)
+        if hasattr(slim_bindings, "PySession"):
+            session_classes.append(("PySession", slim_bindings.PySession))
 
         # Also look for any class that has session-like methods
         for attr_name in dir(slim_bindings):
@@ -501,7 +501,7 @@ class SLIMInstrumentor(BaseInstrumentor):
                 session_classes.append((attr_name, attr))
 
         # Instrument session methods for found classes
-        for _, session_class in session_classes:
+        for class_name, session_class in session_classes:
             # Instrument get_message (v0.6.0+ replacement for receive)
             if hasattr(session_class, "get_message"):
                 self._instrument_session_get_message(session_class)

{ioa_observe_sdk-1.0.23.dist-info → ioa_observe_sdk-1.0.24.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ioa-observe-sdk
-Version: 1.0.23
+Version: 1.0.24
 Summary: IOA Observability SDK
 License-Expression: Apache-2.0
 Requires-Python: >=3.10

{ioa_observe_sdk-1.0.23.dist-info → ioa_observe_sdk-1.0.24.dist-info}/RECORD

@@ -14,10 +14,10 @@ ioa_observe/sdk/decorators/base.py,sha256=pnoj73UrpvaZmOPbcpGRRYCwg7LBAgO8PiBDh5
 ioa_observe/sdk/decorators/helpers.py,sha256=I9HXMBivkZpGDtPe9Ad_UU35p_m_wEPate4r_fU0oOA,2705
 ioa_observe/sdk/decorators/util.py,sha256=IebvH9gwZN1en3LblYJUh4bAV2STl6xmp8WpZzBDH2g,30068
 ioa_observe/sdk/instrumentations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ioa_observe/sdk/instrumentations/a2a.py,sha256=ZpqvPl4u-yheQzSdBfxnZhWFZ8ntbKni_uaW3IDyjqw,6309
+ioa_observe/sdk/instrumentations/a2a.py,sha256=6BOMxrGeMiii-5rptouqkUd2RDpGL01ZyDotgo7pFyM,9088
 ioa_observe/sdk/instrumentations/mcp.py,sha256=vRM3ofnn7AMmry2RrfyZnZVPEutLWiDMghx2TSnm0Wk,18569
 ioa_observe/sdk/instrumentations/nats.py,sha256=UOp2AJlm1JkYkwF3xzU_izzohQVQkByjL-AX4n_JRfo,14476
-ioa_observe/sdk/instrumentations/slim.py,sha256=ONuHU1Wtc15l7qYUBFvB154_I8DVs537nwPaj67sjEI,43912
+ioa_observe/sdk/instrumentations/slim.py,sha256=uU5GgwpCszOoC7cWzyR5S3Zab59STN9vPhQGeTT8dZ8,43933
 ioa_observe/sdk/logging/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ioa_observe/sdk/logging/logging.py,sha256=HZxW9s8Due7jgiNkdI38cIjv5rC9D-Flta3RQMOnpow,2891
 ioa_observe/sdk/metrics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -42,8 +42,8 @@ ioa_observe/sdk/utils/const.py,sha256=d67dUTAH9UpWvUV9GLBUqn1Sc2knJ55dy-e6YoLrvS
 ioa_observe/sdk/utils/in_memory_span_exporter.py,sha256=H_4TRaThMO1H6vUQ0OpQvzJk_fZH0OOsRAM1iZQXsR8,2112
 ioa_observe/sdk/utils/json_encoder.py,sha256=g4NQ0tTqgWssY6I1D7r4zo0G6PiUo61jhofTAw5-jno,639
 ioa_observe/sdk/utils/package_check.py,sha256=1d1MjxhwoEZIx9dumirT2pRsEWgn-m-SI4npDeEalew,576
-ioa_observe_sdk-1.0.23.dist-info/licenses/LICENSE.md,sha256=55VjUfgjWOS4vv3Cf55gfq-RxjPgRIO2vlgYPUuC5lA,11362
-ioa_observe_sdk-1.0.23.dist-info/METADATA,sha256=xb37XMbpNTA7xNLiWyA5O39AHebNB--J7nZ1H9l7kGs,7058
-ioa_observe_sdk-1.0.23.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ioa_observe_sdk-1.0.23.dist-info/top_level.txt,sha256=Yt-6Y1olZEDqCs2REeqI30WjYx0pLGQSVqzYmDd67N8,12
-ioa_observe_sdk-1.0.23.dist-info/RECORD,,
+ioa_observe_sdk-1.0.24.dist-info/licenses/LICENSE.md,sha256=55VjUfgjWOS4vv3Cf55gfq-RxjPgRIO2vlgYPUuC5lA,11362
+ioa_observe_sdk-1.0.24.dist-info/METADATA,sha256=Xq670J9Et2X5XVQl5Ij2hMz_Ovg_5SP7f30A14wgEtk,7058
+ioa_observe_sdk-1.0.24.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ioa_observe_sdk-1.0.24.dist-info/top_level.txt,sha256=Yt-6Y1olZEDqCs2REeqI30WjYx0pLGQSVqzYmDd67N8,12
+ioa_observe_sdk-1.0.24.dist-info/RECORD,,