From e976ccf814b18ada2d3631c1a808aba4bc4f54f8 Mon Sep 17 00:00:00 2001 From: martsforever Date: Sun, 7 Sep 2025 15:14:02 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BD=BF=E7=94=A8simple=20node=20parse?= =?UTF-8?q?r?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/utils/get_embedding_dimension.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/app/utils/get_embedding_dimension.py b/app/utils/get_embedding_dimension.py index 0328977..a936ca3 100644 --- a/app/utils/get_embedding_dimension.py +++ b/app/utils/get_embedding_dimension.py @@ -4,9 +4,9 @@ from app.utils.llm_utils import create_embeddings # 通用测试方法，获取模型的向量维度 def get_embedding_dimension(model, test_text="test"): embedding = model.embed_query(test_text) - print("embedding", embedding) + # print("embedding", embedding) return len(embedding) if hasattr(embedding, '__len__') else embedding.shape[-1] -dimension = get_embedding_dimension(create_embeddings()) +dimension = get_embedding_dimension(create_embeddings("bailian-embedding")) print(f"模型向量维度: {dimension}")