Skip to main content
Open In ColabOpen on GitHub

Clarifai

Clarifai is an AI Platform that provides the full AI lifecycle ranging from data exploration, data labeling, model training, evaluation, and inference.

本示例介绍了如何使用 LangChain 与 Clarifai 模型 进行交互。特别是文本嵌入模型,可以在此处找到。

要使用 Clarifai,您必须拥有账户和个人访问令牌 (PAT) 密钥。 点击此处获取或创建 PAT。

依赖项

# Install required dependencies
%pip install --upgrade --quiet clarifai

导入

在此处,我们将设置个人访问令牌。您可以在 Clarifai 账户的 设置/安全 下找到您的 PAT。

# Please login and get your API key from  https://clarifai.com/settings/security
from getpass import getpass

CLARIFAI_PAT = getpass()
# Import the required modules
from langchain.chains import LLMChain
from langchain_community.embeddings import ClarifaiEmbeddings
from langchain_core.prompts import PromptTemplate

输入

创建一个用于 LLM Chain 的提示模板:

template = """Question: {question}

Answer: Let's think step by step."""

prompt = PromptTemplate.from_template(template)

设置

设置用户ID和应用ID,以指定模型所在的应用程序。您可以在 https://clarifai.com/explore/models 上找到公开模型的列表。

你还需要初始化模型ID,如果需要的话,还有模型版本ID。某些模型有许多版本,你可以根据你的任务选择合适的版本。

USER_ID = "clarifai"
APP_ID = "main"
MODEL_ID = "BAAI-bge-base-en-v15"
MODEL_URL = "https://clarifai.com/clarifai/main/models/BAAI-bge-base-en-v15"

# Further you can also provide a specific model version as the model_version_id arg.
# MODEL_VERSION_ID = "MODEL_VERSION_ID"
# Initialize a Clarifai embedding model
embeddings = ClarifaiEmbeddings(user_id=USER_ID, app_id=APP_ID, model_id=MODEL_ID)

# Initialize a clarifai embedding model using model URL
embeddings = ClarifaiEmbeddings(model_url=MODEL_URL)

# Alternatively you can initialize clarifai class with pat argument.
text = "roses are red violets are blue."
text2 = "Make hay while the sun shines."

你可以使用 embed_query 函数嵌入单行文本!

query_result = embeddings.embed_query(text)

要进一步嵌入文本/文档列表,请使用embed_documents函数。

doc_result = embeddings.embed_documents([text, text2])