omni-cortex 1.0.4__py3-none-any.whl

omni_cortex-1.0.4.data/data/share/omni-cortex/dashboard/backend/websocket_manager.py

@@ -0,0 +1,82 @@
+"""WebSocket manager for real-time updates."""
+
+import asyncio
+import json
+from datetime import datetime
+from typing import Any
+from uuid import uuid4
+
+from fastapi import WebSocket
+
+
+class WebSocketManager:
+    """Manages WebSocket connections and broadcasts."""
+
+    def __init__(self):
+        self.connections: dict[str, WebSocket] = {}
+        self._lock = asyncio.Lock()
+
+    async def connect(self, websocket: WebSocket, client_id: str | None = None) -> str:
+        """Accept a new WebSocket connection."""
+        await websocket.accept()
+        client_id = client_id or str(uuid4())
+        async with self._lock:
+            self.connections[client_id] = websocket
+        print(f"[WS] Client connected: {client_id} (total: {len(self.connections)})")
+        return client_id
+
+    async def disconnect(self, client_id: str):
+        """Remove a WebSocket connection."""
+        async with self._lock:
+            if client_id in self.connections:
+                del self.connections[client_id]
+        print(f"[WS] Client disconnected: {client_id} (total: {len(self.connections)})")
+
+    async def broadcast(self, event_type: str, data: dict[str, Any]):
+        """Broadcast a message to all connected clients."""
+        if not self.connections:
+            return
+
+        message = json.dumps({
+            "event_type": event_type,
+            "data": data,
+            "timestamp": datetime.now().isoformat(),
+        })
+
+        disconnected = []
+        async with self._lock:
+            for client_id, websocket in self.connections.items():
+                try:
+                    await websocket.send_text(message)
+                except Exception as e:
+                    print(f"[WS] Failed to send to {client_id}: {e}")
+                    disconnected.append(client_id)
+
+            # Clean up disconnected clients
+            for client_id in disconnected:
+                del self.connections[client_id]
+
+    async def send_to_client(self, client_id: str, event_type: str, data: dict[str, Any]):
+        """Send a message to a specific client."""
+        message = json.dumps({
+            "event_type": event_type,
+            "data": data,
+            "timestamp": datetime.now().isoformat(),
+        })
+
+        async with self._lock:
+            if client_id in self.connections:
+                try:
+                    await self.connections[client_id].send_text(message)
+                except Exception as e:
+                    print(f"[WS] Failed to send to {client_id}: {e}")
+                    del self.connections[client_id]
+
+    @property
+    def connection_count(self) -> int:
+        """Get the number of active connections."""
+        return len(self.connections)
+
+
+# Global manager instance
+manager = WebSocketManager()

omni_cortex-1.0.4.data/data/share/omni-cortex/hooks/post_tool_use.py

