xAI 提供 API 来与 Grok 模型交互。以下示例介绍如何使用 LangChain 与 xAI 模型进行交互。

安装

pip install -U langchain-xai

环境设置

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

示例

请参阅 ChatXAI 文档 了解更多详情和支持的功能。

# Querying chat models with xAI

from langchain_xai import ChatXAI

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

# stream the response back from the model
stream = chat.stream_events("Tell me fun things to do in NYC", version="v3")
for token in stream.text:
    print(token, 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")