|
|
@ -152,7 +152,7 @@ os.environ["POSTGRES_PASSWORD"] = POSTGRES_PASSWORD
|
|
|
|
os.environ["POSTGRES_DATABASE"] = POSTGRES_DATABASE
|
|
|
|
os.environ["POSTGRES_DATABASE"] = POSTGRES_DATABASE
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async def initialize_pg_rag(WORKING_DIR, namespace_prefix):
|
|
|
|
async def initialize_pg_rag(WORKING_DIR):
|
|
|
|
rag = LightRAG(
|
|
|
|
rag = LightRAG(
|
|
|
|
working_dir=WORKING_DIR,
|
|
|
|
working_dir=WORKING_DIR,
|
|
|
|
llm_model_func=llm_model_func,
|
|
|
|
llm_model_func=llm_model_func,
|
|
|
@ -169,8 +169,7 @@ async def initialize_pg_rag(WORKING_DIR, namespace_prefix):
|
|
|
|
doc_status_storage="PGDocStatusStorage",
|
|
|
|
doc_status_storage="PGDocStatusStorage",
|
|
|
|
graph_storage="PGGraphStorage",
|
|
|
|
graph_storage="PGGraphStorage",
|
|
|
|
vector_storage="PGVectorStorage",
|
|
|
|
vector_storage="PGVectorStorage",
|
|
|
|
auto_manage_storages_states=False,
|
|
|
|
auto_manage_storages_states=False
|
|
|
|
namespace_prefix=namespace_prefix,
|
|
|
|
|
|
|
|
)
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
await rag.initialize_storages()
|
|
|
|
await rag.initialize_storages()
|
|
|
|