Merge branch 'main' into fix-backends

.dockerignore

@@ -0,0 +1 @@
+.env

Dockerfile

@@ -33,7 +33,6 @@ WORKDIR /app
 COPY --from=builder /root/.local /root/.local
 COPY ./lightrag ./lightrag
 COPY setup.py .
-COPY .env .
 
 RUN pip install .
 # Make sure scripts in .local are usable

docker-compose.yml

@@ -8,6 +8,7 @@ services:
     volumes:
       - ./data/rag_storage:/app/data/rag_storage
       - ./data/inputs:/app/data/inputs
+      - .env:/app/.env
     env_file:
       - .env
     environment:

lightrag/api/lightrag_server.py

@@ -247,11 +247,11 @@ def display_splash_screen(args: argparse.Namespace) -> None:
         ASCIIColors.yellow(f"{protocol}://<your-ip-address>:{args.port}")
         ASCIIColors.white("    ├─ API Documentation (local): ", end="")
         ASCIIColors.yellow(f"{protocol}://localhost:{args.port}/docs")
-        ASCIIColors.white("    └─ Alternative Documentation (local): ", end="")
+        ASCIIColors.white("    ├─ Alternative Documentation (local): ", end="")
         ASCIIColors.yellow(f"{protocol}://localhost:{args.port}/redoc")
         ASCIIColors.white("    ├─ WebUI (local): ", end="")
         ASCIIColors.yellow(f"{protocol}://localhost:{args.port}/webui")
-        ASCIIColors.white("    ├─ Graph Viewer (local): ", end="")
+        ASCIIColors.white("    └─ Graph Viewer (local): ", end="")
         ASCIIColors.yellow(f"{protocol}://localhost:{args.port}/graph-viewer")
 
         ASCIIColors.yellow("\n📝 Note:")

lightrag/kg/neo4j_impl.py

@@ -55,9 +55,11 @@ class Neo4JStorage(BaseGraphStorage):
         PASSWORD = os.environ.get(
             "NEO4J_PASSWORD", config.get("neo4j", "password", fallback=None)
         )
-        MAX_CONNECTION_POOL_SIZE = os.environ.get(
+        MAX_CONNECTION_POOL_SIZE = int(
-            "NEO4J_MAX_CONNECTION_POOL_SIZE",
+            os.environ.get(
-            config.get("neo4j", "connection_pool_size", fallback=800),
+                "NEO4J_MAX_CONNECTION_POOL_SIZE",
+                config.get("neo4j", "connection_pool_size", fallback=800),
+            )
         )
         DATABASE = os.environ.get(
             "NEO4J_DATABASE", re.sub(r"[^a-zA-Z0-9-]", "-", namespace)

lightrag/lightrag.py

@@ -69,7 +69,7 @@ STORAGE_IMPLEMENTATIONS = {
     "VECTOR_STORAGE": {
         "implementations": [
             "NanoVectorDBStorage",
-            "MilvusVectorDBStorge",
+            "MilvusVectorDBStorage",
             "ChromaVectorDBStorage",
             "TiDBVectorDBStorage",
             "PGVectorStorage",
@@ -128,7 +128,7 @@ STORAGE_ENV_REQUIREMENTS = {
     ],
     # Vector Storage Implementations
     "NanoVectorDBStorage": [],
-    "MilvusVectorDBStorge": [],
+    "MilvusVectorDBStorage": [],
     "ChromaVectorDBStorage": [],
     "TiDBVectorDBStorage": ["TIDB_USER", "TIDB_PASSWORD", "TIDB_DATABASE"],
     "PGVectorStorage": ["POSTGRES_USER", "POSTGRES_PASSWORD", "POSTGRES_DATABASE"],