@yorha2b-lab/autodev 1.0.4 → 1.0.6

package/package.json

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

package/src/services/llm.js

@@ -1,18 +1,28 @@
 const JSON5 = require('json5')
 const sharp = require('sharp')
+const OpenAI = require('openai')
 const mockPrompt = require('../prompts/mock.js')
+const { getConfig } = require('../utils/utils.js')
 const apiPrompt = require('../prompts/watch-api.js')
-const { getConfig, initOpenAI } = require('../utils/utils.js')
 const { UI_DESIGNER, API_DESIGNER, MOCK_DESIGNER } = require('../prompts/system.js')
 
 const config = getConfig()
-const openai = initOpenAI()
+
+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?```$/, '')

package/src/utils/utils.js

@@ -1,6 +1,5 @@
 const fs = require('fs')
 const path = require('path')
-const OpenAI = require('openai')
 
 const getConfig = () => {
     const localConfigPath = path.join(process.cwd(), 'config.js')
@@ -13,10 +12,6 @@ const getConfig = () => {
     return config
 }
 
-const initOpenAI = () => {
-    return new OpenAI({ apiKey: process.env.API_KEY, baseURL: process.env.BASE_URL })
-}
-
 /**
  * 大模型输出的标准 JSON 无法携带 render: (text) => <Tag> 等箭头函数。
  * 我们在 Prompt 中要求大模型用 _CODE_ 占位符包裹函数字符串，
@@ -80,4 +75,4 @@ const copyComponents = (options, componentsDir = 'src/components') => {
     fs.readdirSync(componentsDirectory).forEach(file => fs.copyFileSync(path.join(componentsDirectory, file), path.join(targetDir, file)))
 }
 
-module.exports = { getConfig, copyHooks, cleanCode, initOpenAI, copyComponents, getExistingMenus, generateSmartImports }
+module.exports = { getConfig, copyHooks, cleanCode, copyComponents, getExistingMenus, generateSmartImports }