crewplus 0.2.63__tar.gz → 0.2.64__tar.gz

{crewplus-0.2.63 → crewplus-0.2.64}/PKG-INFO

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

crewplus-0.2.64/crewplus/callbacks/run_id_handler.py

@@ -0,0 +1,73 @@
+# File: crewplus/callbacks/run_id_handler.py
+
+from typing import Any, Dict
+from uuid import UUID
+import logging
+
+# Langfuse imports with graceful fallback
+try:
+    from langfuse.langchain import CallbackHandler as LangfuseCallbackHandler
+    from .async_langfuse_handler import AsyncLangfuseCallbackHandler
+    from langfuse import get_client
+    LANGFUSE_AVAILABLE = True
+except ImportError:
+    LANGFUSE_AVAILABLE = False
+    # Define dummy base classes if langfuse is not available to avoid runtime errors
+    class LangfuseCallbackHandler: pass
+    class AsyncLangfuseCallbackHandler: pass
+    get_client = None
+
+# --- Custom Callback Handlers to capture the run_id ---
+if LANGFUSE_AVAILABLE:
+    class RunIdCallbackHandler(LangfuseCallbackHandler):
+        """
+        A custom synchronous callback handler that captures the Langchain run_id
+        and adds it to the Langfuse trace metadata.
+        """
+        def on_chain_start(
+            self,
+            serialized: Dict[str, Any],
+            inputs: Dict[str, Any],
+            run_id: UUID,
+            **kwargs: Any,
+        ) -> Any:
+            # First, let the base handler do its work, which includes creating the trace
+            run = super().on_chain_start(serialized, inputs, run_id, **kwargs)
+
+            try:
+                # Then, update the created trace with the run_id as metadata
+                langfuse_client = get_client()
+                langfuse_client.update_current_trace(
+                    metadata={"langchain_run_id": str(run_id)}
+                )
+            except Exception as e:
+                logging.getLogger(__name__).warning(f"Failed to add run_id to Langfuse trace: {e}")
+
+            return run
+
+    class AsyncRunIdCallbackHandler(AsyncLangfuseCallbackHandler):
+        """
+        A custom asynchronous callback handler that captures the Langchain run_id
+        and adds it to the Langfuse trace metadata.
+        """
+        async def on_chain_start(
+            self,
+            serialized: Dict[str, Any],
+            inputs: Dict[str, Any],
+            run_id: UUID,
+            **kwargs: Any,
+        ) -> Any:
+            # First, let the base handler do its work, which includes creating the trace
+            run = await super().on_chain_start(serialized, inputs, run_id, **kwargs)
+
+            try:
+                # The get_client() is thread-safe and can be used in async contexts
+                # for this type of single, atomic update.
+                langfuse_client = get_client()
+                langfuse_client.update_current_trace(
+                    metadata={"langchain_run_id": str(run_id)}
+                )
+            except Exception as e:
+                logging.getLogger(__name__).warning(f"Failed to add run_id to async Langfuse trace: {e}")
+
+            return run

{crewplus-0.2.63 → crewplus-0.2.64}/crewplus/services/tracing_manager.py

@@ -9,11 +9,16 @@ import logging
 try:
     from langfuse.langchain import CallbackHandler as LangfuseCallbackHandler
     from ..callbacks.async_langfuse_handler import AsyncLangfuseCallbackHandler
+    # Import the new custom handlers
+    from ..callbacks.run_id_handler import RunIdCallbackHandler, AsyncRunIdCallbackHandler
     LANGFUSE_AVAILABLE = True
 except ImportError:
     LANGFUSE_AVAILABLE = False
     LangfuseCallbackHandler = None
     AsyncLangfuseCallbackHandler = None
+    # Define dummy classes for the custom handlers to prevent errors if langfuse is not installed
+    class RunIdCallbackHandler: pass
+    class AsyncRunIdCallbackHandler: pass
 
 class TracingContext(Protocol):
     """
@@ -98,13 +103,14 @@ class TracingManager:
         
         if enable_langfuse:
             try:
-                # Create both sync and async handlers. We'll pick one at runtime.
-                sync_handler = LangfuseCallbackHandler()
-                self._sync_handlers.append(sync_handler)
+                # Create and add both the standard and the run_id-capturing handlers.
+                # The standard handler creates the trace, and the custom one updates it.
+                self._sync_handlers.append(LangfuseCallbackHandler())
+                self._sync_handlers.append(RunIdCallbackHandler())
 
                 if AsyncLangfuseCallbackHandler:
-                    async_handler = AsyncLangfuseCallbackHandler()
-                    self._async_handlers.append(async_handler)
+                    self._async_handlers.append(AsyncLangfuseCallbackHandler())
+                    self._async_handlers.append(AsyncRunIdCallbackHandler())
                 
                 self.context.logger.info(f"Langfuse tracing enabled for {self.context.get_model_identifier()}")
             except Exception as e:

{crewplus-0.2.63 → crewplus-0.2.64}/pyproject.toml

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