ioa-observe-sdk 1.0.6__tar.gz → 1.0.8__tar.gz

{ioa_observe_sdk-1.0.6/ioa_observe_sdk.egg-info → ioa_observe_sdk-1.0.8}/PKG-INFO

@@ -1,38 +1,38 @@
 Metadata-Version: 2.4
 Name: ioa-observe-sdk
-Version: 1.0.6
+Version: 1.0.8
 Summary: IOA Observability SDK
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE.md
-Requires-Dist: aiohappyeyeballs==2.4.8
-Requires-Dist: aiohttp==3.11.18
-Requires-Dist: aiosignal==1.3.2
-Requires-Dist: annotated-types==0.7.0
-Requires-Dist: anyio==4.8.0
-Requires-Dist: async-timeout==4.0.3
-Requires-Dist: attrs==25.1.0
-Requires-Dist: backoff==2.2.1
-Requires-Dist: certifi==2025.1.31
-Requires-Dist: charset-normalizer==3.4.1
+Requires-Dist: aiohappyeyeballs>=2.4.8
+Requires-Dist: aiohttp>=3.11.18
+Requires-Dist: aiosignal>=1.3.2
+Requires-Dist: annotated-types>=0.7.0
+Requires-Dist: anyio>=4.8.0
+Requires-Dist: async-timeout>=4.0.3
+Requires-Dist: attrs>=25.1.0
+Requires-Dist: backoff>=2.2.1
+Requires-Dist: certifi>=2025.1.31
+Requires-Dist: charset-normalizer>=3.4.1
 Requires-Dist: colorama==0.4.6
-Requires-Dist: Deprecated==1.2.18
-Requires-Dist: distro==1.9.0
-Requires-Dist: exceptiongroup==1.2.2
-Requires-Dist: frozenlist==1.5.0
-Requires-Dist: googleapis-common-protos==1.69.0
-Requires-Dist: grpcio==1.70.0
-Requires-Dist: h11==0.16.0
-Requires-Dist: httpcore==1.0.9
-Requires-Dist: httpx==0.28.1
-Requires-Dist: idna==3.10
-Requires-Dist: importlib_metadata==8.5.0
-Requires-Dist: Jinja2==3.1.6
-Requires-Dist: jiter==0.8.2
-Requires-Dist: MarkupSafe==3.0.2
-Requires-Dist: monotonic==1.6
-Requires-Dist: multidict==6.1.0
-Requires-Dist: openai==1.75.0
+Requires-Dist: Deprecated>=1.2.18
+Requires-Dist: distro>=1.9.0
+Requires-Dist: exceptiongroup>=1.2.2
+Requires-Dist: frozenlist>=1.5.0
+Requires-Dist: googleapis-common-protos>=1.69.0
+Requires-Dist: grpcio>=1.70.0
+Requires-Dist: h11>=0.16.0
+Requires-Dist: httpcore>=1.0.9
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: idna>=3.10
+Requires-Dist: importlib_metadata>=8.5.0
+Requires-Dist: Jinja2>=3.1.6
+Requires-Dist: jiter>=0.8.2
+Requires-Dist: MarkupSafe>=3.0.2
+Requires-Dist: monotonic>=1.6
+Requires-Dist: multidict>=6.1.0
+Requires-Dist: openai>=1.75.0
 Requires-Dist: opentelemetry-api==1.33.1
 Requires-Dist: opentelemetry-distro
 Requires-Dist: opentelemetry-exporter-otlp==1.33.1
@@ -52,32 +52,32 @@ Requires-Dist: opentelemetry-sdk==1.33.1
 Requires-Dist: opentelemetry-semantic-conventions==0.54b1
 Requires-Dist: opentelemetry-semantic-conventions-ai==0.4.9
 Requires-Dist: opentelemetry-util-http==0.54b1
-Requires-Dist: packaging==24.2
-Requires-Dist: propcache==0.3.0
-Requires-Dist: protobuf==5.29.3
-Requires-Dist: pydantic==2.10.6
-Requires-Dist: pydantic_core==2.27.2
-Requires-Dist: python-dateutil==2.9.0.post0
+Requires-Dist: packaging>=24.2
+Requires-Dist: propcache>=0.3.0
+Requires-Dist: protobuf>=5.29.3
+Requires-Dist: pydantic>=2.10.6
+Requires-Dist: pydantic_core>=2.27.2
+Requires-Dist: python-dateutil>=2.9.0.post0
 Requires-Dist: regex==2024.11.6
