diff --git a/dsRag/Config/Config.py b/dsRag/Config/Config.py index b6b46230..9ce5b393 100644 --- a/dsRag/Config/Config.py +++ b/dsRag/Config/Config.py @@ -21,3 +21,8 @@ MODEL_NAME = "qwen-plus" # Jieba分词自定义词典配置 JIEBA_CUSTOM_WORDS = ['文言虚词', '花呗'] + +# 腾讯 AI Lab 中文词向量模型的路径 +MODEL_PATH = "D:/Tencent_AILab_ChineseEmbedding/Tencent_AILab_ChineseEmbedding.txt" +# 加载词向量模型时限制的词汇数量 +MODEL_LIMIT = 10000 \ No newline at end of file diff --git a/dsRag/Config/__pycache__/Config.cpython-310.pyc b/dsRag/Config/__pycache__/Config.cpython-310.pyc index b4aec3e6..e4ccb80b 100644 Binary files a/dsRag/Config/__pycache__/Config.cpython-310.pyc and b/dsRag/Config/__pycache__/Config.cpython-310.pyc differ diff --git a/dsRag/Doc/9、启动.txt b/dsRag/Doc/9、启动.txt index 93e34ea6..a9b446b6 100644 --- a/dsRag/Doc/9、启动.txt +++ b/dsRag/Doc/9、启动.txt @@ -1 +1 @@ -http://localhost:8000/static/index.html \ No newline at end of file +http://10.10.21.22:8000/static/index.html \ No newline at end of file diff --git a/dsRag/Util/EsSearchUtil.py b/dsRag/Util/EsSearchUtil.py index 6bd9a7b9..8968bfb4 100644 --- a/dsRag/Util/EsSearchUtil.py +++ b/dsRag/Util/EsSearchUtil.py @@ -2,10 +2,9 @@ import logging 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 Config.Config import MODEL_LIMIT, MODEL_PATH from ElasticSearch.Utils.ElasticsearchConnectionPool import ElasticsearchConnectionPool # 初始化日志 @@ -49,7 +48,7 @@ class EsSearchUtil: # 加载预训练模型 - self.model = KeyedVectors.load_word2vec_format(MS_MODEL_PATH, binary=False, limit=MS_MODEL_LIMIT) + self.model = KeyedVectors.load_word2vec_format(MODEL_PATH, binary=False, limit=MODEL_LIMIT) logger.info(f"模型加载成功,词向量维度: {self.model.vector_size}") # 初始化Elasticsearch连接 diff --git a/dsRag/Util/__pycache__/EsSearchUtil.cpython-310.pyc b/dsRag/Util/__pycache__/EsSearchUtil.cpython-310.pyc index 06626bbc..8d10dd86 100644 Binary files a/dsRag/Util/__pycache__/EsSearchUtil.cpython-310.pyc and b/dsRag/Util/__pycache__/EsSearchUtil.cpython-310.pyc differ