From adb0ee68a6b72727d2b8dcf7714b005668d950c7 Mon Sep 17 00:00:00 2001
From: "Edward Z. Yang" <ezyang@mit.edu>
Date: Tue, 25 Mar 2025 18:15:59 +0800
Subject: [PATCH] Update

[ghstack-poisoned]
---
 codemcp.toml     | 1 +
 run_typecheck.sh | 6 ++++--
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/codemcp.toml b/codemcp.toml
index 1aba1892..6393cf80 100644
--- a/codemcp.toml
+++ b/codemcp.toml
@@ -27,6 +27,7 @@ project_prompt = '''
 format = ["./run_format.sh"]
 lint = ["./run_lint.sh"]
 ghstack = ["uv", "tool", "run", "ghstack"]
+typecheck = ["./run_typecheck.sh"]
 [commands.test]
 command = ["./run_test.sh"]
 doc = "Accepts a pytest-style test selector as an argument to run a specific test."
diff --git a/run_typecheck.sh b/run_typecheck.sh
index 26a3fb5d..be770a33 100755
--- a/run_typecheck.sh
+++ b/run_typecheck.sh
@@ -2,6 +2,8 @@
 set -e
 
 echo "Running Pyright type checker with strict settings..."
-python -m pyright codemcp
+SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+cd "$SCRIPT_DIR"
+"${SCRIPT_DIR}/.venv/bin/python" -m pyright $@
 
-echo "Type checking completed successfully!"
\ No newline at end of file
+echo "Type checking completed successfully!"