优化代码，提高解耦

2025-11-19 23:44:14 +08:00
parent 8265c2a7a9
commit a70517f6d8
25 changed files with 143 additions and 355 deletions
@@ -1,12 +1,11 @@
-from langchain_huggingface import HuggingFaceEmbeddings
+from embeddings import get_embeddings

-# 第一次运行可能时间较久
-print('---正在首次加载本地嵌入模型(bge-small-zh-v1.5)...---')
+
+# 首次运行可能时间较久 -- 同时运行本文件需要梯子，不然无法加载到本地
+print('---正在加载本地嵌入模型(bge-small-zh-v1.5)...---')

 # 理论：有embedding的向量模型
-embeddings_model = HuggingFaceEmbeddings(
-    model_name="BAAI/bge-small-zh-v1.5" # 一个中英双语开源模型
-)
+embeddings_model = get_embeddings("bge-small-zh-v1.5")
 print('嵌入模型载入完毕')

 # 演示：将文本转换为向量