crewplus 0.2.26__tar.gz → 0.2.27__tar.gz

{crewplus-0.2.26 → crewplus-0.2.27}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: crewplus
-Version: 0.2.26
+Version: 0.2.27
 Summary: Base services for CrewPlus AI applications
 Author-Email: Tim Liu <tim@opsmateai.com>
 License: MIT

{crewplus-0.2.26 → crewplus-0.2.27}/crewplus/services/tracing_manager.py

@@ -104,9 +104,10 @@ class TracingManager:
         """
         Adds all registered tracing handlers to the request configuration.
         
-        This method is idempotent; it will not add a handler if one of the
-        same type is already present in the configuration's callback list. It
-        also respects a `tracing_disabled` flag in the metadata.
+        This method is robust and handles three scenarios for the 'callbacks' key:
+        1. A list of callbacks.
+        2. A LangChain CallbackManager instance.
+        3. None or a missing key.
         
         Args:
             config: The request configuration dictionary from a LangChain call.
@@ -121,16 +122,27 @@ class TracingManager:
         if not self._handlers or config.get("metadata", {}).get("tracing_disabled"):
             return config
         
-        # Use an empty list as a safe default if 'callbacks' is missing or None.
-        callbacks = config.get("callbacks") or []
-        new_callbacks = list(callbacks)
+        callbacks = config.get("callbacks")
+        
+        # Case 1: The 'callbacks' key holds a CallbackManager instance
+        if hasattr(callbacks, 'add_handler') and hasattr(callbacks, 'handlers'):
+            for handler in self._handlers:
+                if not any(isinstance(cb, type(handler)) for cb in callbacks.handlers):
+                    callbacks.add_handler(handler, inherit=True)
+            return config # Return the original, now-mutated config
+        
+        # Case 2: The 'callbacks' key holds a list or is None
+        current_callbacks = callbacks or []
+        new_callbacks = list(current_callbacks)
         
         for handler in self._handlers:
             if not any(isinstance(cb, type(handler)) for cb in new_callbacks):
                 new_callbacks.append(handler)
         
-        # Only create a new config dictionary if callbacks were actually added.
-        if len(new_callbacks) > len(callbacks):
+        if len(new_callbacks) > len(current_callbacks):
+            # Create a new dictionary with the updated callbacks list.
+            # This is a safe operation that overwrites the existing 'callbacks'
+            # key and avoids mutating the original config object.
             return {**config, "callbacks": new_callbacks}
         
         return config

{crewplus-0.2.26 → crewplus-0.2.27}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "pdm.backend"
 
 [project]
 name = "crewplus"
-version = "0.2.26"
+version = "0.2.27"
 description = "Base services for CrewPlus AI applications"
 authors = [
     { name = "Tim Liu", email = "tim@opsmateai.com" },