From df07c2a8b16cc3bba4cf807f2030c4936728d3cf Mon Sep 17 00:00:00 2001 From: yangdx Date: Wed, 2 Apr 2025 14:43:53 +0800 Subject: [PATCH] Remove Gremlin storage implementaion --- README-zh.md | 2 +- README.md | 2 +- examples/lightrag_ollama_gremlin_demo.py | 4 ++++ lightrag/api/README-zh.md | 1 - lightrag/api/README.md | 1 - lightrag/kg/__init__.py | 6 +++--- 6 files changed, 9 insertions(+), 7 deletions(-) diff --git a/README-zh.md b/README-zh.md index d8678a93..5471137c 100644 --- a/README-zh.md +++ b/README-zh.md @@ -1038,7 +1038,7 @@ rag.clear_cache(modes=["local"]) | **working_dir** | `str` | 存储缓存的目录 | `lightrag_cache+timestamp` | | **kv_storage** | `str` | Storage type for documents and text chunks. Supported types: `JsonKVStorage`,`PGKVStorage`,`RedisKVStorage`,`MongoKVStorage` | `JsonKVStorage` | | **vector_storage** | `str` | Storage type for embedding vectors. Supported types: `NanoVectorDBStorage`,`PGVectorStorage`,`MilvusVectorDBStorage`,`ChromaVectorDBStorage`,`FaissVectorDBStorage`,`MongoVectorDBStorage`,`QdrantVectorDBStorage` | `NanoVectorDBStorage` | -| **graph_storage** | `str` | Storage type for graph edges and nodes. Supported types: `NetworkXStorage`,`Neo4JStorage`,`PGGraphStorage`,`AGEStorage`,`GremlinStorage` | `NetworkXStorage` | +| **graph_storage** | `str` | Storage type for graph edges and nodes. Supported types: `NetworkXStorage`,`Neo4JStorage`,`PGGraphStorage`,`AGEStorage` | `NetworkXStorage` | | **doc_status_storage** | `str` | Storage type for documents process status. Supported types: `JsonDocStatusStorage`,`PGDocStatusStorage`,`MongoDocStatusStorage` | `JsonDocStatusStorage` | | **chunk_token_size** | `int` | 拆分文档时每个块的最大令牌大小 | `1200` | | **chunk_overlap_token_size** | `int` | 拆分文档时两个块之间的重叠令牌大小 | `100` | diff --git a/README.md b/README.md index b97a251e..33bb9278 100644 --- a/README.md +++ b/README.md @@ -1066,7 +1066,7 @@ Valid modes are: | **working_dir** | `str` | Directory where the cache will be stored | `lightrag_cache+timestamp` | | **kv_storage** | `str` | Storage type for documents and text chunks. Supported types: `JsonKVStorage`,`PGKVStorage`,`RedisKVStorage`,`MongoKVStorage` | `JsonKVStorage` | | **vector_storage** | `str` | Storage type for embedding vectors. Supported types: `NanoVectorDBStorage`,`PGVectorStorage`,`MilvusVectorDBStorage`,`ChromaVectorDBStorage`,`FaissVectorDBStorage`,`MongoVectorDBStorage`,`QdrantVectorDBStorage` | `NanoVectorDBStorage` | -| **graph_storage** | `str` | Storage type for graph edges and nodes. Supported types: `NetworkXStorage`,`Neo4JStorage`,`PGGraphStorage`,`AGEStorage`,`GremlinStorage` | `NetworkXStorage` | +| **graph_storage** | `str` | Storage type for graph edges and nodes. Supported types: `NetworkXStorage`,`Neo4JStorage`,`PGGraphStorage`,`AGEStorage` | `NetworkXStorage` | | **doc_status_storage** | `str` | Storage type for documents process status. Supported types: `JsonDocStatusStorage`,`PGDocStatusStorage`,`MongoDocStatusStorage` | `JsonDocStatusStorage` | | **chunk_token_size** | `int` | Maximum token size per chunk when splitting documents | `1200` | | **chunk_overlap_token_size** | `int` | Overlap token size between two chunks when splitting documents | `100` | diff --git a/examples/lightrag_ollama_gremlin_demo.py b/examples/lightrag_ollama_gremlin_demo.py index 893b5606..7ae62810 100644 --- a/examples/lightrag_ollama_gremlin_demo.py +++ b/examples/lightrag_ollama_gremlin_demo.py @@ -1,3 +1,7 @@ +############################################## +# Gremlin storage implementation is deprecated +############################################## + import asyncio import inspect import os diff --git a/lightrag/api/README-zh.md b/lightrag/api/README-zh.md index 745a15f2..03718656 100644 --- a/lightrag/api/README-zh.md +++ b/lightrag/api/README-zh.md @@ -303,7 +303,6 @@ NetworkXStorage NetworkX(默认) Neo4JStorage Neo4J PGGraphStorage Postgres AGEStorage AGE -GremlinStorage Gremlin ``` * VECTOR_STORAGE 支持的实现名称 diff --git a/lightrag/api/README.md b/lightrag/api/README.md index 84efdafc..27f3d14a 100644 --- a/lightrag/api/README.md +++ b/lightrag/api/README.md @@ -314,7 +314,6 @@ NetworkXStorage NetworkX(defualt) Neo4JStorage Neo4J PGGraphStorage Postgres AGEStorage AGE -GremlinStorage Gremlin ``` * VECTOR_STORAGE supported implement-name diff --git a/lightrag/kg/__init__.py b/lightrag/kg/__init__.py index 77e13c76..b917bfa0 100644 --- a/lightrag/kg/__init__.py +++ b/lightrag/kg/__init__.py @@ -16,7 +16,7 @@ STORAGE_IMPLEMENTATIONS = { # "MongoGraphStorage", # "TiDBGraphStorage", "AGEStorage", - "GremlinStorage", + # "GremlinStorage", "PGGraphStorage", ], "required_methods": ["upsert_node", "upsert_edge"], @@ -62,7 +62,7 @@ STORAGE_ENV_REQUIREMENTS: dict[str, list[str]] = { "AGE_POSTGRES_USER", "AGE_POSTGRES_PASSWORD", ], - "GremlinStorage": ["GREMLIN_HOST", "GREMLIN_PORT", "GREMLIN_GRAPH"], + # "GremlinStorage": ["GREMLIN_HOST", "GREMLIN_PORT", "GREMLIN_GRAPH"], "PGGraphStorage": [ "POSTGRES_USER", "POSTGRES_PASSWORD", @@ -104,7 +104,7 @@ STORAGES = { "PGVectorStorage": ".kg.postgres_impl", "AGEStorage": ".kg.age_impl", "PGGraphStorage": ".kg.postgres_impl", - "GremlinStorage": ".kg.gremlin_impl", + # "GremlinStorage": ".kg.gremlin_impl", "PGDocStatusStorage": ".kg.postgres_impl", "FaissVectorDBStorage": ".kg.faiss_impl", "QdrantVectorDBStorage": ".kg.qdrant_impl",