empathy-framework 4.6.6__py3-none-any.whl → 4.7.1__py3-none-any.whl

empathy_os/memory/graph.py

@@ -1,10 +1,10 @@
-"""Memory Graph - Cross-Wizard Knowledge Base
+"""Memory Graph - Cross-Workflow Knowledge Base
 
-A knowledge graph that connects findings across all wizards,
+A knowledge graph that connects findings across all workflows,
 enabling intelligent correlation and learning.
 
 Features:
-- Add findings from any wizard as nodes
+- Add findings from any workflow as nodes
 - Connect related findings with typed edges
 - Query for similar past findings
 - Traverse relationships to find root causes
@@ -29,17 +29,17 @@ from .nodes import Node, NodeType
 
 
 class MemoryGraph:
-    """Knowledge graph for cross-wizard intelligence.
+    """Knowledge graph for cross-workflow intelligence.
 
     Stores nodes (findings) and edges (relationships) discovered
-    by wizards, enabling pattern correlation across sessions.
+    by workflows, enabling pattern correlation across sessions.
 
     Usage:
         graph = MemoryGraph()
 
         # Add a bug finding
         bug_id = graph.add_finding(
-            wizard="bug-predict",
+            workflow="bug-predict",
             finding={
                 "type": "bug",
                 "name": "Null reference in auth.py",
@@ -52,7 +52,7 @@ class MemoryGraph:
 
         # Connect to a fix
         fix_id = graph.add_finding(
-            wizard="bug-predict",
+            workflow="bug-predict",
             finding={
                 "type": "fix",
                 "name": "Add null check",
@@ -80,7 +80,7 @@ class MemoryGraph:
         self._edges_by_source: dict[str, list[Edge]] = defaultdict(list)
         self._edges_by_target: dict[str, list[Edge]] = defaultdict(list)
         self._nodes_by_type: dict[NodeType, list[str]] = defaultdict(list)
-        self._nodes_by_wizard: dict[str, list[str]] = defaultdict(list)
+        self._nodes_by_workflow: dict[str, list[str]] = defaultdict(list)
         self._nodes_by_file: dict[str, list[str]] = defaultdict(list)
 
         self._load()
@@ -133,8 +133,8 @@ class MemoryGraph:
     def _index_node(self, node: Node) -> None:
         """Add node to indexes."""
         self._nodes_by_type[node.type].append(node.id)
-        if node.source_wizard:
-            self._nodes_by_wizard[node.source_wizard].append(node.id)
+        if node.source_workflow:
+            self._nodes_by_workflow[node.source_workflow].append(node.id)
         if node.source_file:
             self._nodes_by_file[node.source_file].append(node.id)
 
@@ -151,11 +151,11 @@ class MemoryGraph:
         timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
         return f"{finding.get('type', 'node')}_{timestamp}_{hash_val}"
 
-    def add_finding(self, wizard: str, finding: dict[str, Any]) -> str:
-        """Add a finding from any wizard, return node ID.
+    def add_finding(self, workflow: str, finding: dict[str, Any]) -> str:
+        """Add a finding from any workflow, return node ID.
 
         Args:
-            wizard: Name of the wizard adding this finding
+            workflow: Name of the workflow adding this finding
             finding: Dict with at least 'type' and 'name' keys
 
         Returns:
@@ -163,7 +163,7 @@ class MemoryGraph:
 
         Example:
             node_id = graph.add_finding(
-                wizard="security-audit",
+                workflow="security-audit",
                 finding={
                     "type": "vulnerability",
                     "name": "SQL Injection in query builder",
@@ -189,7 +189,7 @@ class MemoryGraph:
             type=node_type,
             name=finding.get("name", "Unnamed finding"),
             description=finding.get("description", ""),
-            source_wizard=wizard,
+            source_workflow=workflow,
             source_file=finding.get("file", finding.get("source_file", "")),
             source_line=finding.get("line", finding.get("source_line")),
             severity=finding.get("severity", ""),
@@ -210,7 +210,7 @@ class MemoryGraph:
         target_id: str,
         edge_type: EdgeType,
         description: str = "",
-        wizard: str = "",
+        workflow: str = "",
         weight: float = 1.0,
         bidirectional: bool = False,
     ) -> str:
@@ -221,7 +221,7 @@ class MemoryGraph:
             target_id: Target node ID
             edge_type: Type of relationship
             description: Optional description of the relationship
-            wizard: Wizard that created this edge
+            workflow: Workflow that created this edge
             weight: Strength of relationship (0.0 - 1.0)
             bidirectional: If True, also create reverse edge
 
@@ -239,7 +239,7 @@ class MemoryGraph:
             target_id=target_id,
             type=edge_type,
             description=description,
-            source_wizard=wizard,
+            source_workflow=workflow,
             weight=weight,
         )
 
@@ -254,7 +254,7 @@ class MemoryGraph:
                 target_id=source_id,
                 type=reverse_type,
                 description=description,
-                source_wizard=wizard,
+                source_workflow=workflow,
                 weight=weight,
             )
             self.edges.append(reverse_edge)
@@ -404,9 +404,9 @@ class MemoryGraph:
         node_ids = self._nodes_by_type.get(node_type, [])
         return [self.nodes[nid] for nid in node_ids if nid in self.nodes]
 
-    def find_by_wizard(self, wizard: str) -> list[Node]:
-        """Find all nodes created by a specific wizard."""
-        node_ids = self._nodes_by_wizard.get(wizard, [])
+    def find_by_workflow(self, workflow: str) -> list[Node]:
+        """Find all nodes created by a specific workflow."""
+        node_ids = self._nodes_by_workflow.get(workflow, [])
         return [self.nodes[nid] for nid in node_ids if nid in self.nodes]
 
     def find_by_file(self, file_path: str) -> list[Node]:
@@ -463,13 +463,13 @@ class MemoryGraph:
     def get_statistics(self) -> dict[str, Any]:
         """Get graph statistics."""
         type_counts: dict[str, int] = defaultdict(int)
-        wizard_counts: dict[str, int] = defaultdict(int)
+        workflow_counts: dict[str, int] = defaultdict(int)
         severity_counts: dict[str, int] = defaultdict(int)
 
         for node in self.nodes.values():
             type_counts[node.type.value] += 1
-            if node.source_wizard:
-                wizard_counts[node.source_wizard] += 1
+            if node.source_workflow:
+                workflow_counts[node.source_workflow] += 1
             if node.severity:
                 severity_counts[node.severity] += 1
 
@@ -481,7 +481,7 @@ class MemoryGraph:
             "total_nodes": len(self.nodes),
             "total_edges": len(self.edges),
             "nodes_by_type": dict(type_counts),
-            "nodes_by_wizard": dict(wizard_counts),
+            "nodes_by_workflow": dict(workflow_counts),
             "nodes_by_severity": dict(severity_counts),
             "edges_by_type": dict(edge_type_counts),
             "unique_files": len(self._nodes_by_file),
@@ -537,9 +537,9 @@ class MemoryGraph:
             self._nodes_by_type[node.type] = [
                 nid for nid in self._nodes_by_type[node.type] if nid != node_id
             ]
-        if node.source_wizard in self._nodes_by_wizard:
-            self._nodes_by_wizard[node.source_wizard] = [
-                nid for nid in self._nodes_by_wizard[node.source_wizard] if nid != node_id
+        if node.source_workflow in self._nodes_by_workflow:
+            self._nodes_by_workflow[node.source_workflow] = [
+                nid for nid in self._nodes_by_workflow[node.source_workflow] if nid != node_id
             ]
         if node.source_file in self._nodes_by_file:
             self._nodes_by_file[node.source_file] = [
@@ -565,6 +565,6 @@ class MemoryGraph:
         self._edges_by_source = defaultdict(list)
         self._edges_by_target = defaultdict(list)
         self._nodes_by_type = defaultdict(list)
-        self._nodes_by_wizard = defaultdict(list)
+        self._nodes_by_workflow = defaultdict(list)
         self._nodes_by_file = defaultdict(list)
         self._save()

empathy_os/memory/nodes.py

@@ -1,7 +1,7 @@
 """Memory Graph Node Types
 
-Defines node types for the cross-wizard knowledge graph.
-Each node represents an entity discovered by a wizard.
+Defines node types for the cross-workflow knowledge graph.
+Each node represents an entity discovered by a workflow.
 
 Copyright 2025 Smart AI Memory, LLC
 Licensed under Fair Source 0.9
@@ -52,7 +52,7 @@ class NodeType(Enum):
 class Node:
     """A node in the memory graph.
 
-    Represents any entity that wizards discover or create.
+    Represents any entity that workflows discover or create.
     """
 
     id: str
@@ -61,7 +61,7 @@ class Node:
     description: str = ""
 
     # Where this node came from
-    source_wizard: str = ""
+    source_workflow: str = ""
     source_file: str = ""
     source_line: int | None = None
 
@@ -87,7 +87,7 @@ class Node:
             "type": self.type.value,
             "name": self.name,
             "description": self.description,
-            "source_wizard": self.source_wizard,
+            "source_workflow": self.source_workflow,
             "source_file": self.source_file,
             "source_line": self.source_line,
             "severity": self.severity,
@@ -107,7 +107,7 @@ class Node:
             type=NodeType(data["type"]),
             name=data["name"],
             description=data.get("description", ""),
-            source_wizard=data.get("source_wizard", ""),
+            source_workflow=data.get("source_workflow", data.get("source_wizard", "")),
             source_file=data.get("source_file", ""),
             source_line=data.get("source_line"),
             severity=data.get("severity", ""),

empathy_os/memory/short_term.py

@@ -560,11 +560,15 @@ class RedisShortTermMemory:
 
         if self._config.pii_scrub_enabled:
             self._pii_scrubber = PIIScrubber(enable_name_detection=False)
-            logger.debug("pii_scrubber_enabled", message="PII scrubbing active for short-term memory")
+            logger.debug(
+                "pii_scrubber_enabled", message="PII scrubbing active for short-term memory"
+            )
 
         if self._config.secrets_detection_enabled:
             self._secrets_detector = SecretsDetector()
-            logger.debug("secrets_detector_enabled", message="Secrets detection active for short-term memory")
+            logger.debug(
+                "secrets_detector_enabled", message="Secrets detection active for short-term memory"
+            )
 
         if self.use_mock:
             self._client = None
@@ -735,7 +739,8 @@ class RedisShortTermMemory:
             detections = self._secrets_detector.detect(data_str)
             # Block critical and high severity secrets
             critical_secrets = [
-                d for d in detections
+                d
+                for d in detections
                 if d.severity in (SecretSeverity.CRITICAL, SecretSeverity.HIGH)
             ]
             if critical_secrets:
@@ -1378,17 +1383,18 @@ class RedisShortTermMemory:
 
         """
         if self.use_mock:
+            # Use generator expressions for memory-efficient counting
             return {
                 "mode": "mock",
                 "total_keys": len(self._mock_storage),
-                "working_keys": len(
-                    [k for k in self._mock_storage if k.startswith(self.PREFIX_WORKING)],
+                "working_keys": sum(
+                    1 for k in self._mock_storage if k.startswith(self.PREFIX_WORKING)
                 ),
-                "staged_keys": len(
-                    [k for k in self._mock_storage if k.startswith(self.PREFIX_STAGED)],
+                "staged_keys": sum(
+                    1 for k in self._mock_storage if k.startswith(self.PREFIX_STAGED)
                 ),
-                "conflict_keys": len(
-                    [k for k in self._mock_storage if k.startswith(self.PREFIX_CONFLICT)],
+                "conflict_keys": sum(
+                    1 for k in self._mock_storage if k.startswith(self.PREFIX_CONFLICT)
                 ),
             }