将数据沿袭与 MCP、Gemini 和其他代理搭配使用

本页介绍了如何将数据沿袭连接到开发者工具,例如 Gemini CLI 和其他 Model Context Protocol (MCP) 客户端。 将数据沿袭连接到这些工具后,您就可以直接在开发环境中进行 AI 驱动的沿袭跟踪和数据出处分析。

您可以使用本地 MCP Toolbox for Databases 连接支持 MCP 的 IDE 和开发者工具。然后,您可以在现有 IDE 中使用 AI 代理来查询数据沿袭图,发现上游数据出处,并分析下游对资产的影响。

如需详细了解 MCP,请参阅 Model Context Protocol 简介

本指南演示了适用于以下工具的连接过程:

数据沿袭提供了哪些 MCP 工具?

通过数据沿袭集成,AI 代理可以查询和分析数据沿袭,从而了解源(上游)资产和目标(下游)资产之间的数据流。它同时支持实体级沿袭(跟踪整个资产(例如表格和文件)之间的数据传输)和列级沿袭(跟踪资产内特定字段或列之间的数据传输)。

数据沿袭提供了 datalineage-search-lineage 工具,该工具可检索与所请求资产关联的沿袭链接的流式响应。

如需详细了解数据沿袭来源及其 可用工具,请参阅 数据沿袭来源文档

所需的角色

如需获得使用 MCP Toolbox 连接到数据沿袭所需的权限,请让您的管理员向您授予项目的以下 IAM 角色:

如需详细了解如何授予角色,请参阅管理对项目、文件夹和组织的访问权限

这些预定义角色包含 使用 MCP Toolbox 连接到数据沿袭所需的权限。如需查看所需的确切权限,请展开所需权限部分:

所需权限

如需使用 MCP Toolbox 连接到数据沿袭,需要以下权限:

  • 启用 API: serviceusage.services.enable
  • 使用数据沿袭技能:
    • datalineage.lineage.searchLinks
    • datalineage.processes.get
    • datalineage.runs.get

您也可以使用自定义角色或其他预定义角色来获取这些权限。

启用所需的 API

  1. 在 Google Cloud 控制台中,前往项目选择器页面。

    转到“项目选择器”

  2. 选择或创建 Google Cloud 项目。

    选择或创建项目所需角色

    • 选择项目:选择项目不需要特定的 IAM 角色,您可以选择已获授予角色的任何项目。
    • 创建项目:如需创建项目,您需要 Project Creator 角色 (roles/resourcemanager.projectCreator),该角色包含 resourcemanager.projects.create 权限。了解如何授予 角色
  3. 验证是否已为您的 Google Cloud 项目启用结算功能。

  4. 启用 Data Lineage API。

    启用 API 所需的角色

    如需启用 API,您需要拥有 serviceusage.services.enable 权限。如果您已创建项目,则可能已通过 Owner 角色 (roles/owner) 拥有此权限。否则,您可以通过 Service Usage Admin 角色 (roles/serviceusage.serviceUsageAdmin) 获得此权限。了解如何授予角色

    启用 API

  5. 如果您使用的是本地 shell,请为您的用户 账号创建本地身份验证凭证:

    gcloud auth application-default login

    如果您使用的是 Cloud Shell,则无需执行此操作。

    如果返回了身份验证错误,并且您使用的是外部身份提供方 (IdP),请确认您已 使用联合身份登录 gcloud CLI

安装 MCP Toolbox

如果您只打算使用 Gemini Code Assist, 则无需安装 MCP Toolbox,因为它捆绑了所需的服务器功能。对于其他 IDE 和工具,请按照本部分中的步骤安装 MCP Toolbox。

  1. 以二进制文件形式下载最新版本的 MCP Toolbox。选择与您的操作系统 (OS) 和 CPU 架构对应的 MCP Toolbox 二进制文件版本。您必须使用 MCP Toolbox v0.31.0 或更高版本。

    Linux/amd64

    curl -O https://storage.googleapis.com/mcp-toolbox-for-databases/VERSION/linux/amd64/toolbox

    VERSION 替换为 MCP Toolbox 版本,例如 v0.31.0

    macOS (Darwin)/arm64

    curl -O https://storage.googleapis.com/mcp-toolbox-for-databases/VERSION/darwin/arm64/toolbox

    VERSION 替换为 MCP Toolbox 版本,例如 v0.31.0

    macOS (Darwin)/amd64

    curl -O https://storage.googleapis.com/mcp-toolbox-for-databases/VERSION/darwin/amd64/toolbox

    VERSION 替换为 MCP Toolbox 版本,例如 v0.31.0

    Windows/amd64

    curl -O https://storage.googleapis.com/mcp-toolbox-for-databases/VERSION/windows/amd64/toolbox

    VERSION 替换为 MCP Toolbox 版本,例如 v0.31.0

  2. 将该二进制文件设为可执行文件:

    chmod +x toolbox
    
  3. 验证安装:

    ./toolbox --version
    

    成功安装后,系统会返回版本号,例如 0.15.0

