Merge pull request #100 from neo4j-contrib/memory-updates

Memory updates

Author: a-s-g93

servers/mcp-neo4j-memory/CHANGELOG.md

@@ -2,13 +2,20 @@
 
 ### Fixed
 * Fix bug in `search_nodes` method where query arg wasn't passed properly
+* Fix bug where stdio transport was always selected
+* Fixed argument parsing in server init
 
 ### Changed
 * Implement FastMCP with function decorators to simplify server code
 * Add HTTP transport option
 * Migrate to FastMCP v2.x
+* rename tools to be more clear - `search_nodes` into `search_memories` and `find_nodes` into `find_memories_by_name`
+* Update underlying Pydantic class `ObservationAddition` to have `observations` field to be consistent with `ObservationDeletion` class
+* Update Dockerfile to include `NEO4J_DATABASE`, `NEO4J_TRANSPORT`, `NEO4J_MCP_SERVER_HOST`, `NEO4J_MCP_SERVER_PORT` and `NEO4J_MCP_SERVER_PATH` env variables
 
 ### Added
+* Add compatibility for NEO4J_URI and NEO4J_URL env variables
+* Command in Makefile to easily build and deploy Docker image locally
 
 ## v0.1.5
 

servers/mcp-neo4j-memory/Dockerfile

@@ -10,7 +10,7 @@ RUN pip install --no-cache-dir hatchling
 COPY pyproject.toml /app/
 
 # Install runtime dependencies
-RUN pip install --no-cache-dir mcp>=0.10.0 neo4j>=5.26.0
+RUN pip install --no-cache-dir fastmcp>=2.0.0 neo4j>=5.26.0
 
 # Copy the source code
 COPY src/ /app/src/
@@ -23,6 +23,11 @@ RUN pip install --no-cache-dir -e .
 ENV NEO4J_URL="bolt://host.docker.internal:7687"
 ENV NEO4J_USERNAME="neo4j"
 ENV NEO4J_PASSWORD="password"
+ENV NEO4J_DATABASE="neo4j"
+ENV NEO4J_TRANSPORT="http"
+ENV NEO4J_MCP_SERVER_HOST="0.0.0.0"
+ENV NEO4J_MCP_SERVER_PORT="8000"
+ENV NEO4J_MCP_SERVER_PATH="/api/mcp/"
 
 # Command to run the server using the package entry point
-CMD ["sh", "-c", "mcp-neo4j-memory --db-url ${NEO4J_URL} --username ${NEO4J_USERNAME} --password ${NEO4J_PASSWORD}"]
+CMD ["sh", "-c", "mcp-neo4j-memory"]

servers/mcp-neo4j-memory/Makefile

@@ -1,3 +1,7 @@
+docker-local-build-run:
+	docker build -t mcp-neo4j-memory .
+	docker run -p 8000:8000 mcp-neo4j-memory:latest
+
 install-dev:
 	uv run python3 -m uv pip install -e .
 

servers/mcp-neo4j-memory/src/mcp_neo4j_memory/__init__.py

@@ -7,29 +7,21 @@
 def main():
     """Main entry point for the package."""
     parser = argparse.ArgumentParser(description='Neo4j Memory MCP Server')
-    parser.add_argument('--db-url', 
-                       default=os.getenv("NEO4J_URL", "bolt://localhost:7687"),
-                       help='Neo4j connection URL')
-    parser.add_argument('--username', 
-                       default=os.getenv("NEO4J_USERNAME", "neo4j"),
-                       help='Neo4j username')
-    parser.add_argument('--password', 
-                       default=os.getenv("NEO4J_PASSWORD", "password"),
-                       help='Neo4j password')
-    parser.add_argument("--database",
-                        default=os.getenv("NEO4J_DATABASE", "neo4j"),
-                        help="Neo4j database name")
-    parser.add_argument("--transport", default="stdio", help="Transport type (stdio, sse, http)")
+    parser.add_argument('--db-url', default=None, help='Neo4j connection URL')
+    parser.add_argument('--username', default=None, help='Neo4j username')
+    parser.add_argument('--password', default=None, help='Neo4j password')
+    parser.add_argument("--database", default=None, help="Neo4j database name")
+    parser.add_argument("--transport", default=None, help="Transport type (stdio, sse, http)")
     parser.add_argument("--server-host", default=None, help="HTTP host (default: 127.0.0.1)")
     parser.add_argument("--server-port", type=int, default=None, help="HTTP port (default: 8000)")
     parser.add_argument("--server-path", default=None, help="HTTP path (default: /mcp/)")
 
     args = parser.parse_args()
     asyncio.run(server.main(
-        args.db_url, 
-        args.username, 
-        args.password, 
-        args.database,
+        args.db_url or os.getenv("NEO4J_URL") or os.getenv("NEO4J_URI", "bolt://localhost:7687"),
+        args.username or os.getenv("NEO4J_USERNAME", "neo4j"),
+        args.password or os.getenv("NEO4J_PASSWORD", "password"),
+        args.database or os.getenv("NEO4J_DATABASE", "neo4j"),
         args.transport or os.getenv("NEO4J_TRANSPORT", "stdio"),
         args.server_host or os.getenv("NEO4J_MCP_SERVER_HOST", "127.0.0.1"),
         args.server_port or int(os.getenv("NEO4J_MCP_SERVER_PORT", "8000")),