about summary refs log tree commit diff
path: root/gnqa/src/study2/run_questions.py
diff options
context:
space:
mode:
Diffstat (limited to 'gnqa/src/study2/run_questions.py')
-rw-r--r--gnqa/src/study2/run_questions.py41
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