From 943f18db0ca1f816df84a7e0dae55b23d2652f40 Mon Sep 17 00:00:00 2001 From: "Edward Z. Yang" Date: Tue, 6 May 2025 20:02:06 -0400 Subject: [PATCH] Update [ghstack-poisoned] --- codemcp/tools/__init__.py | 2 ++ codemcp/tools/run_command.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/codemcp/tools/__init__.py b/codemcp/tools/__init__.py index 5cc32f3..ca716ad 100644 --- a/codemcp/tools/__init__.py +++ b/codemcp/tools/__init__.py @@ -8,6 +8,7 @@ from .git_show import git_show from .mv import mv from .rm import rm +from .run_command import run_command __all__ = [ "chmod", @@ -17,4 +18,5 @@ "git_show", "mv", "rm", + "run_command", ] diff --git a/codemcp/tools/run_command.py b/codemcp/tools/run_command.py index ec68619..e4cf91a 100644 --- a/codemcp/tools/run_command.py +++ b/codemcp/tools/run_command.py @@ -5,6 +5,7 @@ from ..code_command import get_command_from_config, run_code_command from ..common import normalize_file_path +from ..mcp import mcp from .commit_utils import append_commit_hash __all__ = [ @@ -12,6 +13,7 @@ ] +@mcp.tool() async def run_command( project_dir: Optional[str] = None, command: str = "",