参考这个
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)
# 网友回复
我知道答案，我要回答
私活外包
