回答

参考这个

from openai import OpenAI
import numpy as np

# 初始化客户端（确保你已设置 OPENAI_API_KEY 环境变量）
client = OpenAI(api_key="你的API密钥")

# 示例文档
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向量
def get_embedding(text):
    response = client.embeddings.create(
        model="text-embedding-3-small",  # 或 "text-embedding-3-large"
        input=text
    )
    return response.data[0].embedding

# 为文档生成向量
doc_embeddings = [get_embedding(doc) for doc in documents]

# 将向量存入数组
doc_embeddings = np.array(doc_embeddings)

#