为数据沿袭设置客户端和连接

本部分介绍了如何将数据沿袭连接到您的工具。

如需将与 MCP 兼容的 IDE 和工具连接到数据沿袭,您 必须先 安装 MCP Toolbox,并为沿袭来源和工具创建自定义配置文件 。

  1. 在项目根目录或配置目录中,使用以下配置创建一个名为 lineage-config.yaml 的 YAML 文件:

    kind: source
    name: lineage-source
    type: datalineage
    project: ${DATALINEAGE_PROJECT}
    ---
    kind: tool
    name: search_lineage
    type: datalineage-search-lineage
    source: lineage-source
    description: Retrieves a streaming response of lineage links connected to requested assets.
    
  2. 设置项目的环境变量: Google Cloud

    export DATALINEAGE_PROJECT=PROJECT_ID
    

    PROJECT_ID 替换为 Google Cloud 项目 ID。

  3. 使用 --config 标志(而不是预构建的配置)配置特定客户端,如以下部分所示。

Gemini CLI

您可以在 Gemini CLI 中使用数据沿袭,方法是使用 MCP Toolbox 和自定义 lineage-config.yaml 文件将其配置为本地 MCP 服务器。

  1. 在项目的工作目录中,创建一个名为 .gemini 的文件夹(或打开全局 ~/.gemini 目录)。
  2. 在该目录中,创建或打开 settings.json 文件。
  3. 添加以下配置:

    {
      "mcpServers": {
        "dataLineage": {
          "command": "./PATH/TO/toolbox",
          "args": ["--config","/PATH/TO/lineage-config.yaml","--stdio"],
          "env": {
            "DATALINEAGE_PROJECT": "PROJECT_ID"
          }
        }
      }
    }
    

    PROJECT_ID 替换为 Google Cloud 项目 ID。

  4. 保存配置。

  5. 以互动模式启动 Gemini CLI:

    gemini
    

    在 Gemini CLI 中,使用 /mcp 命令验证 dataLineage 服务器是否已连接。

Gemini Code Assist

Gemini Code Assist 捆绑了所需的 MCP 服务器功能,因此您无需单独安装 MCP Toolbox。

  1. 在 VS Code 中,安装 Gemini Code Assist 扩展程序。
  2. 在 Gemini Code Assist 对话中启用智能体模式。
  3. 在工作目录中,创建一个名为 .gemini 的文件夹。在该文件夹中,创建一个 settings.json 文件。
  4. 添加以下配置:

    {
      "mcpServers": {
        "dataLineage": {
          "command": "./PATH/TO/toolbox",
          "args": ["--config","/PATH/TO/lineage-config.yaml","--stdio"],
          "env": {
            "DATALINEAGE_PROJECT": "PROJECT_ID"
          }
        }
      }
    }
    

    PROJECT_ID 替换为 Google Cloud 项目 ID。

  5. 保存配置。

Claude Code

虽然官方插件提供了 Knowledge Catalog 的工具,但您可以在 Claude Code 中使用数据沿袭,方法是使用自定义配置文件配置本地 MCP Toolbox 服务器。

  1. 设置环境变量以连接到数据沿袭项目:

    export DATALINEAGE_PROJECT=PROJECT_ID
    

    PROJECT_ID 替换为 Google Cloud 项目 ID。

  2. 将 Claude Code 配置为使用 MCP Toolbox 服务器:

    claude mcp add datalineage -- /PATH/TO/toolbox --config=/PATH/TO/lineage-config.yaml --stdio
    
  3. 启动代理:

    claude
    

Codex

如需在 Codex 中使用数据沿袭,请在 Codex 配置中配置 MCP 服务器连接,以使用自定义 lineage-config.yaml 文件运行 MCP Toolbox:

  1. 设置环境变量以连接到数据沿袭项目:

    export DATALINEAGE_PROJECT="PROJECT_ID"
    

    PROJECT_ID 替换为 Google Cloud 项目 ID。

  2. 在 Codex MCP 配置中,使用 MCP Toolbox 添加服务器:

    {
      "mcpServers": {
        "dataLineage": {
          "command": "./PATH/TO/toolbox",
          "args": ["--config","/PATH/TO/lineage-config.yaml","--stdio"],
          "env": {
            "DATALINEAGE_PROJECT": "PROJECT_ID"
          }
        }
      }
    }
    

    PROJECT_ID 替换为 Google Cloud 项目 ID。

