ioa-observe-sdk 1.0.28__py3-none-any.whl → 1.0.30__py3-none-any.whl

ioa_observe/sdk/tracing/tracing.py

@@ -30,8 +30,15 @@ from opentelemetry.sdk.trace.export import (
     SimpleSpanProcessor,
     BatchSpanProcessor,
 )
-from opentelemetry.trace import get_tracer_provider, ProxyTracerProvider
+from opentelemetry.trace import (
+    get_tracer_provider,
+    ProxyTracerProvider,
+    SpanContext,
+    TraceFlags,
+    NonRecordingSpan,
+)
 from opentelemetry.context import get_value, attach, set_value
+from opentelemetry.trace import set_span_in_context
 from opentelemetry.instrumentation.threading import ThreadingInstrumentor
 from opentelemetry.metrics import get_meter
 from ioa_observe.sdk.metrics.agents.agent_connections import connection_reliability
@@ -137,9 +144,12 @@ class TracerWrapper(object):
             if not TracerWrapper.endpoint:
                 return obj
 
-            # session activity tracking
-            obj._session_last_activity: dict[str, float] = {}
+            # session activity tracking: {session_id: (last_activity_time, trace_id)}
+            obj._session_last_activity: dict[str, tuple[float, int]] = {}
             obj._session_lock = threading.Lock()
+            # Track sessions that have already been ended to prevent duplicates
+            obj._ended_sessions: set[str] = set()
+            obj._ended_sessions_lock = threading.Lock()
             obj.__image_uploader = image_uploader
             # {(agent_name): [success_count, total_count]}
             obj._agent_execution_counts = {}
@@ -300,13 +310,15 @@ class TracerWrapper(object):
         while True:
             time.sleep(SESSION_WATCHER_INTERVAL_SECONDS)
             now = time.time()
-            expired: dict[str, float] = {}
+            expired: dict[str, tuple[float, int]] = {}
 
             # Find idle sessions and remove them from _session_last_activity
             with self._session_lock:
-                for session_id, last_ts in list(self._session_last_activity.items()):
+                for session_id, (last_ts, trace_id) in list(
+                    self._session_last_activity.items()
+                ):
                     if now - last_ts > SESSION_IDLE_TIMEOUT_SECONDS:
-                        expired[session_id] = last_ts
+                        expired[session_id] = (last_ts, trace_id)
                         del self._session_last_activity[session_id]
 
             # Periodic cleanup of processed spans to prevent unbounded memory growth
@@ -314,19 +326,42 @@ class TracerWrapper(object):
                 if len(self._processed_spans) > MAX_PROCESSED_SPANS_SIZE:
                     self._processed_spans.clear()
 
+            # Periodic cleanup of ended sessions to prevent unbounded memory growth
+            with self._ended_sessions_lock:
+                if len(self._ended_sessions) > MAX_PROCESSED_SPANS_SIZE:
+                    self._ended_sessions.clear()
+
             if not expired:
                 continue
 
             tracer = self.get_tracer()
 
             # Iterate over a snapshot and do *not* modify `expired` in the loop
-            for session_id, _last_ts in list(expired.items()):
-                with tracer.start_as_current_span("session.end") as span:
+            for session_id, (last_ts, trace_id) in list(expired.items()):
+                # Check if this session has already been ended to prevent duplicates
+                with self._ended_sessions_lock:
+                    if session_id in self._ended_sessions:
+                        continue
+                    self._ended_sessions.add(session_id)
+
+                # Create a parent context from the stored trace_id to keep session.end under the same trace
+                parent_span_context = SpanContext(
+                    trace_id=trace_id,
+                    span_id=0,  # Use 0 as we don't have a specific parent span
+                    is_remote=False,
+                    trace_flags=TraceFlags(TraceFlags.SAMPLED),
+                )
+                parent_span = NonRecordingSpan(parent_span_context)
+                parent_context = set_span_in_context(parent_span)
+
+                with tracer.start_as_current_span(
+                    "session.end", context=parent_context
+                ) as span:
                     span.set_attribute("session.id", session_id)
                     workflow_name = get_value("workflow_name")
                     if workflow_name:
                         span.set_attribute(OBSERVE_WORKFLOW_NAME, workflow_name)
-                    span.set_attribute("session.ended_at", _last_ts)
+                    span.set_attribute("session.ended_at", last_ts)
 
             # ensure end spans are exported reasonably fast
             self.flush()
@@ -337,11 +372,29 @@ class TracerWrapper(object):
         tracer = self.get_tracer()
 
         with self._session_lock:
-            remaining_ids = list(self._session_last_activity.keys())
+            remaining_sessions = dict(self._session_last_activity)
             self._session_last_activity.clear()
 