@@ -0,0 +1,160 @@
+#!/usr/bin/env python3
+"""PostToolUse hook - logs tool result after execution.
+
+This hook is called by Claude Code after each tool completes.
+It logs the tool output, duration, and success/error status.
+
+Hook configuration for settings.json:
+{
+    "hooks": {
+        "PostToolUse": [
+            {
+                "type": "command",
+                "command": "python hooks/post_tool_use.py"
+            }
+        ]
+    }
+}
+"""
+
+import json
+import sys
+import os
+import sqlite3
+from datetime import datetime, timezone
+from pathlib import Path
+
+
+def get_db_path() -> Path:
+    """Get the database path for the current project."""
+    project_path = os.environ.get("CLAUDE_PROJECT_DIR", os.getcwd())
+    return Path(project_path) / ".omni-cortex" / "cortex.db"
+
+
+def ensure_database(db_path: Path) -> sqlite3.Connection:
+    """Ensure database exists and is initialized.
+
+    Auto-creates the database and schema if it doesn't exist.
+    This enables 'out of the box' functionality.
+    """
+    db_path.parent.mkdir(parents=True, exist_ok=True)
+    conn = sqlite3.connect(str(db_path))
+
+    # Check if schema exists
+    cursor = conn.cursor()
+    cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='activities'")
+    if cursor.fetchone() is None:
+        # Apply minimal schema for activities (full schema applied by MCP)
+        conn.executescript("""
+            CREATE TABLE IF NOT EXISTS activities (
+                id TEXT PRIMARY KEY,
+                session_id TEXT,
+                agent_id TEXT,
+                timestamp TEXT NOT NULL,
+                event_type TEXT NOT NULL,
+                tool_name TEXT,
+                tool_input TEXT,
+                tool_output TEXT,
+                duration_ms INTEGER,
+                success INTEGER DEFAULT 1,
+                error_message TEXT,
+                project_path TEXT,
+                file_path TEXT,
+                metadata TEXT
+            );
+            CREATE INDEX IF NOT EXISTS idx_activities_timestamp ON activities(timestamp DESC);
+            CREATE INDEX IF NOT EXISTS idx_activities_tool ON activities(tool_name);
+        """)
+        conn.commit()
+
+    return conn
+
+
+def generate_id() -> str:
+    """Generate a unique activity ID."""
+    timestamp_ms = int(datetime.now().timestamp() * 1000)
+    random_hex = os.urandom(4).hex()
+    return f"act_{timestamp_ms}_{random_hex}"
+
+
+def truncate(text: str, max_length: int = 10000) -> str:
+    """Truncate text to max length."""
+    if len(text) <= max_length:
+        return text
+    return text[:max_length - 20] + "\n... [truncated]"
+
+
+def main():
+    """Process PostToolUse hook."""
+    try:
+        # Read all input at once (more reliable than json.load on stdin)
+        raw_input = sys.stdin.read()
+        if not raw_input or not raw_input.strip():
+            print(json.dumps({}))
+            return
+
+        input_data = json.loads(raw_input)
+
+        # Extract data from hook input
+        tool_name = input_data.get("tool_name")
+        tool_input = input_data.get("tool_input", {})
+        tool_output = input_data.get("tool_output", {})
+        agent_id = input_data.get("agent_id")
+
+        # Determine success/error
+        is_error = input_data.get("is_error", False)
+        error_message = None
+        if is_error and isinstance(tool_output, dict):
+            error_message = tool_output.get("error") or tool_output.get("message")
+
+        # Skip logging our own tools to prevent recursion
+        # MCP tools are named like "mcp__omni-cortex__cortex_remember"
+        if tool_name and ("cortex_" in tool_name or "omni-cortex" in tool_name):
+            print(json.dumps({}))
+            return
+
+        session_id = os.environ.get("CLAUDE_SESSION_ID")
+        project_path = os.environ.get("CLAUDE_PROJECT_DIR", os.getcwd())
+
+        # Auto-initialize database (creates if not exists)
+        db_path = get_db_path()
+        conn = ensure_database(db_path)
+
+        # Insert activity record
+        cursor = conn.cursor()
+        cursor.execute(
+            """
+            INSERT INTO activities (
+                id, session_id, agent_id, timestamp, event_type,
+                tool_name, tool_input, tool_output, success, error_message, project_path
+            ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+            """,
+            (
+                generate_id(),
+                session_id,
+                agent_id,
+                datetime.now(timezone.utc).isoformat(),
+                "post_tool_use",
+                tool_name,
+                truncate(json.dumps(tool_input, default=str)),
+                truncate(json.dumps(tool_output, default=str)),
+                0 if is_error else 1,
+                error_message,
+                project_path,
+            ),
+        )
+        conn.commit()
+        conn.close()
+
+        # Return empty response (no modification)
+        print(json.dumps({}))
+
+    except Exception as e:
+        # Hooks should never block - log error but continue
+        print(json.dumps({"systemMessage": f"Cortex post_tool_use: {e}"}))
+
+    sys.exit(0)
+
+
+if __name__ == "__main__":
+    main()

omni_cortex-1.0.4.data/data/share/omni-cortex/hooks/pre_tool_use.py

