'commit'

dsRag/ElasticSearch/T4_SelectAllData.py

@@ -1,8 +1,8 @@
 import warnings
 
 from elasticsearch import Elasticsearch
+
 from Config import Config
-import urllib3
 
 # 抑制HTTPS相关警告
 warnings.filterwarnings('ignore', message='Connecting to .* using TLS with verify_certs=False is insecure')

dsRag/ElasticSearch/T6_XiangLiangQuery.py

@@ -14,9 +14,7 @@ esClient = EsSearchUtil(ES_CONFIG)
 warnings.filterwarnings('ignore', message='Connecting to .* using TLS with verify_certs=False is insecure')
 warnings.filterwarnings('ignore', message='Unverified HTTPS request is being made to host')
 
-
-
-def main():
+if __name__ == "__main__":
     # 测试查询
     query = "小学数学中有哪些模型"
     query_tags = ["MATH_1"]  # 默认搜索标签，可修改
@@ -99,12 +97,9 @@ def main():
 
         print("\n文本精确搜索结果:")
         for i, hit in enumerate(text_results['hits']['hits']):
-            print(f"  {i+1}. 文档ID: {hit['_id']}, 匹配分数: {hit['_score']:.2f}")
+            print(f"  {i + 1}. 文档ID: {hit['_id']}, 匹配分数: {hit['_score']:.2f}")
             print(f"      内容: {hit['_source']['user_input']}")
             # print(f"      详细: {hit['_source']['tags']['full_content']}")
 
     finally:
         esClient.es_pool.release_connection(es_conn)
-
-if __name__ == "__main__":
-    main()