获取向量模型模型维度的脚本

#!/bin/bash

# ================== 请修改以下配置 ==================
API_URL="http://api.example.com/v1/embeddings"
API_KEY="sk-5566"
MODEL_NAME="bge-m3"
# ==================================================

echo "正在向 $API_URL 发送请求以查询模型 '$MODEL_NAME' 的维度..."

# 发送请求并获取响应
response=$(curl -s -X POST "$API_URL" \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer $API_KEY" \
  -d "{
    \"model\": \"$MODEL_NAME\",
    \"input\": [\"test_dimension\"]
  }")

# 使用 Python 解析并计算 embedding 数组的长度
dimension=$(echo "$response" | python3 -c "import sys, json; print(len(json.load(sys.stdin)['data'][0]['embedding']))" 2>/dev/null)

# 检查是否成功获取到维度
if [[ "$dimension" =~ ^[0-9]+$ ]]; then
  echo "✅ 查询成功！"
  echo "模型 '$MODEL_NAME' 的向量维度是: $dimension"
else
  echo "❌ 查询失败。请检查 API 地址、密钥或模型名称是否正确。"
  echo "服务器返回的原始信息："
  echo "$response"
fi