4.1 Chroma-OpenAI-python

Chroma为OpenAI的嵌入API提供了一个便捷的封装。这个嵌入函数在OpenAI的服务器上运行，并且需要一个API密钥。您可以通过在 OpenAI注册账户来获得API密钥。

这个嵌入函数依赖于openai python包，您可以使用pip install openai来安装它。

openai_ef = embedding_functions.OpenAIEmbeddingFunction(
                api_key="YOUR_API_KEY",
                model_name="text-embedding-ada-002"
            )

要在其他平台上（例如Azure）使用OpenAI嵌入模型，您可以使用api_base和api_type参数：

openai_ef = embedding_functions.OpenAIEmbeddingFunction(
                api_key="YOUR_API_KEY",
                api_base="YOUR_API_BASE_PATH",
                api_type="azure",
                api_version="YOUR_API_VERSION",
                model_name="text-embedding-ada-002"
            )

您可以传递一个可选的model_name参数，它允许您选择要使用的OpenAI嵌入模型。默认情况下，Chroma使用text-embedding-ada-002模型。您可以在此处查看所有可用模型的列表。