作者
Writer 是一个用于生成不同语言内容的平台。
此示例介绍了如何使用 LangChain 与 Writer 模型 交互。
您需要获取 WRITER_API_KEY 此处。
from getpass import getpass
WRITER_API_KEY = getpass()
········
import os
os.environ["WRITER_API_KEY"] = WRITER_API_KEY
from langchain.chains import LLMChain
from langchain_community.llms import Writer
from langchain_core.prompts import PromptTemplate
template = """Question: {question}
Answer: Let's think step by step."""
prompt = PromptTemplate.from_template(template)
# If you get an error, probably, you need to set up the "base_url" parameter that can be taken from the error log.
llm = Writer()
llm_chain = LLMChain(prompt=prompt, llm=llm)
question = "What NFL team won the Super Bowl in the year Justin Beiber was born?"
llm_chain.run(question)