diff options
Diffstat (limited to 'gnqa/src/study2/run_questions.py')
-rw-r--r-- | gnqa/src/study2/run_questions.py | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/gnqa/src/study2/run_questions.py b/gnqa/src/study2/run_questions.py new file mode 100644 index 00000000..9bac5a23 --- /dev/null +++ b/gnqa/src/study2/run_questions.py @@ -0,0 +1,41 @@ +import json +import sys +import os + +from r2r import R2RClient, RAGResponse +from document_operations import DocOps, QuestionList + +''' +******************************************************************************* +Variables +******************************************************************************* +''' +rag_gen_cfg = {"model": "openai/gpt-4o-mini", "temperature": 0.0, "use_hybrid_search": True} +rag_response = {} +#client = R2RClient("http://localhost:8000") +client = R2RClient("http://localhost:7272") +#health_resp = client.health() + +''' +******************************************************************************* +Commands +******************************************************************************* +''' + +#print("The R2R client's health status is {0}".format(health_resp)) + +try: + read_file = str(sys.argv[1]) + out_file = str(sys.argv[2]) +except: + exit('Example use "python run_questions.py ../../data/study2/lists/human-questions.json ../../data/test_study/human/de/aging_resp.json"') + +qLst = QuestionList(read_file, 1) # second parameter is for verbose output +ndx = 1 +for question in qLst.get("domainexpert","aging"): + print('Getting response for the following question --> {0}'.format(question)) + #rag_response[str(ndx)] = client.rag(question) + rag_response[str(ndx)] = client.retrieval.rag(question, rag_gen_cfg) + ndx += 1 + +DocOps.writeRAGResponses(rag_response, out_file)
\ No newline at end of file |