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

ioa_observe/sdk/client/client.py

@@ -2,6 +2,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import sys
+import threading
 
 from .http import HTTPClient
 from ioa_observe.sdk.version import __version__
@@ -51,21 +52,29 @@ class Client:
 class KVStore(object):
     """
     Key-Value Store for storing key-value pairs (Singleton).
+    Thread-safe implementation for concurrent access.
     """
 
     _instance = None
+    _instance_lock = threading.Lock()
 
     def __new__(cls):
         if cls._instance is None:
-            cls._instance = super(KVStore, cls).__new__(cls)
-            cls._instance.store = {}
+            with cls._instance_lock:
+                # Double-check locking pattern
+                if cls._instance is None:
+                    cls._instance = super(KVStore, cls).__new__(cls)
+                    cls._instance.store = {}
+                    cls._instance._lock = threading.Lock()
         return cls._instance
 
     def set(self, key: str, value: str):
-        self.store[key] = value
+        with self._lock:
+            self.store[key] = value
 
     def get(self, key: str):
-        return self.store.get(key)
+        with self._lock:
+            return self.store.get(key)
 
 
 kv_store = KVStore()

ioa_observe/sdk/tracing/tracing.py

@@ -67,6 +67,7 @@ APP_NAME = ""
 
 SESSION_IDLE_TIMEOUT_SECONDS = 300  # e.g. 5 minutes
 SESSION_WATCHER_INTERVAL_SECONDS = 30
+MAX_PROCESSED_SPANS_SIZE = 100000  # Maximum size before cleanup
 
 
 def determine_reliability_score(span):
@@ -142,8 +143,10 @@ class TracerWrapper(object):
             obj.__image_uploader = image_uploader
             # {(agent_name): [success_count, total_count]}
             obj._agent_execution_counts = {}
+            obj._agent_execution_counts_lock = threading.Lock()
             # Track spans that have been processed to avoid duplicates
             obj._processed_spans = set()
+            obj._processed_spans_lock = threading.Lock()
             TracerWrapper.app_name = TracerWrapper.resource_attributes.get(
                 "service.name", "observe"
             )
@@ -306,6 +309,11 @@ class TracerWrapper(object):
                         expired[session_id] = last_ts
                         del self._session_last_activity[session_id]
 
+            # Periodic cleanup of processed spans to prevent unbounded memory growth
+            with self._processed_spans_lock:
+                if len(self._processed_spans) > MAX_PROCESSED_SPANS_SIZE:
+                    self._processed_spans.clear()
+
             if not expired:
                 continue
 
@@ -313,7 +321,6 @@ class TracerWrapper(object):
 
             # Iterate over a snapshot and do *not* modify `expired` in the loop
             for session_id, _last_ts in list(expired.items()):
-                print("ending session", session_id)
                 with tracer.start_as_current_span("session.end") as span:
                     span.set_attribute("session.id", session_id)
                     workflow_name = get_value("workflow_name")
@@ -420,16 +427,19 @@ class TracerWrapper(object):
         # Added for avoid duplicate on_ending with manual triggers
         # from decorators (@tool, @workflow) in base.py
         span_id = span.context.span_id
-        if span_id in self._processed_spans:
-            # This span was already processed, skip to avoid duplicates
-            return
+        with self._processed_spans_lock:
+            if span_id in self._processed_spans:
+                # This span was already processed, skip to avoid duplicates
+                return
 
-        # Mark this span as processed
-        self._processed_spans.add(span_id)
+            # Mark this span as processed
+            self._processed_spans.add(span_id)
 
         # update last activity per session
+        # Skip session.end spans to avoid infinite loop - these are cleanup spans
+        # that should not re-register the session as active
         session_id = span.attributes.get("session.id")
-        if session_id:
+        if session_id and span.name != "session.end":
             with self._session_lock:
                 self._session_last_activity[session_id] = time.time()
 
