omni-cortex 1.3.0__py3-none-any.whl → 1.11.3__py3-none-any.whl

{omni_cortex-1.3.0.data → omni_cortex-1.11.3.data}/data/share/omni-cortex/hooks/pre_tool_use.py

@@ -18,12 +18,57 @@ Hook configuration for settings.json:
 """
 
 import json
+import re
 import sys
 import os
 import sqlite3
 from datetime import datetime, timezone
 from pathlib import Path
 
+# Import shared session management
+from session_utils import get_or_create_session
+
+
+# Patterns for sensitive field names that should be redacted
+SENSITIVE_FIELD_PATTERNS = [
+    r'(?i)(api[_-]?key|apikey)',
+    r'(?i)(password|passwd|pwd)',
+    r'(?i)(secret|token|credential)',
+    r'(?i)(auth[_-]?token|access[_-]?token)',
+    r'(?i)(private[_-]?key|ssh[_-]?key)',
+]
+
+
+def redact_sensitive_fields(data: dict) -> dict:
+    """Redact sensitive fields from a dictionary for safe logging.
+
+    Recursively processes nested dicts and lists.
+    """
+    if not isinstance(data, dict):
+        return data
+
+    result = {}
+    for key, value in data.items():
+        # Check if key matches sensitive patterns
+        is_sensitive = any(
+            re.search(pattern, str(key))
+            for pattern in SENSITIVE_FIELD_PATTERNS
+        )
+
+        if is_sensitive:
+            result[key] = '[REDACTED]'
+        elif isinstance(value, dict):
+            result[key] = redact_sensitive_fields(value)
+        elif isinstance(value, list):
+            result[key] = [
+                redact_sensitive_fields(item) if isinstance(item, dict) else item
+                for item in value
+            ]
+        else:
+            result[key] = value
+
+    return result
+
 
 def get_db_path() -> Path:
     """Get the database path for the current project."""
@@ -115,13 +160,18 @@ def main():
             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)
 
+        # Get or create session (auto-manages session lifecycle)
+        session_id = get_or_create_session(conn, project_path)
+
+        # Redact sensitive fields before logging
+        safe_input = redact_sensitive_fields(tool_input) if isinstance(tool_input, dict) else tool_input
+
         # Insert activity record
         cursor = conn.cursor()
         cursor.execute(
@@ -138,7 +188,7 @@ def main():
                 datetime.now(timezone.utc).isoformat(),
                 "pre_tool_use",
                 tool_name,
-                truncate(json.dumps(tool_input, default=str)),
+                truncate(json.dumps(safe_input, default=str)),
                 project_path,
             ),
         )

omni_cortex-1.11.3.data/data/share/omni-cortex/hooks/session_utils.py

@@ -0,0 +1,186 @@
+#!/usr/bin/env python3
+"""Shared session management utilities for Claude Code hooks.
+
+This module provides session management functionality that can be shared
+across pre_tool_use.py and post_tool_use.py hooks to ensure consistent
+session tracking.
+
+Session Management Logic:
+1. Check for existing session file at `.omni-cortex/current_session.json`
+2. If session exists and is valid (not timed out), use it
+3. If no valid session, create a new one in both file and database
+4. Update last_activity_at on each use to track session activity
+"""
+
+import json
+import os
+import sqlite3
+import time
+from datetime import datetime, timezone
+from pathlib import Path
+from typing import Optional
+
+
+# Session timeout in seconds (4 hours of inactivity = new session)
+SESSION_TIMEOUT_SECONDS = 4 * 60 * 60
+
+
+def generate_session_id() -> str:
+    """Generate a unique session ID matching the MCP format.
+
+    Returns:
+        Session ID in format: sess_{timestamp_ms}_{random_hex}
+    """
+    timestamp_ms = int(time.time() * 1000)
+    random_hex = os.urandom(4).hex()
+    return f"sess_{timestamp_ms}_{random_hex}"
+
+
+def get_session_file_path() -> Path:
+    """Get the path to the current session file.
+
+    Returns:
+        Path to .omni-cortex/current_session.json
+    """
+    project_path = os.environ.get("CLAUDE_PROJECT_DIR", os.getcwd())
+    return Path(project_path) / ".omni-cortex" / "current_session.json"
+
+
+def load_session_file() -> Optional[dict]:
+    """Load the current session from file if it exists.
+
+    Returns:
+        Session data dict or None if file doesn't exist or is invalid
+    """
+    session_file = get_session_file_path()
+    if not session_file.exists():
+        return None
+
+    try:
+        with open(session_file, "r") as f:
+            return json.load(f)
+    except (json.JSONDecodeError, IOError):
+        return None
+
+
+def save_session_file(session_data: dict) -> None:
+    """Save the current session to file.
+
+    Args:
+        session_data: Dict containing session_id, project_path, started_at, last_activity_at
+    """
+    session_file = get_session_file_path()
+    session_file.parent.mkdir(parents=True, exist_ok=True)
+
+    with open(session_file, "w") as f:
+        json.dump(session_data, f, indent=2)
+
+
+def is_session_valid(session_data: dict) -> bool:
+    """Check if a session is still valid (not timed out).
+
+    A session is valid if:
+    - It has a last_activity_at timestamp
+    - The timestamp is within SESSION_TIMEOUT_SECONDS of now
+
+    Args:
+        session_data: Session dict with last_activity_at field
+
+    Returns:
+        True if session is valid, False otherwise
+    """
+    last_activity = session_data.get("last_activity_at")
+    if not last_activity:
+        return False
+
+    try:
+        last_time = datetime.fromisoformat(last_activity.replace("Z", "+00:00"))
+        now = datetime.now(timezone.utc)
+        elapsed_seconds = (now - last_time).total_seconds()
+        return elapsed_seconds < SESSION_TIMEOUT_SECONDS
+    except (ValueError, TypeError):
+        return False
+
+
+def create_session_in_db(conn: sqlite3.Connection, session_id: str, project_path: str) -> None:
+    """Create a new session record in the database.
+
+    Also creates the sessions table if it doesn't exist (for first-run scenarios).
+
+    Args:
+        conn: SQLite database connection
+        session_id: The session ID to create
+        project_path: The project directory path
+    """
+    cursor = conn.cursor()
+    now = datetime.now(timezone.utc).isoformat()
+
+    # Check if sessions table exists (it might not if only activities table was created)
+    cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='sessions'")
+    if cursor.fetchone() is None:
+        # Create sessions table with minimal schema
+        conn.executescript("""
+            CREATE TABLE IF NOT EXISTS sessions (
+                id TEXT PRIMARY KEY,
+                project_path TEXT NOT NULL,
+                started_at TEXT NOT NULL,
+                ended_at TEXT,
+                summary TEXT,
+                tags TEXT,
+                metadata TEXT
+            );
+            CREATE INDEX IF NOT EXISTS idx_sessions_started ON sessions(started_at DESC);
+            CREATE INDEX IF NOT EXISTS idx_sessions_project ON sessions(project_path);
+        """)
+        conn.commit()
+
+    cursor.execute(
+        """
+        INSERT OR IGNORE INTO sessions (id, project_path, started_at)
+        VALUES (?, ?, ?)
+        """,
+        (session_id, project_path, now),
+    )
+    conn.commit()
+
+
+def get_or_create_session(conn: sqlite3.Connection, project_path: str) -> str:
+    """Get the current session ID, creating a new one if needed.
+
+    Session management logic:
+    1. Check for existing session file
+    2. If exists and not timed out, use it and update last_activity
+    3. If doesn't exist or timed out, create new session
+
+    Args:
+        conn: SQLite database connection
+        project_path: The project directory path
+
+    Returns:
+        The session ID to use for activity logging
+    """
+    session_data = load_session_file()
+    now_iso = datetime.now(timezone.utc).isoformat()
+
+    if session_data and is_session_valid(session_data):
+        # Update last activity time
+        session_data["last_activity_at"] = now_iso
+        save_session_file(session_data)
+        return session_data["session_id"]
+
+    # Create new session
+    session_id = generate_session_id()
+
+    # Create in database
+    create_session_in_db(conn, session_id, project_path)
+
+    # Save to file
+    session_data = {
+        "session_id": session_id,
+        "project_path": project_path,
+        "started_at": now_iso,
+        "last_activity_at": now_iso,
+    }
+    save_session_file(session_data)
+
+    return session_id

{omni_cortex-1.3.0.dist-info → omni_cortex-1.11.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: omni-cortex
-Version: 1.3.0
+Version: 1.11.3
 Summary: Give Claude Code a perfect memory - auto-logs everything, searches smartly, and gets smarter over time
 Project-URL: Homepage, https://github.com/AllCytes/Omni-Cortex
 Project-URL: Repository, https://github.com/AllCytes/Omni-Cortex
@@ -21,9 +21,11 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.10
+Requires-Dist: claude-agent-sdk>=0.1.0
 Requires-Dist: httpx>=0.25.0
 Requires-Dist: mcp>=1.0.0
 Requires-Dist: pydantic>=2.0.0
+Requires-Dist: python-dotenv>=1.0.0
 Requires-Dist: pyyaml>=6.0.0
 Provides-Extra: dev
 Requires-Dist: black>=23.0.0; extra == 'dev'
@@ -76,7 +78,180 @@ A universal memory system for Claude Code that combines activity logging with in
 - **Importance Decay**: Frequently accessed memories naturally surface
 - **Auto Activity Logging**: Automatically logs all tool calls via hooks
 
-## Installation
+## Getting Started (5 Minutes)
+
+A step-by-step guide to get Omni Cortex running on your machine.
+
+### Prerequisites
+
+- **Python 3.10+** - Check with `python --version`
+- **Claude Code CLI** - The Anthropic CLI tool
+- **pip** - Python package manager (comes with Python)
+
+### Step 1: Install the Package
+
+**Option A: From PyPI (Recommended for most users)**
+```bash
+pip install omni-cortex
+```
+
+**Option B: From Source (For development/contributions)**
+```bash
+git clone https://github.com/AllCytes/Omni-Cortex.git
+cd Omni-Cortex
+pip install -e ".[semantic]"
+```
+
+**Expected output:**
+```
+Successfully installed omni-cortex-1.7.1
+```
+
+### Step 2: Run the Setup
+
+```bash
+omni-cortex-setup
+```
+
+This automatically:
+- Adds Omni Cortex as an MCP server in `~/.claude.json`
+- Configures hooks in `~/.claude/settings.json` for activity logging
+
+**Expected output:**
+```
+✓ MCP server configured
+✓ Hooks configured
+Setup complete! Restart Claude Code to activate.
+```
+
+### Step 3: Restart Claude Code
+
+Close and reopen your Claude Code terminal. This loads the new MCP configuration.
+
+### Step 4: Verify It's Working
+
+In Claude Code, try storing a memory:
+
+```
+Ask Claude: "Remember that the database uses SQLite for storage"
+```
+
+Claude should use the `cortex_remember` tool. Then verify:
+
+```
+Ask Claude: "What do you remember about the database?"
+```
+
+Claude should use `cortex_recall` and find your memory.
+
+### Step 5: Start the Dashboard (Optional)
+
+The web dashboard lets you browse and search memories visually.
+
+```bash
+# Start the dashboard (opens http://localhost:5173)
+omni-cortex-dashboard
+```
+
+Or manually:
+```bash
+# Terminal 1: Backend (uses dashboard's own venv)
+cd dashboard/backend
+.venv/Scripts/python -m uvicorn main:app --host 127.0.0.1 --port 8765 --reload
+
+# Terminal 2: Frontend
+cd dashboard/frontend
+npm install
+npm run dev
+```
+
+Open http://localhost:5173 in your browser.
+
+**Note:** The dashboard has its own virtual environment at `dashboard/backend/.venv` with FastAPI and other web dependencies. This is separate from the project root `.venv` which contains the MCP server package.
+
+### Troubleshooting
+
+<details>
+<summary><b>❌ "omni-cortex-setup" command not found</b></summary>
+
+**Cause:** pip installed to a location not in your PATH.
+
+**Solution:**
+```bash
+# Find where pip installed it
+python -m omni_cortex.setup
+
+# Or add Python scripts to PATH (Windows)
+# Add %APPDATA%\Python\Python3x\Scripts to your PATH
+
+# On macOS/Linux, ensure ~/.local/bin is in PATH
+export PATH="$HOME/.local/bin:$PATH"
+```
+</details>
+
+<details>
+<summary><b>❌ Claude doesn't see cortex_* tools</b></summary>
+
+**Cause:** MCP server not configured or Claude Code not restarted.
+
+**Solution:**
+1. Check `~/.claude.json` contains the `omni-cortex` MCP server entry
+2. Fully close and reopen Claude Code (not just the terminal)
+3. Run `omni-cortex-setup` again if needed
+</details>
+
+<details>
+<summary><b>❌ "ModuleNotFoundError: No module named 'omni_cortex'"</b></summary>
+
+**Cause:** Python environment mismatch.
+
+**Solution:**
+```bash
+# Ensure you're using the same Python that pip used
+which python  # or `where python` on Windows
+pip show omni-cortex  # Check if installed
+
+# Reinstall if needed
+pip install --force-reinstall omni-cortex
+```
+</details>
+
+<details>
+<summary><b>❌ Dashboard won't start</b></summary>
+
+**Cause:** Missing dependencies or port conflict.
+
+**Solution:**
+```bash
+# Install backend dependencies
+cd dashboard/backend
+pip install -e .
+
+# Check if port 8765 is in use
+# Windows: netstat -ano | findstr :8765
+# macOS/Linux: lsof -i :8765
+
+# Use a different port if needed
+uvicorn main:app --port 8766
+```
+</details>
+
+<details>
+<summary><b>❌ Semantic search not working</b></summary>
+
+**Cause:** Semantic extras not installed.
+
+**Solution:**
+```bash
+pip install omni-cortex[semantic]
+```
+
+First search will download the embedding model (~100MB).
+</details>
+
+---
+
+## Installation (Detailed)
 
 ### Quick Install (Recommended)
 
@@ -294,18 +469,72 @@ The PDFs use a light theme with blue/purple/green accents. Edit `docs/create_pdf
 
 ## Development
 
+### Quick Setup (with Claude Code)
+
+If you're using Claude Code, just run:
+
 ```bash
-# Install dev dependencies
-pip install -e ".[dev]"
+/dev-setup
+```
+
+This will guide you through setting up the development environment.
+
+### Manual Setup
 
-# Run tests
+```bash
+# Clone and install in editable mode
+git clone https://github.com/AllCytes/Omni-Cortex.git
+cd Omni-Cortex
+pip install -e .
+
+# Dashboard backend has its own venv (already included in repo)
+# If missing, set it up:
+cd dashboard/backend
+python -m venv .venv
+.venv/Scripts/pip install -r requirements.txt  # Windows
+# .venv/bin/pip install -r requirements.txt    # macOS/Linux
+cd ../frontend && npm install
+cd ../..
+
+# Verify installation
+omni-cortex --help
+omni-cortex-dashboard --help
+```
+
+**Important**: Always use `pip install -e .` (editable mode) so changes are immediately reflected without reinstalling.
+
+### Project Structure
+
+```
+omni-cortex/
+├── .venv/                       # Project root venv (omni-cortex MCP package)
+├── src/omni_cortex/             # MCP server source code
+├── dashboard/
+│   ├── backend/
+│   │   ├── .venv/               # Dashboard backend venv (FastAPI, uvicorn)
+│   │   ├── main.py              # FastAPI application
+│   │   └── database.py          # Database queries
+│   └── frontend/                # Vue 3 + Vite frontend
+├── adws/                        # Agentic Development Workflows
+├── specs/                       # Implementation plans
+│   ├── todo/                    # Plans waiting to be built
+│   └── done/                    # Completed plans
+└── tests/                       # Unit tests
+```
+
+**Why two venvs?** The dashboard is a standalone web application that can be packaged/deployed separately from the MCP server. They have different dependencies (MCP server needs `mcp`, dashboard needs `fastapi`).
+
+### Running Tests
+
+```bash
 pytest
 
-# Format code
-black src tests
-ruff check src tests
+# With coverage
+pytest --cov=src/omni_cortex
 ```
 
+See [CONTRIBUTING.md](CONTRIBUTING.md) for full development guidelines.
+
 ## Security
 
 Omni Cortex v1.0.3 has been security reviewed:

omni_cortex-1.11.3.dist-info/RECORD

@@ -0,0 +1,25 @@
+omni_cortex-1.11.3.data/data/share/omni-cortex/hooks/post_tool_use.py,sha256=kHBRCP6VctSk3evApsgDaL3UElGSEBGwRfV4hBZPqgE,14539
+omni_cortex-1.11.3.data/data/share/omni-cortex/hooks/pre_tool_use.py,sha256=mkZ7eeBnjWkIgNnrqfYSXIhhLNYYk4hQx_6F0pNrGoc,6395
+omni_cortex-1.11.3.data/data/share/omni-cortex/hooks/session_utils.py,sha256=3SKPCytqWuRPOupWdzmwBoKBDJqtLcT1Nle_pueDQUY,5746
+omni_cortex-1.11.3.data/data/share/omni-cortex/hooks/stop.py,sha256=T1bwcmbTLj0gzjrVvFBT1zB6wff4J2YkYBAY-ZxZI5g,5336
+omni_cortex-1.11.3.data/data/share/omni-cortex/hooks/subagent_stop.py,sha256=V9HQSFGNOfkg8ZCstPEy4h5V8BP4AbrVr8teFzN1kNk,3314
+omni_cortex-1.11.3.data/data/share/omni-cortex/dashboard/backend/.env.example,sha256=9xS7-UiWlMddRwzlyyyKNHAMlNTsgH-2sPV266guJpQ,372
+omni_cortex-1.11.3.data/data/share/omni-cortex/dashboard/backend/backfill_summaries.py,sha256=ElchfcBv4pmVr2PsePCgFlCyuvf4_jDJj_C3AmMhu7U,8973
+omni_cortex-1.11.3.data/data/share/omni-cortex/dashboard/backend/chat_service.py,sha256=5UCvLayZGeSdGsYAzOeupumclAhoFLusGYLdyl33ANc,9304
+omni_cortex-1.11.3.data/data/share/omni-cortex/dashboard/backend/database.py,sha256=LAN7GSM2tvMcJaL0RrGJurH9-tw3cs2QtPduqCbLvj0,34974
+omni_cortex-1.11.3.data/data/share/omni-cortex/dashboard/backend/image_service.py,sha256=NP6ojFpHb6iNTYRkXqYu1CL6WvooZpZ54mjLiWSWG_g,19205
+omni_cortex-1.11.3.data/data/share/omni-cortex/dashboard/backend/logging_config.py,sha256=WnunFGET9zlsn9WBpVsio2zI7BiUQanE0xzAQQxIhII,3944
+omni_cortex-1.11.3.data/data/share/omni-cortex/dashboard/backend/main.py,sha256=rJrmYJvkGhRsXOdYKOTRPMVnwA00W5QoGJ_Aa3v-TRE,46219
+omni_cortex-1.11.3.data/data/share/omni-cortex/dashboard/backend/models.py,sha256=LkmcYq1imsyDlMYnX3Z_FOTmPsu37MQEfJSI-w5EjvM,7330
+omni_cortex-1.11.3.data/data/share/omni-cortex/dashboard/backend/project_config.py,sha256=ZxGoeRpHvN5qQyf2hRxrAZiHrPSwdQp59f0di6O1LKM,4352
+omni_cortex-1.11.3.data/data/share/omni-cortex/dashboard/backend/project_scanner.py,sha256=lwFXS8iJbOoxf7FAyo2TjH25neaMHiJ8B3jS57XxtDI,5713
+omni_cortex-1.11.3.data/data/share/omni-cortex/dashboard/backend/prompt_security.py,sha256=LcdZhYy1CfpSq_4BPO6lMJ15phc2ZXLUSBAnAvODVCI,3423
+omni_cortex-1.11.3.data/data/share/omni-cortex/dashboard/backend/pyproject.toml,sha256=9pbbGQXLe1Xd06nZAtDySCHIlfMWvPaB-C6tGZR6umc,502
+omni_cortex-1.11.3.data/data/share/omni-cortex/dashboard/backend/security.py,sha256=nQsoPE0n5dtY9ive00d33W1gL48GgK7C5Ae0BK2oW2k,3479
+omni_cortex-1.11.3.data/data/share/omni-cortex/dashboard/backend/uv.lock,sha256=miB9zGGSirBkjDE-OZTPCnv43Yc98xuAz_Ne8vTNFHg,186004
+omni_cortex-1.11.3.data/data/share/omni-cortex/dashboard/backend/websocket_manager.py,sha256=gNQLd94AcC-InumGQmUolREhiogCzilYWpLN8SRZjHI,3645
+omni_cortex-1.11.3.dist-info/METADATA,sha256=oM6RvKOj_J7rCFrqbgZOiYHxFLxy-eLnIlq5QHIEf_o,15712
+omni_cortex-1.11.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+omni_cortex-1.11.3.dist-info/entry_points.txt,sha256=rohx4mFH2ffZmMb9QXPZmFf-ZGjA3jpKVDVeET-ttiM,150
+omni_cortex-1.11.3.dist-info/licenses/LICENSE,sha256=oG_397owMmi-Umxp5sYocJ6RPohp9_bDNnnEu9OUphg,1072
+omni_cortex-1.11.3.dist-info/RECORD,,