-Requires-Dist: requests==2.32.3
-Requires-Dist: six==1.17.0
-Requires-Dist: sniffio==1.3.1
-Requires-Dist: tenacity==9.0.0
-Requires-Dist: tiktoken==0.9.0
-Requires-Dist: tqdm==4.67.1
-Requires-Dist: typing_extensions==4.12.2
-Requires-Dist: urllib3==2.3.0
-Requires-Dist: wrapt==1.17.2
-Requires-Dist: yarl==1.18.3
-Requires-Dist: zipp==3.21.0
+Requires-Dist: requests>=2.32.3
+Requires-Dist: six>=1.17.0
+Requires-Dist: sniffio>=1.3.1
+Requires-Dist: tenacity>=9.0.0
+Requires-Dist: tiktoken>=0.9.0
+Requires-Dist: tqdm>=4.67.1
+Requires-Dist: typing_extensions>=4.12.2
+Requires-Dist: urllib3>=2.3.0
+Requires-Dist: wrapt>=1.17.2
+Requires-Dist: yarl>=1.18.3
+Requires-Dist: zipp>=3.21.0
 Requires-Dist: langgraph>=0.3.2
 Requires-Dist: langchain>=0.3.19
 Requires-Dist: langchain-openai>=0.3.8
 Requires-Dist: langchain-community>=0.3.25
 Requires-Dist: llama-index>=0.12.34
 Requires-Dist: opentelemetry-instrumentation-requests
-Requires-Dist: opentelemetry-instrumentation-transformers>=0.40.2
-Requires-Dist: opentelemetry-instrumentation-crewai>=0.40.2
+Requires-Dist: opentelemetry-instrumentation-transformers>=0.40.8
+Requires-Dist: opentelemetry-instrumentation-crewai>=0.40.8
 Requires-Dist: llama-index-utils-workflow>=0.3.1
 Requires-Dist: pytest
 Requires-Dist: pytest-vcr
@@ -85,6 +85,8 @@ Dynamic: license-file
 
 # Observe-SDK
 
