cli-memory-os 0.1.0__tar.gz → 0.1.3__tar.gz

{cli_memory_os-0.1.0/cli_memory_os.egg-info → cli_memory_os-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cli-memory-os
-Version: 0.1.0
+Version: 0.1.3
 Summary: CLI-based Personal Knowledge Operating System
 Author-email: Anirudh T <anirudh200584@gmail.com>
 License-Expression: MIT

cli_memory_os-0.1.3/cli/commands/ask.py

@@ -0,0 +1,67 @@
+"""
+Command: memory-os ask
+
+Queries the RAG pipeline with a question.
+"""
+
+import sys
+import sqlite3
+from core.llm import run_hybrid_rag
+from storage.db import init_db
+
+
+def run_and_print_ask(query: str):
+    """Run RAG query pipeline and print the structured result blocks."""
+    rag_res = run_hybrid_rag(query)
+    
+    print("========================================")
+    print("ANSWER")
+    print("========================================")
+    print(rag_res["answer"])
+    print("\n========================================")
+    print("SOURCES")
+    print("========================================")
+    if rag_res.get("sources"):
+        for s in rag_res["sources"]:
+            print(f"- {s}")
+    else:
+        print("None")
+    print("\n========================================")
+    print("REPOSITORIES USED")
+    print("========================================")
+    if rag_res.get("repositories"):
+        for r in rag_res["repositories"]:
+            print(f"- {r}")
+    else:
+        print("None")
+    print("\n========================================")
+    confidence = rag_res.get("confidence", 0.0)
+    print(f"Confidence: {confidence:.2f}")
+    print("========================================")
+
+
+def execute(args):
+    """Run the ask command parser handler."""
+    # Ensure database is initialized
+    try:
+        init_db()
+    except sqlite3.OperationalError as e:
+        print("----------------------------------")
+        print("Workspace has not been initialized.")
+        print("")
+        print("Run:")
+        print("memory-os init")
+        print("----------------------------------")
+        sys.exit(1)
+
+    if not args.question:
+        print("Usage: memory-os ask <question>")
+        sys.exit(1)
+
+    query = " ".join(args.question)
+    
+    try:
+        run_and_print_ask(query)
+    except Exception as e:
+        print(f"❌ Error executing query: {e}")
+        sys.exit(1)

{cli_memory_os-0.1.0 → cli_memory_os-0.1.3}/cli/commands/benchmark.py

@@ -6,8 +6,7 @@ graph lookups, LLM invocation, and RAG pipelines.
 """
 
 import time
-import sys
-from infrastructure.health import check_neo4j, check_qdrant, check_groq_api, check_memory_usage
+from infrastructure.health import check_qdrant, check_groq_api, check_memory_usage
 from core.embedder import Embedder
 
 

{cli_memory_os-0.1.0 → cli_memory_os-0.1.3}/cli/commands/doctor.py

@@ -53,8 +53,19 @@ def execute(args):
             "Active Plugins": "No plugins are authenticated. Run 'memory-os init' to set up connector logins.",
         }
         for name, detail in failed:
-            rec = recommendations.get(name, "Check log file logs/memory_os.log for errors.")
-            print(f"  * {name}: {rec} (Detail: {detail})")
+            if name == "Neo4j" and "Authentication failed" in detail:
+                print("  * Neo4j: Neo4j is running but authentication failed.")
+                print("    Recommendations:")
+                print("      memory-os config set neo4j.user <username>")
+                print("      memory-os config set neo4j.password <password>")
+                print("      or")
+                print("      memory-os init")
+            elif name == "Neo4j" and "Unreachable" in detail:
+                rec = recommendations.get(name, "Check log file logs/memory_os.log for errors.")
+                print(f"  * {name}: {rec} (Detail: {detail})")
+            else:
+                rec = recommendations.get(name, "Check log file logs/memory_os.log for errors.")
+                print(f"  * {name}: {rec} (Detail: {detail})")
         print("─────────────────────────────")
     else:
         print("\n  All checks passed.")

{cli_memory_os-0.1.0 → cli_memory_os-0.1.3}/cli/commands/import_cmd.py

@@ -6,7 +6,6 @@ into the active workspace profile.
 """
 
 import sys
-import os
 import json
 import zipfile
 import shutil

{cli_memory_os-0.1.0 → cli_memory_os-0.1.3}/cli/commands/init.py

@@ -1,26 +1,21 @@
 """
 Command: memory-os init
 
-Performs first-time setup and onboarding for Memory-OS.
-Provisions the workspace directory structure, creates config.toml,
-spins up Neo4j and Qdrant via docker compose, initializes SQLite,
-pre-warms the embedding model, and launches Composio toolkit OAuth flows.
+Performs first-time setup and onboarding for Memory-OS using ComposeManager.
 """
 
 import sys
 import uuid
-import time
 import os
-from pathlib import Path
 from infrastructure.workspace import ensure_workspace, get_db_path
 from infrastructure.config import generate_default_config, save_config, load_config
-from infrastructure.docker import check_docker_installed, check_docker_compose_installed, check_docker_running
-from infrastructure.compose import compose_up, wait_for_services
-from infrastructure.health import run_all_checks
+from infrastructure.compose import ComposeManager, wait_for_services
+from infrastructure.health import run_all_checks, check_docker, check_neo4j, check_qdrant
 from storage.db import init_db
 from core.embedder import Embedder
 
 
+
 def get_input(prompt: str, secret: bool = False, default: str = "") -> str:
     """Prompt the user for input with an optional default value."""
     import getpass
@@ -38,14 +33,13 @@ def get_input(prompt: str, secret: bool = False, default: str = "") -> str:
 
 
 def execute(args):
-    """Run the init command."""
+    """Run the init command wizard following the exact 11 steps."""
     print("==================================================")
     print("🧠 MEMORY-OS INITIALIZATION WIZARD")
     print("==================================================")
 
-    # 1. System Dependency Checks
-    print("\n[1/7] Checking system dependencies...")
-    
+    # ── Step 1: Check Python ──────────────────────────────────
+    print("\n[Step 1/11] Checking Python version...")
     python_ok = sys.version_info >= (3, 12)
     py_ver = f"{sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}"
     if not python_ok:
@@ -53,52 +47,73 @@ def execute(args):
         sys.exit(1)
     print(f"  ✓ Python version: {py_ver}")
 
-    docker_ok, docker_ver = check_docker_installed()
+    # ── Step 2: Check Docker daemon ──────────────────────────
+    print("\n[Step 2/11] Checking Docker daemon...")
+    docker_ok, docker_detail = check_docker()
     if not docker_ok:
-        print("----------------------------------")
-        print("Docker is not installed or not in PATH.")
-        print("")
-        print("Please install Docker Desktop and rerun:")
-        print("memory-os init")
-        print("----------------------------------")
-        sys.exit(1)
-    print(f"  ✓ Docker: {docker_ver}")
-
-    compose_ok, compose_ver = check_docker_compose_installed()
-    if not compose_ok:
-        print("----------------------------------")
-        print("Docker Compose is not installed.")
-        print("")
-        print("Please install Docker Compose and rerun:")
-        print("memory-os init")
-        print("----------------------------------")
-        sys.exit(1)
-    print(f"  ✓ Docker Compose: {compose_ver}")
-
-    if not check_docker_running():
         print("----------------------------------")
         print("Docker Desktop is not running.")
         print("")
+        print("stderr:")
+        print(docker_detail)
+        print("")
         print("Start Docker Desktop and rerun:")
         print("memory-os init")
         print("----------------------------------")
         sys.exit(1)
     print("  ✓ Docker Daemon: Running")
 
-
-    # 2. Workspace Tree Setup
-    print("\n[2/7] Creating workspace directories...")
+    # ── Step 3: Create workspace ─────────────────────────────
+    print("\n[Step 3/11] Creating workspace directories...")
     ensure_workspace("default")
     print("  ✓ Workspace path initialized at ~/.memory-os")
 
-    # 3. Configure API Credentials & Settings
-    print("\n[3/7] Configuring credentials...")
-    neo4j_password = get_input("Set Neo4j database password", secret=True, default="memory_neo")
-    groq_api_key = get_input("Enter Groq API Key", secret=True)
-    composio_api_key = get_input("Enter Composio API Key", secret=True)
+    # ── Step 4: Generate docker-compose.yml ──────────────────
+    print("\n[Step 4/11] Generating docker-compose.yml...")
+    from infrastructure.config import get_config
+    existing_config = get_config()
+    existing_password = existing_config.get("neo4j", {}).get("password") or "memory_neo"
+    
+    manager = ComposeManager()
+    manager.generate_compose(profile="default", password=existing_password)
+    print("  ✓ docker-compose.yml generated at ~/.memory-os/docker-compose.yml")
+
+    # ── Step 5: Launch Neo4j + Qdrant ─────────────────────────
+    print("\n[Step 5/11] Launching local services (Neo4j, Qdrant)...")
+    neo4j_ok, _ = check_neo4j()
+    qdrant_ok, _ = check_qdrant()
+    if neo4j_ok and qdrant_ok:
+        print("  ✓ Neo4j already running")
+        print("  ✓ Qdrant already running")
+    else:
+        if not manager.up(profile="default"):
+            print("❌ Failed to start docker compose services.")
+            sys.exit(1)
+        print("  ✓ Containers launched.")
+
+    # ── Step 6: Wait until healthy ────────────────────────────
+    print("\n[Step 6/11] Waiting for services to become healthy...")
+    if neo4j_ok and qdrant_ok:
+        print("  ✓ Services are healthy.")
+    else:
+        if not wait_for_services(timeout=60):
+            print("❌ Services failed to start or respond to health checks in time.")
+            sys.exit(1)
+        print("  ✓ Services are healthy.")
+
+    # ── Step 7: Initialize SQLite ─────────────────────────────
+    print("\n[Step 7/11] Initializing SQLite database...")
+    os.environ["MEMORY_OS_DB_PATH"] = str(get_db_path("default"))
+    init_db()
+    print("  ✓ SQLite database schema initialized.")
+
+    # ── Step 8: Configure API Keys ───────────────────────────
+    print("\n[Step 8/11] Configuring API keys & settings...")
+    neo4j_password = get_input("Set Neo4j database password", secret=True, default=existing_password)
+    groq_api_key = get_input("Enter Groq API Key", secret=True, default=existing_config.get("groq", {}).get("api_key", ""))
+    composio_api_key = get_input("Enter Composio API Key", secret=True, default=existing_config.get("composio", {}).get("api_key", ""))
 
-    # Generate auto Composio User UUID
-    composio_user_id = str(uuid.uuid4())
+    composio_user_id = existing_config.get("composio", {}).get("user_id") or str(uuid.uuid4())
 
     answers = {
         "neo4j_password": neo4j_password,
@@ -109,49 +124,16 @@ def execute(args):
 
     config_dict = generate_default_config(answers)
     save_config(config_dict)
-    
-    # Force reload config in current process memory
     load_config()
     
-    # Overwrite MEMORY_OS_DB_PATH in environment dynamically to point to the workspace DB
-    os.environ["MEMORY_OS_DB_PATH"] = str(get_db_path("default"))
+    if neo4j_password != existing_password:
+        manager.generate_compose(profile="default", password=neo4j_password)
+        print("  ✓ docker-compose.yml updated with new password.")
 
     print("  ✓ Configuration saved to ~/.memory-os/config.toml")
 
-    # 4. Service Provisioning via Docker Compose
-    print("\n[4/7] Provisioning local services (Neo4j, Qdrant)...")
-    if not compose_up():
-        print("❌ Failed to start docker compose services.")
-        sys.exit(1)
-    print("  ✓ Containers launched. Waiting for services to become healthy...")
-    
-    if not wait_for_services(timeout=60):
-        print("❌ Services failed to start or respond to health checks in time.")
-        sys.exit(1)
-    print("  ✓ Services are healthy.")
-
-    # 5. Schema & Database Initialization
-    print("\n[5/7] Initializing SQLite database...")
-    init_db()
-    print("  ✓ SQLite database schema initialized.")
-
-    # 6. Optional Embedding Model Warming
-    print("\n[6/7] Preparing embedding model...")
-    warm_model = get_input("Download and cache embedding model (all-MiniLM-L6-v2) now? (Y/n)", default="y").lower()
-    if warm_model == "y":
-        print("  Downloading model (this may take a minute)...")
-        try:
-            embedder = Embedder()
-            _ = embedder.model
-            print("  ✓ Embedding model cached successfully.")
-        except Exception as e:
-            print(f"  ⚠️ Warning: Failed to cache model during setup: {e}")
-            print("  Model will be downloaded automatically on the first query.")
-    else:
-        print("  Skipped model download.")
-
-    # 7. Optional Composio Integration OAuth
-    print("\n[7/7] Configuring integration connectors...")
+    # ── Step 9: Authenticate Composio ────────────────────────
+    print("\n[Step 9/11] Configuring integration connectors...")
     setup_connectors = get_input("Configure connectors (GitHub, Gmail, Notion) now? (Y/n)", default="y").lower()
     if setup_connectors == "y" and composio_api_key:
         try:
@@ -174,7 +156,6 @@ def execute(args):
                         print(f"  👉 Please open this URL in your browser to complete authentication:\n     {req.redirect_url}")
                         print("  Waiting up to 30 seconds for completion...")
                         try:
-                            # Wait for active connection
                             req.wait_for_connection(timeout=30000)
                             print(f"  ✓ {name} connected successfully!")
                         except Exception:
@@ -186,8 +167,24 @@ def execute(args):
     else:
         print("  Skipped connectors configuration.")
 
-    # Verification Report
-    print("\n==================================================")
+    # ── Step 10: Warm embedding model ─────────────────────────
+    print("\n[Step 10/11] Preparing embedding model...")
+    warm_model = get_input("Download and cache embedding model (all-MiniLM-L6-v2) now? (Y/n)", default="y").lower()
+    if warm_model == "y":
+        print("  Downloading model (this may take a minute)...")
+        try:
+            embedder = Embedder()
+            _ = embedder.model
+            print("  ✓ Embedding model cached successfully.")
+        except Exception as e:
+            print(f"  ⚠️ Warning: Failed to cache model during setup: {e}")
+            print("  Model will be downloaded automatically on the first query.")
+    else:
+        print("  Skipped model download.")
+
+    # ── Step 11: Done ─────────────────────────────────────────
+    print("\n[Step 11/11] Running initial health checks...")
+    print("==================================================")
     print("🏥 RUNNING INITIAL HEALTH CHECKS")
     print("==================================================")
     checks = run_all_checks()

cli_memory_os-0.1.3/cli/commands/restart.py

@@ -0,0 +1,23 @@
+"""
+Command: memory-os restart
+
+Restarts the Docker services (Neo4j and Qdrant) via ComposeManager.
+"""
+
+import sys
+from infrastructure.compose import ComposeManager, wait_for_services
+
+
+def execute(args):
+    """Run the restart command."""
+    print("Restarting Memory-OS local services (Neo4j, Qdrant)...")
+    if not ComposeManager().restart():
+        print("❌ Failed to restart docker compose services.")
+        sys.exit(1)
+        
+    print("Waiting for services to become healthy...")
+    if not wait_for_services(timeout=60):
+        print("❌ Services failed to start or respond to health checks in time.")
+        sys.exit(1)
+        
+    print("✓ Services restarted successfully and are healthy.")

{cli_memory_os-0.1.0 → cli_memory_os-0.1.3}/cli/commands/status.py

@@ -72,6 +72,7 @@ def execute(args):
         print(f"Documents    : 0")
         print(f"Emails       : 0")
         print(f"Vectors      : 0")
+        print(f"Embedding Model : {embedding_model}")
         print("")
         print("Workspace is initialized but contains no indexed data.")
         print("")

cli_memory_os-0.1.3/cli/commands/sync.py

@@ -0,0 +1,70 @@
+"""
+Command: memory-os sync
+
+Synchronizes data from connected sources (GitHub, Gmail, Notion),
+chunks documents, generates embeddings, and syncs databases.
+"""
+
+import sys
+import time
+import logging
+import sqlite3
+from storage.db import init_db, clear_all
+from core.vector_store import run_reindexing, close_qdrant_client
+from storage.graph import GraphStore
+from connectors.github import sync_github
+from connectors.gmail import sync_gmail
+from connectors.notion import sync_notion
+
+
+def execute(args):
+    """Run the synchronization pipeline."""
+    # Ensure database is initialized
+    try:
+        init_db()
+    except sqlite3.OperationalError as e:
+        print("----------------------------------")
+        print("Workspace has not been initialized.")
+        print("")
+        print("Run:")
+        print("memory-os init")
+        print("----------------------------------")
+        sys.exit(1)
+
+    start_time = time.perf_counter()
+    
+    rebuild = getattr(args, "rebuild", False)
+    source = getattr(args, "source", None)
+    
+    if rebuild:
+        print("Performing full rebuild reset...")
+        clear_all()
+        close_qdrant_client()
+        run_reindexing()
+        GraphStore().clear_graph()
+
+    # Sync based on source selection
+    if source == "github":
+        print("Syncing GitHub source...")
+        sync_github()
+    elif source == "gmail":
+        print("Syncing Gmail source...")
+        sync_gmail()
+    elif source == "notion":
+        print("Syncing Notion source...")
+        sync_notion()
+    else:
+        print("Syncing all sources (GitHub, Gmail, Notion)...")
+        sync_github()
+        sync_gmail()
+        sync_notion()
+
+    print("Running document chunking and vector reindexing...")
+    run_reindexing()
+
+    print("Extracting and syncing knowledge graph relationships...")
+    GraphStore().extract_and_sync_graph()
+
+    duration = time.perf_counter() - start_time
+    print(f"\nSync complete. Total Duration: {duration:.2f}s")
+    logging.getLogger("main").info(f"Sync completed in {duration:.2f}s")

{cli_memory_os-0.1.0 → cli_memory_os-0.1.3}/cli/commands/workspace.py

@@ -5,10 +5,8 @@ Provides management of active workspaces and profile structures.
 """
 
 import sys
-import shutil
 import os
 from datetime import datetime
-from pathlib import Path
 from infrastructure.workspace import (
     get_active_profile,
     list_profiles,

{cli_memory_os-0.1.0 → cli_memory_os-0.1.3}/cli/main.py

@@ -115,7 +115,6 @@ def run_interactive():
         close_qdrant_client,
     )
     from core.embedder import Embedder
-    from core.llm import run_hybrid_rag
     from storage.graph import GraphStore
     from connectors.github import sync_github
     from connectors.gmail import sync_gmail
@@ -283,30 +282,11 @@ def run_interactive():
                 if not arg:
                     print("Usage: ask <question>")
                 else:
-                    rag_res = run_hybrid_rag(arg)
-                    print("========================================")
-                    print("ANSWER")
-                    print("========================================")
-                    print(rag_res["answer"])
-                    print("\n========================================")
-                    print("SOURCES")
-                    print("========================================")
-                    if rag_res["sources"]:
-                        for s in rag_res["sources"]:
-                            print(f"- {s}")
-                    else:
-                        print("None")
-                    print("\n========================================")
-                    print("REPOSITORIES USED")
-                    print("========================================")
-                    if rag_res["repositories"]:
-                        for r in rag_res["repositories"]:
-                            print(f"- {r}")
-                    else:
-                        print("None")
-                    print("\n========================================")
-                    print(f"Confidence: {rag_res['confidence']:.2f}")
-                    print("========================================")
+                    from cli.commands.ask import run_and_print_ask
+                    try:
+                        run_and_print_ask(arg)
+                    except Exception as e:
+                        print(f"❌ Error executing query: {e}")
 
             elif cmd == "repo-info":
                 if not arg:
@@ -414,6 +394,9 @@ def route_command(args):
     elif command == "stop":
         from cli.commands.stop import execute
         execute(args)
+    elif command == "restart":
+        from cli.commands.restart import execute
+        execute(args)
     elif command == "plugins":
         from cli.commands.plugins import execute
         execute(args)
@@ -438,11 +421,16 @@ def route_command(args):
     elif command == "logs":
         from cli.commands.logs import execute
         execute(args)
+    elif command == "sync":
+        from cli.commands.sync import execute
+        execute(args)
+    elif command == "ask":
+        from cli.commands.ask import execute
+        execute(args)
     else:
-        # For commands not yet implemented, show a helpful message
-        print(f"Command '{command}' is not yet implemented.")
-        print("Run 'memory-os --help' for available commands.")
-        print("Run 'memory-os' (no arguments) for the interactive REPL.")
+        from cli.parser import build_parser
+        build_parser().print_help()
+        sys.exit(1)
 
 
 

{cli_memory_os-0.1.0 → cli_memory_os-0.1.3}/cli/parser.py

@@ -118,6 +118,12 @@ def build_parser() -> argparse.ArgumentParser:
         help="Stop Docker services (preserves data)",
     )
 
+    # ── restart ────────────────────────────────────────────────
+    subparsers.add_parser(
+        "restart",
+        help="Restart Docker services (Neo4j, Qdrant)",
+    )
+
     # ── version ───────────────────────────────────────────────
     subparsers.add_parser(
         "version",

{cli_memory_os-0.1.0 → cli_memory_os-0.1.3/cli_memory_os.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cli-memory-os
-Version: 0.1.0
+Version: 0.1.3
 Summary: CLI-based Personal Knowledge Operating System
 Author-email: Anirudh T <anirudh200584@gmail.com>
 License-Expression: MIT

{cli_memory_os-0.1.0 → cli_memory_os-0.1.3}/cli_memory_os.egg-info/SOURCES.txt

@@ -6,6 +6,7 @@ cli/__init__.py
 cli/main.py
 cli/parser.py
 cli/commands/__init__.py
+cli/commands/ask.py
 cli/commands/benchmark.py
 cli/commands/config_cmd.py
 cli/commands/doctor.py
@@ -15,9 +16,11 @@ cli/commands/init.py
 cli/commands/logs.py
 cli/commands/monitor.py
 cli/commands/plugins.py
+cli/commands/restart.py
 cli/commands/start.py
 cli/commands/status.py
 cli/commands/stop.py
+cli/commands/sync.py
 cli/commands/version.py
 cli/commands/workspace.py
 cli_memory_os.egg-info/PKG-INFO
@@ -54,5 +57,8 @@ storage/schema.sql
 storage/tech_detector.py
 tests/test_cli_integration.py
 tests/test_commands_phase2.py
+tests/test_compose_manager.py
 tests/test_first_time_user_experience.py
+tests/test_infrastructure_stabilization.py
+tests/test_sync_ask_cli.py
 tests/test_v1_features.py

{cli_memory_os-0.1.0 → cli_memory_os-0.1.3}/core/embedder.py

@@ -1,4 +1,3 @@
-import os
 
 class Embedder:
     _model = None

{cli_memory_os-0.1.0 → cli_memory_os-0.1.3}/core/llm.py

@@ -2,7 +2,6 @@ import os
 import time
 import logging
 from groq import Groq
-from core.vector_store import hybrid_search
 from storage.graph import GraphStore
 
 logger = logging.getLogger("llm")