mirror of
https://chromium.googlesource.com/chromium/tools/depot_tools.git
synced 2026-01-11 18:51:29 +00:00
526aace177
Adds unittest and pylint checks for the mcp/ directory. Change-Id: I7aa6a33fea7ac60500b8c865e18f3cf72d425289 Reviewed-on: https://chromium-review.googlesource.com/c/chromium/tools/depot_tools/+/6847122 Reviewed-by: Struan Shrimpton <sshrimp@google.com> Auto-Submit: Brian Sheedy <bsheedy@chromium.org> Reviewed-by: Scott Lee <ddoman@chromium.org> Commit-Queue: Struan Shrimpton <sshrimp@google.com>
55 lines
1.7 KiB
Python
Executable File
55 lines
1.7 KiB
Python
Executable File
#!/bin/env vpython3
|
|
# Copyright 2025 The Chromium Authors
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
"""The MCP server that provides tools"""
|
|
from collections.abc import Sequence
|
|
import pathlib
|
|
import os
|
|
import sys
|
|
|
|
sys.path.insert(
|
|
0,
|
|
os.path.abspath(
|
|
pathlib.Path(__file__).resolve().parent.parent.joinpath(
|
|
pathlib.Path('infra_lib'))))
|
|
from absl import app
|
|
from mcp.server import fastmcp # pylint: disable=import-self
|
|
import telemetry
|
|
|
|
import buildbucket
|
|
import resultdb
|
|
import git_cl
|
|
|
|
mcp = fastmcp.FastMCP('chrome-infra-mcp')
|
|
|
|
|
|
def main(argv: Sequence[str]) -> None:
|
|
if len(argv) > 1:
|
|
raise app.UsageError('Too many command-line arguments.')
|
|
|
|
# Only initialize telemetry if the user is opted in. The MCP does not
|
|
# currently have the ability to show the banner so we need to rely on other
|
|
# tools to get consent
|
|
if telemetry.opted_in():
|
|
telemetry.initialize('chromium.mcp')
|
|
|
|
mcp.add_tool(buildbucket.get_build)
|
|
mcp.add_tool(buildbucket.get_build_from_build_number)
|
|
mcp.add_tool(buildbucket.get_build_from_id)
|
|
mcp.add_tool(buildbucket.get_build_status)
|
|
mcp.add_tool(buildbucket.get_recent_builds)
|
|
mcp.add_tool(buildbucket.get_recent_failed_builds)
|
|
mcp.add_tool(resultdb.expand_summary_html)
|
|
mcp.add_tool(resultdb.get_non_exonerated_unexpected_results_from_build)
|
|
mcp.add_tool(resultdb.get_test_level_text_artifact)
|
|
mcp.add_tool(git_cl.try_builder_results)
|
|
mcp.add_tool(git_cl.get_current_changes)
|
|
mcp.add_tool(git_cl.format_checkout)
|
|
mcp.add_tool(git_cl.upload_change_list)
|
|
mcp.run()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run(main)
|