Together AI 嵌入

本笔记本展示了如何使用 Together AI 进行嵌入。Together AI 提供了许多最先进的嵌入模型的访问权限。

Visit https://together.ai and sign up to get an API key.

设置

如果您在 Colab 上打开这个笔记本，您可能需要安装 LlamaIndex 🦙。

%pip install llama-index-embeddings-together

!pip install llama-index

# You can set the API key in the embeddings or env
# import os
# os.environ["TOEGETHER_API_KEY"] = "your-api-key"

from llama_index.embeddings.together import TogetherEmbedding

embed_model = TogetherEmbedding(
    model_name="togethercomputer/m2-bert-80M-8k-retrieval", api_key="..."
)

获取嵌入向量

embeddings = embed_model.get_text_embedding("hello world")

print(len(embeddings))

print(embeddings[:5])

[-0.11657876, -0.012690996, 0.24342081, 0.32781482, 0.022501636]