From 036d755e3079a5705ddb8f23c2935f68ef59bbb8 Mon Sep 17 00:00:00 2001
From: "Edward Z. Yang" <ezyang@meta.com>
Date: Tue, 3 Jun 2025 21:34:01 -0400
Subject: [PATCH] Update

[ghstack-poisoned]
---
 codemcp/tools/run_command.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/codemcp/tools/run_command.py b/codemcp/tools/run_command.py
index ec68619..0665b55 100644
--- a/codemcp/tools/run_command.py
+++ b/codemcp/tools/run_command.py
@@ -6,12 +6,14 @@
 from ..code_command import get_command_from_config, run_code_command
 from ..common import normalize_file_path
 from .commit_utils import append_commit_hash
+from ..mcp import mcp
 
 __all__ = [
     "run_command",
 ]
 
 
+@mcp.tool()
 async def run_command(
     project_dir: Optional[str] = None,
     command: str = "",