@@ -0,0 +1,159 @@
+#!/usr/bin/env python3
+"""PreToolUse hook - logs tool call before execution.
+
+This hook is called by Claude Code before each tool is executed.
+It logs the tool name and input to the Cortex activity database.
+
+Hook configuration for settings.json:
+{
+    "hooks": {
+        "PreToolUse": [
+            {
+                "type": "command",
+                "command": "python hooks/pre_tool_use.py"
+            }
+        ]
+    }
+}
+"""
+
+import json
+import sys
+import os
+import sqlite3
+from datetime import datetime, timezone
+from pathlib import Path
+
+
+def get_db_path() -> Path:
+    """Get the database path for the current project."""
+    project_path = os.environ.get("CLAUDE_PROJECT_DIR", os.getcwd())
+    return Path(project_path) / ".omni-cortex" / "cortex.db"
+
+
+def ensure_database(db_path: Path) -> sqlite3.Connection:
+    """Ensure database exists and is initialized.
+
+    Auto-creates the database and schema if it doesn't exist.
+    This enables 'out of the box' functionality.
+    """
+    db_path.parent.mkdir(parents=True, exist_ok=True)
+    conn = sqlite3.connect(str(db_path))
+
+    # Check if schema exists
+    cursor = conn.cursor()
+    cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='activities'")
+    if cursor.fetchone() is None:
+        # Apply minimal schema for activities (full schema applied by MCP)
+        conn.executescript("""
+            CREATE TABLE IF NOT EXISTS activities (
+                id TEXT PRIMARY KEY,
+                session_id TEXT,
+                agent_id TEXT,
+                timestamp TEXT NOT NULL,
+                event_type TEXT NOT NULL,
+                tool_name TEXT,
+                tool_input TEXT,
+                tool_output TEXT,
+                duration_ms INTEGER,
+                success INTEGER DEFAULT 1,
+                error_message TEXT,
+                project_path TEXT,
+                file_path TEXT,
+                metadata TEXT
+            );
+            CREATE INDEX IF NOT EXISTS idx_activities_timestamp ON activities(timestamp DESC);
+            CREATE INDEX IF NOT EXISTS idx_activities_tool ON activities(tool_name);
+        """)
+        conn.commit()
+
+    return conn
+
+
+def generate_id() -> str:
+    """Generate a unique activity ID."""
+    timestamp_ms = int(datetime.now().timestamp() * 1000)
+    random_hex = os.urandom(4).hex()
+    return f"act_{timestamp_ms}_{random_hex}"
+
+
+def truncate(text: str, max_length: int = 10000) -> str:
+    """Truncate text to max length."""
+    if len(text) <= max_length:
+        return text
+    return text[:max_length - 20] + "\n... [truncated]"
+
+
+def main():
+    """Process PreToolUse hook."""
+    try:
+        # Read input from stdin with timeout protection
+        import select
+        if sys.platform != "win32":
+            # Unix: use select for timeout
+            ready, _, _ = select.select([sys.stdin], [], [], 5.0)
+            if not ready:
+                print(json.dumps({}))
+                return
+
+        # Read all input at once
+        raw_input = sys.stdin.read()
+        if not raw_input or not raw_input.strip():
+            print(json.dumps({}))
+            return
+
+        input_data = json.loads(raw_input)
+
+        # Extract data from hook input
+        tool_name = input_data.get("tool_name")
+        tool_input = input_data.get("tool_input", {})
+        agent_id = input_data.get("agent_id")
+
+        # Skip logging our own tools to prevent recursion
+        # MCP tools are named like "mcp__omni-cortex__cortex_remember"
+        if tool_name and ("cortex_" in tool_name or "omni-cortex" in tool_name):
+            print(json.dumps({}))
+            return
+
+        session_id = os.environ.get("CLAUDE_SESSION_ID")
+        project_path = os.environ.get("CLAUDE_PROJECT_DIR", os.getcwd())
+
+        # Auto-initialize database (creates if not exists)
+        db_path = get_db_path()
+        conn = ensure_database(db_path)
+
+        # Insert activity record
+        cursor = conn.cursor()
+        cursor.execute(
+            """
+            INSERT INTO activities (
+                id, session_id, agent_id, timestamp, event_type,
+                tool_name, tool_input, project_path
+            ) VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+            """,
+            (
+                generate_id(),
+                session_id,
+                agent_id,
+                datetime.now(timezone.utc).isoformat(),
+                "pre_tool_use",
+                tool_name,
+                truncate(json.dumps(tool_input, default=str)),
+                project_path,
+            ),
+        )
+        conn.commit()
+        conn.close()
+
+        # Return empty response (no modification to tool call)
+        print(json.dumps({}))
+
+    except Exception as e:
+        # Hooks should never block - log error but continue
+        print(json.dumps({"systemMessage": f"Cortex pre_tool_use: {e}"}))
+
+    sys.exit(0)
+
+
+if __name__ == "__main__":
+    main()

omni_cortex-1.0.4.data/data/share/omni-cortex/hooks/stop.py

