Chroma 提供了一个便捷的包装器,用于 OpenAI 的嵌入 API。此嵌入函数在 OpenAI 的服务器上远程运行,并需要一个 API 密钥。您可以通过在 OpenAI 注册账户来获取 API 密钥。
以下 OpenAI 嵌入模型受支持:
text-embedding-ada-002
text-embedding-3-small
text-embedding-3-large
更多信息
访问 OpenAI 嵌入 文档 获取更多信息。
此嵌入函数依赖于 openai
Python 包,您可以使用 pip install openai
安装它。
您可以传入一个可选的 model_name
参数,该参数允许您选择使用哪个 OpenAI 嵌入模型。默认情况下,Chroma 使用 text-embedding-ada-002
。
要在其他平台(如 Azure)上使用 OpenAI 嵌入模型,您可以使用 api_base
和 api_type
参数: