diff --git a/package.json b/package.json
index aa9892e..4a1cf45 100644
--- a/package.json
+++ b/package.json
@@ -14,7 +14,8 @@
   },
   "pi": {
     "extensions": [
-      "./packages/pi-permission-inner-cmd/src/index.ts"
+      "./packages/pi-permission-inner-cmd/src/index.ts",
+      "./packages/pi-permission-ai-judge/src/index.ts"
     ]
   },
   "devDependencies": {