dasein-core 0.2.4__py3-none-any.whl → 0.2.6__py3-none-any.whl

dasein/api.py

@@ -2163,12 +2163,14 @@ Follow these rules when planning your actions."""
         tool_calls = len([step for step in trace if step.get('step_type') == 'tool_start'])
         total_turns = len(trace)
         
-        # Sum up tokens and time
+        # Sum up tokens and calculate average time
         input_tokens = sum(step.get('tokens_input', 0) for step in trace)
         output_tokens = sum(step.get('tokens_output', 0) for step in trace)
         total_tokens = input_tokens + output_tokens
         
-        trace_time_ms = sum(step.get('duration_ms', 0) for step in trace)
+        # Calculate average duration_ms across all steps that have timing data
+        durations = [step.get('duration_ms', 0) for step in trace if step.get('duration_ms', 0) > 0]
+        trace_time_ms = int(sum(durations) / len(durations)) if durations else 0
         
         # Calculate wall time from timestamps (they are ISO format strings)
         if len(trace) > 1:
@@ -2823,6 +2825,81 @@ Follow these rules when planning your actions."""
                     return getattr(agent, 'agent_id', None) or f"agent_{id(agent)}"
 
             agent_fingerprint = _minimal_agent_fingerprint(self._agent)
+            
+            # Extract tool metadata for Stage 3.5 tool grounding
+            def _extract_tool_metadata(agent):
+                """
+                Extract tool metadata (name, description, args_schema) from agent.
+                
+                CRITICAL: Extracts ALL available tools from the agent, not just tools used in trace.
+                Why: If agent used wrong tool (e.g., extract_text instead of get_elements), 
+                the trace won't show the correct tool. Stage 3.5 needs to see all options
+                to suggest better alternatives.
+                """
+                tools_metadata = []
+                tools_to_process = []
+                
+                # Get ALL tools from agent (LangChain or LangGraph) - not filtered by trace usage
+                tools_attr = getattr(agent, 'tools', None)
+                if tools_attr:
+                    try:
+                        tools_to_process = list(tools_attr)
+                    except Exception:
+                        pass
+                elif getattr(agent, 'toolkit', None):
+                    tk = getattr(agent, 'toolkit')
+                    tk_tools = getattr(tk, 'tools', None) or getattr(tk, 'get_tools', None)
+                    try:
+                        tools_to_process = list(tk_tools() if callable(tk_tools) else tk_tools or [])
+                    except Exception:
+                        pass
+                
+                # Also try LangGraph tools from compiled graph
+                if hasattr(agent, 'nodes') and 'tools' in agent.nodes:
+                    tools_node = agent.nodes['tools']
+                    if hasattr(tools_node, 'node') and hasattr(tools_node.node, 'steps'):
+                        for step in tools_node.node.steps:
+                            if hasattr(step, 'tools_by_name'):
+                                tools_to_process.extend(step.tools_by_name.values())
+                                break
+                
+                # Extract metadata from each tool
+                for tool in tools_to_process:
+                    try:
+                        tool_meta = {
+                            'name': getattr(tool, 'name', str(tool.__class__.__name__)),
+                            'description': getattr(tool, 'description', ''),
+                        }
+                        
+                        # Extract args_schema if available
+                        if hasattr(tool, 'args_schema') and tool.args_schema:
+                            try:
+                                # Try Pydantic v2 method
+                                if hasattr(tool.args_schema, 'model_json_schema'):
+                                    tool_meta['args_schema'] = tool.args_schema.model_json_schema()
+                                # Fallback to Pydantic v1 method
+                                elif hasattr(tool.args_schema, 'schema'):
+                                    tool_meta['args_schema'] = tool.args_schema.schema()
+                                else:
+                                    tool_meta['args_schema'] = {}
+                            except Exception:
+                                tool_meta['args_schema'] = {}
+                        else:
+                            tool_meta['args_schema'] = {}
+                        
+                        tools_metadata.append(tool_meta)
+                    except Exception as e:
+                        # Skip tools that fail to extract
+                        pass
+                
+                return tools_metadata
+            
+            tools_metadata = _extract_tool_metadata(self._agent)
+            print(f"[DASEIN] Extracted metadata for {len(tools_metadata)} tools")
+            if tools_metadata:
+                print(f"[DASEIN] Sample tool: {tools_metadata[0].get('name', 'unknown')}")
+            else:
+                print(f"[DASEIN] WARNING: No tools extracted! Agent type: {type(self._agent)}")
 
             response = self._service_adapter.synthesize_rules(
                 run_id=None,  # Will use stored run_id from pre-run phase
@@ -2837,7 +2914,8 @@ Follow these rules when planning your actions."""
                 agent_fingerprint=agent_fingerprint,  # Reuse fingerprint from pre-run (line 2613)
                 step_id=self._current_step_id,  # Pass step_id for parallel execution tracking
                 post_run_mode=self._post_run,  # Pass post_run mode ("full" or "kpi_only")
-                wait_for_synthesis=wait_for_synthesis  # Wait for synthesis on retry runs (except last)
+                wait_for_synthesis=wait_for_synthesis,  # Wait for synthesis on retry runs (except last)
+                tools_metadata=tools_metadata  # Tool metadata for Stage 3.5 tool grounding
             )
 
             # response is a dict from ServiceAdapter; handle accordingly

dasein/services/post_run_client.py

@@ -32,6 +32,7 @@ class RuleSynthesisRequest:
     skip_synthesis: bool = False
     wait_for_synthesis: bool = False
     step_id: Optional[str] = None
+    tools_metadata: Optional[List[Dict[str, Any]]] = None  # Tool metadata for Stage 3.5 grounding
 
 
 @dataclass
@@ -95,6 +96,7 @@ class PostRunClient:
             "performance_tracking_id": request.performance_tracking_id,
             "skip_synthesis": request.skip_synthesis,
             "wait_for_synthesis": request.wait_for_synthesis,
+            "tools_metadata": request.tools_metadata or [],  # Tool metadata for Stage 3.5 grounding
         }
         
         logger.info(f"Synthesizing rules for run: {request.run_id}")

dasein/services/service_adapter.py

@@ -95,7 +95,7 @@ class ServiceAdapter:
                         max_rules: Optional[int] = 5, performance_tracking_id: Optional[str] = None,
                         skip_synthesis: bool = False, agent_fingerprint: Optional[str] = None,
                         step_id: Optional[str] = None, post_run_mode: str = "full",
-                        wait_for_synthesis: bool = False) -> Dict[str, Any]:
+                        wait_for_synthesis: bool = False, tools_metadata: Optional[List[Dict[str, Any]]] = None) -> Dict[str, Any]:
         """
         Synthesize rules from run telemetry (replaces local rule synthesis)
         
@@ -138,7 +138,8 @@ class ServiceAdapter:
                 performance_tracking_id=performance_tracking_id,
                 skip_synthesis=should_skip_synthesis,
                 wait_for_synthesis=wait_for_synthesis,
-                step_id=step_id
+                step_id=step_id,
+                tools_metadata=tools_metadata
             )
             
             response = self.post_run_client.synthesize_rules(request)

