Added database viewer

Added Memgraph lab

Author: nullchimp

docker/docker-compose.yml

@@ -9,7 +9,7 @@ services:
       - "7687:7687" # Bolt protocol
       - "7444:7444" # HTTP API
     volumes:
-      - memgraph_data:/data
+      - memgraph_data:/var/lib/memgraph
       - memgraph_logs:/var/log/memgraph
       - memgraph_conf:/etc/memgraph
     healthcheck:
@@ -18,7 +18,28 @@ services:
       timeout: 10s
       retries: 3
     restart: unless-stopped
-    command: ["--log-level=TRACE", "--also-log-to-stderr"]
+    command: [
+      "--log-level=TRACE",
+      "--also-log-to-stderr",
+      "--storage-snapshot-interval-sec=300",
+      "--storage-wal-enabled=true",
+      "--storage-snapshot-on-exit=true",
+      "--data-directory=/var/lib/memgraph"
+    ]
+
+  memgraph-lab:
+    image: memgraph/lab:latest
+    container_name: ai-agent-memgraph-lab
+    ports:
+      - "3000:3000"  # Memgraph Lab UI
+    environment:
+      - QUICK_CONNECT_MG_HOST=memgraph
+      - QUICK_CONNECT_MG_PORT=7687
+      - QUICK_CONNECT_MG_USERNAME=${MEMGRAPH_USERNAME}
+      - QUICK_CONNECT_MG_PASSWORD=${MEMGRAPH_PASSWORD}
+    depends_on:
+      - memgraph
+    restart: unless-stopped
 
   neo4j-browser:
     image: neo4j:5.15-community

scripts/check_database.py

@@ -0,0 +1,52 @@
+from dotenv import load_dotenv
+load_dotenv(override=True)
+
+import os
+from core.db.handler.memgraph import MemGraphClient
+
+db = MemGraphClient(
+    host=os.environ.get("MEMGRAPH_URI", "localhost"),
+    port=int(os.environ.get("MEMGRAPH_PORT", 7687)),
+    username=os.environ.get("MEMGRAPH_USERNAME", "nullchimp"),
+    password=os.environ.get("MEMGRAPH_PASSWORD", "nullchimp"),
+).connect()
+
+print("=" * 60)
+print("DATABASE INSPECTION")
+print("=" * 60)
+
+print("\n1. Total nodes:")
+db._cur.execute("MATCH (n) RETURN count(n) as total")
+result = db._cur.fetchone()
+print(f"   Total nodes: {result[0]}")
+
+print("\n2. Node labels and counts:")
+db._cur.execute("MATCH (n) RETURN labels(n) as labels, count(*) as count ORDER BY count DESC")
+for row in db._cur.fetchall():
+    labels = row[0] if row[0] else ['(no label)']
+    print(f"   {labels}: {row[1]}")
+
+print("\n3. Sample nodes (first 5):")
+db._cur.execute("MATCH (n) RETURN n LIMIT 5")
+for i, row in enumerate(db._cur.fetchall(), 1):
+    node = row[0]
+    print(f"\n   Node {i}:")
+    print(f"   Labels: {node.labels}")
+    print(f"   Properties: {dict(node.properties)}")
+
+print("\n4. All property keys:")
+db._cur.execute("MATCH (n) UNWIND keys(n) as key RETURN DISTINCT key ORDER BY key")
+keys = [row[0] for row in db._cur.fetchall()]
+print(f"   {', '.join(keys)}")
+
+print("\n5. Relationships:")
+db._cur.execute("MATCH ()-[r]->() RETURN DISTINCT type(r) as rel_type, count(*) as count ORDER BY count DESC")
+rels = db._cur.fetchall()
+if rels:
+    for row in rels:
+        print(f"   {row[0]}: {row[1]}")
+else:
+    print("   No relationships found")
+
+db.close()
+print("\n" + "=" * 60)