diff --git a/.dockerignore b/.dockerignore new file mode 100644 index 00000000..4c49bd78 --- /dev/null +++ b/.dockerignore @@ -0,0 +1 @@ +.env diff --git a/Dockerfile b/Dockerfile index 8c9474de..4532ee61 100644 --- a/Dockerfile +++ b/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 diff --git a/docker-compose.yml b/docker-compose.yml index a4713279..b5659692 100644 --- a/docker-compose.yml +++ b/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: diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py index a442074d..28085723 100644 --- a/lightrag/api/lightrag_server.py +++ b/lightrag/api/lightrag_server.py @@ -247,11 +247,11 @@ def display_splash_screen(args: argparse.Namespace) -> None: ASCIIColors.yellow(f"{protocol}://:{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:") diff --git a/lightrag/kg/neo4j_impl.py b/lightrag/kg/neo4j_impl.py index c1c313ca..e9a53110 100644 --- a/lightrag/kg/neo4j_impl.py +++ b/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( - "NEO4J_MAX_CONNECTION_POOL_SIZE", - config.get("neo4j", "connection_pool_size", fallback=800), + MAX_CONNECTION_POOL_SIZE = int( + os.environ.get( + "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) diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py index 6db6e752..73716621 100644 --- a/lightrag/lightrag.py +++ b/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"],