Skip to main content
Open In ColabOpen on GitHub

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 LLMs进行交互

from langchain_community.llms import Writer as WriterLLM

llm = WriterLLM(
temperature=0.7,
max_tokens=1000,
# other params...
)
API Reference: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参考


这个页面有帮助吗?