ioa-observe-sdk 1.0.23__tar.gz → 1.0.24__tar.gz

{ioa_observe_sdk-1.0.23 → ioa_observe_sdk-1.0.24}/PKG-INFO

@@ -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 → ioa_observe_sdk-1.0.24}/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-1.0.23 → ioa_observe_sdk-1.0.24}/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 → ioa_observe_sdk-1.0.24}/ioa_observe_sdk.egg-info/PKG-INFO

@@ -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 → ioa_observe_sdk-1.0.24}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ioa-observe-sdk"
-version = "1.0.23"
+version = "1.0.24"
 license = "Apache-2.0"
 description = "IOA Observability SDK"
 readme = "README.md"