From dec2c65b6bb864a9511fda2802ff09e0754a720e Mon Sep 17 00:00:00 2001 From: martsforever Date: Sat, 6 Sep 2025 23:08:36 +0800 Subject: [PATCH] feat: get_embedding_dimension --- app/utils/get_embedding_dimension.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) create mode 100644 app/utils/get_embedding_dimension.py diff --git a/app/utils/get_embedding_dimension.py b/app/utils/get_embedding_dimension.py new file mode 100644 index 0000000..0328977 --- /dev/null +++ b/app/utils/get_embedding_dimension.py @@ -0,0 +1,12 @@ + +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}")