跳到主要内容

Writer LLM

Writer 是一个生成不同语言内容的平台。

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

设置

要访问 Writer 模型,您需要创建一个 Writer 帐户,获取 API 密钥,并安装 writer-sdklangchain-community 包。

凭据

前往 Writer AI Studio 注册 OpenAI 并生成 API 密钥。完成后,设置 WRITER_API_KEY 环境变量

import getpass
import os

if not os.environ.get("WRITER_API_KEY"):
os.environ["WRITER_API_KEY"] = getpass.getpass("Enter your Writer API key:")

安装

LangChain Writer 集成位于 langchain-community 包中

%pip install -qU langchain-community writer-sdk

[notice] A new release of pip is available: 24.2 -> 24.3.1
[notice] To update, run: pip install --upgrade pip
Note: you may need to restart the kernel to use updated packages.

现在我们可以初始化我们的模型对象来与 writer LLM 进行交互

from langchain_community.llms import Writer as WriterLLM

llm = WriterLLM(
temperature=0.7,
max_tokens=1000,
# other params...
)
API 参考:Writer

调用

response_text = llm.invoke(input="Write a poem")
print(response_text)

流式传输

stream_response = llm.stream(input="Tell me a fairytale")
for chunk in stream_response:
print(chunk, end="")

异步

Writer 通过 ainvoke()astream() 方法支持异步调用

API 参考

有关所有 Writer 功能的详细文档,请访问我们的 API 参考


此页面是否对您有帮助?