OpenAI Embeddings（OpenAI嵌入）¶
如果您在colab上打开这个笔记本，您可能需要安装LlamaIndex 🦙。
In [ ]:
%pip install llama-index-embeddings-openai
In [ ]:
!pip install llama-index
In [ ]:
import os
os.environ["OPENAI_API_KEY"] = "sk-..."
In [ ]:
from llama_index.embeddings.openai import OpenAIEmbedding
from llama_index.core import Settings
embed_model = OpenAIEmbedding(embed_batch_size=10)
Settings.embed_model = embed_model
使用OpenAI的text-embedding-3-large和text-embedding-3-small¶
注意，你可能需要更新你的OpenAI客户端：pip install -U openai
In [ ]:
# 获取API密钥并创建嵌入
from llama_index.embeddings.openai import OpenAIEmbedding
embed_model = OpenAIEmbedding(model="text-embedding-3-large")
embeddings = embed_model.get_text_embedding(
"Open AI new Embeddings models is great."
)
In [ ]:
print(embeddings[:5])
[-0.011500772088766098, 0.02457442320883274, -0.01760469563305378, -0.017763426527380943, 0.029841400682926178]
In [ ]:
print(len(embeddings))
3072
In [ ]:
# 获取API密钥并创建嵌入
from llama_index.embeddings.openai import OpenAIEmbedding
embed_model = OpenAIEmbedding(
model="text-embedding-3-small",
)
embeddings = embed_model.get_text_embedding(
"Open AI new Embeddings models is awesome."
)
In [ ]:
print(len(embeddings))
1536
更改输出嵌入的维度¶
注意：确保你安装了最新的OpenAI客户端。
In [ ]:
# 获取API密钥并创建嵌入
from llama_index.embeddings.openai import OpenAIEmbedding
embed_model = OpenAIEmbedding(
model="text-embedding-3-large",
dimensions=512,
)
embeddings = embed_model.get_text_embedding(
"Open AI new Embeddings models with different dimensions is awesome."
)
print(len(embeddings))
512