OpenAI

Chroma 提供了一个便捷的包装器,用于 OpenAI 的嵌入 API。此嵌入函数在 OpenAI 的服务器上远程运行,并需要一个 API 密钥。您可以通过在 OpenAI 注册账户来获取 API 密钥。

以下 OpenAI 嵌入模型受支持:

  • text-embedding-ada-002
  • text-embedding-3-small
  • text-embedding-3-large

此嵌入函数依赖于 openai Python 包,您可以使用 pip install openai 安装它。

您可以传入一个可选的 model_name 参数,该参数允许您选择使用哪个 OpenAI 嵌入模型。默认情况下,Chroma 使用 text-embedding-ada-002

python

要在其他平台(如 Azure)上使用 OpenAI 嵌入模型,您可以使用 api_baseapi_type 参数:

python