blob: 9bac5a2336eae153a655c7eadc682f366d356a5d (
about) (
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
|
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)
|
