ioa-observe-sdk 1.0.13__tar.gz → 1.0.15__tar.gz

{ioa_observe_sdk-1.0.13 → ioa_observe_sdk-1.0.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ioa-observe-sdk
-Version: 1.0.13
+Version: 1.0.15
 Summary: IOA Observability SDK
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
@@ -15,17 +15,28 @@ Requires-Dist: opentelemetry-exporter-otlp-proto-grpc==1.33.1
 Requires-Dist: opentelemetry-exporter-otlp-proto-http==1.33.1
 Requires-Dist: opentelemetry-instrumentation
 Requires-Dist: opentelemetry-instrumentation-logging==0.54b1
-Requires-Dist: opentelemetry-instrumentation-openai==0.40.8
-Requires-Dist: opentelemetry-instrumentation-llamaindex==0.40.8
-Requires-Dist: opentelemetry-instrumentation-ollama==0.40.8
-Requires-Dist: opentelemetry-instrumentation-anthropic==0.40.8
-Requires-Dist: opentelemetry-instrumentation-langchain==0.40.8
-Requires-Dist: opentelemetry-instrumentation-threading==00.54b1
+Requires-Dist: opentelemetry-instrumentation-openai==0.43.1
+Requires-Dist: opentelemetry-instrumentation-llamaindex==0.43.1
+Requires-Dist: opentelemetry-instrumentation-ollama==0.43.1
+Requires-Dist: opentelemetry-instrumentation-anthropic==0.43.1
+Requires-Dist: opentelemetry-instrumentation-langchain==0.43.1
+Requires-Dist: opentelemetry-instrumentation-bedrock==0.43.1
+Requires-Dist: opentelemetry-instrumentation-cohere==0.43.1
+Requires-Dist: opentelemetry-instrumentation-crewai==0.43.1
+Requires-Dist: opentelemetry-instrumentation-google-generativeai==0.43.1
+Requires-Dist: opentelemetry-instrumentation-groq==0.43.1
+Requires-Dist: opentelemetry-instrumentation-mistralai==0.43.1
+Requires-Dist: opentelemetry-instrumentation-requests==0.54b1
+Requires-Dist: opentelemetry-instrumentation-sagemaker==0.43.1
+Requires-Dist: opentelemetry-instrumentation-threading==0.54b1
+Requires-Dist: opentelemetry-instrumentation-together==0.43.1
+Requires-Dist: opentelemetry-instrumentation-transformers==0.43.1
 Requires-Dist: opentelemetry-instrumentation-urllib3==0.54b1
+Requires-Dist: opentelemetry-instrumentation-vertexai==0.43.1
 Requires-Dist: opentelemetry-proto==1.33.1
 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-semantic-conventions-ai>=0.4.11
 Requires-Dist: opentelemetry-util-http==0.54b1
 Requires-Dist: langgraph>=0.3.2
 Requires-Dist: langchain>=0.3.19

{ioa_observe_sdk-1.0.13 → ioa_observe_sdk-1.0.15}/ioa_observe/sdk/decorators/__init__.py

@@ -21,6 +21,7 @@ def task(
     name: Optional[str] = None,
     description: Optional[str] = None,
     version: Optional[int] = None,
+    protocol: Optional[str] = None,
     method_name: Optional[str] = None,
     tlp_span_kind: Optional[ObserveSpanKindValues] = ObserveSpanKindValues.TASK,
 ) -> Callable[[F], F]:
@@ -29,6 +30,7 @@ def task(
             name=name,
             description=description,
             version=version,
+            protocol=protocol,
             tlp_span_kind=tlp_span_kind,
         )
     else:
@@ -36,6 +38,7 @@ def task(
             name=name,
             description=description,
             version=version,
+            protocol=protocol,
             method_name=method_name,
             tlp_span_kind=tlp_span_kind,
         )
@@ -45,6 +48,7 @@ def workflow(
     name: Optional[str] = None,
     description: Optional[str] = None,
     version: Optional[int] = None,
+    protocol: Optional[str] = None,
     method_name: Optional[str] = None,
     tlp_span_kind: Optional[
         Union[ObserveSpanKindValues, str]
@@ -57,6 +61,7 @@ def workflow(
                 name=name,
                 description=description,
                 version=version,
+                protocol=protocol,
                 method_name=method_name,
                 tlp_span_kind=tlp_span_kind,
             )(target)
@@ -66,6 +71,7 @@ def workflow(
                 name=name,
                 description=description,
                 version=version,
+                protocol=protocol,
                 tlp_span_kind=tlp_span_kind,
             )(target)
 
@@ -77,14 +83,18 @@ def graph(
     description: Optional[str] = None,
     version: Optional[int] = None,
     method_name: Optional[str] = None,
+    protocol: Optional[str] = None,
 ) -> Callable[[F], F]:
     if method_name is None:
-        return entity_method(name=name, version=version, tlp_span_kind="graph")
+        return entity_method(
+            name=name, version=version, protocol=protocol, tlp_span_kind="graph"
+        )
     else:
         return entity_class(
             name=name,
             version=version,
             method_name=method_name,
+            protocol=protocol,
             tlp_span_kind="graph",
         )
 
@@ -93,12 +103,14 @@ def agent(
     name: Optional[str] = None,
     description: Optional[str] = None,
     version: Optional[int] = None,
+    protocol: Optional[str] = None,
     method_name: Optional[str] = None,
 ) -> Callable[[F], F]:
     return workflow(
         name=name,
         description=description,
         version=version,
+        protocol=protocol,
         method_name=method_name,
         tlp_span_kind=ObserveSpanKindValues.AGENT,
     )

{ioa_observe_sdk-1.0.13 → ioa_observe_sdk-1.0.15}/ioa_observe/sdk/decorators/base.py

@@ -7,7 +7,7 @@ import traceback
 from functools import wraps
 import os
 import types
-from typing import Optional, TypeVar, Callable, Awaitable, Any, cast, Union
+from typing import Optional, TypeVar, Callable, Awaitable, Any, Union
 import inspect
 
 from ioa_observe.sdk.decorators.helpers import (
@@ -285,13 +285,23 @@ def _cleanup_span(span, ctx_token):
     context_api.detach(ctx_token)
 
 
+def _unwrap_structured_tool(fn):
+    # Unwraps StructuredTool or similar wrappers to get the underlying function
+    if hasattr(fn, "func") and callable(fn.func):
+        return fn.func
+    return fn
+
+
 def entity_method(
     name: Optional[str] = None,
     description: Optional[str] = None,
     version: Optional[int] = None,
+    protocol: Optional[str] = None,
     tlp_span_kind: Optional[ObserveSpanKindValues] = ObserveSpanKindValues.TASK,
 ) -> Callable[[F], F]:
     def decorate(fn: F) -> F:
+        # Unwrap StructuredTool if present
+        fn = _unwrap_structured_tool(fn)
         is_async = _is_async_method(fn)
         entity_name = name or _get_original_function_name(fn)
         if is_async:
@@ -367,7 +377,7 @@ def entity_method(
                                     agent_recovery_tracker.record_agent_recovery(
                                         entity_name
                                     )
-                        _handle_graph_response(span, res, tlp_span_kind)
+                        _handle_graph_response(span, res, protocol, tlp_span_kind)
                         # span will be ended in the generator
                         if isinstance(res, types.GeneratorType):
                             return _handle_generator(span, res)
@@ -413,7 +423,7 @@ def entity_method(
                         _cleanup_span(span, ctx_token)
                     return res
 
-                return cast(F, async_wrap)
+            decorated = async_wrap
         else:
 
             @wraps(fn)
@@ -477,7 +487,7 @@ def entity_method(
                                 agent_recovery_tracker.record_agent_recovery(
                                     entity_name
                                 )
-                    _handle_graph_response(span, res, tlp_span_kind)
+                    _handle_graph_response(span, res, protocol, tlp_span_kind)
 
                     # span will be ended in the generator
                     if isinstance(res, types.GeneratorType):
@@ -524,7 +534,12 @@ def entity_method(
                     _cleanup_span(span, ctx_token)
                 return res
 
-            return cast(F, sync_wrap)
+            decorated = sync_wrap
+        # # If the original fn was a StructuredTool, re-wrap
+        if hasattr(fn, "func") and callable(fn.func):
+            fn.func = decorated
+            return fn
+        return decorated
 
     return decorate
 
@@ -533,6 +548,7 @@ def entity_class(
     name: Optional[str],
     description: Optional[str],
     version: Optional[int],
+    protocol: Optional[str],
     method_name: Optional[str],
     tlp_span_kind: Optional[ObserveSpanKindValues] = ObserveSpanKindValues.TASK,
 ):
@@ -577,16 +593,18 @@ def entity_class(
             if hasattr(cls, method_to_wrap):
                 original_method = getattr(cls, method_to_wrap)
                 # Only wrap actual functions defined in this class
-                if inspect.isfunction(original_method):
+                unwrapped_method = _unwrap_structured_tool(original_method)
+                if inspect.isfunction(unwrapped_method):
                     try:
                         # Verify the method has a proper signature
-                        sig = inspect.signature(original_method)
+                        sig = inspect.signature(unwrapped_method)
                         wrapped_method = entity_method(
                             name=f"{task_name}.{method_to_wrap}",
                             description=description,
                             version=version,
+                            protocol=protocol,
                             tlp_span_kind=tlp_span_kind,
-                        )(original_method)
+                        )(unwrapped_method)
                         # Set the wrapped method on the class
                         setattr(cls, method_to_wrap, wrapped_method)
                     except Exception:
@@ -646,15 +664,17 @@ def _handle_execution_result(span, success):
     return
 
 
-def _handle_graph_response(span, res, tlp_span_kind):
+def _handle_graph_response(span, res, protocol, tlp_span_kind):
     if tlp_span_kind == "graph":
+        if protocol:
+            protocol = protocol.upper()
+            span.set_attribute("gen_ai.ioa.graph.protocol", protocol)
         # Check if the response is a Llama Index Workflow object
         graph = determine_workflow_type(res)
         if graph is not None:
             # Convert the graph to JSON string
             graph_json = json.dumps(graph, indent=2)
             span.set_attribute("gen_ai.ioa.graph", graph_json)
-
             # get graph dynamism
             dynamism = topology_dynamism(graph)
             span.set_attribute("gen_ai.ioa.graph_dynamism", dynamism)

{ioa_observe_sdk-1.0.13 → ioa_observe_sdk-1.0.15}/ioa_observe/sdk/decorators/util.py

@@ -13,7 +13,7 @@ from llama_index.core.workflow.utils import (
 )
 
 
-def _serialize_object(obj, max_depth=5, current_depth=0):
+def _serialize_object(obj, max_depth=3, current_depth=0):
     """
     Intelligently serialize an object to a more meaningful representation
     """

{ioa_observe_sdk-1.0.13 → ioa_observe_sdk-1.0.15}/ioa_observe_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ioa-observe-sdk
-Version: 1.0.13
+Version: 1.0.15
 Summary: IOA Observability SDK
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
@@ -15,17 +15,28 @@ Requires-Dist: opentelemetry-exporter-otlp-proto-grpc==1.33.1
 Requires-Dist: opentelemetry-exporter-otlp-proto-http==1.33.1
 Requires-Dist: opentelemetry-instrumentation
 Requires-Dist: opentelemetry-instrumentation-logging==0.54b1
-Requires-Dist: opentelemetry-instrumentation-openai==0.40.8
-Requires-Dist: opentelemetry-instrumentation-llamaindex==0.40.8
-Requires-Dist: opentelemetry-instrumentation-ollama==0.40.8
-Requires-Dist: opentelemetry-instrumentation-anthropic==0.40.8
-Requires-Dist: opentelemetry-instrumentation-langchain==0.40.8
-Requires-Dist: opentelemetry-instrumentation-threading==00.54b1
+Requires-Dist: opentelemetry-instrumentation-openai==0.43.1
+Requires-Dist: opentelemetry-instrumentation-llamaindex==0.43.1
+Requires-Dist: opentelemetry-instrumentation-ollama==0.43.1
+Requires-Dist: opentelemetry-instrumentation-anthropic==0.43.1
+Requires-Dist: opentelemetry-instrumentation-langchain==0.43.1
+Requires-Dist: opentelemetry-instrumentation-bedrock==0.43.1
+Requires-Dist: opentelemetry-instrumentation-cohere==0.43.1
+Requires-Dist: opentelemetry-instrumentation-crewai==0.43.1
+Requires-Dist: opentelemetry-instrumentation-google-generativeai==0.43.1
+Requires-Dist: opentelemetry-instrumentation-groq==0.43.1
+Requires-Dist: opentelemetry-instrumentation-mistralai==0.43.1
+Requires-Dist: opentelemetry-instrumentation-requests==0.54b1
+Requires-Dist: opentelemetry-instrumentation-sagemaker==0.43.1
+Requires-Dist: opentelemetry-instrumentation-threading==0.54b1
+Requires-Dist: opentelemetry-instrumentation-together==0.43.1
+Requires-Dist: opentelemetry-instrumentation-transformers==0.43.1
 Requires-Dist: opentelemetry-instrumentation-urllib3==0.54b1
+Requires-Dist: opentelemetry-instrumentation-vertexai==0.43.1
 Requires-Dist: opentelemetry-proto==1.33.1
 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-semantic-conventions-ai>=0.4.11
 Requires-Dist: opentelemetry-util-http==0.54b1
 Requires-Dist: langgraph>=0.3.2
 Requires-Dist: langchain>=0.3.19

ioa_observe_sdk-1.0.15/ioa_observe_sdk.egg-info/requires.txt

@@ -0,0 +1,39 @@
+colorama==0.4.6
+requests>=2.32.3
+opentelemetry-api==1.33.1
+opentelemetry-distro
+opentelemetry-exporter-otlp==1.33.1
+opentelemetry-exporter-otlp-proto-common==1.33.1
+opentelemetry-exporter-otlp-proto-grpc==1.33.1
+opentelemetry-exporter-otlp-proto-http==1.33.1
+opentelemetry-instrumentation
+opentelemetry-instrumentation-logging==0.54b1
+opentelemetry-instrumentation-openai==0.43.1
+opentelemetry-instrumentation-llamaindex==0.43.1
+opentelemetry-instrumentation-ollama==0.43.1
+opentelemetry-instrumentation-anthropic==0.43.1
+opentelemetry-instrumentation-langchain==0.43.1
+opentelemetry-instrumentation-bedrock==0.43.1
+opentelemetry-instrumentation-cohere==0.43.1
+opentelemetry-instrumentation-crewai==0.43.1
+opentelemetry-instrumentation-google-generativeai==0.43.1
+opentelemetry-instrumentation-groq==0.43.1
+opentelemetry-instrumentation-mistralai==0.43.1
+opentelemetry-instrumentation-requests==0.54b1
+opentelemetry-instrumentation-sagemaker==0.43.1
+opentelemetry-instrumentation-threading==0.54b1
+opentelemetry-instrumentation-together==0.43.1
+opentelemetry-instrumentation-transformers==0.43.1
+opentelemetry-instrumentation-urllib3==0.54b1
+opentelemetry-instrumentation-vertexai==0.43.1
+opentelemetry-proto==1.33.1
+opentelemetry-sdk==1.33.1
+opentelemetry-semantic-conventions==0.54b1
+opentelemetry-semantic-conventions-ai>=0.4.11
+opentelemetry-util-http==0.54b1
+langgraph>=0.3.2
+langchain>=0.3.19
+langchain-openai>=0.3.8
+llama-index>=0.12.34
+pytest
+pytest-vcr

{ioa_observe_sdk-1.0.13 → ioa_observe_sdk-1.0.15}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ioa-observe-sdk"
-version = "1.0.13"
+version = "1.0.15"
 description = "IOA Observability SDK"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -21,17 +21,28 @@ dependencies = [
     "opentelemetry-exporter-otlp-proto-http==1.33.1",
     "opentelemetry-instrumentation",
     "opentelemetry-instrumentation-logging==0.54b1",
-    "opentelemetry-instrumentation-openai==0.40.8",
-    "opentelemetry-instrumentation-llamaindex==0.40.8",
-    "opentelemetry-instrumentation-ollama==0.40.8",
-    "opentelemetry-instrumentation-anthropic==0.40.8",
-    "opentelemetry-instrumentation-langchain==0.40.8",
-    "opentelemetry-instrumentation-threading==00.54b1",
+    "opentelemetry-instrumentation-openai==0.43.1",
+    "opentelemetry-instrumentation-llamaindex==0.43.1",
+    "opentelemetry-instrumentation-ollama==0.43.1",
+    "opentelemetry-instrumentation-anthropic==0.43.1",
+    "opentelemetry-instrumentation-langchain==0.43.1",
+    "opentelemetry-instrumentation-bedrock==0.43.1",
+    "opentelemetry-instrumentation-cohere==0.43.1",
+    "opentelemetry-instrumentation-crewai==0.43.1",
+    "opentelemetry-instrumentation-google-generativeai==0.43.1",
+    "opentelemetry-instrumentation-groq==0.43.1",
+    "opentelemetry-instrumentation-mistralai==0.43.1",
+    "opentelemetry-instrumentation-requests==0.54b1",
+    "opentelemetry-instrumentation-sagemaker==0.43.1",
+    "opentelemetry-instrumentation-threading==0.54b1",
+    "opentelemetry-instrumentation-together==0.43.1",
+    "opentelemetry-instrumentation-transformers==0.43.1",
     "opentelemetry-instrumentation-urllib3==0.54b1",
+    "opentelemetry-instrumentation-vertexai==0.43.1",
     "opentelemetry-proto==1.33.1",
     "opentelemetry-sdk==1.33.1",
     "opentelemetry-semantic-conventions==0.54b1",
-    "opentelemetry-semantic-conventions-ai==0.4.9",
+    "opentelemetry-semantic-conventions-ai>=0.4.11",
     "opentelemetry-util-http==0.54b1",
     "langgraph>=0.3.2",
     "langchain>=0.3.19",

ioa_observe_sdk-1.0.13/ioa_observe_sdk.egg-info/requires.txt

@@ -1,28 +0,0 @@
-colorama==0.4.6
-requests>=2.32.3
-opentelemetry-api==1.33.1
-opentelemetry-distro
-opentelemetry-exporter-otlp==1.33.1
-opentelemetry-exporter-otlp-proto-common==1.33.1
-opentelemetry-exporter-otlp-proto-grpc==1.33.1
-opentelemetry-exporter-otlp-proto-http==1.33.1
-opentelemetry-instrumentation
-opentelemetry-instrumentation-logging==0.54b1
-opentelemetry-instrumentation-openai==0.40.8
-opentelemetry-instrumentation-llamaindex==0.40.8
-opentelemetry-instrumentation-ollama==0.40.8
-opentelemetry-instrumentation-anthropic==0.40.8
-opentelemetry-instrumentation-langchain==0.40.8
-opentelemetry-instrumentation-threading==00.54b1
-opentelemetry-instrumentation-urllib3==0.54b1
-opentelemetry-proto==1.33.1
-opentelemetry-sdk==1.33.1
-opentelemetry-semantic-conventions==0.54b1
-opentelemetry-semantic-conventions-ai==0.4.9
-opentelemetry-util-http==0.54b1
-langgraph>=0.3.2
-langchain>=0.3.19
-langchain-openai>=0.3.8
-llama-index>=0.12.34
-pytest
-pytest-vcr