@@ -0,0 +1,184 @@
+#!/usr/bin/env python3
+"""Stop hook - logs session end when Claude Code stops.
+
+This hook is called when Claude Code exits or the session ends.
+It finalizes the session and generates a summary.
+
+Hook configuration for settings.json:
+{
+    "hooks": {
+        "Stop": [
+            {
+                "type": "command",
+                "command": "python hooks/stop.py"
+            }
+        ]
+    }
+}
+"""
+
+import json
+import sys
+import os
+import sqlite3
+from datetime import datetime, timezone
+from pathlib import Path
+
+
+def get_db_path() -> Path:
+    """Get the database path for the current project."""
+    project_path = os.environ.get("CLAUDE_PROJECT_DIR", os.getcwd())
+    return Path(project_path) / ".omni-cortex" / "cortex.db"
+
+
+def generate_id(prefix: str) -> str:
+    """Generate a unique ID."""
+    timestamp_ms = int(datetime.now().timestamp() * 1000)
+    random_hex = os.urandom(4).hex()
+    return f"{prefix}_{timestamp_ms}_{random_hex}"
+
+
+def main():
+    """Process Stop hook."""
+    try:
+        # Read input from stdin
+        input_data = json.load(sys.stdin)
+
+        db_path = get_db_path()
+
+        # Only process if database exists
+        if not db_path.exists():
+            print(json.dumps({}))
+            return
+
+        session_id = os.environ.get("CLAUDE_SESSION_ID")
+        if not session_id:
+            print(json.dumps({}))
+            return
+
+        now = datetime.now(timezone.utc).isoformat()
+
+        # Connect to database
+        conn = sqlite3.connect(str(db_path))
+        conn.row_factory = sqlite3.Row
+        cursor = conn.cursor()
+
+        # Check if session exists
+        cursor.execute("SELECT id FROM sessions WHERE id = ?", (session_id,))
+        if not cursor.fetchone():
+            print(json.dumps({}))
+            conn.close()
+            return
+
+        # End the session
+        cursor.execute(
+            "UPDATE sessions SET ended_at = ? WHERE id = ? AND ended_at IS NULL",
+            (now, session_id),
+        )
+
+        # Gather session statistics
+        cursor.execute(
+            "SELECT COUNT(*) as cnt FROM activities WHERE session_id = ?",
+            (session_id,),
+        )
+        total_activities = cursor.fetchone()["cnt"]
+
+        cursor.execute(
+            "SELECT COUNT(*) as cnt FROM memories WHERE source_session_id = ?",
+            (session_id,),
+        )
+        total_memories = cursor.fetchone()["cnt"]
+
+        # Get tools used
+        cursor.execute(
+            """
+            SELECT tool_name, COUNT(*) as cnt
+            FROM activities
+            WHERE session_id = ? AND tool_name IS NOT NULL
+            GROUP BY tool_name
+            """,
+            (session_id,),
+        )
+        tools_used = {row["tool_name"]: row["cnt"] for row in cursor.fetchall()}
+
+        # Get files modified
+        cursor.execute(
+            """
+            SELECT DISTINCT file_path
+            FROM activities
+            WHERE session_id = ? AND file_path IS NOT NULL
+            """,
+            (session_id,),
+        )
+        files_modified = [row["file_path"] for row in cursor.fetchall()]
+
+        # Get errors
+        cursor.execute(
+            """
+            SELECT error_message
+            FROM activities
+            WHERE session_id = ? AND success = 0 AND error_message IS NOT NULL
+            LIMIT 10
+            """,
+            (session_id,),
+        )
+        key_errors = [row["error_message"] for row in cursor.fetchall()]
+
+        # Create or update summary
+        cursor.execute(
+            "SELECT id FROM session_summaries WHERE session_id = ?",
+            (session_id,),
+        )
+        existing = cursor.fetchone()
+
+        if existing:
+            cursor.execute(
+                """
+                UPDATE session_summaries
+                SET key_errors = ?, files_modified = ?, tools_used = ?,
+                    total_activities = ?, total_memories_created = ?
+                WHERE session_id = ?
+                """,
+                (
+                    json.dumps(key_errors) if key_errors else None,
+                    json.dumps(files_modified) if files_modified else None,
+                    json.dumps(tools_used) if tools_used else None,
+                    total_activities,
+                    total_memories,
+                    session_id,
+                ),
+            )
+        else:
+            cursor.execute(
+                """
+                INSERT INTO session_summaries (
+                    id, session_id, key_errors, files_modified, tools_used,
+                    total_activities, total_memories_created, created_at
+                ) VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+                """,
+                (
+                    generate_id("sum"),
+                    session_id,
+                    json.dumps(key_errors) if key_errors else None,
+                    json.dumps(files_modified) if files_modified else None,
+                    json.dumps(tools_used) if tools_used else None,
+                    total_activities,
+                    total_memories,
+                    now,
+                ),
+            )
+
+        conn.commit()
+        conn.close()
+
+        print(json.dumps({}))
+
+    except Exception as e:
+        # Hooks should never block
+        print(json.dumps({"systemMessage": f"Cortex stop: {e}"}))
+
+    sys.exit(0)
+
+
+if __name__ == "__main__":
+    main()

