Chroma-Embedding

Chroma-Embedding

使用：

from chromadb.utils import embedding_functions

嵌入方法

默认嵌入：all-MiniLM-L6-v2

默认情况下，Chroma使用all-MiniLM-L6-v2模型进行嵌入

Chroma会下载模型文件，然后完成嵌入：

default_ef = embedding_functions.DefaultEmbeddingFunction()

使用default_ef函数实现embedding：

default_ef("你好")

openai嵌入

当你提供api密钥时，Chroma可以非常方便的使用openai相关模型进行嵌入：

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

当你使用Azure平台的openai模型时，通过下面的代码来实现：

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

使用openai_ef函数实现embedding：

openai_ef("你好")

Google PaLM嵌入

palm_embedding = embedding_functions.GooglePalmEmbeddingFunction(
    api_key=api_key)
palm_embedding("你好")