feat: 使用simple node parser

2025-09-07 15:14:02 +08:00
parent 178d0ddf24
commit e976ccf814
1 changed files with 2 additions and 2 deletions
@@ -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}")