{dasein_core-0.2.4.dist-info → dasein_core-0.2.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dasein-core
-Version: 0.2.4
+Version: 0.2.6
 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.4.dist-info → dasein_core-0.2.6.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=ysyZ29JdMuayHWeYP1gc-hB_zdYEc3UfNsmJEornHO0,172227
+dasein/api.py,sha256=Ii4_RNFtml3fA2Kt2Et_19RkXoL6OkQ91gkvvFNnP5g,176478
 dasein/capture.py,sha256=XrEPsteG5__csqcqXzOmBSzPYgeI-OFzu3IRVMPYj3w,83814
 dasein/config.py,sha256=lXO8JG4RXbodn3gT5yEnuB0VRwWdrRVwhX3Rm06IZmU,1957
 dasein/events.py,sha256=mG-lnOvQoZUhXbrPSjrG4RME6ywUcbSZ04PscoJ15GI,12896
@@ -10,12 +10,12 @@ dasein/injector.py,sha256=EItWhlG6oMAf_D7YJnRNyDwAQIK5MsaATu1ig3OENqM,7256
 dasein/trace_buffer.py,sha256=bIyTpU8ZrNFR_TCwS43HvzUrDHpZ2F8pLVDeUE9jpwM,4117
 dasein/types.py,sha256=FjGXZowiRYZzWj5GzSnAnA_-xwYqqE7WmXFCosVyGI8,2974
 dasein/services/__init__.py,sha256=0o6vKEVSYgGo-u-xDFf7Z4cQr8gIht2YovD6eEXUquE,356
-dasein/services/post_run_client.py,sha256=bzWKiu-FGo36r6J-QBv6zjLpjI7mMjKmjhvfiddWFno,4352
+dasein/services/post_run_client.py,sha256=qYT9qp2O1MecYiTLRjFBIJy8mfZ_FTntmK-_I-7-YlI,4552
 dasein/services/pre_run_client.py,sha256=tXmz_PQaSfq0xwypiWUAqNkXOmREZ6EwXLC4OM89J-A,4317
-dasein/services/service_adapter.py,sha256=nVvPPioYKoxYmR33w1ssYx422UPBeTsq8yIKZum8YHg,7128
+dasein/services/service_adapter.py,sha256=0oMDoKDWkZ17jMCVBuDMJ9TMVmfRie4q4PLMIpiV_uw,7230
 dasein/services/service_config.py,sha256=8_4tpV4mZvfaOc5_yyHbOyL4rYsPHzkLTEY1rtYgLs8,1629
-dasein_core-0.2.4.dist-info/licenses/LICENSE,sha256=7FHjIFEKl_3hSc3tGUVEWmufC_3oi8rh_2zVuL7jMKs,1091
-dasein_core-0.2.4.dist-info/METADATA,sha256=u1OEGtlVRM2WYxUpT_pHstT5NoQPjjrcdA6D-fSw_oM,10192
-dasein_core-0.2.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dasein_core-0.2.4.dist-info/top_level.txt,sha256=6yYY9kltjvvPsg9K6KyMKRtzEr5qM7sHXN7VzmrDtp0,7
-dasein_core-0.2.4.dist-info/RECORD,,
+dasein_core-0.2.6.dist-info/licenses/LICENSE,sha256=7FHjIFEKl_3hSc3tGUVEWmufC_3oi8rh_2zVuL7jMKs,1091
+dasein_core-0.2.6.dist-info/METADATA,sha256=Z30TeYgJc6ms5g7gaKCpfvOV9TiOehgXI9sO-uuowdQ,10192
+dasein_core-0.2.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dasein_core-0.2.6.dist-info/top_level.txt,sha256=6yYY9kltjvvPsg9K6KyMKRtzEr5qM7sHXN7VzmrDtp0,7
+dasein_core-0.2.6.dist-info/RECORD,,