gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update Step_3.py and openai compatible script

Browse Source
This commit is contained in:
KIM Jae Boum
2024-10-18 06:06:47 +08:00
parent 7163f70924
commit a8f2e6a478
3 changed files with 167 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
reproduce/Step_3.py
View File

@@ -53,10 +53,10 @@ def run_queries_and_save_to_json(queries, rag_instance, query_param, output_file
if __name__ == "__main__":
cls = "agriculture"
mode = "hybrid"
WORKING_DIR = "../{cls}"
WORKING_DIR = f"../{cls}"
rag = LightRAG(working_dir=WORKING_DIR)
query_param = QueryParam(mode=mode)
queries = extract_queries(f"../datasets/questions/{cls}_questions.txt")
run_queries_and_save_to_json(queries, rag, query_param, "result.json", "errors.json")
run_queries_and_save_to_json(queries, rag, query_param, f"{cls}_result.json", f"{cls}_errors.json")