wuzhiqiang/ai-admin-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 使用simple node parser

Browse Source
This commit is contained in:
martsforever
2025-09-07 15:14:02 +08:00
parent 178d0ddf24
commit e976ccf814
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/utils/get_embedding_dimension.py
+2 -2
View File
@@ -4,9 +4,9 @@ from app.utils.llm_utils import create_embeddings
# 通用测试方法，获取模型的向量维度 # 通用测试方法，获取模型的向量维度
def get_embedding_dimension(model, test_text="test"): def get_embedding_dimension(model, test_text="test"):
embedding = model.embed_query(test_text) embedding = model.embed_query(test_text)
print("embedding", embedding) # print("embedding", embedding)
return len(embedding) if hasattr(embedding, '__len__') else embedding.shape[-1] 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}") print(f"模型向量维度: {dimension}")