我使用与此 Git Repo 中几乎相同的代码从 OpenAI 获取嵌入: https://gist.github.com/limcheekin/997de2ae0757cd46db796f162c3dd58c
oai = OpenAI(
# This is the default and can be omitted
api_key="sk-.....",
)
def get_embedding(text_to_embed, openai):
response = openai.embeddings.create(
model= "text-embedding-ada-002",
input=[text_to_embed]
)
return response
embedding_raw = get_embedding(text,oai)
根据 Git Repo,Vector 应该位于
response['data'][0]['embedding']
中。但这不是我的情况。
当我打印响应变量时,我得到了这个:
print(embedding_raw)
输出:
CreateEmbeddingResponse(data=[Embedding(embedding=[0.009792150929570198, -0.01779201813042164, 0.011846082285046577, -0.0036859565880149603, -0.0013213189085945487, 0.00037509595858864486,..... -0.0121011883020401, -0.015751168131828308], index=0, object='embedding')], model='text-embedding-ada-002', object='list', usage=Usage(prompt_tokens=360, total_tokens=360))
抱歉,我是Python新手,但是如何访问矢量数据?
为什么不简单地返回嵌入呢?
def get_embedding(text_to_embed, openai):
response = openai.embeddings.create(
model= "text-embedding-ada-002",
input=[text_to_embed]
)
return response.data[0].embedding
embedding_raw = get_embedding(text,oai)