我正在尝试运行 openai 网站上的示例代码来获取数据集的嵌入:https://platform.openai.com/docs/guides/embeddings/use-cases。但是,代码返回一个错误,我无法查看历史帖子来解决该错误
我尝试运行此代码,其中 df 是我用自己的数据创建的数据框,已成功加载。
from openai import OpenAI
client = OpenAI()
def get_embedding(text, model="text-embedding-ada-002"):
text = text.replace("\n", " ")
return client.embeddings.create(input = [text], model=model)['data'][0]['embedding']
df['embedding'] = df.ITEM_DESCRIPTION.apply(lambda x: get_embedding(x, model='text-embedding-ada-002'))
df.to_csv('embedded_output.csv', index=False)
Run Code Online (Sandbox Code Playgroud)