dasein-core 0.2.5__py3-none-any.whl → 0.2.7__py3-none-any.whl

dasein/api.py

@@ -2825,6 +2825,97 @@ 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)
+            
+            # Reuse existing graph analysis (already extracted in __init__)
+            graph_metadata = None
+            if self._is_langgraph and hasattr(self._agent, 'get_graph'):
+                try:
+                    graph = self._agent.get_graph()
+                    graph_metadata = {
+                        'nodes': list(graph.nodes.keys()),
+                        'edges': [{'source': e.source, 'target': e.target} for e in graph.edges],
+                        'is_multi_agent': len(graph.nodes) > 1
+                    }
+                    print(f"[DASEIN] Extracted graph metadata: {len(graph_metadata['nodes'])} nodes, {len(graph_metadata['edges'])} edges")
+                except Exception:
+                    pass
+            
+            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
@@ -2839,7 +2930,9 @@ 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
+                graph_metadata=graph_metadata  # Graph metadata for Stage 3.5 node grounding
             )
 
             # response is a dict from ServiceAdapter; handle accordingly

dasein/services/post_run_client.py

@@ -32,6 +32,8 @@ 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 tool grounding
+    graph_metadata: Optional[Dict[str, Any]] = None  # Graph metadata for Stage 3.5 node grounding
 
 
 @dataclass
@@ -95,6 +97,8 @@ 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 tool grounding
+            "graph_metadata": request.graph_metadata or {},  # Graph metadata for Stage 3.5 node grounding
         }
         
         logger.info(f"Synthesizing rules for run: {request.run_id}")

dasein/services/service_adapter.py

@@ -95,7 +95,8 @@ 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,
+                        graph_metadata: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
         """
         Synthesize rules from run telemetry (replaces local rule synthesis)
         
@@ -138,7 +139,9 @@ 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,
+                graph_metadata=graph_metadata
             )
             
             response = self.post_run_client.synthesize_rules(request)

{dasein_core-0.2.5.dist-info → dasein_core-0.2.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dasein-core
-Version: 0.2.5
+Version: 0.2.7
 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.5.dist-info → dasein_core-0.2.7.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=GQpZIRFDYCoZvDN2T4Gsh2BBXfCJ7ui3WTxi_ZQGb0Q,172432
+dasein/api.py,sha256=KIWZ5I_8in2xas50IAfUOCmZgjaaXYxPcM0BgMx7Bs0,177347
 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=UjK3eqf7oWGSuWkKe0vQmeMS0yUUOhYFD4-SZ7uj8WY,4768
 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=YHk41lR3PXh8WTmxOzzwKf6hwPYGqIdApI92vQKlkAY,7350
 dasein/services/service_config.py,sha256=8_4tpV4mZvfaOc5_yyHbOyL4rYsPHzkLTEY1rtYgLs8,1629
-dasein_core-0.2.5.dist-info/licenses/LICENSE,sha256=7FHjIFEKl_3hSc3tGUVEWmufC_3oi8rh_2zVuL7jMKs,1091
-dasein_core-0.2.5.dist-info/METADATA,sha256=k-g7KsEhfAp3b_FgiAwh0i8Swuof9olOFnjJlGGj_zs,10192
-dasein_core-0.2.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dasein_core-0.2.5.dist-info/top_level.txt,sha256=6yYY9kltjvvPsg9K6KyMKRtzEr5qM7sHXN7VzmrDtp0,7
-dasein_core-0.2.5.dist-info/RECORD,,
+dasein_core-0.2.7.dist-info/licenses/LICENSE,sha256=7FHjIFEKl_3hSc3tGUVEWmufC_3oi8rh_2zVuL7jMKs,1091
+dasein_core-0.2.7.dist-info/METADATA,sha256=M7zLIt6gXOfna7oFIMtqECJMqSU4WufP8g2IoeMJqKk,10192
+dasein_core-0.2.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dasein_core-0.2.7.dist-info/top_level.txt,sha256=6yYY9kltjvvPsg9K6KyMKRtzEr5qM7sHXN7VzmrDtp0,7
+dasein_core-0.2.7.dist-info/RECORD,,