Claude Desktop

  1. 打开 Claude Desktop,然后前往 设置
  2. 如需打开配置文件,请在开发者 标签页中点击修改配置
  3. 添加以下配置:

    {
      "mcpServers": {
        "dataLineage": {
          "command": "./PATH/TO/toolbox",
          "args": ["--config","/PATH/TO/lineage-config.yaml","--stdio"],
          "env": {
            "DATALINEAGE_PROJECT": "PROJECT_ID"
          }
        }
      }
    }
    

    PROJECT_ID 替换为 Google Cloud 项目 ID。

  4. 保存配置。

  5. 重启 Claude Desktop。新聊天界面会显示 MCP 图标,表示新的 MCP 服务器。

Cline

  1. 在 VS Code 中,打开 Cline 扩展程序,然后点击 MCP 服务器 图标。
  2. 如需打开配置文件,请点按配置 MCP 服务器
  3. 添加以下配置:

    {
      "mcpServers": {
        "dataLineage": {
          "command": "./PATH/TO/toolbox",
          "args": ["--config","/PATH/TO/lineage-config.yaml","--stdio"],
          "env": {
            "DATALINEAGE_PROJECT": "PROJECT_ID"
          }
        }
      }
    }
    

    PROJECT_ID 替换为 Google Cloud 项目 ID。

  4. 保存配置。服务器成功连接后,系统会显示绿色的活跃状态。

Cursor

  1. 在项目根目录中创建 .cursor 目录(如果尚不存在)。
  2. 创建 .cursor/mcp.json 文件(如果尚不存在),然后打开该文件。
  3. 添加以下配置:

    {
      "mcpServers": {
        "dataLineage": {
          "command": "./PATH/TO/toolbox",
          "args": ["--config","/PATH/TO/lineage-config.yaml","--stdio"],
          "env": {
            "DATALINEAGE_PROJECT": "PROJECT_ID"
          }
        }
      }
    }
    

    PROJECT_ID 替换为 Google Cloud 项目 ID。

  4. 保存配置。

  5. 打开 Cursor,然后依次前往 设置 > Cursor 设置 > MCP。服务器连接时,系统会显示绿色的活跃状态。

VS Code (Copilot)

  1. 打开 VS Code ,并在项目根目录中创建 .vscode 目录(如果尚不 存在)。
  2. 创建 .vscode/mcp.json 文件(如果尚不存在),然后打开该文件。
  3. 添加以下配置:

    {
      "servers": {
        "dataLineage": {
          "command": "./PATH/TO/toolbox",
          "args": ["--config","/PATH/TO/lineage-config.yaml","--stdio"],
          "env": {
            "DATALINEAGE_PROJECT": "PROJECT_ID"
          }
        }
      }
    }
    

    PROJECT_ID 替换为 Google Cloud 项目 ID。

  4. 保存配置。

Windsurf

  1. 打开 Windsurf 并前往 Cascade 助理。
  2. 如需打开配置文件,请点击 MCP 图标,然后点击配置
  3. 添加以下配置:

    {
      "mcpServers": {
        "dataLineage": {
          "command": "./PATH/TO/toolbox",
          "args": ["--config","/PATH/TO/lineage-config.yaml","--stdio"],
          "env": {
            "DATALINEAGE_PROJECT": "PROJECT_ID"
          }
        }
      }
    }
    

    PROJECT_ID 替换为 Google Cloud 项目 ID。

  4. 保存配置。

使用技能

您的 AI 助理现已连接到数据沿袭。尝试让 AI 助理跟踪资产之间的上游和下游数据沿袭。

例如,您可以让 AI 助理执行以下操作:

  • 跟踪 BigQuery 表的数据来源(上游沿袭)。
  • 发现哪些下游表格或报告依赖于特定数据资产(下游沿袭)。
  • 检查资产之间特定字段的列级沿袭。

可选:添加系统指令

系统指令可用于为 LLM 提供特定准则,帮助其了解上下文并更准确地回答问题。根据数据沿袭推荐的系统提示设置系统 指令

例如,您可以添加指令来指导 LLM 如何使用数据沿袭技能:

  • 当被要求跟踪资产或列之间的上游或下游数据传输时,请使用 search_lineage 技能或 datalineage-search-lineage 工具。

如需详细了解如何配置指令,请参阅 使用指令获取符合您编码风格的 AI 编辑内容

后续步骤