diff --git a/apps/ai-cli/gen/data.ts b/apps/ai-cli/gen/data.ts
index b169223..929ffed 100644
--- a/apps/ai-cli/gen/data.ts
+++ b/apps/ai-cli/gen/data.ts
@@ -1,7 +1,7 @@
import glob from 'fast-glob';
import * as fs from 'node:fs/promises';
import * as path from 'node:path';
-
+import { Command } from 'commander';
export interface FileInfo {
name: string;
path: string;
@@ -45,4 +45,15 @@ export const saveJsonToFile = async (dir: string, outputPath: string = 'data.jso
}
};
-saveJsonToFile('.')
\ No newline at end of file
+const program = new Command();
+program
+ .description('保存JSON数据')
+ .option('-d, --dir
', '指定目录')
+ .option('-o, --output