@yorha2b-lab/autodev 1.0.3 → 1.0.4

package/package.json

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

package/src/services/llm.js

@@ -1,13 +1,12 @@
 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 = new OpenAI({ apiKey: process.env.API_KEY, baseURL: process.env.BASE_URL })
+const openai = initOpenAI()
 
 const askAI = async (model, messages, retryCount = 0) => {
 

package/src/utils/utils.js

@@ -1,5 +1,6 @@
 const fs = require('fs')
 const path = require('path')
+const OpenAI = require('openai')
 
 const getConfig = () => {
     const localConfigPath = path.join(process.cwd(), 'config.js')
@@ -12,6 +13,10 @@ 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_ 占位符包裹函数字符串，
@@ -75,4 +80,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, copyComponents, getExistingMenus, generateSmartImports }
+module.exports = { getConfig, copyHooks, cleanCode, initOpenAI, copyComponents, getExistingMenus, generateSmartImports }