AI21

AI21 Studio 提供对 Jurassic-2 大型语言模型的 API 访问。

本示例介绍如何使用 LangChain 与 AI21 模型 进行交互。

# 安装包：
pip install ai21

# 获取 AI21_API_KEY。使用 https://studio.ai21.com/account/account

from getpass import getpass

AI21_API_KEY = getpass()

     ········

from langchain.llms import AI21
from langchain import PromptTemplate, LLMChain

template = """问题：{question}

回答：让我们逐步思考。"""

prompt = PromptTemplate(template=template, input_variables=["question"])

llm = AI21(ai21_api_key=AI21_API_KEY)

llm_chain = LLMChain(prompt=prompt, llm=llm)

question = "贾斯汀·比伯出生的那一年，哪支 NFL 球队赢得了超级碗？"

llm_chain.run(question)

    '\n1. 贾斯汀·比伯是在哪一年出生的？\n贾斯汀·比伯出生于1994年。\n2. 1994年超级碗的冠军是哪支球队？\n达拉斯牛仔队赢得了1994年的超级碗。'