ioa-observe-sdk 1.0.18__py3-none-any.whl → 1.0.19__py3-none-any.whl

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/decorators/base.py

@@ -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,6 +106,7 @@ 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}"
 
@@ -111,9 +114,6 @@ def _setup_span(
         span = tracer.start_span(span_name)
         ctx = trace.set_span_in_context(span)
         ctx_token = context_api.attach(ctx)
-        span.set_attribute(
-            "agent_id", entity_name
-        ) if tlp_span_kind == ObserveSpanKindValues.AGENT else None
         if tlp_span_kind == ObserveSpanKindValues.AGENT:
             with trace.get_tracer(__name__).start_span(
                 "agent_start_event", context=trace.set_span_in_context(span)
@@ -148,7 +148,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 +301,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 +324,7 @@ def entity_method(
                         tlp_span_kind,
                         version,
                         description,
+                        application_id,
                     )
                     _handle_span_input(span, args, kwargs, cls=JSONEncoder)
 
@@ -340,6 +346,7 @@ def entity_method(
                         tlp_span_kind,
                         version,
                         description,
+                        application_id,
                     )
 
                     # Handle case where span setup failed
@@ -436,6 +443,7 @@ def entity_method(
                     tlp_span_kind,
                     version,
                     description,
+                    application_id,
                 )
 
                 # Handle case where span setup failed
@@ -549,6 +557,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 +612,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 +629,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/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.dist-info → ioa_observe_sdk-1.0.19.dist-info}/METADATA

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

{ioa_observe_sdk-1.0.18.dist-info → ioa_observe_sdk-1.0.19.dist-info}/RECORD

@@ -9,8 +9,8 @@ ioa_observe/sdk/client/http.py,sha256=LdLYSQPFIhKN5BTB-N78jLO7ITl7jGjA0-qpewEIvO
 ioa_observe/sdk/config/__init__.py,sha256=8aVNaw0yRNLFPxlf97iOZLlJVcV81ivSDnudH2m1OIo,572
 ioa_observe/sdk/connectors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ioa_observe/sdk/connectors/slim.py,sha256=NwbKEV7d5NIOqmG8zKqtgGigSJl7kf3QJ65z2gxpsY8,8498
-ioa_observe/sdk/decorators/__init__.py,sha256=GUZs_HA57bQTLSgo7GAnaofAapk2Y-NuE6md0HPSH3s,3603
-ioa_observe/sdk/decorators/base.py,sha256=FLICXVOFzv6yuCQL99G5Do5h1WrKJNsegvfm5KHe9v8,30173
+ioa_observe/sdk/decorators/__init__.py,sha256=qCpJAv98eLKs3I5EMXJVTV0s49Nc6QDSOHNh5rW5vLg,4268
+ioa_observe/sdk/decorators/base.py,sha256=eYep1xHXscdxIGisXgXwSGbkCrdpqqT_2s5BMP07ldg,30646
 ioa_observe/sdk/decorators/helpers.py,sha256=I9HXMBivkZpGDtPe9Ad_UU35p_m_wEPate4r_fU0oOA,2705
 ioa_observe/sdk/decorators/util.py,sha256=IebvH9gwZN1en3LblYJUh4bAV2STl6xmp8WpZzBDH2g,30068
 ioa_observe/sdk/instrumentations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -34,15 +34,15 @@ ioa_observe/sdk/tracing/content_allow_list.py,sha256=1fAkpIwUQ7vDwCTkIVrqeltWQtr
 ioa_observe/sdk/tracing/context_manager.py,sha256=O0JEXYa9h8anhW78R8KKBuqS0j4by1E1KXxNIMPnLr8,400
 ioa_observe/sdk/tracing/context_utils.py,sha256=-sYS9vPLI87davV9ubneq5xqbV583CC_c0SmOQS1TAs,2933
 ioa_observe/sdk/tracing/manual.py,sha256=KS6WN-zw9vAACzXYmnMoJm9d1fenYMfvzeK1GrGDPDE,1937
-ioa_observe/sdk/tracing/tracing.py,sha256=lwmatGwviSrf-hlBxRyxa0kv9AQdP302R3duBgNKsSU,47127
+ioa_observe/sdk/tracing/tracing.py,sha256=MgKclJe_IRBh_EKTKwVJkrgrsBX1NUU05eRNcfYnTzE,47380
 ioa_observe/sdk/tracing/transform_span.py,sha256=XTApi_gJxum7ynvhtcoCfDyK8VVOj91Q1DT6hAeLHA8,8419
 ioa_observe/sdk/utils/__init__.py,sha256=UPn182U-UblF_XwXaFpx8F-TmQTbm1LYf9y89uSp5Hw,704
 ioa_observe/sdk/utils/const.py,sha256=d67dUTAH9UpWvUV9GLBUqn1Sc2knJ55dy-e6YoLrvSo,1318
 ioa_observe/sdk/utils/in_memory_span_exporter.py,sha256=H_4TRaThMO1H6vUQ0OpQvzJk_fZH0OOsRAM1iZQXsR8,2112
 ioa_observe/sdk/utils/json_encoder.py,sha256=g4NQ0tTqgWssY6I1D7r4zo0G6PiUo61jhofTAw5-jno,639
 ioa_observe/sdk/utils/package_check.py,sha256=1d1MjxhwoEZIx9dumirT2pRsEWgn-m-SI4npDeEalew,576
-ioa_observe_sdk-1.0.18.dist-info/licenses/LICENSE.md,sha256=55VjUfgjWOS4vv3Cf55gfq-RxjPgRIO2vlgYPUuC5lA,11362
-ioa_observe_sdk-1.0.18.dist-info/METADATA,sha256=o4RDNJkldVU6TtdOHM4ISncUc61APJ_2KITjsmlqKAc,7027
-ioa_observe_sdk-1.0.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ioa_observe_sdk-1.0.18.dist-info/top_level.txt,sha256=Yt-6Y1olZEDqCs2REeqI30WjYx0pLGQSVqzYmDd67N8,12
-ioa_observe_sdk-1.0.18.dist-info/RECORD,,
+ioa_observe_sdk-1.0.19.dist-info/licenses/LICENSE.md,sha256=55VjUfgjWOS4vv3Cf55gfq-RxjPgRIO2vlgYPUuC5lA,11362
+ioa_observe_sdk-1.0.19.dist-info/METADATA,sha256=ynlTiy3cP00dX-e_PjBgTP3wB8FoT20ztUbOLeSKQEg,7027
+ioa_observe_sdk-1.0.19.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ioa_observe_sdk-1.0.19.dist-info/top_level.txt,sha256=Yt-6Y1olZEDqCs2REeqI30WjYx0pLGQSVqzYmDd67N8,12
+ioa_observe_sdk-1.0.19.dist-info/RECORD,,