Chroma为OpenAI的嵌入API提供了一个便捷的封装。这个嵌入函数在OpenAI的服务器上运行,并且需要一个API密钥。您可以通过在 OpenAI注册账户来获得API密钥。
const {OpenAIEmbeddingFunction} = require('chromadb');
const embedder = new OpenAIEmbeddingFunction({openai_api_key: "apiKey"})
// use directly
const embeddings = embedder.generate(["document1","document2"])
// pass documents to query for .add and .query
const collection = await client.createCollection({name: "name", embeddingFunction: embedder})
const collection = await client.getCollection({name: "name", embeddingFunction: embedder})
您可以传递一个可选的model_name参数,它允许您选择要使用的OpenAI嵌入模型。默认情况下,Chroma使用text-embedding-ada-002模型。您可以在此处查看所有可用模型的列表。