-        for session_id in remaining_ids:
-            with tracer.start_as_current_span("session.end") as span:
+        for session_id, (_, trace_id) in remaining_sessions.items():
+            # Check if this session has already been ended to prevent duplicates
+            with self._ended_sessions_lock:
+                if session_id in self._ended_sessions:
+                    continue
+                self._ended_sessions.add(session_id)
+
+            # Create a parent context from the stored trace_id to keep session.end under the same trace
+            parent_span_context = SpanContext(
+                trace_id=trace_id,
+                span_id=0,  # Use 0 as we don't have a specific parent span
+                is_remote=False,
+                trace_flags=TraceFlags(TraceFlags.SAMPLED),
+            )
+            parent_span = NonRecordingSpan(parent_span_context)
+            parent_context = set_span_in_context(parent_span)
+
+            with tracer.start_as_current_span(
+                "session.end", context=parent_context
+            ) as span:
                 span.set_attribute("session.id", session_id)
                 workflow_name = get_value("workflow_name")
                 if workflow_name:
@@ -441,7 +494,11 @@ class TracerWrapper(object):
         session_id = span.attributes.get("session.id")
         if session_id and span.name != "session.end":
             with self._session_lock:
-                self._session_last_activity[session_id] = time.time()
+                # Store both the last activity time and the trace_id for this session
+                self._session_last_activity[session_id] = (
+                    time.time(),
+                    span.context.trace_id,
+                )
 
         determine_reliability_score(span)
         # start_time = span.attributes.get("ioa_start_time")

