本指南展示如何在 LangSmith 中追踪和记录 Google 的 Gemini 模型。您将使用最新的 google-genai SDK (Python) 或 @google/genai SDK (JavaScript) 来为 Gemini 调用添加插桩,包装 Gemini 客户端以进行追踪,并尝试包含基本提示、元数据标记和多轮对话在内的示例。
LangSmith 的 Gemini 包装器目前处于 测试版。API 可能会在未来版本中发生变化。

安装

使用您偏好的包管理器安装所需的包:

设置

设置您的 API 密钥 和项目名称:
要创建 Google API 密钥,请参考 Google AI Studio。

配置追踪

要追踪 Gemini API 调用,请使用 LangSmith 的 wrap_gemini (Python) 或 wrapGemini (JavaScript) 包装器函数。此包装器拦截对 Gemini 客户端的调用,并自动将它们作为追踪记录在 LangSmith 中。包装器在保留原始客户端所有功能的同时,增加了可观测性:
您可以通过在调用 wrap_gemini() 时传递 tracing_extra 来自定义追踪。此参数适用于您使用该包装客户端发出的所有后续请求,允许您附加标签和元数据,以便在 LangSmith UI 中筛选和组织追踪记录。tracing_extra 参数接受:
  • tags:一个用于分类追踪的字符串列表(例如,["production", "gemini"])。
  • metadata:一个包含附加上下文的键值对字典(例如,{"team": "ml-research", "integration": "google-genai"})。
  • client:一个可选的自定义 LangSmith 客户端实例。
这些设置一致地应用于包装客户端的所有追踪记录,因此您可以包含在整个应用程序中应保持不变的环境级标签或团队元数据。

在 LangSmith 中查看追踪

运行应用程序后,您可以在 LangSmith UI 中查看追踪记录,其中包括:
  • 模型请求:发送给 Gemini 模型的完整提示
  • 模型响应:生成的文本和结构化输出
  • 函数调用:使用函数调用时的工具调用及其结果
  • 聊天会话:多轮对话的上下文
  • 性能指标:延迟和 token 使用信息