dasein-core 0.2.12__tar.gz → 0.2.14__tar.gz

{dasein_core-0.2.12/src/dasein_core.egg-info → dasein_core-0.2.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dasein-core
-Version: 0.2.12
+Version: 0.2.14
 Summary: Universal memory for agentic AI. Attach a brain to any LangChain/LangGraph agent in a single line.
 Author-email: Dasein Team <support@dasein.ai>
 License: MIT

{dasein_core-0.2.12 → dasein_core-0.2.14}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dasein-core"
-version = "0.2.12"
+version = "0.2.14"
 description = "Universal memory for agentic AI. Attach a brain to any LangChain/LangGraph agent in a single line."
 readme = "README.md"
 requires-python = ">=3.8"

{dasein_core-0.2.12 → dasein_core-0.2.14}/src/dasein/api.py

@@ -887,6 +887,11 @@ class CognateProxy:
         # Wrap the agent's LLM with our trace capture wrapper
         self._wrap_agent_llm()
         
+        # CRITICAL: Update langgraph_params to use wrapped LLM for recreation
+        if self._is_langgraph and self._langgraph_params and self._wrapped_llm:
+            print(f" [DASEIN][WRAPPER] Updating langgraph_params to use wrapped LLM")
+            self._langgraph_params['model'] = self._wrapped_llm
+        
         # Inject universal dead-letter tool
         self._inject_deadletter_tool()
     
@@ -1754,14 +1759,24 @@ Follow these rules when planning your actions."""
             # 1. Search in agent
             agent_llm = self._find_llm_recursively(self._agent, max_depth=5)
             if agent_llm:
-                all_llms.append(('agent', agent_llm))
+                # CRITICAL: If this is a DaseinLLMWrapper, we need to patch the INNER LLM for pipecleaner!
+                # But we skip patching the wrapper itself to avoid double callbacks
+                if isinstance(agent_llm, DaseinLLMWrapper) and hasattr(agent_llm, '_llm'):
+                    print(f"[DASEIN][WRAPPER] Found DaseinLLMWrapper, patching inner LLM: {type(agent_llm._llm).__name__}")
+                    all_llms.append(('agent_inner', agent_llm._llm))
+                else:
+                    all_llms.append(('agent', agent_llm))
             
             # 2. Search in tools (where Summary LLM lives!)
             if hasattr(self._agent, 'tools'):
                 for i, tool in enumerate(self._agent.tools or []):
                     tool_llm = self._find_llm_recursively(tool, max_depth=3, path=f"tools[{i}]")
                     if tool_llm:
-                        all_llms.append((f'tool_{i}_{getattr(tool, "name", "unknown")}', tool_llm))
+                        # Skip if it's a DaseinLLMWrapper (already handles callbacks)
+                        if isinstance(tool_llm, DaseinLLMWrapper):
+                            print(f"[DASEIN][WRAPPER] Found DaseinLLMWrapper in tool - skipping")
+                        else:
+                            all_llms.append((f'tool_{i}_{getattr(tool, "name", "unknown")}', tool_llm))
             
             print(f"[DASEIN][WRAPPER] Found {len(all_llms)} LLM(s)")
             for location, llm in all_llms:
@@ -2045,43 +2060,10 @@ Follow these rules when planning your actions."""
                                         traceback.print_exc()
                                 
                                 try:
-                                    # CRITICAL FIX: Ensure callbacks propagate to _generate/_agenerate
-                                    # AgentExecutor doesn't pass run_manager to these internal methods
-                                    # So we need to manually inject the callback handler
-                                    if meth_name in ['_generate', '_agenerate'] and callback_handler:
-                                        if 'run_manager' not in kwargs and hasattr(callback_handler, 'on_llm_start'):
-                                            # Manually trigger on_llm_start since no run_manager
-                                            import uuid
-                                            run_id = uuid.uuid4()
-                                            # Extract messages for on_llm_start
-                                            messages = args[0] if args else []
-                                            prompts = []
-                                            for msg in (messages if isinstance(messages, list) else [messages]):
-                                                if hasattr(msg, 'content'):
-                                                    prompts.append(str(msg.content))
-                                                else:
-                                                    prompts.append(str(msg))
-                                            
-                                            # Call on_llm_start
-                                            callback_handler.on_llm_start(
-                                                serialized={'name': type(self_llm).__name__},
-                                                prompts=prompts,
-                                                run_id=run_id
-                                            )
-                                            
-                                            # Store run_id for on_llm_end
-                                            if not hasattr(self_llm, '_dasein_pending_run_ids'):
-                                                self_llm._dasein_pending_run_ids = []
-                                            self_llm._dasein_pending_run_ids.append(run_id)
-                                    
+                                    # NOTE: Manual callback injection DISABLED - DaseinLLMWrapper handles callbacks
+                                    # The patched methods ONLY do pipecleaner deduplication, not callbacks
                                     result = await orig_method(self_llm, *args, **kwargs)
                                     
-                                    # Call on_llm_end if we called on_llm_start
-                                    if meth_name in ['_generate', '_agenerate'] and callback_handler:
-                                        if hasattr(self_llm, '_dasein_pending_run_ids') and self_llm._dasein_pending_run_ids:
-                                            run_id = self_llm._dasein_pending_run_ids.pop(0)
-                                            callback_handler.on_llm_end(result, run_id=run_id)
-                                    
                                     # 🚨 MICROTURN ENFORCEMENT - DISABLED
                                     # Microturn can interfere with tool execution, so it's disabled
                                     # TODO: Re-enable with proper gating if needed for specific use cases
@@ -2310,43 +2292,10 @@ Follow these rules when planning your actions."""
                                         traceback.print_exc()
                                 
                                 try:
-                                    # CRITICAL FIX: Ensure callbacks propagate to _generate/_agenerate
-                                    # AgentExecutor doesn't pass run_manager to these internal methods
-                                    # So we need to manually inject the callback handler
-                                    if meth_name in ['_generate', '_agenerate'] and callback_handler:
-                                        if 'run_manager' not in kwargs and hasattr(callback_handler, 'on_llm_start'):
-                                            # Manually trigger on_llm_start since no run_manager
-                                            import uuid
-                                            run_id = uuid.uuid4()
-                                            # Extract messages for on_llm_start
-                                            messages = args[0] if args else []
-                                            prompts = []
-                                            for msg in (messages if isinstance(messages, list) else [messages]):
-                                                if hasattr(msg, 'content'):
-                                                    prompts.append(str(msg.content))
-                                                else:
-                                                    prompts.append(str(msg))
-                                            
-                                            # Call on_llm_start
-                                            callback_handler.on_llm_start(
-                                                serialized={'name': type(self_llm).__name__},
-                                                prompts=prompts,
-                                                run_id=run_id
-                                            )
-                                            
-                                            # Store run_id for on_llm_end
-                                            if not hasattr(self_llm, '_dasein_pending_run_ids'):
-                                                self_llm._dasein_pending_run_ids = []
-                                            self_llm._dasein_pending_run_ids.append(run_id)
-                                    
+                                    # NOTE: Manual callback injection DISABLED - DaseinLLMWrapper handles callbacks
+                                    # The patched methods ONLY do pipecleaner deduplication, not callbacks
                                     result = orig_method(self_llm, *args, **kwargs)
                                     
-                                    # Call on_llm_end if we called on_llm_start
-                                    if meth_name in ['_generate', '_agenerate'] and callback_handler:
-                                        if hasattr(self_llm, '_dasein_pending_run_ids') and self_llm._dasein_pending_run_ids:
-                                            run_id = self_llm._dasein_pending_run_ids.pop(0)
-                                            callback_handler.on_llm_end(result, run_id=run_id)
-                                    
                                     # 🚨 MICROTURN ENFORCEMENT - DISABLED (can interfere with tool execution)
                                     # TODO: Re-enable with proper gating if needed
                                     
@@ -3841,7 +3790,11 @@ Follow these rules when planning your actions."""
                     print(f"[DASEIN] 🔧 Pre-loading embedding model for pipecleaner (found filter search rules)...")
                     from .pipecleaner import _get_embedding_model
                     try:
-                        _get_embedding_model()  # Warm up the model
+                        # Suppress protobuf warnings from sentence-transformers
+                        import warnings
+                        with warnings.catch_warnings():
+                            warnings.filterwarnings('ignore', category=Warning)
+                            _get_embedding_model()  # Warm up the model
                         print(f"[DASEIN] ✅ Embedding model pre-loaded successfully")
                     except Exception as e:
                         print(f"[DASEIN] ⚠️  Failed to pre-load embedding model: {e}")

{dasein_core-0.2.12 → dasein_core-0.2.14/src/dasein_core.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dasein-core
-Version: 0.2.12
+Version: 0.2.14
 Summary: Universal memory for agentic AI. Attach a brain to any LangChain/LangGraph agent in a single line.
 Author-email: Dasein Team <support@dasein.ai>
 License: MIT