'commit'

dsLightRag/Test/T1_RW_Excel.py

@@ -0,0 +1,15 @@
+from openpyxl import load_workbook
+
+excel_file = r'D:\dsWork\dsProject\dsLightRag\Doc\T1、史校长资料\测试结果\100题测试结果.xlsx'
+
+# 1、利用EXCEL工具，读取O1的内容
+wb = load_workbook(excel_file)
+ws = wb.active  # 获取第一个sheet
+O1_content = ws['O1'].value  # 获取O1单元格内容
+print(f"O1单元格内容为: {O1_content}")
+
+# 2、写入示例
+ws['O1'] = "【理想】召回文本"  # 修改O1单元格内容
+wb.save(excel_file)  # 保存修改
+
+wb.close()

dsLightRag/Test/T2_AnswerFromRAG.py

@@ -2,6 +2,7 @@ import json
 import logging
 
 import requests
+from openpyxl import load_workbook
 
 # 更详细地控制日志输出
 logger = logging.getLogger('MathRag')
@@ -19,7 +20,7 @@ def getAnswer(query_text):
     }
     data = {
         "query": query_text,
-        "topic":"ChuZhongShuXue"
+        "topic": "ChuZhongShuXue"
     }
     try:
         response = requests.post(url, headers=headers, data=json.dumps(data))
@@ -30,6 +31,19 @@ def getAnswer(query_text):
         return None
 
 
+excel_file = r'D:\dsWork\dsProject\dsLightRag\Doc\T1、史校长资料\测试结果\100题测试结果.xlsx'
+
 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()