opc_agent/agent-craft
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore(embeddings): minor code optimization

Browse Source
This commit is contained in:
Annyfee
2026-01-15 15:32:21 +08:00
parent 3b8aab8984
commit cf504582be
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
embeddings.py
+1 -1
View File
@@ -2,6 +2,7 @@
from pathlib import Path from pathlib import Path
from langchain_huggingface import HuggingFaceEmbeddings from langchain_huggingface import HuggingFaceEmbeddings
from huggingface_hub import snapshot_download
# 获取embeddings模型 -- 首次调用时自动下载 # 获取embeddings模型 -- 首次调用时自动下载
def get_embeddings(model_name="BAAI/bge-small-zh-v1.5",device="cpu",**kwargs): def get_embeddings(model_name="BAAI/bge-small-zh-v1.5",device="cpu",**kwargs):
@@ -10,7 +11,6 @@ def get_embeddings(model_name="BAAI/bge-small-zh-v1.5",device="cpu",**kwargs):
if not local_dir.exists(): if not local_dir.exists():
print(f'⚠️ 首次使用嵌入模型，正在下载到{local_dir.absolute()}') print(f'⚠️ 首次使用嵌入模型，正在下载到{local_dir.absolute()}')
print("💡 提示：需要联网(必需梯子)，完成后可离线使用") print("💡 提示：需要联网(必需梯子)，完成后可离线使用")
from huggingface_hub import snapshot_download
# 模型下载工具 # 模型下载工具
snapshot_download( snapshot_download(
repo_id = model_name, repo_id = model_name,