对 PostgreSQL 运行支持查询

此 Helm 仓库包含一些查询，用于产出 LangSmith UI 目前不直接支持的结果（例如在单次查询中获取多个组织的追踪计数）。

该命令接收一个包含内嵌用户名和密码的 PostgreSQL 连接字符串（可以从 secrets manager 调用中传入），并从输入文件执行查询。以下示例使用 support_queries/postgres 目录中的 pg_get_historic_trace_counts_daily.sql 输入文件。

前置条件

请确保准备好以下工具/项目。

1. kubectl

   • https://kubernetes.io/docs/tasks/tools/

2. PostgreSQL 客户端

   • https://www.postgresql.org/download/

3. PostgreSQL 数据库连接：

   • 主机
   • 端口
   • 用户名
     • 如果使用捆绑版本，则为 postgres
   • 密码
     • 如果使用捆绑版本，则为 postgres
   • 数据库名称
     • 如果使用捆绑版本，则为 postgres

4. 从你将运行迁移脚本的机器到 PostgreSQL 数据库的连通性。

   • 如果使用捆绑版本，你可能需要将 postgresql 服务端口转发到本地机器。
   • 运行 kubectl port-forward svc/langsmith-postgres 5432:5432 将 postgresql 服务端口转发到本地机器。

5. 运行支持查询的脚本

   • 下载 PostgreSQL 支持查询脚本

运行查询脚本

运行以下命令以执行所需的查询：

sh run_support_query_pg.sh <postgres_url> --input path/to/query.sql

例如，如果你使用捆绑版本并进行端口转发，命令可能如下：

sh run_support_query_pg.sh "postgres://postgres:postgres@localhost:5432/postgres" --input support_queries/postgres/pg_get_historic_trace_counts_daily.sql

它会按工作区 ID 和组织 ID 输出每日追踪计数。要将其导出到文件，请添加 --output path/to/file.csv 标志

导出用量数据

所有导出方式都会产生相同的数据：LangSmith 追踪计数、LangSmith Deployments 节点用量，以及所有工作区和组织的 Fleet run 计数。

NOTE

UI 和 API 导出都要求满足以下两项：

• organization:manage 权限。
• 调用者的邮箱必须列在 USAGE_EXPORT_ADMIN_EMAILS 中，或者 ORG_ADMINS_INSTALLATION_USAGE_EXPORT_ENABLED 必须设为 true。

USAGE_EXPORT_ADMIN_EMAILS='["admin@example.com", "admin2@example.com"]'
ORG_ADMINS_INSTALLATION_USAGE_EXPORT_ENABLED=true

从 UI 导出（推荐）

1. 导航到 Settings > Usage and billing > Usage export。
2. 点击 Export usage data。
3. 包含所有用量数据的 ZIP 文件将开始下载。

通过 API 导出

如果你希望以编程方式导出用量数据，可以直接调用导出 API 端点。

curl -OJ \
  -H "X-API-Key: <your_api_key>" \
  https://<langsmith_url>/api/v1/orgs/current/usage/backfill-export

通过 SQL 脚本导出

你也可以直接对 PostgreSQL 数据库运行 SQL 脚本以导出用量数据。这需要数据库访问凭据——不适用任何应用层权限。

要导出追踪用量（需要 Helm chart 0.11.4 或更高版本）：

sh run_support_query_pg.sh <postgres_url> \
  --input support_queries/postgres/pg_usage_traces_full_export.sql \
  --output ls_export.csv

要导出节点用量（需要 Helm chart 0.11.4 或更高版本）：

sh run_support_query_pg.sh <postgres_url> \
  --input support_queries/postgres/pg_usage_nodes_full_export.sql \
  --output lgp_export.csv

要导出 Fleet run 计数（需要 Helm chart 0.13.25 或更高版本）：

sh run_support_query_pg.sh <postgres_url> \
  --input support_queries/postgres/pg_usage_agent_builder_full_export.sql \
  --output ab_export.csv

要导出用量快照（每日实体计数，例如工作区、项目、数据集、提示词和活跃用户）：

sh run_support_query_pg.sh <postgres_url> \
  --input support_queries/postgres/pg_usage_snapshots_full_export.sql \
  --output usage_snapshots_export.csv

---