OpenAI 笔记:获取embedding

1 输入openai的api key

from openai import OpenAI

client = OpenAI(
    api_key=**)

2 举例

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

3 可选参数

input

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

model

  • 类型: 字符串
  • 必需: 是
  • 描述: 要使用的模型的 ID。

encoding_format

  • 类型: 字符串
  • 可选
  • 默认值: float
  • 描述: 返回嵌入的格式。可以是 float 或 base64。

dimensions

  • 类型: 整数
  • 可选
  • 描述: 结果输出嵌入应具有的维数。仅在 text-embedding-3 及以后的模型中支持。

看起来也不是直接截断

上一篇:Docker资源管理-数据管理


下一篇:怎么将音频合并到一起?安利音频合并的3个方法