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

dasein/api.py

@@ -881,6 +881,9 @@ class CognateProxy:
         # Initialize KPI tracking
         self._last_run_kpis = None
         
+        # Initialize wrapped LLM (will be set by _wrap_agent_llm if applicable)
+        self._wrapped_llm = None
+        
         # Wrap the agent's LLM with our trace capture wrapper
         self._wrap_agent_llm()
         
@@ -1680,7 +1683,69 @@ Follow these rules when planning your actions."""
             self._deadletter_fn = None
     
     def _wrap_agent_llm(self):
-        """Monkey-patch ALL LLM classes found in agent + tools."""
+        """Wrap LLM instance (for SQL agent callbacks) AND monkey-patch (for Pipecleaner)."""
+        try:
+            # STEP 1: Wrap the main agent LLM with DaseinLLMWrapper (captures all _generate calls)
+            # This is critical for SQL agents where callbacks don't propagate properly
+            llm = self._find_llm_recursively(self._agent, max_depth=5)
+            if llm:
+                wrapped_llm = DaseinLLMWrapper(llm, self._callback_handler, verbose=self._verbose)
+                # Replace the original LLM with our wrapped version
+                self._replace_llm_in_structure(self._agent, llm, wrapped_llm, max_depth=5)
+                self._wrapped_llm = wrapped_llm
+                self._vprint(f"[DASEIN][WRAPPER] Successfully wrapped {type(llm).__name__} LLM")
+            else:
+                self._vprint(f"[DASEIN][WRAPPER] Could not find any LLM in agent structure")
+                self._wrapped_llm = None
+            
+            # STEP 2: Monkey-patch LLM classes for Pipecleaner deduplication
+            # This is critical for research agents with Summary calls that need deduplication
+            self._monkey_patch_llm_classes()
+            
+        except Exception as e:
+            self._vprint(f"[DASEIN][WRAPPER] Failed to wrap agent LLM: {e}")
+            import traceback
+            traceback.print_exc()
+            self._wrapped_llm = None
+    
+    def _replace_llm_in_structure(self, obj, original_llm, wrapped_llm, max_depth=5, path=""):
+        """Replace the original LLM with wrapped LLM in the structure."""
+        if max_depth <= 0:
+            return
+        
+        # Special handling for RunnableSequence - check steps
+        if hasattr(obj, 'steps') and hasattr(obj, '__iter__'):
+            for i, step in enumerate(obj.steps):
+                if step is original_llm:
+                    self._vprint(f"[DASEIN][WRAPPER] Replacing LLM at {path}.steps[{i}]")
+                    obj.steps[i] = wrapped_llm
+                    return
+                # Check if step has bound attribute (RunnableBinding)
+                if hasattr(step, 'bound') and step.bound is original_llm:
+                    self._vprint(f"[DASEIN][WRAPPER] Replacing LLM at {path}.steps[{i}].bound")
+                    step.bound = wrapped_llm
+                    return
+                # Recursively search in the step
+                self._replace_llm_in_structure(step, original_llm, wrapped_llm, max_depth - 1, f"{path}.steps[{i}]")
+        
+        # Search in attributes
+        for attr_name in dir(obj):
+            if attr_name.startswith('_'):
+                continue
+            try:
+                attr_value = getattr(obj, attr_name)
+                if attr_value is original_llm:
+                    self._vprint(f"[DASEIN][WRAPPER] Replacing LLM at {path}.{attr_name}")
+                    setattr(obj, attr_name, wrapped_llm)
+                    return
+                # Recursively search in the attribute
+                if hasattr(attr_value, '__dict__') or hasattr(attr_value, '__iter__'):
+                    self._replace_llm_in_structure(attr_value, original_llm, wrapped_llm, max_depth - 1, f"{path}.{attr_name}")
+            except:
+                continue
+    
+    def _monkey_patch_llm_classes(self):
+        """Monkey-patch ALL LLM classes found in agent + tools for Pipecleaner deduplication."""
         try:
             # Find ALL LLMs in agent structure + tools
             print(f"[DASEIN][WRAPPER] Searching for ALL LLMs in agent+tools...")
@@ -1713,10 +1778,10 @@ Follow these rules when planning your actions."""
                 print(f"[DASEIN][WRAPPER] Patching {llm_class.__name__} (found in {location})...")
                 
                 # Check what methods the LLM class has
-                # Only patch TOP-LEVEL methods to avoid double-deduplication from internal calls
+                # Patch both user-facing AND internal methods since SQL agents bypass invoke
                 print(f"[DASEIN][WRAPPER] Checking LLM methods...")
                 methods_to_patch = []
-                for method in ['invoke', 'ainvoke']:  # Only patch user-facing methods, not internal _generate
+                for method in ['invoke', 'ainvoke', '_generate', '_agenerate']:  # Include internal methods for SQL agents
                     if hasattr(llm_class, method):
                         print(f"[DASEIN][WRAPPER]   - Has {method}")
                         methods_to_patch.append(method)
@@ -1811,7 +1876,7 @@ Follow these rules when planning your actions."""
                                                     prompt_strings.append(msg.content)
                                                 elif isinstance(msg, str):
                                                     prompt_strings.append(msg)
-                                                else:
+                                            else:
                                                     prompt_strings.append(str(msg))
                                             
                                             # =============================================================
@@ -1929,6 +1994,7 @@ Follow these rules when planning your actions."""
                                                             break
                                             
                                             if should_dedupe:
+                                                try:
                                                     # Deduplicate each prompt
                                                     from .pipecleaner import get_or_create_corpus
                                                     import hashlib
@@ -1969,14 +2035,53 @@ Follow these rules when planning your actions."""
                                                         kwargs['messages'] = messages_to_dedupe
                                                     elif 'prompts' in kwargs:
                                                         kwargs['prompts'] = messages_to_dedupe
+                                                except Exception as e:
+                                                    print(f"[🔥 HOTPATH] ⚠️ Deduplication error: {e}")
+                                                    import traceback
+                                                    traceback.print_exc()
                                     except Exception as e:
-                                        print(f"[🔥 HOTPATH] ⚠️ Deduplication error: {e}")
+                                        print(f"[🔥 HOTPATH] ⚠️ Error in pipecleaner preprocessing: {e}")
                                         import traceback
                                         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)
+                                    
                                     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
@@ -2205,8 +2310,43 @@ 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)
+                                    
                                     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
                                     

{dasein_core-0.2.10.dist-info → dasein_core-0.2.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dasein-core
-Version: 0.2.10
+Version: 0.2.12
 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.10.dist-info → dasein_core-0.2.12.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=tQ4OtvWRG6rzYuXPKRj6DkcgIOM0kg3L1xNjOTYBf1Q,247127
+dasein/api.py,sha256=dDOtZVXnfZ3VbiY6N_1u2m3JtiSUwmKC9AO6gxevHxY,256483
 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.10.dist-info/licenses/LICENSE,sha256=7FHjIFEKl_3hSc3tGUVEWmufC_3oi8rh_2zVuL7jMKs,1091
+dasein_core-0.2.12.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.10.dist-info/METADATA,sha256=BotwsA09ka3tRHCKiI0Ms8Jb8Zss87Q5zBG0OL_3VuA,10297
-dasein_core-0.2.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dasein_core-0.2.10.dist-info/top_level.txt,sha256=6yYY9kltjvvPsg9K6KyMKRtzEr5qM7sHXN7VzmrDtp0,7
-dasein_core-0.2.10.dist-info/RECORD,,
+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,,