Skip to content

Commit 422dde8

Browse files
committed
feat: System application generate prompt
1 parent d47dbc3 commit 422dde8

File tree

2 files changed

+41
-2
lines changed

2 files changed

+41
-2
lines changed

ui/src/api/system-resource-management/application.ts

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -124,6 +124,23 @@ const open: (application_id: string, loading?: Ref<boolean>) => Promise<Result<s
124124
return get(`${prefix}/${application_id}/open`, {}, loading)
125125
}
126126

127+
/**
128+
* 生成提示词
129+
* @param application_id
130+
* @param model_id
131+
* @param data
132+
* @returns
133+
*/
134+
const generate_prompt: (application_id:string, model_id:string, data: any) => Promise<any> = (
135+
application_id,
136+
model_id,
137+
data
138+
) => {
139+
const prefix = (window.MaxKB?.prefix ? window.MaxKB?.prefix : '/admin') + '/api'
140+
return postStream(`${prefix}/system/resource/application/${application_id}/model/${model_id}/prompt_generate`, data)
141+
}
142+
143+
127144
/**
128145
* 应用发布
129146
* @param application_id
@@ -302,5 +319,6 @@ export default {
302319
postTextToSpeech,
303320
speechToText,
304321
getMcpTools,
305-
putXpackAccessToken
322+
putXpackAccessToken,
323+
generate_prompt
306324
}

ui/src/views/application/component/GeneratePromptDialog.vue

Lines changed: 22 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,13 +75,23 @@
7575

7676
<script setup lang="ts">
7777
import { computed, reactive, ref } from 'vue'
78+
import { useRoute } from 'vue-router'
79+
import systemGeneratePromptAPI from '@/api/system-resource-management/application'
7880
import generatePromptAPI from '@/api/application/application'
7981
import useStore from '@/stores'
8082
const emit = defineEmits(['replace'])
8183
const { user } = useStore()
84+
const route = useRoute()
8285
8386
const chatMessages = ref<Array<any>>([])
8487
88+
const apiType = computed(() => {
89+
if (route.path.includes('resource-management')) {
90+
return 'systemManage'
91+
} else {
92+
return 'workspace'
93+
}
94+
})
8595
// 原始输入
8696
const originalUserInput = ref<string>('')
8797
const modelID = ref('')
@@ -204,10 +214,21 @@ function generatePrompt(inputValue: any) {
204214
messages: chatMessages.value,
205215
prompt: promptTemplates.INIT_TEMPLATE,
206216
}
207-
generatePromptAPI.generate_prompt(workspaceId, modelID.value, applicationID.value,requestData).then((response) => {
217+
if (apiType.value === 'workspace') {
218+
generatePromptAPI.generate_prompt(workspaceId, modelID.value, applicationID.value,requestData)
219+
.then((response) => {
220+
const reader = response.body.getReader()
221+
reader.read().then(getWrite(reader))
222+
})
223+
} else if (apiType.value === 'systemManage') {
224+
console.log(apiType.value)
225+
systemGeneratePromptAPI.generate_prompt(applicationID.value, modelID.value, requestData)
226+
.then((response) => {
208227
const reader = response.body.getReader()
209228
reader.read().then(getWrite(reader))
210229
})
230+
}
231+
211232
}
212233
213234
// 重新生成点击

0 commit comments

Comments
 (0)