|
|
@ -2,6 +2,7 @@ import json
|
|
|
|
import logging
|
|
|
|
import logging
|
|
|
|
|
|
|
|
|
|
|
|
import requests
|
|
|
|
import requests
|
|
|
|
|
|
|
|
from openpyxl import load_workbook
|
|
|
|
|
|
|
|
|
|
|
|
# 更详细地控制日志输出
|
|
|
|
# 更详细地控制日志输出
|
|
|
|
logger = logging.getLogger('MathRag')
|
|
|
|
logger = logging.getLogger('MathRag')
|
|
|
@ -19,7 +20,7 @@ def getAnswer(query_text):
|
|
|
|
}
|
|
|
|
}
|
|
|
|
data = {
|
|
|
|
data = {
|
|
|
|
"query": query_text,
|
|
|
|
"query": query_text,
|
|
|
|
"topic":"ChuZhongShuXue"
|
|
|
|
"topic": "ChuZhongShuXue"
|
|
|
|
}
|
|
|
|
}
|
|
|
|
try:
|
|
|
|
try:
|
|
|
|
response = requests.post(url, headers=headers, data=json.dumps(data))
|
|
|
|
response = requests.post(url, headers=headers, data=json.dumps(data))
|
|
|
@ -30,6 +31,19 @@ def getAnswer(query_text):
|
|
|
|
return None
|
|
|
|
return None
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
excel_file = r'D:\dsWork\dsProject\dsLightRag\Doc\T1、史校长资料\测试结果\100题测试结果.xlsx'
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
if __name__ == '__main__':
|
|
|
|
query_text = '什么是数学的性质命题'
|
|
|
|
# query_text = '什么是数学的性质命题'
|
|
|
|
print(getAnswer(query_text))
|
|
|
|
# print(getAnswer(query_text))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
wb = load_workbook(excel_file)
|
|
|
|
|
|
|
|
ws = wb.active # 获取第一个sheet
|
|
|
|
|
|
|
|
# 读取C列第2行到101行的数据
|
|
|
|
|
|
|
|
idx = 1
|
|
|
|
|
|
|
|
for row in range(2, 102): # 行号从2到101
|
|
|
|
|
|
|
|
cell_value = ws[f'C{row}'].value
|
|
|
|
|
|
|
|
logger.info("问题{}:{}".format(idx, cell_value))
|
|
|
|
|
|
|
|
idx = idx + 1
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
wb.close()
|