回答

参考这个

#!/usr/local/python3/bin/python3
# -*- coding: utf-8 -*-
import os
import numpy as np
from openai import OpenAI

# 初始化 DashScope 兼容 OpenAI API 的客户端
client = OpenAI(
    api_key=os.getenv("DASHSCOPE_API_KEY"),  # 请确保已设置环境变量，或替换为你的实际 API Key
    base_url="https://dashscope.aliyuncs.com/compatible-mode/v1"
)

# 示例文档（可以替换成你自己的数据）
documents = [
    "OpenAI is a company that develops artificial intelligence models.",
    "Python is a programming language used for AI and data science.",
    "The capital of France is Paris.",
    "Neural networks are a fundamental concept in deep learning."
]

# 获取文本 embedding 的函数（使用 Qwen text-embedding-v4）
def get_embedding(text: str):
    response = client.embeddings.create(
        model="text-embedding-v4",
        input=text,
        dimensions=1024,           # Qwen v4 支持指定维度（1024 是推荐值）
        encoding_format="float"
    )
    return response.data[0].embedding

# 为所有文档生成 embedding 向量
print("