@@ -567,21 +577,23 @@ class TracerWrapper(object):
         return self.__tracer_provider.get_tracer(TRACER_NAME)
 
     def record_agent_execution(self, agent_name: str, success: bool):
-        counts = self._agent_execution_counts.setdefault(agent_name, [0, 0])
-        if success:
-            counts[0] += 1  # success count
-        counts[1] += 1  # total count
+        with self._agent_execution_counts_lock:
+            counts = self._agent_execution_counts.setdefault(agent_name, [0, 0])
+            if success:
+                counts[0] += 1  # success count
+            counts[1] += 1  # total count
 
     def _observe_agent_execution_success_rate(self, observer):
         measurements = []
-        for agent_name, (
-            success_count,
-            total_count,
-        ) in self._agent_execution_counts.items():
-            rate = (success_count / total_count) if total_count > 0 else 0.0
-            measurements.append(
-                Observation(value=rate, attributes={"agent_name": agent_name})
-            )
+        with self._agent_execution_counts_lock:
+            for agent_name, (
+                success_count,
+                total_count,
+            ) in self._agent_execution_counts.items():
+                rate = (success_count / total_count) if total_count > 0 else 0.0
+                measurements.append(
+                    Observation(value=rate, attributes={"agent_name": agent_name})
+                )
         return measurements
 
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ioa-observe-sdk
-Version: 1.0.27
+Version: 1.0.28
 Summary: IOA Observability SDK
 License-Expression: Apache-2.0
 Requires-Python: >=3.10

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

@@ -4,7 +4,7 @@ ioa_observe/sdk/instruments.py,sha256=cA5Yq1BYFovMrYUNYQXua-JXsMtMOa_YOn6yiJZNwL
 ioa_observe/sdk/telemetry.py,sha256=6wwaOYhZMjAZ6dXDdBA2LUWo3LLptTcy93BJqDdbqBM,3103
 ioa_observe/sdk/version.py,sha256=oriNAY8huVDPw5N_rv5F_PehFrcGo37FSGBCfZCM81M,121
 ioa_observe/sdk/client/__init__.py,sha256=V4Rt-Z1EHlM12Lx3hGd0Ew70V1JKAQZXNb9ABtdWHEI,224
-ioa_observe/sdk/client/client.py,sha256=6TVOo_E1ulE3WO_CYG7oPgeucs-qegOA09uTO3yQiyk,2112
+ioa_observe/sdk/client/client.py,sha256=i4ZlYrhFcXfg5BTROLAKxhfAAsb-7RS0ch9gucsWXd0,2480
 ioa_observe/sdk/client/http.py,sha256=LdLYSQPFIhKN5BTB-N78jLO7ITl7jGjA0-qpewEIvO4,1724
 ioa_observe/sdk/config/__init__.py,sha256=8aVNaw0yRNLFPxlf97iOZLlJVcV81ivSDnudH2m1OIo,572
 ioa_observe/sdk/connectors/__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=7eJAEqaAEBZ2gjIXrdb7SyUidA8P10GnpBLzr8InLKI,50007
+ioa_observe/sdk/tracing/tracing.py,sha256=ywvT5Z1DcaLYZQvUP30YKz6m6v76Idlvb2qszupcxS4,50774
 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.27.dist-info/licenses/LICENSE.md,sha256=55VjUfgjWOS4vv3Cf55gfq-RxjPgRIO2vlgYPUuC5lA,11362
-ioa_observe_sdk-1.0.27.dist-info/METADATA,sha256=OrhfkqzR0Qj2qwSGXtSmL9rYjS1OMk-IYmTU5wHMHdE,7997
-ioa_observe_sdk-1.0.27.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-ioa_observe_sdk-1.0.27.dist-info/top_level.txt,sha256=Yt-6Y1olZEDqCs2REeqI30WjYx0pLGQSVqzYmDd67N8,12
-ioa_observe_sdk-1.0.27.dist-info/RECORD,,
+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,,