better-mem0-mcp 1.1.0b22__py3-none-any.whl → 1.1.2__py3-none-any.whl

better_mem0_mcp/config.py

@@ -33,36 +33,43 @@ class Settings(BaseSettings):
     embedder_models: str = "gemini/gemini-embedding-001"
 
     def setup_api_keys(self) -> dict[str, list[str]]:
-        """Parse API_KEYS (format: ENV_VAR:key,...) and set env vars.
-
-        Example:
-            API_KEYS="GOOGLE_API_KEY:abc,GOOGLE_API_KEY:def,OPENAI_API_KEY:xyz"
+        """
+        Parse API_KEYS and set environment variables for LiteLLM.
 
         Returns:
-            Dict mapping env var name to list of API keys.
+            Dict mapping provider to list of API keys.
         """
-        keys_by_env: dict[str, list[str]] = {}
+        env_map = {
+            "gemini": "GOOGLE_API_KEY",
+            "openai": "OPENAI_API_KEY",
+            "anthropic": "ANTHROPIC_API_KEY",
+            "groq": "GROQ_API_KEY",
+            "deepseek": "DEEPSEEK_API_KEY",
+            "mistral": "MISTRAL_API_KEY",
+        }
+
+        keys_by_provider: dict[str, list[str]] = {}
 
         for pair in self.api_keys.split(","):
             pair = pair.strip()
             if ":" not in pair:
                 continue
 
-            env_var, key = pair.split(":", 1)
-            env_var = env_var.strip()
+            provider, key = pair.split(":", 1)
+            provider = provider.strip().lower()
             key = key.strip()
 
             if not key:
                 continue
 
-            keys_by_env.setdefault(env_var, []).append(key)
+            keys_by_provider.setdefault(provider, []).append(key)
 
-        # Set first key of each env var (LiteLLM reads from env)
-        for env_var, keys in keys_by_env.items():
-            if keys:
-                os.environ[env_var] = keys[0]
+        # Set first key of each provider as env var (LiteLLM reads from env)
+        for provider, keys in keys_by_provider.items():
+            if provider in env_map and keys:
+                os.environ[env_map[provider]] = keys[0]
 
-        return keys_by_env
+        return keys_by_provider
 
     def parse_database_url(self) -> dict:
         """Parse DATABASE_URL into connection parameters."""

better_mem0_mcp/docs/memory.md

@@ -17,8 +17,6 @@ Save information to long-term memory. Mem0 automatically:
 {"action": "add", "content": "User prefers dark mode and uses FastAPI"}
 ```
 
-**Response:** `Saved: {results: [{id, memory, event}]}`
-
 ### search
 Semantic search across stored memories. Combines vector search with graph context.
 
@@ -26,8 +24,6 @@ Semantic search across stored memories. Combines vector search with graph contex
 {"action": "search", "query": "coding preferences", "limit": 5}
 ```
 
-**Response:** Formatted list of matching memories with optional graph context.
-
 ### list
 Get all stored memories for a user.
 
@@ -35,8 +31,6 @@ Get all stored memories for a user.
 {"action": "list"}
 ```
 
-**Response:** `Memories (N): - [id] memory text`
-
 ### delete
 Remove a memory by ID.
 
@@ -44,17 +38,10 @@ Remove a memory by ID.
 {"action": "delete", "memory_id": "abc12345-..."}
 ```
 
-**Response:** `Deleted: {memory_id}`
-
 ## Parameters
 - `action` - Required: add, search, list, delete
 - `content` - Required for add: information to remember
 - `query` - Required for search: what to search for
 - `memory_id` - Required for delete: ID of memory to remove
 - `limit` - Optional for search: max results (default: 5)
-- `user_id` - Optional: scope memories to a specific user (default: "default")
-
-## Technical Notes
-- Mem0 API returns `{"results": [...]}` for both `search()` and `get_all()`
-- Graph context is automatically included in search results when available
-- Embeddings use 1536 dimensions for pgvector HNSW compatibility
+- `user_id` - Optional: scope memories to a specific user

better_mem0_mcp/server.py

@@ -108,24 +108,21 @@ async def memory(
             if not query:
                 return "Error: 'query' required for search action"
 
-            # Vector search - returns {"results": [...]} (same as get_all)
-            response = _memory.search(query, user_id=user_id, limit=limit)
-            memories = (
-                response.get("results", []) if isinstance(response, dict) else response
-            )
+            # Vector search
+            results = _memory.search(query, user_id=user_id, limit=limit)
 
             # Add graph context
             graph_context = ""
             if _graph:
                 graph_context = _graph.get_context(query, user_id)
 
-            if not memories and not graph_context:
+            if not results and not graph_context:
                 return "No memories found."
 
             output = ""
-            if memories:
+            if results:
                 output = "Memories:\n" + "\n".join(
-                    [f"- {m.get('memory', str(m))}" for m in memories]
+                    [f"- {r.get('memory', str(r))}" for r in results]
                 )
             if graph_context:
                 output += f"\n\n{graph_context}"
@@ -133,17 +130,13 @@ async def memory(
             return output
 
         elif action == "list":
-            # get_all returns {"results": [...]} or list directly
-            response = _memory.get_all(user_id=user_id)
-            memories = (
-                response.get("results", []) if isinstance(response, dict) else response
-            )
+            results = _memory.get_all(user_id=user_id)
 
-            if not memories:
+            if not results:
                 return "No memories stored."
 
-            lines = [f"- [{m['id'][:8]}] {m['memory']}" for m in memories]
-            return f"Memories ({len(memories)}):\n" + "\n".join(lines)
+            lines = [f"- [{r['id'][:8]}] {r['memory']}" for r in results]
+            return f"Memories ({len(results)}):\n" + "\n".join(lines)
 
         elif action == "delete":
             if not memory_id:

{better_mem0_mcp-1.1.0b22.dist-info → better_mem0_mcp-1.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: better-mem0-mcp
-Version: 1.1.0b22
+Version: 1.1.2
 Summary: Zero-setup MCP Server for AI memory - works with Neon/Supabase
 Project-URL: Homepage, https://github.com/n24q02m/better-mem0-mcp
 Project-URL: Repository, https://github.com/n24q02m/better-mem0-mcp.git
@@ -17,7 +17,7 @@ Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Requires-Python: ==3.13.*
+Requires-Python: >=3.13
 Requires-Dist: google-genai>=1.0.0
 Requires-Dist: litellm>=1.0.0
 Requires-Dist: loguru>=0.7.0
@@ -29,28 +29,16 @@ Description-Content-Type: text/markdown
 
 # better-mem0-mcp
 
-**Self-hosted MCP Server for AI memory with PostgreSQL (pgvector).**
+**Zero-setup** MCP Server for AI memory. Works with Neon/Supabase free tier.
 
-[![PyPI](https://img.shields.io/pypi/v/better-mem0-mcp)](https://pypi.org/project/better-mem0-mcp/)
-[![Docker](https://img.shields.io/docker/v/n24q02m/better-mem0-mcp?label=docker)](https://hub.docker.com/r/n24q02m/better-mem0-mcp)
-[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
-
-## Features
-
-- **Self-hosted PostgreSQL** - Your data stays with you (Neon/Supabase free tier supported)
-- **Graph Memory** - SQL-based relationship tracking alongside vector memory
-- **Multi-provider LLM** - Gemini, OpenAI, Anthropic, Groq, DeepSeek, Mistral
-- **Fallback chains** - Multi-key per provider + multi-model fallback
-- **Zero manual setup** - Just `DATABASE_URL` + `API_KEYS`
-
----
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
 
 ## Quick Start
 
 ### 1. Get Prerequisites
 
-- **Database**: [Neon](https://neon.tech) or [Supabase](https://supabase.com) (free tier works)
-- **API Key**: Any supported provider ([Google AI Studio](https://aistudio.google.com/apikey) is free)
+- **Database**: [Neon](https://neon.tech) or [Supabase](https://supabase.com) (free tier)
+- **API Key**: [Google AI Studio](https://aistudio.google.com/apikey) (free tier)
 
 ### 2. Add to mcp.json
 
@@ -59,12 +47,12 @@ Description-Content-Type: text/markdown
 ```json
 {
   "mcpServers": {
-    "better-mem0": {
+    "memory": {
       "command": "uvx",
-      "args": ["better-mem0-mcp@latest"],
+      "args": ["better-mem0-mcp"],
       "env": {
         "DATABASE_URL": "postgresql://user:pass@xxx.neon.tech/neondb?sslmode=require",
-        "API_KEYS": "GOOGLE_API_KEY:AIza..."
+        "API_KEYS": "gemini:AIza..."
       }
     }
   }
@@ -76,12 +64,12 @@ Description-Content-Type: text/markdown
 ```json
 {
   "mcpServers": {
-    "better-mem0": {
+    "memory": {
       "command": "docker",
       "args": ["run", "-i", "--rm", "-e", "DATABASE_URL", "-e", "API_KEYS", "n24q02m/better-mem0-mcp:latest"],
       "env": {
         "DATABASE_URL": "postgresql://...",
-        "API_KEYS": "GOOGLE_API_KEY:AIza..."
+        "API_KEYS": "gemini:AIza..."
       }
     }
   }
@@ -90,7 +78,7 @@ Description-Content-Type: text/markdown
 
 ### 3. Done!
 
-Ask your AI: "Remember that I prefer dark mode and use FastAPI"
+Ask Claude: "Remember that I prefer dark mode and use FastAPI"
 
 ---
 
@@ -98,25 +86,22 @@ Ask your AI: "Remember that I prefer dark mode and use FastAPI"
 
 | Variable | Required | Description |
 |----------|----------|-------------|
-| `DATABASE_URL` | Yes | PostgreSQL with pgvector extension |
-| `API_KEYS` | Yes | `ENV_VAR:key` pairs, comma-separated |
-| `LLM_MODELS` | No | Model fallback chain |
-| `EMBEDDER_MODELS` | No | Embedding model chain |
+| `DATABASE_URL` | Yes | PostgreSQL connection string |
+| `API_KEYS` | Yes | `provider:key,...` (multi-key per provider OK) |
+| `LLM_MODELS` | No | `provider/model,...` (fallback chain) |
+| `EMBEDDER_MODELS` | No | `provider/model,...` (fallback chain) |
 
-### Supported LiteLLM Providers
+### Examples
 
-Use environment variable names from [LiteLLM docs](https://docs.litellm.ai/):
-`GOOGLE_API_KEY`, `OPENAI_API_KEY`, `ANTHROPIC_API_KEY`, `GROQ_API_KEY`, etc.
-
-**Single provider:**
-```bash
-API_KEYS=GOOGLE_API_KEY:AIza...
+**Minimal (Gemini only):**
+```
+API_KEYS=gemini:AIza...
 ```
 
 **Multi-key with fallback:**
-```bash
-API_KEYS=GOOGLE_API_KEY:AIza-1,GOOGLE_API_KEY:AIza-2,OPENAI_API_KEY:sk-xxx
-LLM_MODELS=gemini/gemini-3-flash-preview,openai/gpt-4o-mini
+```
+API_KEYS=gemini:AIza-1,gemini:AIza-2,openai:sk-xxx
+LLM_MODELS=gemini/gemini-2.5-flash,openai/gpt-4o-mini
 EMBEDDER_MODELS=gemini/gemini-embedding-001,openai/text-embedding-3-small
 ```
 
@@ -124,7 +109,7 @@ EMBEDDER_MODELS=gemini/gemini-embedding-001,openai/text-embedding-3-small
 
 | Setting | Default |
 |---------|---------|
-| `LLM_MODELS` | `gemini/gemini-3-flash-preview` |
+| `LLM_MODELS` | `gemini/gemini-2.5-flash` |
 | `EMBEDDER_MODELS` | `gemini/gemini-embedding-001` |
 
 ---
@@ -133,41 +118,32 @@ EMBEDDER_MODELS=gemini/gemini-embedding-001,openai/text-embedding-3-small
 
 | Tool | Description |
 |------|-------------|
-| `memory` | Memory operations: `add`, `search`, `list`, `delete` |
-| `help` | Get full documentation for tools |
+| `memory` | `action`: add, search, list, delete |
+| `help` | Detailed documentation |
 
-### Usage Examples
+### Usage
 
 ```json
 {"action": "add", "content": "I prefer TypeScript over JavaScript"}
-{"action": "search", "query": "programming preferences"}
+{"action": "search", "query": "preferences"}
 {"action": "list"}
 {"action": "delete", "memory_id": "abc123"}
 ```
 
 ---
 
-## Build from Source
+## Why better-mem0-mcp?
 
-```bash
-git clone https://github.com/n24q02m/better-mem0-mcp
-cd better-mem0-mcp
-
-# Setup (requires mise: https://mise.jdx.dev/)
-mise run setup
-
-# Run
-uv run better-mem0-mcp
-```
-
-**Requirements:** Python 3.13+
+| Feature | Official mem0-mcp | better-mem0-mcp |
+|---------|-------------------|-----------------|
+| Storage | Mem0 Cloud | **Self-hosted PostgreSQL** |
+| Graph Memory | No | **Yes (SQL-based)** |
+| LLM Provider | OpenAI only | **Any (Gemini/OpenAI/Ollama/...)** |
+| Fallback | No | **Yes (multi-key + multi-model)** |
+| Setup | API Key | **DATABASE_URL + API_KEYS** |
 
 ---
 
-## Contributing
-
-See [CONTRIBUTING.md](CONTRIBUTING.md)
-
 ## License
 
-MIT - See [LICENSE](LICENSE)
+MIT

better_mem0_mcp-1.1.2.dist-info/RECORD

@@ -0,0 +1,11 @@
+better_mem0_mcp/__init__.py,sha256=fcqgbz2HvMCPidqqoPvtRky5pGIHP2w9oVim7UQkuBc,106
+better_mem0_mcp/config.py,sha256=vgKLIw3jHyeBLTzzBnuHV5x6Nra0Rbav1IJRL8rLCuk,5576
+better_mem0_mcp/graph.py,sha256=rE9z6XECiAktEqDNgmwqCpFpvKSn3azO9H4sRBhj8UU,6195
+better_mem0_mcp/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+better_mem0_mcp/server.py,sha256=O6s0L1FggB9FDfX4OKF6govrRznwUIAh9RkFRCrYX3Y,5124
+better_mem0_mcp/docs/memory.md,sha256=jqoxBYHVh2N9TZbt77qY5zhL1vSU-Ro-yuaczT1r7Mo,1126
+better_mem0_mcp-1.1.2.dist-info/METADATA,sha256=-Ccmz2b162rXT7n0AnE6wbTRS2R1n3o7FG4vWOYWcZU,3871
+better_mem0_mcp-1.1.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+better_mem0_mcp-1.1.2.dist-info/entry_points.txt,sha256=2b7E3D6yo94mQXP2Ms0bhUlWkK9f664f0GrstImOq30,57
+better_mem0_mcp-1.1.2.dist-info/licenses/LICENSE,sha256=d7xQ6sRyeGus6gnvwgqiQtSY7XdFw0Jd0w5-Co_xHnk,1064
+better_mem0_mcp-1.1.2.dist-info/RECORD,,

better_mem0_mcp/__main__.py

@@ -1,6 +0,0 @@
-"""Entry point for `python -m better_mem0_mcp`."""
-
-from .server import main
-
-if __name__ == "__main__":
-    main()

better_mem0_mcp-1.1.0b22.dist-info/RECORD

@@ -1,12 +0,0 @@
-better_mem0_mcp/__init__.py,sha256=fcqgbz2HvMCPidqqoPvtRky5pGIHP2w9oVim7UQkuBc,106
-better_mem0_mcp/__main__.py,sha256=IeeRidmZF4oBaamjQM6FUbhXpsSdE4sQF_6y_D2GEE4,116
-better_mem0_mcp/config.py,sha256=cqLGzxYno7SgRMn7Y8GOBR0P4CrBuFjsmFpK6eGZhs0,5312
-better_mem0_mcp/graph.py,sha256=rE9z6XECiAktEqDNgmwqCpFpvKSn3azO9H4sRBhj8UU,6195
-better_mem0_mcp/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-better_mem0_mcp/server.py,sha256=QYLP46pz0U9gkxQKB_g8FMga77ywElMMjSCU-hGZQ-M,5499
-better_mem0_mcp/docs/memory.md,sha256=198dDuAGccG5Ca7rhEXIU03ZhxNKK44B_Brl2glurGc,1608
-better_mem0_mcp-1.1.0b22.dist-info/METADATA,sha256=GPpfk3pfbGIzWDVgHn6WdZ2G61sStge0k-EYUlrrUS8,4679
-better_mem0_mcp-1.1.0b22.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-better_mem0_mcp-1.1.0b22.dist-info/entry_points.txt,sha256=2b7E3D6yo94mQXP2Ms0bhUlWkK9f664f0GrstImOq30,57
-better_mem0_mcp-1.1.0b22.dist-info/licenses/LICENSE,sha256=d7xQ6sRyeGus6gnvwgqiQtSY7XdFw0Jd0w5-Co_xHnk,1064
-better_mem0_mcp-1.1.0b22.dist-info/RECORD,,