node.js - OpenAI 嵌入 API : How to change the embedding output dimension?

Question

在官方 OpenAI Node 库中创建嵌入，例如使用模型 text-embedding-ada-002返回的嵌入是一个大约 1536 的数组.

import {Configuration, OpenAIApi} from 'openai'

openai = new OpenAIApi(this.configuration)
const parameters= {
      model: 'text-embedding-ada-002',
      input: text,
    }
// Make the embedding request and return the result
const resp = await openai.createEmbedding(parameters)
const embeddings = embedding?.data.data[0].embedding

我希望能够限制返回的嵌入列表的长度。

Answer 1

这不是 OpenAI Embeddings API 端点的工作方式。

您无法更改嵌入输出维度。如果您使用 text-embedding-ada-002 模型，您将始终获得 1536 维嵌入向量(即里面有 1536 个数字)。

它是预定义的，如官方OpenAI documentation所述: