Add Memgraph authentication support

.env.example

@@ -49,5 +49,7 @@ CYPHER_ENDPOINT=http://localhost:11434/v1
 MEMGRAPH_HOST=localhost
 MEMGRAPH_PORT=7687
 MEMGRAPH_HTTP_PORT=7444
+MEMGRAPH_USERNAME=
+MEMGRAPH_PASSWORD=
 LAB_PORT=3000
 TARGET_REPO_PATH=.

codebase_rag/config.py

@@ -42,6 +42,8 @@ class AppConfig(BaseSettings):
     MEMGRAPH_HOST: str = "localhost"
     MEMGRAPH_PORT: int = 7687
     MEMGRAPH_HTTP_PORT: int = 7444
+    MEMGRAPH_USERNAME: str | None = None
+    MEMGRAPH_PASSWORD: str | None = None
     LAB_PORT: int = 3000
     MEMGRAPH_BATCH_SIZE: int = 1000
 

codebase_rag/main.py

@@ -785,6 +785,8 @@ async def main_async(repo_path: str, batch_size: int) -> None:
         host=settings.MEMGRAPH_HOST,
         port=settings.MEMGRAPH_PORT,
         batch_size=batch_size,
+        username=settings.MEMGRAPH_USERNAME,
+        password=settings.MEMGRAPH_PASSWORD,
     ) as ingestor:
         console.print("[bold green]Successfully connected to Memgraph.[/bold green]")
         console.print(
@@ -870,6 +872,8 @@ def start(
             host=settings.MEMGRAPH_HOST,
             port=settings.MEMGRAPH_PORT,
             batch_size=effective_batch_size,
+            username=settings.MEMGRAPH_USERNAME,
+            password=settings.MEMGRAPH_PASSWORD,
         ) as ingestor:
             if clean:
                 console.print("[bold yellow]Cleaning database...[/bold yellow]")
@@ -930,6 +934,8 @@ def export(
             host=settings.MEMGRAPH_HOST,
             port=settings.MEMGRAPH_PORT,
             batch_size=effective_batch_size,
+            username=settings.MEMGRAPH_USERNAME,
+            password=settings.MEMGRAPH_PASSWORD,
         ) as ingestor:
             console.print("[bold cyan]Exporting graph data...[/bold cyan]")
             if not _export_graph_to_file(ingestor, output):
@@ -970,6 +976,8 @@ async def main_optimize_async(
         host=settings.MEMGRAPH_HOST,
         port=settings.MEMGRAPH_PORT,
         batch_size=effective_batch_size,
+        username=settings.MEMGRAPH_USERNAME,
+        password=settings.MEMGRAPH_PASSWORD,
     ) as ingestor:
         console.print("[bold green]Successfully connected to Memgraph.[/bold green]")
 

codebase_rag/mcp/server.py

@@ -103,6 +103,8 @@ def create_server() -> tuple[Server, MemgraphIngestor]:
         host=settings.MEMGRAPH_HOST,
         port=settings.MEMGRAPH_PORT,
         batch_size=settings.MEMGRAPH_BATCH_SIZE,
+        username=settings.MEMGRAPH_USERNAME,
+        password=settings.MEMGRAPH_PASSWORD,
     )
 
     # CypherGenerator gets config from settings automatically

codebase_rag/services/graph_service.py

@@ -9,9 +9,18 @@
 class MemgraphIngestor:
     """Handles all communication and query execution with the Memgraph database."""
 
-    def __init__(self, host: str, port: int, batch_size: int = 1000):
+    def __init__(
+        self,
+        host: str,
+        port: int,
+        batch_size: int = 1000,
+        username: str | None = None,
+        password: str | None = None,
+    ):
         self._host = host
         self._port = port
+        self._username = username
+        self._password = password
         if batch_size < 1:
             raise ValueError("batch_size must be a positive integer")
         self.batch_size = batch_size
@@ -32,7 +41,12 @@ def __init__(self, host: str, port: int, batch_size: int = 1000):
 
     def __enter__(self) -> "MemgraphIngestor":
         logger.info(f"Connecting to Memgraph at {self._host}:{self._port}...")
-        self.conn = mgclient.connect(host=self._host, port=self._port)
+        self.conn = mgclient.connect(
+            host=self._host,
+            port=self._port,
+            username=self._username,
+            password=self._password,
+        )
         self.conn.autocommit = True
         logger.info("Successfully connected to Memgraph.")
         return self

realtime_updater.py

@@ -94,6 +94,8 @@ def start_watcher(
         host=host,
         port=port,
         batch_size=effective_batch_size,
+        username=settings.MEMGRAPH_USERNAME,
+        password=settings.MEMGRAPH_PASSWORD,
     ) as ingestor:
         updater = GraphUpdater(ingestor, repo_path_obj, parsers, queries)