ioa-observe-sdk 1.0.18__tar.gz → 1.0.20__tar.gz

{ioa_observe_sdk-1.0.18 → ioa_observe_sdk-1.0.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ioa-observe-sdk
-Version: 1.0.18
+Version: 1.0.20
 Summary: IOA Observability SDK
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{ioa_observe_sdk-1.0.18 → ioa_observe_sdk-1.0.20}/ioa_observe/sdk/decorators/__init__.py

@@ -22,6 +22,7 @@ def task(
     description: Optional[str] = None,
     version: Optional[int] = None,
     protocol: Optional[str] = None,
+    application_id: Optional[str] = None,
     method_name: Optional[str] = None,
     tlp_span_kind: Optional[ObserveSpanKindValues] = ObserveSpanKindValues.TASK,
 ) -> Callable[[F], F]:
@@ -31,6 +32,7 @@ def task(
             description=description,
             version=version,
             protocol=protocol,
+            application_id=application_id,
             tlp_span_kind=tlp_span_kind,
         )
     else:
@@ -39,6 +41,7 @@ def task(
             description=description,
             version=version,
             protocol=protocol,
+            application_id=application_id,
             method_name=method_name,
             tlp_span_kind=tlp_span_kind,
         )
@@ -49,6 +52,7 @@ def workflow(
     description: Optional[str] = None,
     version: Optional[int] = None,
     protocol: Optional[str] = None,
+    application_id: Optional[str] = None,
     method_name: Optional[str] = None,
     tlp_span_kind: Optional[
         Union[ObserveSpanKindValues, str]
@@ -62,6 +66,7 @@ def workflow(
                 description=description,
                 version=version,
                 protocol=protocol,
+                application_id=application_id,
                 method_name=method_name,
                 tlp_span_kind=tlp_span_kind,
             )(target)
@@ -72,6 +77,7 @@ def workflow(
                 description=description,
                 version=version,
                 protocol=protocol,
+                application_id=application_id,
                 tlp_span_kind=tlp_span_kind,
             )(target)
 
@@ -82,19 +88,27 @@ def graph(
     name: Optional[str] = None,
     description: Optional[str] = None,
     version: Optional[int] = None,
+    application_id: Optional[str] = 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, protocol=protocol, tlp_span_kind="graph"
+            name=name,
+            description=description,
+            version=version,
+            protocol=protocol,
+            application_id=application_id,
+            tlp_span_kind="graph",
         )
     else:
         return entity_class(
             name=name,
+            description=description,
             version=version,
             method_name=method_name,
             protocol=protocol,
+            application_id=application_id,
             tlp_span_kind="graph",
         )
 
@@ -104,6 +118,7 @@ def agent(
     description: Optional[str] = None,
     version: Optional[int] = None,
     protocol: Optional[str] = None,
+    application_id: Optional[str] = None,
     method_name: Optional[str] = None,
 ) -> Callable[[F], F]:
     return workflow(
@@ -111,6 +126,7 @@ def agent(
         description=description,
         version=version,
         protocol=protocol,
+        application_id=application_id,
         method_name=method_name,
         tlp_span_kind=ObserveSpanKindValues.AGENT,
     )
@@ -120,12 +136,14 @@ def tool(
     name: Optional[str] = None,
     description: Optional[str] = None,
     version: Optional[int] = None,
+    application_id: Optional[str] = None,
     method_name: Optional[str] = None,
 ) -> Callable[[F], F]:
     return task(
         name=name,
         description=description,
         version=version,
+        application_id=application_id,
         method_name=method_name,
         tlp_span_kind=ObserveSpanKindValues.TOOL,
     )

{ioa_observe_sdk-1.0.18 → ioa_observe_sdk-1.0.20}/ioa_observe/sdk/decorators/base.py

@@ -16,10 +16,10 @@ from ioa_observe.sdk.decorators.helpers import (
     _is_async_generator,
 )
 
-
 from langgraph.graph.state import CompiledStateGraph
 from opentelemetry import trace
 from opentelemetry import context as context_api
+from opentelemetry.context import get_value, attach, set_value
 from pydantic_core import PydanticSerializationError
 from typing_extensions import ParamSpec
 
@@ -36,6 +36,7 @@ from ioa_observe.sdk.tracing.tracing import (
     set_entity_path,
     get_chained_entity_path,
     set_agent_id_event,
+    set_application_id,
 )
 from ioa_observe.sdk.metrics.agents.agent_connections import connection_reliability
 from ioa_observe.sdk.utils import camel_to_snake
@@ -91,6 +92,7 @@ def _setup_span(
     tlp_span_kind: Optional[ObserveSpanKindValues] = None,
     version: Optional[int] = None,
     description: Optional[str] = None,
+    application_id: Optional[str] = None,
 ):
     """Sets up the OpenTelemetry span and context"""
     if tlp_span_kind in [
@@ -104,16 +106,56 @@ def _setup_span(
         #     set_session_id(session_id)
     if tlp_span_kind == "graph":
         span_name = f"{entity_name}.{tlp_span_kind}"
+
     else:
         span_name = f"{entity_name}.{tlp_span_kind.value}"
 
     with get_tracer() as tracer:
         span = tracer.start_span(span_name)
         ctx = trace.set_span_in_context(span)
+
+        # Preserve existing context values before attaching new context
+        session_id = get_value("session.id")
+        current_traceparent = get_value("current_traceparent")
+        agent_id = get_value("agent_id")
+        application_id_ctx = get_value("application_id")
+        association_properties = get_value("association_properties")
+        managed_prompt = get_value("managed_prompt")
+        prompt_key = get_value("prompt_key")
+        prompt_version = get_value("prompt_version")
+        prompt_version_name = get_value("prompt_version_name")
+        prompt_version_hash = get_value("prompt_version_hash")
+        prompt_template = get_value("prompt_template")
+        prompt_template_variables = get_value("prompt_template_variables")
+
         ctx_token = context_api.attach(ctx)
-        span.set_attribute(
-            "agent_id", entity_name
-        ) if tlp_span_kind == ObserveSpanKindValues.AGENT else None
+
+        # Re-attach preserved context values to the new context
+        if session_id is not None:
+            attach(set_value("session.id", session_id))
+        if current_traceparent is not None:
+            attach(set_value("current_traceparent", current_traceparent))
+        if agent_id is not None:
+            attach(set_value("agent_id", agent_id))
+        if application_id_ctx is not None:
+            attach(set_value("application_id", application_id_ctx))
+        if association_properties is not None:
+            attach(set_value("association_properties", association_properties))
+        if managed_prompt is not None:
+            attach(set_value("managed_prompt", managed_prompt))
+        if prompt_key is not None:
+            attach(set_value("prompt_key", prompt_key))
+        if prompt_version is not None:
+            attach(set_value("prompt_version", prompt_version))
+        if prompt_version_name is not None:
+            attach(set_value("prompt_version_name", prompt_version_name))
+        if prompt_version_hash is not None:
+            attach(set_value("prompt_version_hash", prompt_version_hash))
+        if prompt_template is not None:
+            attach(set_value("prompt_template", prompt_template))
+        if prompt_template_variables is not None:
+            attach(set_value("prompt_template_variables", prompt_template_variables))
+
         if tlp_span_kind == ObserveSpanKindValues.AGENT:
             with trace.get_tracer(__name__).start_span(
                 "agent_start_event", context=trace.set_span_in_context(span)
@@ -127,9 +169,6 @@ def _setup_span(
                     },
                 )
             # start_span.end()  # end the span immediately
-        # 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,
@@ -148,7 +187,11 @@ def _setup_span(
 
         if tlp_span_kind == ObserveSpanKindValues.AGENT:
             span.set_attribute("agent_chain_start_time", time.time())
-
+        if application_id:
+            set_application_id(application_id)
+            span.set_attribute(
+                "application_id", application_id
+            )  # set application id attribute
     return span, ctx, ctx_token
 
 
@@ -297,6 +340,7 @@ def entity_method(
     description: Optional[str] = None,
     version: Optional[int] = None,
     protocol: Optional[str] = None,
+    application_id: Optional[str] = None,
     tlp_span_kind: Optional[ObserveSpanKindValues] = ObserveSpanKindValues.TASK,
 ) -> Callable[[F], F]:
     def decorate(fn: F) -> F:
@@ -319,6 +363,7 @@ def entity_method(
                         tlp_span_kind,
                         version,
                         description,
+                        application_id,
                     )
                     _handle_span_input(span, args, kwargs, cls=JSONEncoder)
 
@@ -340,6 +385,7 @@ def entity_method(
                         tlp_span_kind,
                         version,
                         description,
+                        application_id,
                     )
 
                     # Handle case where span setup failed
@@ -436,6 +482,7 @@ def entity_method(
                     tlp_span_kind,
                     version,
                     description,
+                    application_id,
                 )
 
                 # Handle case where span setup failed
@@ -549,6 +596,7 @@ def entity_class(
     description: Optional[str],
     version: Optional[int],
     protocol: Optional[str],
+    application_id: Optional[str],
     method_name: Optional[str],
     tlp_span_kind: Optional[ObserveSpanKindValues] = ObserveSpanKindValues.TASK,
 ):
@@ -603,6 +651,7 @@ def entity_class(
                             description=description,
                             version=version,
                             protocol=protocol,
+                            application_id=application_id,
                             tlp_span_kind=tlp_span_kind,
                         )(unwrapped_method)
                         # Set the wrapped method on the class
@@ -619,6 +668,7 @@ def entity_class(
 def _handle_agent_span(span, entity_name, description, tlp_span_kind):
     if tlp_span_kind == ObserveSpanKindValues.AGENT:
         try:
+            span.set_attribute("agent_id", entity_name)  # set the agent id attribute
             set_agent_id_event(entity_name)
             span.add_event(
                 "agent_start_event",

{ioa_observe_sdk-1.0.18 → ioa_observe_sdk-1.0.20}/ioa_observe/sdk/instrumentations/slim.py

@@ -349,8 +349,17 @@ class SLIMInstrumentor(BaseInstrumentor):
                             # Store in kv_store with thread safety
                             with _kv_lock:
                                 kv_store.set(f"execution.{traceparent}", session_id)
-                    finally:
+
+                        # DON'T detach the context yet - we need it to persist for the callback
+                        # The context will be cleaned up later or by the garbage collector
+
+                    except Exception as e:
+                        # Only detach on error
                         context.detach(token)
+                        raise e
+                elif traceparent and session_id and session_id != "None":
+                    # Even without carrier context, set session ID if we have the data
+                    set_session_id(session_id, traceparent=traceparent)
 
                 # Fallback: check stored execution ID if not found in headers
                 if traceparent and (not session_id or session_id == "None"):

{ioa_observe_sdk-1.0.18 → ioa_observe_sdk-1.0.20}/ioa_observe/sdk/tracing/tracing.py

@@ -291,6 +291,10 @@ class TracerWrapper(object):
         if agent_id is not None:
             span.set_attribute("agent_id", agent_id)
 
+        application_id = get_value("application_id")
+        if application_id is not None:
+            span.set_attribute("application_id", application_id)
+
         if is_llm_span(span):
             self.llm_call_counter.add(1, attributes=span.attributes)
 
@@ -736,6 +740,10 @@ def set_agent_id_event(agent_id: str) -> None:
     attach(set_value("agent_id", agent_id))
 
 
+def set_application_id(app_id: str) -> None:
+    attach(set_value("application_id", app_id))
+
+
 def set_entity_path(entity_path: str) -> None:
     attach(set_value("entity_path", entity_path))
 

{ioa_observe_sdk-1.0.18 → ioa_observe_sdk-1.0.20}/ioa_observe_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ioa-observe-sdk
-Version: 1.0.18
+Version: 1.0.20
 Summary: IOA Observability SDK
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{ioa_observe_sdk-1.0.18 → ioa_observe_sdk-1.0.20}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ioa-observe-sdk"
-version = "1.0.18"
+version = "1.0.20"
 description = "IOA Observability SDK"
 readme = "README.md"
 requires-python = ">=3.10"