13 lines
415 B
Python
13 lines
415 B
Python
|
|
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)
|
|
return len(embedding) if hasattr(embedding, '__len__') else embedding.shape[-1]
|
|
|
|
|
|
dimension = get_embedding_dimension(create_embeddings())
|
|
print(f"模型向量维度: {dimension}")
|