This commit is contained in:
stardrophere
2026-03-12 01:50:08 +08:00
parent 966bcfbba4
commit e28b893a12
7 changed files with 123 additions and 14 deletions
+1 -1
View File
@@ -35,7 +35,7 @@ def generate_md5(text: str) -> str:
def generate_embedding_json(text: str) -> str:
"""辅助函数:调用大模型生成向量,并序列化为 JSON 字符串"""
raw_vec = embedder_model.encode([text], normalize_embeddings=True)[0]
raw_vec = embedder_model.encode([text], normalize_embeddings=True, show_progress_bar=False)[0]
truncated_vec = [round(float(x), 5) for x in raw_vec]
return json.dumps(truncated_vec, separators=(',', ':'))