LightRAG/examples/insert.py
2024-10-07 18:48:45 +08:00

19 lines
318 B
Python

import os
import sys
sys.path.append('xxx/xxx/LightRAG')
from lightrag import LightRAG
os.environ["OPENAI_API_KEY"] = ""
WORKING_DIR = ""
if not os.path.exists(WORKING_DIR):
os.mkdir(WORKING_DIR)
rag = LightRAG(working_dir=WORKING_DIR)
with open('./text.txt', 'r') as f:
text = f.read()
rag.insert(text)