feat: 更新AI功能

pixelmaxQm · pixelmaxQm · commit 2112caeac11b · 2024-10-10T21:43:09.000+08:00
diff --git a/server/api/v1/system/sys_auto_code.go b/server/api/v1/system/sys_auto_code.go
@@ -1,6 +1,7 @@
 package system
 
 import (
+	systemReq "github.com/flipped-aurora/gin-vue-admin/server/model/system/request"
 	"github.com/goccy/go-json"
 	"io"
 	"strings"
@@ -105,18 +106,23 @@ func (autoApi *AutoCodeApi) GetColumn(c *gin.Context) {
 }
 
 func (autoApi *AutoCodeApi) LLMAuto(c *gin.Context) {
-	prompt := c.Query("prompt")
-	mode := c.Query("mode")
-	params := make(map[string]string)
-	params["prompt"] = prompt
-	params["mode"] = mode
+	var llm systemReq.LLMAutoCode
+	err := c.ShouldBindJSON(&llm)
+	if err != nil {
+		response.FailWithMessage(err.Error(), c)
+		return
+	}
+	if global.GVA_CONFIG.AutoCode.AiPath == "" {
+		response.FailWithMessage("请先前往插件市场个人中心获取AiPath并填入config.yaml中", c)
+		return
+	}
 	path := strings.ReplaceAll(global.GVA_CONFIG.AutoCode.AiPath, "{FUNC}", "api/chat/ai")
 	res, err := request.HttpRequest(
 		path,
 		"POST",
 		nil,
-		params,
 		nil,
+		llm,
 	)
 	if err != nil {
 		global.GVA_LOG.Error("大模型生成失败!", zap.Error(err))
diff --git a/server/model/system/request/sys_auto_code.go b/server/model/system/request/sys_auto_code.go
@@ -259,3 +259,8 @@ type InitApi struct {
 	PlugName string `json:"plugName"`
 	APIs     []uint `json:"apis"`
 }
+
+type LLMAutoCode struct {
+	Prompt string `json:"prompt" form:"prompt" gorm:"column:prompt;comment:提示语;type:text;"` //提示语
+	Mode   string `json:"mode" form:"mode" gorm:"column:mode;comment:模式;type:text;"`        //模式
+}
diff --git a/web/src/api/autoCode.js b/web/src/api/autoCode.js
@@ -140,20 +140,16 @@ export const pubPlug = (params) => {
 }
 
 
-export const llmAuto = (params) => {
-  let modeName = {
-    "xiaoqi": "小奇",
-    "xiaomiao": "小淼",
-  }
+export const llmAuto = (data) => {
   return service({
     url: '/autoCode/llmAuto',
     method: 'post',
-    params,
+    data,
     timeout: 1000 * 60 * 10,
     loadingOption:{
       lock: true,
       fullscreen:true,
-      text: `${modeName[params.mode]}正在思考，请稍候...`,
+      text: `小淼正在思考，请稍候...`,
     }
   })
 }
@@ -182,6 +178,3 @@ export const initAPI = (data) => {
     data
   })
 }
-
-
-
diff --git a/web/src/pathInfo.json b/web/src/pathInfo.json
@@ -1,5 +1,6 @@
 {
   "/src/view/about/index.vue": "About",
+  "/src/view/dashboard/index.vue": "Dashboard",
   "/src/view/error/index.vue": "Error",
   "/src/view/error/reload.vue": "Reload",
   "/src/view/example/breakpoint/breakpoint.vue": "BreakPoint",
@@ -32,6 +33,7 @@
   "/src/view/superAdmin/menu/icon.vue": "Icon",
   "/src/view/superAdmin/menu/menu.vue": "Menus",
   "/src/view/superAdmin/operation/sysOperationRecord.vue": "SysOperationRecord",
+  "/src/view/superAdmin/params/sysParams.vue": "SysParams",
   "/src/view/superAdmin/user/user.vue": "User",
   "/src/view/system/state.vue": "State",
   "/src/view/systemTools/autoCode/component/fieldDialog.vue": "FieldDialog",
diff --git a/web/src/view/systemTools/autoCode/index.vue b/web/src/view/systemTools/autoCode/index.vue
@@ -5,22 +5,23 @@
       title="此功能为开发环境使用，不建议发布到生产，具体使用效果请点我观看。"
     />
     <div class="gva-search-box">
-      <div class="text-lg mb-2 text-gray-600">使用AI创建</div>
+      <div class="text-lg mb-2 text-gray-600">使用AI创建<a class="text-blue-600 text-sm ml-4" href="https://plugin.gin-vue-admin.com/#/layout/userInfo/center" target="_blank">获取AiPath</a></div>
       <div class="relative">
-        <el-input v-model="prompt" type="textarea" :rows="5" :maxlength="100" :placeholder="`【Beta】试试描述你的表，让AI帮你完成。\n目前正在测试阶段，遇到问题请及时反馈。\n此功能需要到插件市场个人中心获取自己的AI-Path，把AI-Path填入config.yaml下的autocode-->ai-path，重启项目即可使用。`" resize="none" />
+        <el-input v-model="prompt"
+                  type="textarea"
+                  :rows="5"
+                  :maxlength="100"
+                  :placeholder="`现已完全免费\n试试描述你的表，让AI帮你完成。\n此功能需要到插件市场个人中心获取自己的AI-Path，把AI-Path填入config.yaml下的autocode-->ai-path，重启项目即可使用。\n按下 Ctrl+Enter 或 Cmd+Enter 直接生成`"
+                  resize="none"
+                  @focus="handleFocus"
+                  @blur="handleBlur"/>
         <div class="flex absolute right-2 bottom-2">
           <el-tooltip
-            content="小奇存在失败概率，面向所有用户开放使用（失败了重新生成一下就好）。"
-          >
-            <el-button type="primary" @click="llmAutoFunc('xiaoqi')">
-              <el-icon size="18">
-                <ai-gva />
-              </el-icon> 小奇
-            </el-button>
-          </el-tooltip>
-          <el-tooltip
-            content="小淼基本啥也能设计出来，但是需要消耗积分，测试阶段授权用户自动获得基础积分，开源用户需要填表申请。"
+            effect="light"
           >
+            <template #content>
+              <div>【完全免费】前往<a class="text-blue-600" href="https://plugin.gin-vue-admin.com/#/layout/userInfo/center" target="_blank">插件市场个人中心</a>申请AIPath，填入config.yaml的ai-path属性即可使用。</div>
+            </template>
             <el-button type="primary" @click="llmAutoFunc('xiaomiao')">
               <el-icon size="18">
                 <ai-gva />
@@ -420,6 +421,7 @@
           row-key="fieldName"
         >
           <el-table-column
+            fixed="left"
             align="left"
             type="index"
             width="60"
@@ -431,12 +433,14 @@
             </template>
           </el-table-column>
           <el-table-column
+            fixed="left"
             align="left"
             type="index"
             label="序列"
             width="60"
           />
           <el-table-column
+            fixed="left"
             align="left"
             type="index"
             label="主键"
@@ -447,6 +451,7 @@
             </template>
           </el-table-column>
           <el-table-column
+            fixed="left"
             align="left"
             prop="fieldName"
             label="字段名称"
@@ -525,6 +530,7 @@
           <el-table-column
               align="left"
               prop="excel"
+              width="100"
               label="导入/导出"
           >
             <template #default="{row}">
@@ -791,6 +797,21 @@ import { ElMessage, ElMessageBox } from 'element-plus'
 import WarningBar from '@/components/warningBar/warningBar.vue'
 import Sortable from 'sortablejs'
 
+const handleFocus = () => {
+  document.addEventListener('keydown', handleKeydown);
+};
+
+const handleBlur = () => {
+  document.removeEventListener('keydown', handleKeydown);
+};
+
+
+const handleKeydown = (event) => {
+  if ((event.ctrlKey || event.metaKey) && event.key === 'Enter') {
+    llmAutoFunc('xiaomiao');
+  }
+};
+
 const getOnlyNumber = () => {
   let randomNumber = '';
   while (randomNumber.length < 16) {
@@ -802,6 +823,10 @@ const getOnlyNumber = () => {
 const prompt = ref("")
 
 const llmAutoFunc = async (mode) =>{
+  if (!prompt.value) {
+    ElMessage.error('请输入描述')
+    return
+  }
   const res = await llmAuto({prompt:prompt.value,mode:mode})
   if (res.code === 0) {
     form.value.fields = []

Original file line number	Diff line number	Diff line change
`@@ -259,3 +259,8 @@ type InitApi struct {`
`259`	`259`	PlugName string `json:"plugName"`
`260`	`260`	APIs []uint `json:"apis"`
`261`	`261`	`}`
	`262`	`+`
	`263`	`+type LLMAutoCode struct {`
	`264`	+ Prompt string `json:"prompt" form:"prompt" gorm:"column:prompt;comment:提示语;type:text;"` //提示语
	`265`	+ Mode string `json:"mode" form:"mode" gorm:"column:mode;comment:模式;type:text;"` //模式
	`266`	`+}`