crewplus 0.2.42__py3-none-any.whl → 0.2.44__py3-none-any.whl

crewplus/callbacks/async_langfuse_handler.py

@@ -50,10 +50,18 @@ class AsyncLangfuseCallbackHandler(AsyncCallbackHandler):
     async def on_llm_start(
         self, serialized: Dict[str, Any], prompts: List[str], **kwargs: Any
     ) -> None:
+        # --- DEBUGGING: Inspect the arguments from LangChain ---
+        print("--- [DEBUG] AsyncLangfuseCallbackHandler.on_llm_start ---")
+        print(f"Received prompts type: {type(prompts)}")
+        print(f"Received prompts value: {prompts!r}") # Using !r to see quotes if it's a string
+        print("----------------------------------------------------------")
+        # --- END DEBUGGING ---
+
         # WORKAROUND: LangChain's async implementation can sometimes pass a raw
         # string for prompts instead of a list. We wrap it in a list to ensure
         # compatibility with the synchronous handler.
         corrected_prompts = prompts if isinstance(prompts, list) else [prompts]
+        
         await asyncio.to_thread(
             self.sync_handler.on_llm_start, serialized, corrected_prompts, **kwargs
         )

crewplus/services/azure_chat_model.py

@@ -5,7 +5,6 @@ from typing import Any, Optional
 from langchain_openai.chat_models.azure import AzureChatOpenAI
 from pydantic import Field
 from .tracing_manager import TracingManager, TracingContext
-from ..callbacks.async_langfuse_handler import async_context
 
 class TracedAzureChatOpenAI(AzureChatOpenAI):
     """
@@ -103,13 +102,12 @@ class TracedAzureChatOpenAI(AzureChatOpenAI):
         return f"{self.__class__.__name__} (deployment='{self.deployment_name}')"
 
     def invoke(self, input, config=None, **kwargs):
-        config = self._tracing_manager.add_callbacks_to_config(config)
+        config = self._tracing_manager.add_sync_callbacks_to_config(config)
         return super().invoke(input, config=config, **kwargs)
 
     async def ainvoke(self, input, config=None, **kwargs):
-        with async_context():
-            config = self._tracing_manager.add_callbacks_to_config(config)
-            return await super().ainvoke(input, config=config, **kwargs)
+        config = self._tracing_manager.add_async_callbacks_to_config(config)
+        return await super().ainvoke(input, config=config, **kwargs)
 
     def stream(self, input, config=None, **kwargs):
         # Add stream_options to get usage data for Langfuse
@@ -117,7 +115,7 @@ class TracedAzureChatOpenAI(AzureChatOpenAI):
         stream_options["include_usage"] = True
         kwargs["stream_options"] = stream_options
         
-        config = self._tracing_manager.add_callbacks_to_config(config)
+        config = self._tracing_manager.add_sync_callbacks_to_config(config)
         yield from super().stream(input, config=config, **kwargs)
 
     async def astream(self, input, config=None, **kwargs) :
@@ -126,7 +124,6 @@ class TracedAzureChatOpenAI(AzureChatOpenAI):
         stream_options["include_usage"] = True
         kwargs["stream_options"] = stream_options
         
-        with async_context():
-            config = self._tracing_manager.add_callbacks_to_config(config)
-            async for chunk in super().astream(input, config=config, **kwargs):
-                yield chunk
+        config = self._tracing_manager.add_async_callbacks_to_config(config)
+        async for chunk in super().astream(input, config=config, **kwargs):
+            yield chunk

crewplus/services/tracing_manager.py

@@ -8,13 +8,12 @@ import logging
 # even if the langfuse library is not installed.
 try:
     from langfuse.langchain import CallbackHandler as LangfuseCallbackHandler
-    from ..callbacks.async_langfuse_handler import AsyncLangfuseCallbackHandler, in_async_context
+    from ..callbacks.async_langfuse_handler import AsyncLangfuseCallbackHandler
     LANGFUSE_AVAILABLE = True
 except ImportError:
     LANGFUSE_AVAILABLE = False
     LangfuseCallbackHandler = None
     AsyncLangfuseCallbackHandler = None
-    in_async_context = None
 
 class TracingContext(Protocol):
     """
@@ -111,42 +110,22 @@ class TracingManager:
             except Exception as e:
                 self.context.logger.warning(f"Failed to initialize Langfuse: {e}")
     
