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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cli-memory-os
-Version: 0.1.0
+Version: 0.1.2
 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.2}/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.2}/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.2}/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.2}/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.2/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.2}/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.0 → cli_memory_os-0.1.2}/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.2}/cli/main.py

@@ -414,6 +414,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)

{cli_memory_os-0.1.0 → cli_memory_os-0.1.2}/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.2/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.2
 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.2}/cli_memory_os.egg-info/SOURCES.txt

@@ -15,6 +15,7 @@ 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
@@ -54,5 +55,7 @@ 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_v1_features.py

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

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

{cli_memory_os-0.1.0 → cli_memory_os-0.1.2}/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")

{cli_memory_os-0.1.0 → cli_memory_os-0.1.2}/core/vector_store.py

@@ -226,55 +226,6 @@ def get_vector_index_stats() -> dict:
             "exists": False
         }
 
-def get_vector_chunks(repo_name: str, limit: int = 5) -> list:
-    """Retrieve the first `limit` chunks for a given repository from Qdrant.
-    Returns a list of Record objects.
-    """
-    client = get_qdrant_client()
-    try:
-        filter_expr = models.Filter(
-            must=[
-                models.FieldCondition(
-                    key="repository_name",
-                    match=models.MatchValue(value=repo_name)
-                )
-            ]
-        )
-        points, _ = client.scroll(
-            collection_name=COLLECTION_NAME,
-            scroll_filter=filter_expr,
-            limit=limit,
-            with_payload=True,
-            with_vectors=False
-        )
-        return points
-    except Exception as e:
-        import logging
-        logging.getLogger("vector_store").error(f"Error fetching vector chunks: {e}")
-        return []
-
-def count_vector_chunks(repo_name: str) -> int:
-    """Count the total number of vectors for a given repository in Qdrant."""
-    client = get_qdrant_client()
-    try:
-        filter_expr = models.Filter(
-            must=[
-                models.FieldCondition(
-                    key="repository_name",
-                    match=models.MatchValue(value=repo_name)
-                )
-            ]
-        )
-        result = client.count(
-            collection_name=COLLECTION_NAME,
-            count_filter=filter_expr,
-            exact=True
-        )
-        return result.count
-    except Exception as e:
-        import logging
-        logging.getLogger("vector_store").error(f"Error counting vector chunks: {e}")
-        return 0
 
 def run_semantic_search(query: str, limit: int = 5, source_filter: str = None, raw_scores: bool = False, repo_filter: str = None) -> list:
     """Perform raw vector search on the Qdrant database, with optional source and repository filters."""