'commit'

dsRag/ElasticSearch/T6_XiangLiangQuery.py

@@ -1,16 +1,12 @@
 import logging
 
 from Config.Config import ES_CONFIG
+from Util.EsSearchUtil import EsSearchUtil
 
 # 初始化日志
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.INFO)
 
-
-
-# 导入EsSearchUtil
-from Util.EsSearchUtil import EsSearchUtil
-
 # 初始化EsSearchUtil
 es_search_util = EsSearchUtil(ES_CONFIG)
 

dsRag/Util/EsSearchUtil.py

@@ -3,6 +3,8 @@ import os
 from logging.handlers import RotatingFileHandler
 import jieba
 from elasticsearch import Elasticsearch
+from gensim.models import KeyedVectors
+from Config.Config import MS_MODEL_PATH, MS_MODEL_LIMIT
 
 from ElasticSearch.Utils.ElasticsearchConnectionPool import ElasticsearchConnectionPool
 
@@ -16,7 +18,6 @@ handler.setFormatter(logging.Formatter('%(asctime)s - %(levelname)s - %(message)
 logger.addHandler(handler)
 
 class EsSearchUtil:
-
     def __init__(self, es_config):
         """
                 初始化Elasticsearch搜索工具
@@ -43,9 +44,9 @@ class EsSearchUtil:
         # 确保es_conn属性存在以兼容旧代码
         self.es_conn = self.es
 
-    def __init__(self, es_config):
-        from gensim.models import KeyedVectors
-        from Config.Config import MS_MODEL_PATH, MS_MODEL_LIMIT
+        # 确保es_conn属性存在以兼容旧代码
+        self.es_conn = self.es
+
 
         # 加载预训练模型
         self.model = KeyedVectors.load_word2vec_format(MS_MODEL_PATH, binary=False, limit=MS_MODEL_LIMIT)