fix bug
This commit is contained in:
@@ -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
|
||||
)
|
||||
|
||||
|
||||
|
Reference in New Issue
Block a user