This commit is contained in:
LarFii
2024-10-15 21:11:12 +08:00
parent f2e8732c43
commit cc4af0d75f
3 changed files with 14 additions and 21 deletions

View File

@@ -5,15 +5,15 @@ from lightrag import LightRAG, QueryParam
from lightrag.llm import gpt_4o_mini_complete, gpt_4o_complete
from transformers import AutoModel,AutoTokenizer
WORKING_DIR = "./dickens"
WORKING_DIR = "/home/zrguo/code/myrag/agriculture"
if not os.path.exists(WORKING_DIR):
os.mkdir(WORKING_DIR)
rag = LightRAG(
working_dir=WORKING_DIR,
llm_model_func=gpt_4o_complete
# llm_model_func=gpt_4o_mini_complete
llm_model_func=gpt_4o_mini_complete
# llm_model_func=gpt_4o_complete
)