@yorha2b-lab/autodev 1.0.5 → 1.0.7

package/bin/autodev.js

@@ -3,6 +3,7 @@ const fs = require('fs')
 const path = require('path')
 const chalk = require('chalk')
 const figlet = require('figlet')
+const { program } = require('commander')
 
 console.log(chalk.cyan(figlet.textSync('AutoDev', { horizontalLayout: 'full' })))
 console.log(chalk.gray('--------------------------------------------------'))
@@ -10,35 +11,43 @@ console.log(chalk.white(' [System] ') + chalk.green('YoRHa No.2 Type B Unit: ')
 console.log(chalk.white(' [Mission] ') + chalk.yellow('Generate Frontend CRUD: ') + chalk.cyan('Awaiting Command'))
 console.log(chalk.gray('--------------------------------------------------\n'))
 
-const localEnv = path.resolve(process.cwd(), '.env')
-const pkgEnv = path.resolve(__dirname, '../.env')
-
-require("dotenv").config({
-    path: fs.existsSync(localEnv) ? localEnv : pkgEnv
-})
-
-const { program } = require('commander')
-const watchApi = require('../src/commands/watch-api')
-const watchPage = require('../src/commands/watch-page')
-const watchPart = require('../src/commands/watch-part')
-
 program.version('1.0.0').description('AI驱动的前端CRUD代码生成器')
-
 program.option('-t, --template <type>', '指定前端框架模板', 'react')
 
+program
+    .command('init')
+    .description('在当前目录初始化 .env 和 config.js 配置文件')
+    .action(() => {
+        const tplConfig = path.resolve(__dirname, '../config.js')
+        const tplEnv = path.resolve(__dirname, '../.env.example')
+        fs.copyFileSync(tplConfig, path.join(process.cwd(), 'config.js'))
+        fs.copyFileSync(tplEnv, path.join(process.cwd(), '.env'))
+        console.log('✅ 已生成 config.js')
+        console.log('✅ 已生成 .env')
+    })
+
 program
     .command('watch:page')
     .description('监听截图，自动生成完整的增删改查页面')
-    .action(() => watchPage(program.opts()))
+    .action(() => {
+        const watchPage = require('../src/commands/watch-page')
+        watchPage(program.opts())
+    })
 
 program
     .command('watch:part')
     .description('监听截图，自动生成局部 UI 组件')
-    .action(() => watchPart(program.opts()))
+    .action(() => {
+        const watchPart = require('../src/commands/watch-part')
+        watchPart(program.opts())
+    })
 
 program
     .command('watch:api')
     .description('监听 Swagger，自动对齐真实接口字段')
-    .action(() => watchApi())
+    .action(() => {
+        const watchApi = require('../src/commands/watch-api')
+        watchApi()
+    })
 
 program.parse(process.argv)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yorha2b-lab/autodev",
-  "version": "1.0.5",
+  "version": "1.0.7",
   "description": "基于视觉大模型的前端(Umi+Antd)全自动 CRUD 代码生成器",
   "bin": {
     "autodev": "bin/autodev.js"

package/src/services/llm.js

@@ -1,3 +1,5 @@
+const fs = require('fs')
+const path = require('path')
 const JSON5 = require('json5')
 const sharp = require('sharp')
 const OpenAI = require('openai')
@@ -7,13 +9,28 @@ const apiPrompt = require('../prompts/watch-api.js')
 const { UI_DESIGNER, API_DESIGNER, MOCK_DESIGNER } = require('../prompts/system.js')
 
 const config = getConfig()
-const openai = new OpenAI({ apiKey: process.env.API_KEY, baseURL: process.env.BASE_URL })
+
+let client = null
+
+const getOpenAI = () => {
+    if (!client) {
+
+        const localEnv = path.resolve(process.cwd(), '.env')
+        const pkgEnv = path.resolve(__dirname, '../.env')
+        require("dotenv").config({
+            path: fs.existsSync(localEnv) ? localEnv : pkgEnv
+        })
+        client = new OpenAI({ apiKey: process.env.API_KEY, baseURL: process.env.BASE_URL })
+    }
+    return client
+}
 
 const askAI = async (model, messages, retryCount = 0) => {
 
     if (retryCount > 3) throw new Error('AI 重试次数耗尽，请检查网络或图片')
 
     try {
+        const openai = getOpenAI()
         const response = await openai.chat.completions.create({ model, messages, response_format: { type: 'json_object' } })
         let raw = response.choices[0].message.content.trim()
         raw = raw.replace(/^```json\n?/, '').replace(/\n?```$/, '')