LightRAG/examples/insert.py

19 lines
318 B
Python
Raw Normal View History

2024-10-07 18:48:45 +08:00
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)