From 422dde8ba3ed231f625571f8fa6b2a1adcc2ed02 Mon Sep 17 00:00:00 2001
From: zhangzhanwei <zhanwei.zhang@fit2cloud.com>
Date: Mon, 15 Sep 2025 11:42:55 +0800
Subject: [PATCH] feat: System application generate prompt

---
 .../system-resource-management/application.ts | 20 +++++++++++++++-
 .../component/GeneratePromptDialog.vue        | 23 ++++++++++++++++++-
 2 files changed, 41 insertions(+), 2 deletions(-)

diff --git a/ui/src/api/system-resource-management/application.ts b/ui/src/api/system-resource-management/application.ts
index 9e94c8ccd..b3ae5a515 100644
--- a/ui/src/api/system-resource-management/application.ts
+++ b/ui/src/api/system-resource-management/application.ts
@@ -124,6 +124,23 @@ const open: (application_id: string, loading?: Ref<boolean>) => Promise<Result<s
   return get(`${prefix}/${application_id}/open`, {}, loading)
 }
 
+/**
+ * 生成提示词
+ * @param application_id 
+ * @param model_id 
+ * @param data 
+ * @returns 
+ */
+const generate_prompt: (application_id:string, model_id:string, data: any) => Promise<any> = (
+  application_id,
+  model_id,
+  data
+) => {
+  const prefix = (window.MaxKB?.prefix ? window.MaxKB?.prefix : '/admin') + '/api'
+  return postStream(`${prefix}/system/resource/application/${application_id}/model/${model_id}/prompt_generate`, data)
+}
+
+
 /**
  * 应用发布
  * @param application_id
@@ -302,5 +319,6 @@ export default {
   postTextToSpeech,
   speechToText,
   getMcpTools,
-  putXpackAccessToken
+  putXpackAccessToken,
+  generate_prompt
 }
diff --git a/ui/src/views/application/component/GeneratePromptDialog.vue b/ui/src/views/application/component/GeneratePromptDialog.vue
index 00ea01587..fe029b5fb 100644
--- a/ui/src/views/application/component/GeneratePromptDialog.vue
+++ b/ui/src/views/application/component/GeneratePromptDialog.vue
@@ -75,13 +75,23 @@
 
 <script setup lang="ts">
 import { computed, reactive, ref } from 'vue'
+import { useRoute } from 'vue-router'
+import systemGeneratePromptAPI from '@/api/system-resource-management/application'
 import generatePromptAPI from '@/api/application/application'
 import useStore from '@/stores'
 const emit = defineEmits(['replace'])
 const { user } = useStore()
+const route = useRoute()
 
 const chatMessages = ref<Array<any>>([])
 
+const apiType = computed(() => {
+  if (route.path.includes('resource-management')) {
+    return 'systemManage'
+  } else {
+    return 'workspace'
+  }
+})
 // 原始输入
 const originalUserInput = ref<string>('')
 const modelID = ref('')
@@ -204,10 +214,21 @@ function generatePrompt(inputValue: any) {
     messages: chatMessages.value,
     prompt: promptTemplates.INIT_TEMPLATE,
   }
-  generatePromptAPI.generate_prompt(workspaceId, modelID.value, applicationID.value,requestData).then((response) => {
+  if (apiType.value === 'workspace') {
+        generatePromptAPI.generate_prompt(workspaceId, modelID.value, applicationID.value,requestData)
+  .then((response) => {
     const reader = response.body.getReader()
     reader.read().then(getWrite(reader))
   })
+  } else if (apiType.value === 'systemManage') {
+    console.log(apiType.value)
+    systemGeneratePromptAPI.generate_prompt(applicationID.value, modelID.value, requestData)
+    .then((response) => {
+    const reader = response.body.getReader()
+    reader.read().then(getWrite(reader))
+  })
+  }
+  
 }
 
 // 重新生成点击