blob: 5e0b28b2b00663501d138bd9dc3bf06c7541efe9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
import json
import sys
import os
from r2r import R2R, Document, GenerationConfig, R2RClient
def writeDatasetFile(responses, outp_file):
print(outp_file)
output = json.dumps(responses, indent=2)
if os.path.exists(outp_file):
with open(outp_file, "a") as the_data:
the_data.write('' + output)
else:
with open(outp_file, "a") as the_data:
the_data.write(output)
client = R2RClient("http://localhost:8000")
health_resp = client.health()
print("The R2R client's health status is {0}".format(health_resp))
questions = [
"List as many studies as you can that include rapamycin.",
"Why is it so difficult to map gene loci that control aging in humans?"
]
rag_response = {}
ndx = 1
for question in questions:
rag_response[str(ndx)] = client.rag(question)
ndx = ndx + 1
#print(json.dumps(rag_response, indent=2))
try:
read_file = str(sys.argv[1])
out_file = str(sys.argv[2])
except:
exit('Example use "python run_questions.py ../data/questions/human/de/aging.json ../data/responses/human/de/aging_resp.json"')
writeDatasetFile(rag_response, out_file)
|