From e1d9dba20de356ba4e8eb214c361ad5f1fecbdae Mon Sep 17 00:00:00 2001 From: yangdx Date: Thu, 13 Feb 2025 20:40:29 +0800 Subject: [PATCH 1/2] Fix typos in documentation tree structure and storage implementation names MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit • Correct menu tree structure in splash screen • Fix MilvusVectorDBStorage typo • Update storage env requirements --- lightrag/api/lightrag_server.py | 4 ++-- lightrag/lightrag.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py index fe52f592..94d59f69 100644 --- a/lightrag/api/lightrag_server.py +++ b/lightrag/api/lightrag_server.py @@ -246,11 +246,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/lightrag.py b/lightrag/lightrag.py index d53a252d..daf4de20 100644 --- a/lightrag/lightrag.py +++ b/lightrag/lightrag.py @@ -69,7 +69,7 @@ STORAGE_IMPLEMENTATIONS = { "VECTOR_STORAGE": { "implementations": [ "NanoVectorDBStorage", - "MilvusVectorDBStorge", + "MilvusVectorDBStorage", "ChromaVectorDBStorage", "TiDBVectorDBStorage", "PGVectorStorage", @@ -123,7 +123,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"], From 47268c2ab167fd6d2ed28141dd665f238d549829 Mon Sep 17 00:00:00 2001 From: yangdx Date: Thu, 13 Feb 2025 21:01:34 +0800 Subject: [PATCH 2/2] Fix linting --- lightrag/api/lightrag_server.py | 1 + 1 file changed, 1 insertion(+) diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py index 94d59f69..28085723 100644 --- a/lightrag/api/lightrag_server.py +++ b/lightrag/api/lightrag_server.py @@ -101,6 +101,7 @@ def estimate_tokens(text: str) -> int: return int(tokens) + def get_default_host(binding_type: str) -> str: default_hosts = { "ollama": os.getenv("LLM_BINDING_HOST", "http://localhost:11434"),