fix: AI导出表格模板获取表格信息错误的bug

Author: pixelmaxQm

web/src/view/systemTools/exportTemplate/exportTemplate.vue

@@ -249,6 +249,25 @@
           </el-select>
         </el-form-item>
 
+        <el-form-item
+          label="AI用关联表"
+          prop="tables"
+        >
+          <el-select
+            multiple
+            v-model="tables"
+            clearable
+            placeholder="关联表使用AI关联的情况下请选择"
+          >
+              <el-option
+                  v-for="item in tableOptions"
+                  :key="item.tableName"
+                  :label="item.tableName"
+                  :value="item.tableName"
+              />
+          </el-select>
+        </el-form-item>
+
         <el-form-item
             label="AI帮写:"
             prop="ai"
@@ -488,6 +507,7 @@ const formData = ref({
 })
 
 const prompt = ref('')
+const tables = ref([])
 
 const typeSearchOptions = ref([
   {
@@ -614,11 +634,26 @@ const dbList = ref([])
 const tableOptions = ref([])
 const aiLoading = ref(false)
 
+const getTablesCloumn = async() => {
+  const tablesMap = {}
+  const promises = tables.value.map(async item => {
+    const res = await getColumn({
+      businessDB: formData.value.dbName,
+      tableName: item
+    })
+    if(res.code === 0) {
+      tablesMap[item] = res.data.columns
+    }
+  })
+  await Promise.all(promises)
+  return tablesMap
+}
+
 
 const autoExport = async () => {
   aiLoading.value = true
-  const tables = tableOptions.value.map(item => item.tableName)
-  const aiRes = await butler({prompt:prompt.value,businessDB: formData.value.dbName||"",tables:tables,command:'autoExportTemplate'})
+  const tableMap = await getTablesCloumn()
+  const aiRes = await butler({prompt:prompt.value,businessDB: formData.value.dbName||"",tableMap:tableMap,command:'autoExportTemplate'})
   aiLoading.value = false
   if (aiRes.code === 0) {
     const aiData = JSON.parse(aiRes.data)
@@ -643,6 +678,7 @@ getDbFunc()
 const dbNameChange = () => {
   formData.value.tableName = ''
   formData.value.templateInfo = ''
+  tables.value = []
   getTableFunc()
 }