{ioa_observe_sdk-1.0.28.dist-info → ioa_observe_sdk-1.0.30.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ioa-observe-sdk
-Version: 1.0.28
+Version: 1.0.30
 Summary: IOA Observability SDK
 License-Expression: Apache-2.0
 Requires-Python: >=3.10
@@ -8,37 +8,37 @@ Description-Content-Type: text/markdown
 License-File: LICENSE.md
 Requires-Dist: colorama==0.4.6
 Requires-Dist: requests>=2.32.3
-Requires-Dist: opentelemetry-api==1.37.0
+Requires-Dist: opentelemetry-api==1.39.1
 Requires-Dist: opentelemetry-distro
-Requires-Dist: opentelemetry-exporter-otlp==1.37.0
-Requires-Dist: opentelemetry-exporter-otlp-proto-common==1.37.0
-Requires-Dist: opentelemetry-exporter-otlp-proto-grpc==1.37.0
-Requires-Dist: opentelemetry-exporter-otlp-proto-http==1.37.0
+Requires-Dist: opentelemetry-exporter-otlp==1.39.1
+Requires-Dist: opentelemetry-exporter-otlp-proto-common==1.39.1
+Requires-Dist: opentelemetry-exporter-otlp-proto-grpc==1.39.1
+Requires-Dist: opentelemetry-exporter-otlp-proto-http==1.39.1
 Requires-Dist: opentelemetry-instrumentation
-Requires-Dist: opentelemetry-instrumentation-logging==0.58b0
-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.58b0
-Requires-Dist: opentelemetry-instrumentation-sagemaker==0.43.1
-Requires-Dist: opentelemetry-instrumentation-threading==0.58b0
-Requires-Dist: opentelemetry-instrumentation-together==0.43.1
-Requires-Dist: opentelemetry-instrumentation-transformers==0.43.1
-Requires-Dist: opentelemetry-instrumentation-urllib3==0.58b0
-Requires-Dist: opentelemetry-instrumentation-vertexai==0.43.1
-Requires-Dist: opentelemetry-proto==1.37.0
-Requires-Dist: opentelemetry-sdk==1.37.0
-Requires-Dist: opentelemetry-semantic-conventions==0.58b0
+Requires-Dist: opentelemetry-instrumentation-logging==0.60b1
+Requires-Dist: opentelemetry-instrumentation-openai==0.52.1
+Requires-Dist: opentelemetry-instrumentation-llamaindex==0.52.1
+Requires-Dist: opentelemetry-instrumentation-ollama==0.52.1
+Requires-Dist: opentelemetry-instrumentation-anthropic==0.52.1
+Requires-Dist: opentelemetry-instrumentation-langchain==0.52.1
+Requires-Dist: opentelemetry-instrumentation-bedrock==0.52.1
+Requires-Dist: opentelemetry-instrumentation-cohere==0.52.1
+Requires-Dist: opentelemetry-instrumentation-crewai==0.52.1
+Requires-Dist: opentelemetry-instrumentation-google-generativeai==0.52.1
+Requires-Dist: opentelemetry-instrumentation-groq==0.52.1
+Requires-Dist: opentelemetry-instrumentation-mistralai==0.52.1
+Requires-Dist: opentelemetry-instrumentation-requests==0.60b1
+Requires-Dist: opentelemetry-instrumentation-sagemaker==0.52.1
+Requires-Dist: opentelemetry-instrumentation-threading==0.60b1
+Requires-Dist: opentelemetry-instrumentation-together==0.52.1
+Requires-Dist: opentelemetry-instrumentation-transformers==0.52.1
+Requires-Dist: opentelemetry-instrumentation-urllib3==0.60b1
+Requires-Dist: opentelemetry-instrumentation-vertexai==0.52.1
+Requires-Dist: opentelemetry-proto==1.39.1
+Requires-Dist: opentelemetry-sdk==1.39.1
+Requires-Dist: opentelemetry-semantic-conventions==0.60b1
 Requires-Dist: opentelemetry-semantic-conventions-ai>=0.4.11
-Requires-Dist: opentelemetry-util-http==0.58b0
+Requires-Dist: opentelemetry-util-http==0.60b1
 Requires-Dist: langgraph>=0.3.2
 Requires-Dist: langchain>=0.3.19
 Requires-Dist: langchain-openai>=0.3.8

{ioa_observe_sdk-1.0.28.dist-info → ioa_observe_sdk-1.0.30.dist-info}/RECORD

@@ -14,10 +14,10 @@ ioa_observe/sdk/decorators/base.py,sha256=gezoLMLOo-CWsxzsXbQMsj5HxmchUEmlhTspbq
 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
-ioa_observe/sdk/instrumentations/a2a.py,sha256=LHvvQUluPtmvbRNuoqf-EkWlZJa_NicerTpM3t1UZws,15299
+ioa_observe/sdk/instrumentations/a2a.py,sha256=smPMHhFONi9B7r2qVHe-YWoFctjHWehO_MR0kY_DgVI,27897
 ioa_observe/sdk/instrumentations/mcp.py,sha256=UG82fWitFT-K-ecnkTKZa-uFTDAs-j2HymvhUwCBRLY,18721
 ioa_observe/sdk/instrumentations/nats.py,sha256=MEYCFqHyRK5HtkjrbGVV5qIWplZ1ZjJjFCB8vtQl6qE,14736
-ioa_observe/sdk/instrumentations/slim.py,sha256=c_Q1cb6pdjZLOw1MbclUf1eRFrc_sUEe9pQJvAyoD1Y,45712
+ioa_observe/sdk/instrumentations/slim.py,sha256=CE0wUGlRyxbAgP9oI69kGQ9M2yu9HmufrQNua4L0jTE,19510
 ioa_observe/sdk/logging/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ioa_observe/sdk/logging/logging.py,sha256=HZxW9s8Due7jgiNkdI38cIjv5rC9D-Flta3RQMOnpow,2891
 ioa_observe/sdk/metrics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -35,15 +35,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=ywvT5Z1DcaLYZQvUP30YKz6m6v76Idlvb2qszupcxS4,50774
+ioa_observe/sdk/tracing/tracing.py,sha256=pGiIa1S3562QRlTG5mANtU94wkFjKspnX21N_bJ7SfY,53359
 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.28.dist-info/licenses/LICENSE.md,sha256=55VjUfgjWOS4vv3Cf55gfq-RxjPgRIO2vlgYPUuC5lA,11362
-ioa_observe_sdk-1.0.28.dist-info/METADATA,sha256=xEbYDkX8xS0u72TgyF-DvEM9R7BxEA8wp3khv4MWHAE,7997
-ioa_observe_sdk-1.0.28.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-ioa_observe_sdk-1.0.28.dist-info/top_level.txt,sha256=Yt-6Y1olZEDqCs2REeqI30WjYx0pLGQSVqzYmDd67N8,12
-ioa_observe_sdk-1.0.28.dist-info/RECORD,,
+ioa_observe_sdk-1.0.30.dist-info/licenses/LICENSE.md,sha256=55VjUfgjWOS4vv3Cf55gfq-RxjPgRIO2vlgYPUuC5lA,11362
+ioa_observe_sdk-1.0.30.dist-info/METADATA,sha256=KxVA9-iwTZF37H5BMdAeRxk_HeJ_uSJXRzVcsbeMc7c,7997
+ioa_observe_sdk-1.0.30.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+ioa_observe_sdk-1.0.30.dist-info/top_level.txt,sha256=Yt-6Y1olZEDqCs2REeqI30WjYx0pLGQSVqzYmDd67N8,12
+ioa_observe_sdk-1.0.30.dist-info/RECORD,,