xai-org/xai-sdk-python

The official Python SDK for the xAI API

解决的问题

提供一种标准化、开发者友好的方式,与 xAI 的 AI 模型套件进行交互。无需手动处理原始 gRPC 调用,即可简化将文本生成、图像理解、视频生成/编辑集成到 Python 应用程序中的过程。

工作原理

SDK 使用基于 gRPC 的架构与 xAI 的 API 进行通信。提供同步(Client)和异步(AsyncClient)两种接口,支持非阻塞操作。内置了多轮对话历史管理(通过 append 方法)、实时输出的流式响应,以及视频生成等长时间运行任务的自动轮询功能。

适用人群

希望利用 Grok 模型实现多模态能力、推理或代理工具调用的 Python 开发者。

主要亮点

  • 多模态能力:支持文本、图像理解、图像生成,以及全面的视频生成(文本到视频、图像到视频、视频编辑和视频扩展)。
  • 代理工具调用:使模型能够自主使用网络搜索、𝕏 或代码执行等工具。
  • 结构化输出:支持将模型响应以 Pydantic 模型形式返回,便于数据解析。
  • 可观测性:集成 OpenTelemetry 支持,可将追踪数据导出至控制台或 OTLP 后端(如 Langfuse),以监控 token 使用量和性能。
  • 开发者体验:内置指数退避重试机制、可配置超时、同时支持同步和异步工作流。

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • 项目