omni_cortex-1.0.4.data/data/share/omni-cortex/hooks/subagent_stop.py

@@ -0,0 +1,120 @@
+#!/usr/bin/env python3
+"""SubagentStop hook - logs when a subagent completes.
+
+This hook is called when a subagent (spawned by the Task tool) finishes.
+It logs the subagent completion and any results.
+
+Hook configuration for settings.json:
+{
+    "hooks": {
+        "SubagentStop": [
+            {
+                "type": "command",
+                "command": "python hooks/subagent_stop.py"
+            }
+        ]
+    }
+}
+"""
+
+import json
+import sys
+import os
+import sqlite3
+from datetime import datetime, timezone
+from pathlib import Path
+
+
+def get_db_path() -> Path:
+    """Get the database path for the current project."""
+    project_path = os.environ.get("CLAUDE_PROJECT_DIR", os.getcwd())
+    return Path(project_path) / ".omni-cortex" / "cortex.db"
+
+
+def generate_id() -> str:
+    """Generate a unique activity ID."""
+    timestamp_ms = int(datetime.now().timestamp() * 1000)
+    random_hex = os.urandom(4).hex()
+    return f"act_{timestamp_ms}_{random_hex}"
+
+
+def truncate(text: str, max_length: int = 10000) -> str:
+    """Truncate text to max length."""
+    if len(text) <= max_length:
+        return text
+    return text[:max_length - 20] + "\n... [truncated]"
+
+
+def main():
+    """Process SubagentStop hook."""
+    try:
+        # Read input from stdin
+        input_data = json.load(sys.stdin)
+
+        db_path = get_db_path()
+
+        # Only log if database exists
+        if not db_path.exists():
+            print(json.dumps({}))
+            return
+
+        # Extract data from hook input
+        subagent_id = input_data.get("subagent_id")
+        subagent_type = input_data.get("subagent_type", "subagent")
+        result = input_data.get("result", {})
+
+        session_id = os.environ.get("CLAUDE_SESSION_ID")
+        project_path = os.environ.get("CLAUDE_PROJECT_DIR", os.getcwd())
+        now = datetime.now(timezone.utc).isoformat()
+
+        # Connect to database
+        conn = sqlite3.connect(str(db_path))
+        cursor = conn.cursor()
+
+        # Log the subagent completion as an activity
+        cursor.execute(
+            """
+            INSERT INTO activities (
+                id, session_id, agent_id, timestamp, event_type,
+                tool_name, tool_output, success, project_path
+            ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
+            """,
+            (
+                generate_id(),
+                session_id,
+                subagent_id,
+                now,
+                "subagent_stop",
+                f"subagent_{subagent_type}",
+                truncate(json.dumps(result, default=str)),
+                1,
+                project_path,
+            ),
+        )
+
+        # Update or create agent record
+        cursor.execute(
+            """
+            INSERT INTO agents (id, name, type, first_seen, last_seen, total_activities)
+            VALUES (?, ?, ?, ?, ?, 1)
+            ON CONFLICT(id) DO UPDATE SET
+                last_seen = ?,
+                total_activities = total_activities + 1
+            """,
+            (subagent_id, None, "subagent", now, now, now),
+        )
+
+        conn.commit()
+        conn.close()
+
+        print(json.dumps({}))
+
+    except Exception as e:
+        # Hooks should never block
+        print(json.dumps({"systemMessage": f"Cortex subagent_stop: {e}"}))
+
+    sys.exit(0)
+
+
+if __name__ == "__main__":
+    main()