mem0-open-mcp 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl

{mem0_open_mcp-0.1.2.dist-info → mem0_open_mcp-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mem0-open-mcp
-Version: 0.1.2
+Version: 0.1.3
 Summary: Open-source MCP server for mem0 - local LLMs, self-hosted, Docker-free
 Author: Alex
 License-Expression: Apache-2.0
@@ -83,10 +83,18 @@ mem0-open-mcp configure
 # Start the server
 mem0-open-mcp serve
 
+# Test configuration before starting (recommended for initial setup)
+mem0-open-mcp serve --test
+
 # With options
 mem0-open-mcp serve --port 8765 --user-id alice
 ```
 
+The `--test` flag runs connectivity and memory tests before starting the server:
+- Checks Vector Store, LLM, and Embedder connections
+- Performs actual memory add/search operations
+- Cleans up test data automatically
+
 ## Configuration
 
 Create `mem0-open-mcp.yaml`:

{mem0_open_mcp-0.1.2.dist-info → mem0_open_mcp-0.1.3.dist-info}/RECORD

@@ -1,15 +1,15 @@
-mem0_server/__init__.py,sha256=J3DIJ4cwmlNijrW0AXOxNElD0eKN1V443Ew-DGp0-Gg,232
-mem0_server/cli.py,sha256=96uFUoUBZDc-dqO4OTTs3xu0Xf3vTz5X7PeLHUIYqmc,15564
+mem0_server/__init__.py,sha256=FrcCqmcA_VSR57xjgdQ1obSrsSzxDsDsXdEGSdVhUq0,379
+mem0_server/cli.py,sha256=TVjbm-pSj4-yS0uZ1qZcqMlGtL__JyZm6S5gBkkqvdU,24363
 mem0_server/server.py,sha256=Obc2eZaAYuZdUgYpOY8qWwbOXgc3w1vRmmEF3tc0gdI,12494
 mem0_server/api/__init__.py,sha256=JOOnrq3v7yym3xotMkOUAoZj1zjhI8C_a1wEdbpCakQ,122
 mem0_server/api/routes.py,sha256=fBOeQ1kckEa9Nr57cutCunhG9D1J5zYrv9sFiYcZ1yQ,21743
 mem0_server/config/__init__.py,sha256=Z0GWnr8I7VbrUfVLeA37K9qHOsK7HJOHiGELbWp8YcI,733
 mem0_server/config/loader.py,sha256=LPobv8BSyPShJ_27d_hwsx9EhAJPDdzWJMrUGJ4Qdt8,8373
-mem0_server/config/schema.py,sha256=44TQKNavwkH8btdJBXSPWUcFcMKTpzb-Pb03pZ3dImI,12651
+mem0_server/config/schema.py,sha256=HnFWUR4KxwHkyukFw_ptMCHt7K26Zl7AU02-5jdtjL4,12731
 mem0_server/mcp/__init__.py,sha256=8BWWjlJj5_jdeUuQFN4i5hm5Zd3Pk0cCR2Vf0G4eUB4,127
 mem0_server/mcp/server.py,sha256=Y_f8upEtvjr0JjZaUQ4YWLRrof9btR2ds8FL2DucCZI,184
 mem0_server/utils/__init__.py,sha256=GvVR3Tz4OmU2gRGCbTyQg5Xip5cAx7KW6ToIh2sKA4w,41
-mem0_open_mcp-0.1.2.dist-info/METADATA,sha256=y7dIo0XWvnOpdGMspJQszXDEQP_t99fxmFfB16IGsWc,4622
-mem0_open_mcp-0.1.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-mem0_open_mcp-0.1.2.dist-info/entry_points.txt,sha256=WXqVdvwhFvMkzAmNtdlHlRZV23yikM43BtB8S4F9ByE,54
-mem0_open_mcp-0.1.2.dist-info/RECORD,,
+mem0_open_mcp-0.1.3.dist-info/METADATA,sha256=2P1mwFuVOFhugcDwJ7KmF_nRLszRkJDAbCvO1eveuBg,4937
+mem0_open_mcp-0.1.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+mem0_open_mcp-0.1.3.dist-info/entry_points.txt,sha256=WXqVdvwhFvMkzAmNtdlHlRZV23yikM43BtB8S4F9ByE,54
+mem0_open_mcp-0.1.3.dist-info/RECORD,,

mem0_server/__init__.py

@@ -5,4 +5,9 @@ This package provides a CLI tool to run mem0 as an MCP server without Docker,
 with optional web UI for configuration management.
 """
 
-__version__ = "0.1.0"
+from importlib.metadata import version, PackageNotFoundError
+
+try:
+    __version__ = version("mem0-open-mcp")
+except PackageNotFoundError:
+    __version__ = "0.0.0-dev"

mem0_server/cli.py

@@ -2,7 +2,9 @@
 
 from __future__ import annotations
 
+import json
 import logging
+import time
 from pathlib import Path
 from typing import Annotated
 
@@ -29,6 +31,190 @@ app = typer.Typer(
 
 console = Console()
 
+UPDATE_CHECK_CACHE = Path.home() / ".cache" / "mem0-open-mcp" / "update_check.json"
+UPDATE_CHECK_INTERVAL = 86400  # 24 hours
+
+
+def _check_for_updates() -> None:
+    """Check PyPI for newer version (once per day)."""
+    try:
+        UPDATE_CHECK_CACHE.parent.mkdir(parents=True, exist_ok=True)
+        
+        now = time.time()
+        if UPDATE_CHECK_CACHE.exists():
+            cache = json.loads(UPDATE_CHECK_CACHE.read_text())
+            if now - cache.get("last_check", 0) < UPDATE_CHECK_INTERVAL:
+                if cache.get("latest") and cache["latest"] != __version__:
+                    console.print(
+                        f"[yellow]Update available: {__version__} → {cache['latest']}[/yellow]\n"
+                        f"[dim]  pip install --upgrade mem0-open-mcp[/dim]\n"
+                    )
+                return
+        
+        import httpx
+        resp = httpx.get("https://pypi.org/pypi/mem0-open-mcp/json", timeout=3)
+        resp.raise_for_status()
+        latest = resp.json()["info"]["version"]
+        
+        UPDATE_CHECK_CACHE.write_text(json.dumps({"last_check": now, "latest": latest}))
+        
+        if latest != __version__:
+            console.print(
+                f"[yellow]Update available: {__version__} → {latest}[/yellow]\n"
+                f"[dim]  pip install --upgrade mem0-open-mcp[/dim]\n"
+            )
+    except Exception:
+        pass
+
+
+def _run_connectivity_tests(config: Mem0ServerConfig) -> bool:
+    """Run connectivity tests for LLM, Embedder, and Vector Store."""
+    console.print("[bold]Running connectivity tests...[/bold]\n")
+    
+    all_passed = True
+    
+    # Test Vector Store
+    console.print("  [dim]Vector Store...[/dim]", end=" ")
+    try:
+        vs_config = config.vector_store
+        if vs_config.provider.value == "qdrant":
+            from qdrant_client import QdrantClient
+            host = vs_config.config.host or "localhost"
+            port = vs_config.config.port or 6333
+            client = QdrantClient(host=host, port=port, timeout=5)
+            client.get_collections()
+            console.print("[green]✓ Connected[/green]")
+        elif vs_config.provider.value == "chroma":
+            import chromadb
+            if vs_config.config.host:
+                client = chromadb.HttpClient(host=vs_config.config.host, port=vs_config.config.port or 8000)
+            else:
+                client = chromadb.Client()
+            client.heartbeat()
+            console.print("[green]✓ Connected[/green]")
+        else:
+            console.print(f"[yellow]⚠ Skip (no test for {vs_config.provider.value})[/yellow]")
+    except Exception as e:
+        console.print(f"[red]✗ Failed: {e}[/red]")
+        all_passed = False
+    
+    # Test LLM
+    console.print("  [dim]LLM...[/dim]", end=" ")
+    try:
+        llm_config = config.llm
+        if llm_config.provider.value == "ollama":
+            import httpx
+            base_url = llm_config.config.base_url or "http://localhost:11434"
+            resp = httpx.get(f"{base_url}/api/tags", timeout=5)
+            resp.raise_for_status()
+            models = [m["name"] for m in resp.json().get("models", [])]
+            if llm_config.config.model in models or any(llm_config.config.model in m for m in models):
+                console.print(f"[green]✓ Connected ({llm_config.config.model})[/green]")
+            else:
+                console.print(f"[yellow]⚠ Connected but model '{llm_config.config.model}' not found[/yellow]")
+                console.print(f"      [dim]Available: {', '.join(models[:5])}{'...' if len(models) > 5 else ''}[/dim]")
+        elif llm_config.provider.value in ("openai", "lmstudio"):
+            import httpx
+            base_url = llm_config.config.base_url or "https://api.openai.com/v1"
+            headers = {}
+            if llm_config.config.api_key:
+                headers["Authorization"] = f"Bearer {llm_config.config.api_key}"
+            resp = httpx.get(f"{base_url}/models", headers=headers, timeout=5)
+            resp.raise_for_status()
+            console.print(f"[green]✓ Connected ({llm_config.config.model})[/green]")
+        else:
+            console.print(f"[yellow]⚠ Skip (no test for {llm_config.provider.value})[/yellow]")
+    except Exception as e:
+        console.print(f"[red]✗ Failed: {e}[/red]")
+        all_passed = False
+    
+    # Test Embedder
+    console.print("  [dim]Embedder...[/dim]", end=" ")
+    try:
+        emb_config = config.embedder
+        if emb_config.provider.value == "ollama":
+            import httpx
+            base_url = emb_config.config.base_url or "http://localhost:11434"
+            resp = httpx.get(f"{base_url}/api/tags", timeout=5)
+            resp.raise_for_status()
+            models = [m["name"] for m in resp.json().get("models", [])]
+            if emb_config.config.model in models or any(emb_config.config.model in m for m in models):
+                console.print(f"[green]✓ Connected ({emb_config.config.model})[/green]")
+            else:
+                console.print(f"[yellow]⚠ Connected but model '{emb_config.config.model}' not found[/yellow]")
+                console.print(f"      [dim]Available: {', '.join(models[:5])}{'...' if len(models) > 5 else ''}[/dim]")
+        elif emb_config.provider.value in ("openai", "lmstudio"):
+            import httpx
+            base_url = emb_config.config.base_url or "https://api.openai.com/v1"
+            headers = {}
+            if emb_config.config.api_key:
+                headers["Authorization"] = f"Bearer {emb_config.config.api_key}"
+            resp = httpx.get(f"{base_url}/models", headers=headers, timeout=5)
+            resp.raise_for_status()
+            console.print(f"[green]✓ Connected ({emb_config.config.model})[/green]")
+        else:
+            console.print(f"[yellow]⚠ Skip (no test for {emb_config.provider.value})[/yellow]")
+    except Exception as e:
+        console.print(f"[red]✗ Failed: {e}[/red]")
+        all_passed = False
+    
+    console.print()
+    if all_passed:
+        console.print("[bold green]All connectivity tests passed![/bold green]\n")
+    else:
+        console.print("[bold red]Some connectivity tests failed. Please check your configuration.[/bold red]\n")
+    
+    return all_passed
+
+
+def _run_memory_tests(config: Mem0ServerConfig) -> bool:
+    """Run actual mem0 memory add/search tests."""
+    import uuid
+    console.print("[bold]Running memory tests...[/bold]\n")
+    
+    test_user_id = f"__test_user_{uuid.uuid4().hex[:8]}"
+    test_memory_text = "This is a test memory for connectivity verification."
+    
+    try:
+        console.print("  [dim]Initializing mem0 client...[/dim]", end=" ")
+        from mem0 import Memory
+        mem0_config = config.to_mem0_config()
+        memory = Memory.from_config(mem0_config)
+        console.print("[green]✓[/green]")
+        
+        console.print("  [dim]Adding test memory...[/dim]", end=" ")
+        add_result = memory.add(test_memory_text, user_id=test_user_id)
+        if add_result and add_result.get("results"):
+            first_result = add_result["results"][0]
+            memory_id = first_result.get("id") if first_result else None
+            if memory_id:
+                console.print(f"[green]✓ Added (id: {memory_id[:8]}...)[/green]")
+            else:
+                console.print("[green]✓ Added[/green]")
+        else:
+            console.print("[green]✓ Added[/green]")
+        
+        console.print("  [dim]Searching memories...[/dim]", end=" ")
+        search_result = memory.search("test memory verification", user_id=test_user_id, limit=5)
+        if search_result and search_result.get("results"):
+            console.print(f"[green]✓ Found {len(search_result['results'])} result(s)[/green]")
+        else:
+            console.print("[yellow]⚠ No results (may be expected for new setup)[/yellow]")
+        
+        console.print("  [dim]Cleaning up test data...[/dim]", end=" ")
+        memory.delete_all(user_id=test_user_id)
+        console.print("[green]✓ Cleaned[/green]")
+        
+        console.print()
+        console.print("[bold green]All memory tests passed![/bold green]\n")
+        return True
+        
+    except Exception as e:
+        console.print(f"[red]✗ Failed: {e}[/red]")
+        console.print()
+        console.print("[bold red]Memory test failed. Check your LLM/Embedder/VectorStore configuration.[/bold red]\n")
+        return False
+
 
 def version_callback(value: bool) -> None:
     """Show version and exit."""
@@ -83,6 +269,10 @@ def serve(
         str,
         typer.Option("--log-level", "-l", help="Logging level."),
     ] = "info",
+    test: Annotated[
+        bool,
+        typer.Option("--test", "-t", help="Run connectivity tests before starting server."),
+    ] = False,
 ) -> None:
     """Start the MCP server.
     
@@ -120,7 +310,9 @@ def serve(
         border_style="green",
     ))
     
-    console.print("\n[bold]Configuration:[/bold]")
+    _check_for_updates()
+    
+    console.print("[bold]Configuration:[/bold]")
     console.print(f"  Host: [cyan]{config.server.host}[/cyan]")
     console.print(f"  Port: [cyan]{config.server.port}[/cyan]")
     console.print(f"  User ID: [cyan]{config.server.user_id}[/cyan]")
@@ -129,6 +321,12 @@ def serve(
     console.print(f"  Vector Store: [cyan]{config.vector_store.provider.value}[/cyan]")
     console.print()
     
+    if test:
+        if not _run_connectivity_tests(config):
+            raise typer.Exit(1)
+        if not _run_memory_tests(config):
+            raise typer.Exit(1)
+    
     # Start the server
     try:
         from mem0_server.server import run_server

mem0_server/config/schema.py

@@ -323,6 +323,7 @@ def get_default_config() -> Mem0ServerConfig:
             config=EmbedderConfig(
                 model="text-embedding-3-small",
                 api_key="env:OPENAI_API_KEY",
+                embedding_dims=1536,
             ),
         ),
         vector_store=VectorStoreProvider(
@@ -331,6 +332,7 @@ def get_default_config() -> Mem0ServerConfig:
                 collection_name="mem0_memories",
                 host="localhost",
                 port=6333,
+                embedding_model_dims=1536,
             ),
         ),
     )