feat: System application generate prompt

This commit is contained in:
zhangzhanwei 2025-09-15 11:42:55 +08:00 committed by zhanweizhang7
parent d47dbc3a50
commit 422dde8ba3
2 changed files with 41 additions and 2 deletions

View File

@ -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
}

View File

@ -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))
})
}
}
//