-    def add_callbacks_to_config(self, config: Optional[dict]) -> dict:
-        """
-        Adds all registered tracing handlers to the request configuration.
-        
-        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.
-            
-        Returns:
-            The updated configuration dictionary with tracing callbacks added.
-        """
+    def _add_callbacks_to_config(self, config: Optional[dict], handlers: List[Any]) -> dict:
+        """A generic helper to add a list of handlers to a config object."""
         if config is None:
             config = {}
         
-        # Decide which handlers to use based on the async context flag.
-        is_async = in_async_context.get() if in_async_context else False
-        handlers = self._async_handlers if is_async else self._sync_handlers
-
-        # Respect a global disable flag for this specific call.
         if not handlers or config.get("metadata", {}).get("tracing_disabled"):
             return config
         
         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 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
+            return config
         
-        # Case 2: The 'callbacks' key holds a list or is None
         current_callbacks = callbacks or []
         new_callbacks = list(current_callbacks)
         
@@ -161,3 +140,11 @@ class TracingManager:
             return {**config, "callbacks": new_callbacks}
         
         return config
+
+    def add_sync_callbacks_to_config(self, config: Optional[dict]) -> dict:
+        """Adds synchronous tracing handlers to the request configuration."""
+        return self._add_callbacks_to_config(config, self._sync_handlers)
+
+    def add_async_callbacks_to_config(self, config: Optional[dict]) -> dict:
+        """Adds asynchronous tracing handlers to the request configuration."""
+        return self._add_callbacks_to_config(config, self._async_handlers)

{crewplus-0.2.42.dist-info → crewplus-0.2.44.dist-info}/METADATA

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

{crewplus-0.2.42.dist-info → crewplus-0.2.44.dist-info}/RECORD

@@ -1,16 +1,16 @@
-crewplus-0.2.42.dist-info/METADATA,sha256=NMEudKtv137fmfM09RfvDIhT0y3ha-DHJuxG8GlOqzI,5362
-crewplus-0.2.42.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-crewplus-0.2.42.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
-crewplus-0.2.42.dist-info/licenses/LICENSE,sha256=2_NHSHRTKB_cTcT_GXgcenOCtIZku8j343mOgAguTfc,1087
+crewplus-0.2.44.dist-info/METADATA,sha256=KdBsfThzg0wjMW0ydjEsKw_Kp5c-nX76GQ721K4FOHY,5362
+crewplus-0.2.44.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+crewplus-0.2.44.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+crewplus-0.2.44.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=YZhwyLTQ3IcL9K9cAfNQxDNZYY365mIPuzKTM1vtn-Y,4194
+crewplus/callbacks/async_langfuse_handler.py,sha256=XnHpQ-sqcUCb6BCrE_JgldtTbaV61lH9aJnfy_IUp9Y,4614
 crewplus/services/__init__.py,sha256=V1CG8b2NOmRzNgQH7BPl4KVxWSYJH5vfEsW1wVErKNE,375
-crewplus/services/azure_chat_model.py,sha256=LXTd1g6OBV-YEaGokVNMddd1P5BU8nfV4k5tG_GcH04,5643
+crewplus/services/azure_chat_model.py,sha256=iWzJ2GQFSNmwJx-2O5_xKPSB6VVc-7T6bcfFI8_WezA,5521
 crewplus/services/gemini_chat_model.py,sha256=VsOB_st1qRmDkwLXzo-gCShhUsZHpk0V-G-ulQXGN3g,40081
 crewplus/services/init_services.py,sha256=7oZ1GmesK32EDB_DYnTzW17MEpXjXK41_U_1pmqu_m4,2183
 crewplus/services/model_load_balancer.py,sha256=Q9Gx3GrbKworU-Ytxeqp0ggHSgZ1Q6brtTk-nCl4sak,12095
-crewplus/services/tracing_manager.py,sha256=vT-7zerq6v0x-cwEBWAsB9NHdul4mPDnI60azcngTr8,7058
+crewplus/services/tracing_manager.py,sha256=0KR-F0BKYEMdADANWofFZH9D9jcWDHzDICUE7nDhzJc,6579
 crewplus/utils/__init__.py,sha256=2Gk1n5srFJQnFfBuYTxktdtKOVZyNrFcNaZKhXk35Pw,142
 crewplus/utils/schema_action.py,sha256=GDaBoVFQD1rXqrLVSMTfXYW1xcUu7eDcHsn57XBSnIg,422
 crewplus/utils/schema_document_updater.py,sha256=frvffxn2vbi71fHFPoGb9hq7gH2azmmdq17p-Fumnvg,7322
@@ -22,4 +22,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.42.dist-info/RECORD,,
+crewplus-0.2.44.dist-info/RECORD,,