mem0-open-mcp 0.1.5__py3-none-any.whl → 0.1.7__py3-none-any.whl

{mem0_open_mcp-0.1.5.dist-info → mem0_open_mcp-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mem0-open-mcp
-Version: 0.1.5
+Version: 0.1.7
 Summary: Open-source MCP server for mem0 - local LLMs, self-hosted, Docker-free
 Author: Alex
 License-Expression: Apache-2.0
@@ -86,9 +86,6 @@ mem0-open-mcp test
 # Start the server
 mem0-open-mcp serve
 
-# Test and start server
-mem0-open-mcp serve --test
-
 # With options
 mem0-open-mcp serve --port 8765 --user-id alice
 ```

{mem0_open_mcp-0.1.5.dist-info → mem0_open_mcp-0.1.7.dist-info}/RECORD

@@ -1,5 +1,5 @@
 mem0_server/__init__.py,sha256=FrcCqmcA_VSR57xjgdQ1obSrsSzxDsDsXdEGSdVhUq0,379
-mem0_server/cli.py,sha256=IcR7eTQjJvYVwENLctbNGd7Ej1AAtX3BekTIDbhdj5A,25962
+mem0_server/cli.py,sha256=1ZWZIch8RIJJ6w-ffprvs0Fz9qg9AwVeLlYSm7M_sRU,27633
 mem0_server/server.py,sha256=Obc2eZaAYuZdUgYpOY8qWwbOXgc3w1vRmmEF3tc0gdI,12494
 mem0_server/api/__init__.py,sha256=JOOnrq3v7yym3xotMkOUAoZj1zjhI8C_a1wEdbpCakQ,122
 mem0_server/api/routes.py,sha256=fBOeQ1kckEa9Nr57cutCunhG9D1J5zYrv9sFiYcZ1yQ,21743
@@ -9,7 +9,7 @@ mem0_server/config/schema.py,sha256=HnFWUR4KxwHkyukFw_ptMCHt7K26Zl7AU02-5jdtjL4,
 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.5.dist-info/METADATA,sha256=aXza5vrdT6hLelN4pLFGGMfTwT-sl_R4UYjZ3epZ8Mg,4960
-mem0_open_mcp-0.1.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-mem0_open_mcp-0.1.5.dist-info/entry_points.txt,sha256=WXqVdvwhFvMkzAmNtdlHlRZV23yikM43BtB8S4F9ByE,54
-mem0_open_mcp-0.1.5.dist-info/RECORD,,
+mem0_open_mcp-0.1.7.dist-info/METADATA,sha256=IOtypdi-yunILO6wgrSEvZQ-b6mdZ0TWLx2QhTyu8qs,4908
+mem0_open_mcp-0.1.7.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+mem0_open_mcp-0.1.7.dist-info/entry_points.txt,sha256=WXqVdvwhFvMkzAmNtdlHlRZV23yikM43BtB8S4F9ByE,54
+mem0_open_mcp-0.1.7.dist-info/RECORD,,

mem0_server/cli.py

@@ -76,6 +76,26 @@ def _check_for_updates() -> None:
         pass
 
 
+def _model_matches(config_model: str, available_models: list[str]) -> bool:
+    """Check if configured model matches any available model.
+    
+    Handles :latest suffix - 'llama3.2' matches 'llama3.2:latest'
+    """
+    config_lower = config_model.lower()
+    config_with_latest = f"{config_lower}:latest" if ":" not in config_lower else config_lower
+    
+    for m in available_models:
+        m_lower = m.lower()
+        if config_lower == m_lower:
+            return True
+        if config_with_latest == m_lower:
+            return True
+        m_without_latest = m_lower.replace(":latest", "") if m_lower.endswith(":latest") else m_lower
+        if config_lower == m_without_latest:
+            return True
+    return False
+
+
 def _run_connectivity_tests(config: Mem0ServerConfig) -> bool:
     """Run connectivity tests for LLM, Embedder, and Vector Store."""
     console.print("[bold]Running connectivity tests...[/bold]\n")
@@ -117,7 +137,7 @@ def _run_connectivity_tests(config: Mem0ServerConfig) -> bool:
             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):
+            if _model_matches(llm_config.config.model, 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]")
@@ -147,7 +167,7 @@ def _run_connectivity_tests(config: Mem0ServerConfig) -> bool:
             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):
+            if _model_matches(emb_config.config.model, 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]")
@@ -178,6 +198,7 @@ def _run_connectivity_tests(config: Mem0ServerConfig) -> bool:
 
 def _run_memory_tests(config: Mem0ServerConfig) -> bool:
     """Run actual mem0 memory add/search tests."""
+    import os
     import uuid
     console.print("[bold]Running memory tests...[/bold]\n")
     
@@ -186,13 +207,16 @@ def _run_memory_tests(config: Mem0ServerConfig) -> bool:
     
     try:
         console.print("  [dim]Initializing mem0 client...[/dim]", end=" ")
+        os.environ["MEM0_TELEMETRY"] = "false"
         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=" ")
+        # 1. Add memory
+        console.print("  [dim]1. Adding test memory...[/dim]", end=" ")
         add_result = memory.add(test_memory_text, user_id=test_user_id)
+        memory_id = None
         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
@@ -203,14 +227,40 @@ def _run_memory_tests(config: Mem0ServerConfig) -> bool:
         else:
             console.print("[green]✓ Added[/green]")
         
-        console.print("  [dim]Searching memories...[/dim]", end=" ")
+        # 2. List memories
+        console.print("  [dim]2. Listing memories...[/dim]", end=" ")
+        list_result = memory.get_all(user_id=test_user_id)
+        if list_result and list_result.get("results"):
+            stored_count = len(list_result["results"])
+            console.print(f"[green]✓ Found {stored_count} memory(s)[/green]")
+            # Get memory_id from list if not captured from add
+            if not memory_id and stored_count > 0:
+                memory_id = list_result["results"][0].get("id")
+        else:
+            console.print("[red]✗ Memory not stored[/red]")
+            return False
+        
+        # 3. Search memories
+        console.print("  [dim]3. 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("[yellow]⚠ No search results (indexing may be delayed)[/yellow]")
         
-        console.print("  [dim]Cleaning up test data...[/dim]", end=" ")
+        # 4. Delete single memory
+        console.print("  [dim]4. Deleting single memory...[/dim]", end=" ")
+        if memory_id:
+            delete_result = memory.delete(memory_id)
+            if delete_result and delete_result.get("message"):
+                console.print(f"[green]✓ Deleted (id: {memory_id[:8]}...)[/green]")
+            else:
+                console.print("[green]✓ Deleted[/green]")
+        else:
+            console.print("[yellow]⚠ Skipped (no memory_id)[/yellow]")
+        
+        # 5. Cleanup remaining test data
+        console.print("  [dim]5. Cleaning up test data...[/dim]", end=" ")
         memory.delete_all(user_id=test_user_id)
         console.print("[green]✓ Cleaned[/green]")
         
@@ -278,10 +328,6 @@ 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.
     
@@ -330,12 +376,6 @@ 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