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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ioa-observe-sdk
-Version: 1.0.23
+Version: 1.0.25
 Summary: IOA Observability SDK
 License-Expression: Apache-2.0
 Requires-Python: >=3.10
@@ -76,6 +76,37 @@ Alternatively, to download the SDK from git, you could also use the following co
 uv add "git+https://github.com/agntcy/observe"
 ```
 
+### Quick Start
+
+After installation, import and initialize the SDK:
+
+```python
+import os
+from ioa_observe.sdk import Observe
+from ioa_observe.sdk.decorators import agent
+from ioa_observe.sdk.tracing import session_start
+
+# Initialize Observe
+Observe.init(
+    app_name="your_app_name",
+    api_endpoint=os.getenv("OTLP_HTTP_ENDPOINT", "http://localhost:4318")
+)
+
+# Use decorators to instrument your agents
+@agent(name="my_agent", description="Example agent")
+def my_agent_function(state):
+    # Your agent logic here
+    return {"result": "success"}
+
+# Start a session for tracking
+with session_start() as session_id:
+    result = my_agent_function({"input": "data"})
+```
+
+**Note:** The package name for installation is `ioa_observe_sdk`, but imports use `ioa_observe` (underscore, not hyphen).
+
+For comprehensive integration examples with LangGraph, LlamaIndex, and other frameworks, see the [Getting Started Guide](GETTING-STARTED.md).
+
 ## Schema
 
 The AGNTCY observability schema is an extension of the OTel LLM Semantic Conventions for Generative AI systems.

{ioa_observe_sdk-1.0.23 → ioa_observe_sdk-1.0.25}/README.md

@@ -27,6 +27,37 @@ Alternatively, to download the SDK from git, you could also use the following co
 uv add "git+https://github.com/agntcy/observe"
 ```
 
+### Quick Start
+
+After installation, import and initialize the SDK:
+
+```python
+import os
+from ioa_observe.sdk import Observe
+from ioa_observe.sdk.decorators import agent
+from ioa_observe.sdk.tracing import session_start
+
+# Initialize Observe
+Observe.init(
+    app_name="your_app_name",
+    api_endpoint=os.getenv("OTLP_HTTP_ENDPOINT", "http://localhost:4318")
+)
+
+# Use decorators to instrument your agents
+@agent(name="my_agent", description="Example agent")
+def my_agent_function(state):
+    # Your agent logic here
+    return {"result": "success"}
+
+# Start a session for tracking
+with session_start() as session_id:
+    result = my_agent_function({"input": "data"})
+```
+
+**Note:** The package name for installation is `ioa_observe_sdk`, but imports use `ioa_observe` (underscore, not hyphen).
+
+For comprehensive integration examples with LangGraph, LlamaIndex, and other frameworks, see the [Getting Started Guide](GETTING-STARTED.md).
+
 ## Schema
 
 The AGNTCY observability schema is an extension of the OTel LLM Semantic Conventions for Generative AI systems.

{ioa_observe_sdk-1.0.23 → ioa_observe_sdk-1.0.25}/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.25}/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)
@@ -686,7 +686,13 @@ class SLIMInstrumentor(BaseInstrumentor):
                             baggage.set_baggage(f"execution.{traceparent}", session_id)
 
                         # Wrap the message (first argument for publish, second for publish_to)
-                        message_idx = 1 if method_name == "publish_to" else 0
+                        # If the session_class is SessionContext, the message is always in second position
+                        message_idx = (
+                            1
+                            if method_name == "publish_to"
+                            or session_class.__name__ == "SessionContext"
+                            else 0
+                        )
                         if len(args) > message_idx:
                             args_list = list(args)
                             message = args_list[message_idx]

{ioa_observe_sdk-1.0.23 → ioa_observe_sdk-1.0.25}/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.25
 Summary: IOA Observability SDK
 License-Expression: Apache-2.0
 Requires-Python: >=3.10
@@ -76,6 +76,37 @@ Alternatively, to download the SDK from git, you could also use the following co
 uv add "git+https://github.com/agntcy/observe"
 ```
 
+### Quick Start
+
+After installation, import and initialize the SDK:
+
+```python
+import os
+from ioa_observe.sdk import Observe
+from ioa_observe.sdk.decorators import agent
+from ioa_observe.sdk.tracing import session_start
+
+# Initialize Observe
+Observe.init(
+    app_name="your_app_name",
+    api_endpoint=os.getenv("OTLP_HTTP_ENDPOINT", "http://localhost:4318")
+)
+
+# Use decorators to instrument your agents
+@agent(name="my_agent", description="Example agent")
+def my_agent_function(state):
+    # Your agent logic here
+    return {"result": "success"}
+
+# Start a session for tracking
+with session_start() as session_id:
+    result = my_agent_function({"input": "data"})
+```
+
+**Note:** The package name for installation is `ioa_observe_sdk`, but imports use `ioa_observe` (underscore, not hyphen).
+
+For comprehensive integration examples with LangGraph, LlamaIndex, and other frameworks, see the [Getting Started Guide](GETTING-STARTED.md).
+
 ## Schema
 
 The AGNTCY observability schema is an extension of the OTel LLM Semantic Conventions for Generative AI systems.

{ioa_observe_sdk-1.0.23 → ioa_observe_sdk-1.0.25}/pyproject.toml

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