crewplus 0.2.65__py3-none-any.whl → 0.2.66__py3-none-any.whl

crewplus/callbacks/run_id_handler.py

@@ -22,8 +22,16 @@ 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.
+        and adds it to the Langfuse trace metadata. It also stores the mapping
+        between the run_id and the generated trace_id.
         """
+        def __init__(self, *args, **kwargs):
+            super().__init__(*args, **kwargs)
+            self.run_trace_map = {}
+            # Use a named logger for better context
+            self.logger = logging.getLogger(__name__)
+            self.logger.info("RunIdCallbackHandler initialized.")
+
         def on_chain_start(
             self,
             serialized: Dict[str, Any],
@@ -31,17 +39,28 @@ if LANGFUSE_AVAILABLE:
             run_id: UUID,
             **kwargs: Any,
         ) -> Any:
+            self.logger.debug(f"on_chain_start triggered for run_id: {run_id}")
             # First, let the base handler do its work, which includes creating the trace
             run = super().on_chain_start(serialized, inputs, run_id, **kwargs)
+            self.logger.debug(f"Base handler's on_chain_start completed. Current trace_id: {self.last_trace_id}")
 
             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)}
-                )
+                # The base handler sets `self.last_trace_id` when a trace is created
+                if self.last_trace_id:
+                    self.run_trace_map[str(run_id)] = self.last_trace_id
+                    self.logger.info(f"Mapping run_id '{run_id}' to trace_id '{self.last_trace_id}'.")
+
+                    # 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)}
+                    )
+                    self.logger.info(f"Successfully updated trace '{self.last_trace_id}' with metadata for run_id '{run_id}'.")
+                else:
+                    self.logger.warning(f"Could not find 'last_trace_id' for run_id '{run_id}'. Metadata will not be updated.")
+
             except Exception as e:
-                logging.getLogger(__name__).warning(f"Failed to add run_id to Langfuse trace: {e}")
+                self.logger.error(f"Failed to process run_id '{run_id}' in RunIdCallbackHandler: {e}", exc_info=True)
 
             return run
 
@@ -50,6 +69,12 @@ if LANGFUSE_AVAILABLE:
         A custom asynchronous callback handler that captures the Langchain run_id
         and adds it to the Langfuse trace metadata.
         """
+        def __init__(self, *args, **kwargs):
+            super().__init__(*args, **kwargs)
+            self.run_trace_map = {}
+            self.logger = logging.getLogger(__name__)
+            self.logger.info("AsyncRunIdCallbackHandler initialized.")
+
         async def on_chain_start(
             self,
             serialized: Dict[str, Any],
@@ -57,17 +82,25 @@ if LANGFUSE_AVAILABLE:
             run_id: UUID,
             **kwargs: Any,
         ) -> Any:
-            # First, let the base handler do its work, which includes creating the trace
+            self.logger.debug(f"Async on_chain_start triggered for run_id: {run_id}")
+            # First, let the base handler do its work
             run = await super().on_chain_start(serialized, inputs, run_id, **kwargs)
-
+            self.logger.debug(f"Async base handler's on_chain_start completed. Current trace_id: {self.last_trace_id}")
+            
             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)}
-                )
+                if self.last_trace_id:
+                    self.run_trace_map[str(run_id)] = self.last_trace_id
+                    self.logger.info(f"Async: Mapping run_id '{run_id}' to trace_id '{self.last_trace_id}'.")
+
+                    langfuse_client = get_client()
+                    langfuse_client.update_current_trace(
+                        metadata={"langchain_run_id": str(run_id)}
+                    )
+                    self.logger.info(f"Async: Successfully updated trace '{self.last_trace_id}' with metadata for run_id '{run_id}'.")
+                else:
+                    self.logger.warning(f"Async: Could not find 'last_trace_id' for run_id '{run_id}'. Metadata will not be updated.")
+
             except Exception as e:
-                logging.getLogger(__name__).warning(f"Failed to add run_id to async Langfuse trace: {e}")
+                self.logger.error(f"Async: Failed to process run_id '{run_id}' in AsyncRunIdCallbackHandler: {e}", exc_info=True)
 
             return run

{crewplus-0.2.65.dist-info → crewplus-0.2.66.dist-info}/METADATA

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

{crewplus-0.2.65.dist-info → crewplus-0.2.66.dist-info}/RECORD

@@ -1,11 +1,11 @@
-crewplus-0.2.65.dist-info/METADATA,sha256=2wszVd5gRVaS18s9lGbfkKw12xDFLA-dCiZ2aBj4Ei4,5424
-crewplus-0.2.65.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-crewplus-0.2.65.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
-crewplus-0.2.65.dist-info/licenses/LICENSE,sha256=2_NHSHRTKB_cTcT_GXgcenOCtIZku8j343mOgAguTfc,1087
+crewplus-0.2.66.dist-info/METADATA,sha256=nfG_3A6490ii2Hmp9cfDZgWd_aIOebzVa7cyhAsCDao,5424
+crewplus-0.2.66.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+crewplus-0.2.66.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+crewplus-0.2.66.dist-info/licenses/LICENSE,sha256=2_NHSHRTKB_cTcT_GXgcenOCtIZku8j343mOgAguTfc,1087
 crewplus/__init__.py,sha256=m46HkZL1Y4toD619NL47Sn2Qe084WFFSFD7e6VoYKZc,284
 crewplus/callbacks/__init__.py,sha256=YG7ieeb91qEjp1zF0-inEN7mjZ7yT_D2yzdWFT8Z1Ws,63
 crewplus/callbacks/async_langfuse_handler.py,sha256=A4uFeLpvOUdc58M7sZoE65_C1V98u0QCvx5jUquM0pM,7006
-crewplus/callbacks/run_id_handler.py,sha256=k3HxARO3oWMELoP2ZjvF4x5xHWbICZ_DmKQ71fkIWd0,2887
+crewplus/callbacks/run_id_handler.py,sha256=UurImqJYJg_EnOKB8nzBmG377KL4wI7fl8lz-98_VXY,4964
 crewplus/services/__init__.py,sha256=V1CG8b2NOmRzNgQH7BPl4KVxWSYJH5vfEsW1wVErKNE,375
 crewplus/services/azure_chat_model.py,sha256=iWzJ2GQFSNmwJx-2O5_xKPSB6VVc-7T6bcfFI8_WezA,5521
 crewplus/services/gemini_chat_model.py,sha256=DYqz01H2TIHiCDQesSozVfOsMigno6QGwOtIweg7UHk,40103
@@ -23,4 +23,4 @@ docs/GeminiChatModel.md,sha256=zZYyl6RmjZTUsKxxMiC9O4yV70MC4TD-IGUmWhIDBKA,8677
 docs/ModelLoadBalancer.md,sha256=aGHES1dcXPz4c7Y8kB5-vsCNJjriH2SWmjBkSGoYKiI,4398
 docs/VDBService.md,sha256=Dw286Rrf_fsi13jyD3Bo4Sy7nZ_G7tYm7d8MZ2j9hxk,9375
 docs/index.md,sha256=3tlc15uR8lzFNM5WjdoZLw0Y9o1P1gwgbEnOdIBspqc,1643
-crewplus-0.2.65.dist-info/RECORD,,
+crewplus-0.2.66.dist-info/RECORD,,