MosaicML

MosaicML提供了一个托管的推理服务。您可以使用各种开源模型，或者部署自己的模型。

本示例介绍如何使用LangChain与MosaicML推理服务进行文本嵌入交互。

# 注册账号：https://forms.mosaicml.com/demo?utm_source=langchain

from getpass import getpass

MOSAICML_API_TOKEN = getpass()

import os

os.environ["MOSAICML_API_TOKEN"] = MOSAICML_API_TOKEN

from langchain.embeddings import MosaicMLInstructorEmbeddings

embeddings = MosaicMLInstructorEmbeddings(
    query_instruction="为检索表示查询："
)

query_text = "这是一个测试查询。"
query_result = embeddings.embed_query(query_text)

document_text = "这是一个测试文档。"
document_result = embeddings.embed_documents([document_text])

import numpy as np

query_numpy = np.array(query_result)
document_numpy = np.array(document_result[0])
similarity = np.dot(query_numpy, document_numpy) / (
    np.linalg.norm(query_numpy) * np.linalg.norm(document_numpy)
)
print(f"文档和查询之间的余弦相似度：{similarity}")