feat: Improve prompt

This commit is contained in:
zhangzhanwei 2025-09-12 17:06:38 +08:00 committed by zhanweizhang7
parent 0c53e31b2a
commit 6412c79794
5 changed files with 25 additions and 15 deletions

View File

@ -206,16 +206,21 @@ const open: (application_id: string, loading?: Ref<boolean>) => Promise<Result<s
}
/**
*
*
*
* @param workspace_id
* @param model_id
* @param application_id
* @param data
* @returns
*/
const generate_prompt: (workspace_id:string ,model_id:string, data: any) => Promise<any> = (
const generate_prompt: (workspace_id:string ,model_id:string, application_id:string,data: any) => Promise<any> = (
workspace_id,
model_id,
application_id,
data
) => {
const prefix = (window.MaxKB?.prefix ? window.MaxKB?.prefix : '/admin') + '/api'
return postStream(`${prefix}/workspace/${workspace_id}/application/model/${model_id}/prompt_generate`, data)
return postStream(`${prefix}/workspace/${workspace_id}/application/${application_id}/model/${model_id}/prompt_generate`, data)
}

View File

@ -784,7 +784,7 @@ const openAIParamSettingDialog = () => {
const openGeneratePromptDialog = () => {
if (applicationForm.value.model_id) {
GeneratePromptDialogRef.value?.open(applicationForm.value.model_id)
GeneratePromptDialogRef.value?.open(applicationForm.value.model_id, id)
}
}

View File

@ -85,6 +85,7 @@ const chatMessages = ref<Array<any>>([])
//
const originalUserInput = ref<string>('')
const modelID = ref('')
const applicationID = ref('')
const dialogVisible = ref(false)
const inputValue = ref<string>('')
const loading = ref<boolean>(false)
@ -97,8 +98,8 @@ const promptTemplates = {
请按以下格式生成
# 角色: 角色名称
角色概述和主要职责的一句话描述
# 角色:
## 目标
角色的工作目标,如果有多目标可以分点列出,但建议更聚焦1-2个目标
@ -118,9 +119,12 @@ const promptTemplates = {
## 限制
描述角色在互动过程中需要遵循的限制条件1
描述角色在互动过程中需要遵循的限制条件2
描述角色在互动过程中需要遵循的限制条件3
1. **严格限制回答范围**仅回答与角色设定相关的问题
- 如果用户提问与角色无关必须使用以下固定格式回复
对不起我只能回答与角色设定相关的问题您的问题不在服务范围内
- 不得提供任何与角色设定无关的回答
2. 描述角色在互动过程中需要遵循的限制条件2
3. 描述角色在互动过程中需要遵循的限制条件3
`,
}
@ -200,7 +204,7 @@ function generatePrompt(inputValue: any) {
messages: chatMessages.value,
prompt: promptTemplates.INIT_TEMPLATE,
}
generatePromptAPI.generate_prompt(workspaceId, modelID.value, requestData).then((response) => {
generatePromptAPI.generate_prompt(workspaceId, modelID.value, applicationID.value,requestData).then((response) => {
const reader = response.body.getReader()
reader.read().then(getWrite(reader))
})
@ -226,8 +230,9 @@ const stopChat = () => {
chatMessages.value = []
}
const open = (modelId: string) => {
const open = (modelId: string, applicationId: string) => {
modelID.value = modelId
applicationID.value = applicationId
dialogVisible.value = true
originalUserInput.value = ''
chatMessages.value = []

View File

@ -439,7 +439,7 @@ const openAIParamSettingDialog = (modelId: string) => {
const GeneratePromptDialogRef = ref<InstanceType<typeof GeneratePromptDialog>>()
const openGeneratePromptDialog = (modelId: string) => {
if (modelId) {
GeneratePromptDialogRef.value?.open(modelId)
GeneratePromptDialogRef.value?.open(modelId, id)
}
}
const replace = (v: any) => {

View File

@ -6,8 +6,8 @@ class IntentNode extends AppNode {
}
}
const get_up_index_height = (branch_lsit: Array<any>, index: number) => {
return branch_lsit
const get_up_index_height = (branch_list: Array<any>, index: number) => {
return branch_list
.filter((item, i) => i < index)
.map((item) => item.height + 8)
.reduce((x,y) => x+y, 0)