+[![PyPI version](https://img.shields.io/pypi/v/ioa-observe-sdk.svg)](https://pypi.org/project/ioa-observe-sdk/)
+
 IOA observability SDK for your multi-agentic application.
 
 ## Table of Contents
@@ -119,6 +121,20 @@ Link: [AGNTCY Observability Schema](https://github.com/agntcy/observe/blob/main/
 
 ## Dev
 
+Any Opentelemetry compatible backend can be used, but for this guide, we will use ClickhouseDB as the backend database.
+
+### Opentelemetry collector
+
+The OpenTelemetry Collector offers a vendor-agnostic implementation of how to receive, process and export telemetry data. It removes the need to run, operate, and maintain multiple agents/collectors.
+
+### Clickhouse DB
+
+ClickhouseDB is used as a backend database to store and query the collected telemetry data efficiently, enabling you to analyze and visualize observability information for your multi-agentic applications.
+
+### Grafana (optional)
+
+Grafana can be used to visualize the telemetry data collected by the OpenTelemetry Collector and stored in ClickhouseDB.
+
 To get started with development, start a Clickhouse DB and an OTel collector container locally using docker-compose like so:
 
 ```
@@ -126,6 +142,8 @@ cd deploy/
 docker compose up -d
 ```
 
+Running both locally allows you to test, monitor, and debug your observability setup in a development environment before deploying to production.
+
 Ensure the contents of `otel-collector.yaml` is correct.
 
 Check the logs of the collector to ensure it is running correctly:
@@ -134,6 +152,18 @@ Check the logs of the collector to ensure it is running correctly:
 docker logs -f otel-collector
 ```
 
+Viewing data in Clickhouse DB can be done using the Clickhouse client. The collector is configured to export telemetry data to Clickhouse.
+
+The clickhouse exporter creates various tables in the Clickhouse DB, including `otel_traces`, which is used to store trace data.
+
+For more info, refer to the [OpenTelemetry Clickhouse Exporter documentation](https://github.com/open-telemetry/opentelemetry-collector-contrib/blob/main/exporter/clickhouseexporter/README.md)
+
+```bash
+docker exec -it clickhouse-server clickhouse-client
+
+select * from otel_traces LIMIT 10;
+```
+
 Create a `.env` file with the following content:
 
 ```bash
@@ -165,6 +195,28 @@ OPENAI_API_KEY=<KEY> make test
 For getting started with the SDK, please refer to the [Getting Started](https://github.com/agntcy/observe/blob/main/GETTING-STARTED.md)
  file. It contains detailed instructions on how to set up and use the SDK effectively.
 
+### Grafana
+
+To configure Grafana to visualize the telemetry data, follow these steps:
+
+1. Spin up Grafana locally using Docker:
+
+```bash
+docker run -d -p 3000:3000 --name=grafana grafana/grafana
+```
+2. Access Grafana by navigating to `http://localhost:3000` in your web browser.
+   - Default username: `admin`
+   - Default password: `admin`
+
+3. Add a new data source:
+   - Choose "ClickHouse" as the data source type.
+   - Set the URL to `http://0.0.0.0:8123`.
+   - Configure the authentication settings if necessary.
+   - Save and test the connection to ensure it works correctly.
+
+Refer to the [Grafana ClickHouse plugin documentation](https://grafana.com/grafana/plugins/grafana-clickhouse-datasource/) for more details on configuring ClickHouse as a data source.
+
+
 ## Contributing
 
 Contributions are welcome! Please follow these steps to contribute:

ioa_observe_sdk-1.0.8/README.md

@@ -0,0 +1,143 @@
+# Observe-SDK
+
+[![PyPI version](https://img.shields.io/pypi/v/ioa-observe-sdk.svg)](https://pypi.org/project/ioa-observe-sdk/)
+
+IOA observability SDK for your multi-agentic application.
+
+## Table of Contents
+
+- [Installation](#installation)
+- [Schema](#schema)
+- [Dev](#dev)
+- [Testing](#testing)
+- [Getting Started](#getting-started)
+- [Contributing](#contributing)
+
+## Installation
+
+To install the package via PyPI, simply run:
+
+```bash
+pip install ioa_observe_sdk
+```
+
+Alternatively, to download the SDK from git, you could also use the following command. Ensure you have `uv` installed in your environment.
+
+```bash
+uv add "git+https://github.com/agntcy/observe"
+```
+
+## Schema
+
+The AGNTCY observability schema is an extension of the OTel LLM Semantic Conventions for Generative AI systems.
+This schema is designed to provide comprehensive observability for Multi-Agent Systems (MAS).
+
+Link: [AGNTCY Observability Schema](https://github.com/agntcy/observe/blob/main/schema/)
+
+## Dev
+
+Any Opentelemetry compatible backend can be used, but for this guide, we will use ClickhouseDB as the backend database.
+
+### Opentelemetry collector
+
+The OpenTelemetry Collector offers a vendor-agnostic implementation of how to receive, process and export telemetry data. It removes the need to run, operate, and maintain multiple agents/collectors.
+
+### Clickhouse DB
+
+ClickhouseDB is used as a backend database to store and query the collected telemetry data efficiently, enabling you to analyze and visualize observability information for your multi-agentic applications.
+
+### Grafana (optional)
+
+Grafana can be used to visualize the telemetry data collected by the OpenTelemetry Collector and stored in ClickhouseDB.
+
+To get started with development, start a Clickhouse DB and an OTel collector container locally using docker-compose like so:
+
+```
+cd deploy/
+docker compose up -d
+```
+
+Running both locally allows you to test, monitor, and debug your observability setup in a development environment before deploying to production.
+
+Ensure the contents of `otel-collector.yaml` is correct.
+
+Check the logs of the collector to ensure it is running correctly:
+
+```
+docker logs -f otel-collector
+```
+
+Viewing data in Clickhouse DB can be done using the Clickhouse client. The collector is configured to export telemetry data to Clickhouse.
+
+The clickhouse exporter creates various tables in the Clickhouse DB, including `otel_traces`, which is used to store trace data.
+
+For more info, refer to the [OpenTelemetry Clickhouse Exporter documentation](https://github.com/open-telemetry/opentelemetry-collector-contrib/blob/main/exporter/clickhouseexporter/README.md)
+
+```bash
+docker exec -it clickhouse-server clickhouse-client
+
+select * from otel_traces LIMIT 10;
+```
+
+Create a `.env` file with the following content:
+
+```bash
+OTLP_HTTP_ENDPOINT=http://localhost:4318
+```
+
+Install the dependencies and activate the virtual environment:
+
+```bash
+set -a
+source .env
+set +a
+
+python3 -m venv .venv
+source .venv/bin/activate
+uv sync
+```
+
+## Testing
+
+To run the unit tests, ensure you have the `OPENAI_API_KEY` set in your environment. You can run the tests using the following command:
+
+```bash
+OPENAI_API_KEY=<KEY> make test
+```
+
+## 🚀 Getting Started
+
+For getting started with the SDK, please refer to the [Getting Started](https://github.com/agntcy/observe/blob/main/GETTING-STARTED.md)
+ file. It contains detailed instructions on how to set up and use the SDK effectively.
+
+### Grafana
+
+To configure Grafana to visualize the telemetry data, follow these steps:
+
+1. Spin up Grafana locally using Docker:
+
+```bash
+docker run -d -p 3000:3000 --name=grafana grafana/grafana
+```
+2. Access Grafana by navigating to `http://localhost:3000` in your web browser.
+   - Default username: `admin`
+   - Default password: `admin`
+
+3. Add a new data source:
+   - Choose "ClickHouse" as the data source type.
+   - Set the URL to `http://0.0.0.0:8123`.
+   - Configure the authentication settings if necessary.
+   - Save and test the connection to ensure it works correctly.
+
+Refer to the [Grafana ClickHouse plugin documentation](https://grafana.com/grafana/plugins/grafana-clickhouse-datasource/) for more details on configuring ClickHouse as a data source.
+
+
+## Contributing
+
+Contributions are welcome! Please follow these steps to contribute:
+
+1. Fork the repository.
+2. Create a new branch (`git checkout -b feature-branch`).
+3. Commit your changes (`git commit -am 'Add new feature'`).
+4. Push to the branch (`git push origin feature-branch`).
+5. Create a new Pull Request.

{ioa_observe_sdk-1.0.6 → ioa_observe_sdk-1.0.8}/ioa_observe/sdk/decorators/__init__.py

@@ -19,15 +19,22 @@ F = TypeVar("F", bound=Callable[P, Union[R, Awaitable[R]]])
 
 def task(
     name: Optional[str] = None,
+    description: Optional[str] = None,
     version: Optional[int] = None,
     method_name: Optional[str] = None,
     tlp_span_kind: Optional[ObserveSpanKindValues] = ObserveSpanKindValues.TASK,
 ) -> Callable[[F], F]:
     if method_name is None:
-        return entity_method(name=name, version=version, tlp_span_kind=tlp_span_kind)
+        return entity_method(
+            name=name,
+            description=description,
+            version=version,
+            tlp_span_kind=tlp_span_kind,
+        )
     else:
         return entity_class(
             name=name,
+            description=description,
             version=version,
             method_name=method_name,
             tlp_span_kind=tlp_span_kind,
@@ -36,21 +43,21 @@ def task(
 
 def workflow(
     name: Optional[str] = None,
+    description: Optional[str] = None,
     version: Optional[int] = None,
     method_name: Optional[str] = None,
     tlp_span_kind: Optional[
         Union[ObserveSpanKindValues, str]
     ] = ObserveSpanKindValues.WORKFLOW,
 ) -> Callable[[F], F]:
-    if method_name is None:
-        return entity_method(name=name, version=version, tlp_span_kind=tlp_span_kind)
-    else:
-        return entity_class(
-            name=name,
-            version=version,
-            method_name=method_name,
-            tlp_span_kind=tlp_span_kind,
-        )
+    # Always use entity_class for class decorators
+    return entity_class(
+        name=name,
+        description=description,
+        version=version,
+        method_name=method_name,
+        tlp_span_kind=tlp_span_kind,
+    )
 
 
 def graph(
@@ -60,6 +67,7 @@ def graph(
 ) -> Callable[[F], F]:
     return workflow(
         name=name,
+        description=None,
         version=version,
         method_name=method_name,
         tlp_span_kind="graph",
@@ -68,11 +76,13 @@ def graph(
 
 def agent(
     name: Optional[str] = None,
+    description: Optional[str] = None,
     version: Optional[int] = None,
     method_name: Optional[str] = None,
 ) -> Callable[[F], F]:
     return workflow(
         name=name,
+        description=description,
         version=version,
         method_name=method_name,
         tlp_span_kind=ObserveSpanKindValues.AGENT,
@@ -81,11 +91,13 @@ def agent(
 
 def tool(
     name: Optional[str] = None,
+    description: Optional[str] = None,
     version: Optional[int] = None,
     method_name: Optional[str] = None,
 ) -> Callable[[F], F]:
     return task(
         name=name,
+        description=description,
         version=version,
         method_name=method_name,
         tlp_span_kind=ObserveSpanKindValues.TOOL,

{ioa_observe_sdk-1.0.6 → ioa_observe_sdk-1.0.8}/ioa_observe/sdk/decorators/base.py

@@ -2,6 +2,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import json
+import time
 import traceback
 from functools import wraps
 import os
@@ -91,6 +92,7 @@ def _setup_span(
     entity_name,
     tlp_span_kind: Optional[ObserveSpanKindValues] = None,
     version: Optional[int] = None,
+    description: Optional[str] = None,
 ):
     """Sets up the OpenTelemetry span and context"""
     if tlp_span_kind in [
@@ -100,8 +102,8 @@ def _setup_span(
     ]:
         set_workflow_name(entity_name)
         # if tlp_span_kind == "graph":
-        #     execution_id = entity_name + "_" + str(uuid.uuid4())
-        #     set_execution_id(execution_id)
+        #     session_id = entity_name + "_" + str(uuid.uuid4())
+        #     set_session_id(session_id)
     if tlp_span_kind == "graph":
         span_name = f"{entity_name}.{tlp_span_kind}"
     else:
@@ -122,13 +124,14 @@ def _setup_span(
                     "agent_start_event",
                     {
                         "agent_name": entity_name,
+                        "description": description if description else "",
                         "type": tlp_span_kind.value,
                     },
                 )
             # start_span.end()  # end the span immediately
-        # execution_id = get_value("execution.id")
-        # if execution_id is not None:
-        #     span.set_attribute("execution.id", execution_id)
+        # session_id = get_value("session.id")
+        # if session_id is not None:
+        #     span.set_attribute("session.id", session_id)
         if tlp_span_kind in [
             ObserveSpanKindValues.TASK,
             ObserveSpanKindValues.TOOL,
@@ -145,9 +148,8 @@ def _setup_span(
         if version:
             span.set_attribute(OBSERVE_ENTITY_VERSION, version)
 
-        # if execution_id:
-        #     print(f"Execution ID: {execution_id}")
-        #     span.set_attribute("execution.id", execution_id)
+        if tlp_span_kind == ObserveSpanKindValues.AGENT:
+            span.set_attribute("agent_chain_start_time", time.time())
 
     return span, ctx, ctx_token
 
@@ -243,12 +245,27 @@ def _handle_span_output(span, tlp_span_kind, res, cls=None):
 
 def _cleanup_span(span, ctx_token):
     """End the span process and detach the context token"""
+
+    # Calculate agent chain completion time before ending span
+    span_kind = span.attributes.get(OBSERVE_SPAN_KIND)
+    if span_kind == ObserveSpanKindValues.AGENT.value:
+        start_time = span.attributes.get("agent_chain_start_time")
+        if start_time is not None:
+            import time
+
+            completion_time = time.time() - start_time
+
+            # Emit the metric
+            TracerWrapper().agent_chain_completion_time_histogram.record(
+                completion_time, attributes=span.attributes
+            )
     span.end()
     context_api.detach(ctx_token)
 
 
 def entity_method(
     name: Optional[str] = None,
+    description: Optional[str] = None,
     version: Optional[int] = None,
     tlp_span_kind: Optional[ObserveSpanKindValues] = ObserveSpanKindValues.TASK,
 ) -> Callable[[F], F]:
@@ -266,7 +283,10 @@ def entity_method(
                         return
 
                     span, ctx, ctx_token = _setup_span(
-                        entity_name, tlp_span_kind, version
+                        entity_name,
+                        tlp_span_kind,
+                        version,
+                        description if description else None,
                     )
                     _handle_span_input(span, args, kwargs, cls=JSONEncoder)
 
@@ -284,7 +304,10 @@ def entity_method(
                         return await fn(*args, **kwargs)
 
                     span, ctx, ctx_token = _setup_span(
-                        entity_name, tlp_span_kind, version
+                        entity_name,
+                        tlp_span_kind,
+                        version,
+                        description if description else None,
                     )
                     _handle_span_input(span, args, kwargs, cls=JSONEncoder)
                     success = False
@@ -359,6 +382,8 @@ def entity_method(
                         _handle_agent_failure_event(str(e), span, tlp_span_kind)
                         raise e
                     finally:
+                        if tlp_span_kind == ObserveSpanKindValues.AGENT:
+                            TracerWrapper().record_agent_execution(entity_name, success)
                         _cleanup_span(span, ctx_token)
                     return res
 
@@ -370,10 +395,15 @@ def entity_method(
                 if not TracerWrapper.verify_initialized():
                     return fn(*args, **kwargs)
 
-                span, ctx, ctx_token = _setup_span(entity_name, tlp_span_kind, version)
+                span, ctx, ctx_token = _setup_span(
+                    entity_name,
+                    tlp_span_kind,
+                    version,
+                    description if description else None,
+                )
 
                 _handle_span_input(span, args, kwargs, cls=JSONEncoder)
-                _handle_agent_span(span, entity_name, tlp_span_kind)
+                _handle_agent_span(span, entity_name, description, tlp_span_kind)
                 success = False
 
                 # Record heartbeat for agent
@@ -459,6 +489,8 @@ def entity_method(
                     _handle_agent_failure_event(str(e), span, tlp_span_kind)
                     raise e
                 finally:
+                    if tlp_span_kind == ObserveSpanKindValues.AGENT:
+                        TracerWrapper().record_agent_execution(entity_name, success)
                     _cleanup_span(span, ctx_token)
                 return res
 
@@ -469,26 +501,51 @@ def entity_method(
 
 def entity_class(
     name: Optional[str],
+    description: Optional[str],
     version: Optional[int],
-    method_name: str,
+    method_name: Optional[str],
     tlp_span_kind: Optional[ObserveSpanKindValues] = ObserveSpanKindValues.TASK,
 ):
     def decorator(cls):
         task_name = name if name else camel_to_snake(cls.__qualname__)
-        method = getattr(cls, method_name)
-        setattr(
-            cls,
-            method_name,
-            entity_method(name=task_name, version=version, tlp_span_kind=tlp_span_kind)(
-                method
-            ),
-        )
+
+        methods_to_wrap = []
+
+        if method_name:
+            # Specific method specified - existing behavior
+            methods_to_wrap = [method_name]
+        else:
+            # No method specified - wrap all public methods
+            for attr_name in dir(cls):
+                if (
+                    not attr_name.startswith("_")  # Skip private/built-in methods
+                    and attr_name != "mro"  # Skip class method
+                    and hasattr(cls, attr_name)
+                    and callable(getattr(cls, attr_name))
+                    and not isinstance(
+                        getattr(cls, attr_name), (classmethod, staticmethod, property)
+                    )
+                ):
+                    methods_to_wrap.append(attr_name)
+
+        # Wrap all detected methods
+        for method_to_wrap in methods_to_wrap:
+            if hasattr(cls, method_to_wrap):
+                method = getattr(cls, method_to_wrap)
+                wrapped_method = entity_method(
+                    name=f"{task_name}.{method_to_wrap}",
+                    description=description,
+                    version=version,
+                    tlp_span_kind=tlp_span_kind,
+                )(method)
+                setattr(cls, method_to_wrap, wrapped_method)
+
         return cls
 
     return decorator
 
 
-def _handle_agent_span(span, entity_name, tlp_span_kind):
+def _handle_agent_span(span, entity_name, description, tlp_span_kind):
     if tlp_span_kind == ObserveSpanKindValues.AGENT:
         try:
             set_agent_id_event(entity_name)
@@ -496,6 +553,7 @@ def _handle_agent_span(span, entity_name, tlp_span_kind):
                 "agent_start_event",
                 {
                     "agent_name": entity_name,
+                    "description": description if description else "",
                     "type": tlp_span_kind.value
                     if tlp_span_kind != "graph"
                     else "graph",
@@ -581,7 +639,8 @@ def _handle_graph_response(span, res, tlp_span_kind):
             }
 
             # Convert to JSON string
-            s_graph_json = json.dumps(graph_dict, indent=2)
+            s_graph_json = json.dumps(graph_dict)
+            # convert to JSON and set as attribute
             span.set_attribute("gen_ai.ioa.graph", s_graph_json)
 
             # get graph dynamism