gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: New ini file reading database configuration method for production environment convenience, modified Lightrag ainsert method - add.doc_keys retrieval method to only filter existing ones, but this will prevent failed documents from being stored again. Added -- chunk_Size and -- chunkoverlapped size for production environment convenience, and added llm_binding: openai-ollama for easy use of openai while also using ollama embedding

Browse Source
This commit is contained in:
hyb
2025-01-24 17:10:19 +08:00
parent afb30fe6c1
commit 7b02fac7b9
1 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
lightrag/api/lightrag_server.py
View File

@@ -63,17 +63,17 @@ DOC_STATUS_STORAGE = "JsonDocStatusStorage"
GRAPH_STORAGE = "NetworkXStorage" GRAPH_STORAGE = "NetworkXStorage"
VECTOR_STORAGE = "NanoVectorDBStorage" VECTOR_STORAGE = "NanoVectorDBStorage"
# 读取配置文件 # read config.ini
config = configparser.ConfigParser() config = configparser.ConfigParser()
config.read("config.ini") config.read("config.ini")
# Redis 配置 # Redis config
redis_uri = config.get("redis", "uri", fallback=None) redis_uri = config.get("redis", "uri", fallback=None)
if redis_uri: if redis_uri:
os.environ["REDIS_URI"] = redis_uri os.environ["REDIS_URI"] = redis_uri
KV_STORAGE = "RedisKVStorage" KV_STORAGE = "RedisKVStorage"
DOC_STATUS_STORAGE = "RedisKVStorage" DOC_STATUS_STORAGE = "RedisKVStorage"
# Neo4j 配置 # Neo4j config
neo4j_uri = config.get("neo4j", "uri", fallback=None) neo4j_uri = config.get("neo4j", "uri", fallback=None)
neo4j_username = config.get("neo4j", "username", fallback=None) neo4j_username = config.get("neo4j", "username", fallback=None)
neo4j_password = config.get("neo4j", "password", fallback=None) neo4j_password = config.get("neo4j", "password", fallback=None)
@@ -83,7 +83,7 @@ if neo4j_uri:
os.environ["NEO4J_PASSWORD"] = neo4j_password os.environ["NEO4J_PASSWORD"] = neo4j_password
GRAPH_STORAGE = "Neo4JStorage" GRAPH_STORAGE = "Neo4JStorage"
# Milvus 配置 # Milvus config
milvus_uri = config.get("milvus", "uri", fallback=None) milvus_uri = config.get("milvus", "uri", fallback=None)
milvus_user = config.get("milvus", "user", fallback=None) milvus_user = config.get("milvus", "user", fallback=None)
milvus_password = config.get("milvus", "password", fallback=None) milvus_password = config.get("milvus", "password", fallback=None)
@@ -95,7 +95,7 @@ if milvus_uri:
os.environ["MILVUS_DB_NAME"] = milvus_db_name os.environ["MILVUS_DB_NAME"] = milvus_db_name
VECTOR_STORAGE = "MilvusVectorDBStorge" VECTOR_STORAGE = "MilvusVectorDBStorge"
# MongoDB 配置 # MongoDB config
mongo_uri = config.get("mongodb", "uri", fallback=None) mongo_uri = config.get("mongodb", "uri", fallback=None)
mongo_database = config.get("mongodb", "LightRAG", fallback=None) mongo_database = config.get("mongodb", "LightRAG", fallback=None)
if mongo_uri: if mongo_uri:
@@ -611,8 +611,13 @@ def get_api_key_dependency(api_key: Optional[str]):
def create_app(args): def create_app(args):
# Verify that bindings arer correctly setup # Verify that bindings arer correctly setup
if args.llm_binding not in ["lollms", "ollama", "openai", "openai-ollama", "azure_openai"]: if args.llm_binding not in [
"lollms",
"ollama",
"openai",
"openai-ollama",
"azure_openai",
]:
raise Exception("llm binding not supported") raise Exception("llm binding not supported")
if args.embedding_binding not in ["lollms", "ollama", "openai", "azure_openai"]: if args.embedding_binding not in ["lollms", "ollama", "openai", "azure_openai"]: