OpenAI 笔记：获取embedding

类型: 字符串或数组
必需: 是
描述: 要嵌入的输入文本&#xff0c;可以编码为一个字符串或令牌数组。要在单个请求中嵌入多个输入&#xff0c;请传递字符串数组或令牌数组。输入不能超过模型的最大输入令牌数&#xff08;例如 text-embedding-ada-002 的 8192 令牌&#xff09;&#xff0c;不能是空字符串&#xff0c;任何数组必须是 2048 维或更少。

1 输入openai的api key

from openai import OpenAI

client = OpenAI(
    api_key=**)

response = client.embeddings.create(
    input="Hello",
    model="text-embedding-3-small"
)

print(response.data[0].embedding)

默认情况下，text-embedding-3-small 的嵌入向量长度为 1536，text-embedding-3-large 的长度为 3072

len(response.data[0].embedding)
#1536

input	类型: 字符串或数组必需: 是描述: 要嵌入的输入文本，可以编码为一个字符串或令牌数组。要在单个请求中嵌入多个输入，请传递字符串数组或令牌数组。输入不能超过模型的最大输入令牌数（例如 text-embedding-ada-002 的 8192 令牌），不能是空字符串，任何数组必须是 2048 维或更少。
model	类型: 字符串必需: 是描述: 要使用的模型的 ID。
encoding_format	类型: 字符串可选默认值: float 描述: 返回嵌入的格式。可以是 float 或 base64。
dimensions	类型: 整数可选描述: 结果输出嵌入应具有的维数。仅在 text-embedding-3 及以后的模型中支持。看起来也不是直接截断