@yorha2b-lab/autodev 1.0.5 → 1.0.6

package/bin/autodev.js

@@ -26,6 +26,18 @@ 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('监听截图，自动生成完整的增删改查页面')

package/package.json

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

package/src/services/llm.js

@@ -7,13 +7,22 @@ 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) {
+        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?```$/, '')