xAI

xAI 提供了一个 API 来与 Grok 模型互动。

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

安装

%pip install --upgrade langchain-xai

环境

要使用 xAI，您需要创建一个 API 密钥。API 密钥可以作为 init 参数 xai_api_key 传入，或设置为环境变量 XAI_API_KEY。

示例

# Querying chat models with xAI

from langchain_xai import ChatXAI

chat = ChatXAI(
    # xai_api_key="YOUR_API_KEY",
    model="grok-beta",
)

# stream the response back from the model
for m in chat.stream("Tell me fun things to do in NYC"):
    print(m.content, end="", flush=True)

# if you don't want to do streaming, you can use the invoke method
# chat.invoke("Tell me fun things to do in NYC")

API 参考:ChatXAI