dasein-core 0.2.12__py3-none-any.whl → 0.2.14__py3-none-any.whl

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.dist-info → dasein_core-0.2.14.dist-info}/METADATA

@@ -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.dist-info → dasein_core-0.2.14.dist-info}/RECORD

@@ -1,6 +1,6 @@
 dasein/__init__.py,sha256=RY0lhaaWB6yJ_5YMRmaHDvQ0eFbc0BGbYNe5OVyxzYE,2316
 dasein/advice_format.py,sha256=5-h4J24L_B2Y9dlmyDuIYtmPCWOGAYoinBEXqpcNg2s,5386
-dasein/api.py,sha256=dDOtZVXnfZ3VbiY6N_1u2m3JtiSUwmKC9AO6gxevHxY,256483
+dasein/api.py,sha256=8WYkFUnMqkzOL3NDvowH_WD6A5fd27H39akujnKO99c,252752
 dasein/capture.py,sha256=D4DvknI2wbmVup5WqvNcgw-zW5riEstYG81Rl98uz6o,110942
 dasein/config.py,sha256=lXO8JG4RXbodn3gT5yEnuB0VRwWdrRVwhX3Rm06IZmU,1957
 dasein/events.py,sha256=mG-lnOvQoZUhXbrPSjrG4RME6ywUcbSZ04PscoJ15GI,12896
@@ -45,7 +45,7 @@ dasein/services/post_run_client.py,sha256=UjK3eqf7oWGSuWkKe0vQmeMS0yUUOhYFD4-SZ7
 dasein/services/pre_run_client.py,sha256=tXmz_PQaSfq0xwypiWUAqNkXOmREZ6EwXLC4OM89J-A,4317
 dasein/services/service_adapter.py,sha256=YHk41lR3PXh8WTmxOzzwKf6hwPYGqIdApI92vQKlkAY,7350
 dasein/services/service_config.py,sha256=8_4tpV4mZvfaOc5_yyHbOyL4rYsPHzkLTEY1rtYgLs8,1629
-dasein_core-0.2.12.dist-info/licenses/LICENSE,sha256=7FHjIFEKl_3hSc3tGUVEWmufC_3oi8rh_2zVuL7jMKs,1091
+dasein_core-0.2.14.dist-info/licenses/LICENSE,sha256=7FHjIFEKl_3hSc3tGUVEWmufC_3oi8rh_2zVuL7jMKs,1091
 dasein/models/en_core_web_sm/en_core_web_sm-3.7.1.dist-info/LICENSE,sha256=OTPBdpebaLxtC8yQLH1sEw8dEn9Hbxe6XNuo2Zz9ABI,1056
 dasein/models/en_core_web_sm/en_core_web_sm-3.7.1.dist-info/LICENSES_SOURCES,sha256=INnfrNIVESJR8VNB7dGkex-Yvzk6IS8Q8ZT_3H7pipA,2347
 dasein/models/en_core_web_sm/en_core_web_sm-3.7.1.dist-info/METADATA,sha256=-vGqRxa_M2RwKtLjBhc4JlBQdJ3k7CwOnseT_ReYcic,2958
@@ -53,7 +53,7 @@ dasein/models/en_core_web_sm/en_core_web_sm-3.7.1.dist-info/RECORD,sha256=dDb6U7
 dasein/models/en_core_web_sm/en_core_web_sm-3.7.1.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
 dasein/models/en_core_web_sm/en_core_web_sm-3.7.1.dist-info/entry_points.txt,sha256=OkWs-KxPJtDdpvIFCVXzDC9ECtejhPxv7pP3Tgk2cNg,47
 dasein/models/en_core_web_sm/en_core_web_sm-3.7.1.dist-info/top_level.txt,sha256=56OIuRbEuhr12HsM9XpCMnTtHRMgNC5Hje4Xeo8wF2c,15
-dasein_core-0.2.12.dist-info/METADATA,sha256=43WrPvrzNviksaZxsnZNnLh0nPU7vBO8-rdI2J9R-JM,10297
-dasein_core-0.2.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dasein_core-0.2.12.dist-info/top_level.txt,sha256=6yYY9kltjvvPsg9K6KyMKRtzEr5qM7sHXN7VzmrDtp0,7
-dasein_core-0.2.12.dist-info/RECORD,,
+dasein_core-0.2.14.dist-info/METADATA,sha256=Wor24FjmtzERNq7zLbq-b9oP4EVqlKPASshkGCgAO0c,10297
+dasein_core-0.2.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dasein_core-0.2.14.dist-info/top_level.txt,sha256=6yYY9kltjvvPsg9K6KyMKRtzEr5qM7sHXN7VzmrDtp0,7
+dasein_core-0.2.14.dist-info/RECORD,,