339 lines
11 KiB
Vue
339 lines
11 KiB
Vue
<template>
|
||
<NodeContainer :nodeModel="nodeModel">
|
||
<h5 class="title-decoration-1 mb-8">{{ $t('views.applicationWorkflow.nodeSetting') }}</h5>
|
||
<el-card shadow="never" class="card-never" style="--el-card-padding: 12px">
|
||
<el-form
|
||
@submit.prevent
|
||
:model="chat_data"
|
||
label-position="top"
|
||
require-asterisk-position="right"
|
||
label-width="auto"
|
||
ref="aiChatNodeFormRef"
|
||
hide-required-asterisk
|
||
>
|
||
<el-form-item
|
||
:label="$t('views.application.applicationForm.form.aiModel.label')"
|
||
prop="model_id"
|
||
:rules="{
|
||
required: true,
|
||
message: $t('views.application.applicationForm.form.aiModel.placeholder'),
|
||
trigger: 'change'
|
||
}"
|
||
>
|
||
<template #label>
|
||
<div class="flex-between w-full">
|
||
<div>
|
||
<span
|
||
>{{ $t('views.application.applicationForm.form.aiModel.label')
|
||
}}<span class="danger">*</span></span
|
||
>
|
||
</div>
|
||
|
||
<el-button
|
||
:disabled="!chat_data.model_id"
|
||
type="primary"
|
||
link
|
||
@click="openAIParamSettingDialog(chat_data.model_id)"
|
||
@refreshForm="refreshParam"
|
||
>
|
||
<el-icon><Setting /></el-icon>
|
||
</el-button>
|
||
</div>
|
||
</template>
|
||
<ModelSelect
|
||
@change="model_change"
|
||
@wheel="wheel"
|
||
:teleported="false"
|
||
v-model="chat_data.model_id"
|
||
:placeholder="$t('views.application.applicationForm.form.aiModel.placeholder')"
|
||
:options="modelOptions"
|
||
@submitModel="getModel"
|
||
showFooter
|
||
:model-type="'LLM'"
|
||
></ModelSelect>
|
||
</el-form-item>
|
||
|
||
<el-form-item :label="$t('views.application.applicationForm.form.roleSettings.label')">
|
||
<MdEditorMagnify
|
||
:title="$t('views.application.applicationForm.form.roleSettings.label')"
|
||
v-model="chat_data.system"
|
||
style="height: 100px"
|
||
@submitDialog="submitSystemDialog"
|
||
:placeholder="$t('views.application.applicationForm.form.roleSettings.label')"
|
||
/>
|
||
</el-form-item>
|
||
<el-form-item
|
||
:label="$t('views.application.applicationForm.form.prompt.label')"
|
||
prop="prompt"
|
||
:rules="{
|
||
required: true,
|
||
message: $t('views.application.applicationForm.form.prompt.requiredMessage'),
|
||
trigger: 'blur'
|
||
}"
|
||
>
|
||
<template #label>
|
||
<div class="flex align-center">
|
||
<div class="mr-4">
|
||
<span
|
||
>{{ $t('views.application.applicationForm.form.prompt.label')
|
||
}}<span class="danger">*</span></span
|
||
>
|
||
</div>
|
||
<el-tooltip effect="dark" placement="right" popper-class="max-w-200">
|
||
<template #content
|
||
>{{ $t('views.application.applicationForm.form.prompt.tooltip') }}
|
||
</template>
|
||
<AppIcon iconName="app-warning" class="app-warning-icon"></AppIcon>
|
||
</el-tooltip>
|
||
</div>
|
||
</template>
|
||
<MdEditorMagnify
|
||
@wheel="wheel"
|
||
:title="$t('views.application.applicationForm.form.prompt.label')"
|
||
v-model="chat_data.prompt"
|
||
style="height: 150px"
|
||
@submitDialog="submitDialog"
|
||
/>
|
||
</el-form-item>
|
||
<el-form-item :label="$t('views.application.applicationForm.form.historyRecord.label')">
|
||
<template #label>
|
||
<div class="flex-between">
|
||
<div>{{ $t('views.application.applicationForm.form.historyRecord.label') }}</div>
|
||
<el-select v-model="chat_data.dialogue_type" type="small" style="width: 100px">
|
||
<el-option :label="$t('views.applicationWorkflow.node')" value="NODE" />
|
||
<el-option :label="$t('views.applicationWorkflow.workflow')" value="WORKFLOW" />
|
||
</el-select>
|
||
</div>
|
||
</template>
|
||
<el-input-number
|
||
v-model="chat_data.dialogue_number"
|
||
:min="0"
|
||
:value-on-clear="0"
|
||
controls-position="right"
|
||
class="w-full"
|
||
:step="1"
|
||
:step-strictly="true"
|
||
/>
|
||
</el-form-item>
|
||
|
||
<div class="flex-between mb-16">
|
||
<div class="lighter">{{ $t('views.applicationWorkflow.nodes.mcpNode.tool') }}</div>
|
||
<el-button type="primary" link @click="openMcpServersDialog" @refreshForm="refreshParam">
|
||
<el-icon><Setting /></el-icon>
|
||
</el-button>
|
||
</div>
|
||
|
||
<el-form-item @click.prevent>
|
||
<template #label>
|
||
<div class="flex-between w-full">
|
||
<div>
|
||
<span>{{
|
||
$t('views.application.applicationForm.form.reasoningContent.label')
|
||
}}</span>
|
||
</div>
|
||
<el-button
|
||
type="primary"
|
||
link
|
||
@click="openReasoningParamSettingDialog"
|
||
@refreshForm="refreshParam"
|
||
>
|
||
<el-icon><Setting /></el-icon>
|
||
</el-button>
|
||
</div>
|
||
</template>
|
||
<el-switch size="small" v-model="chat_data.model_setting.reasoning_content_enable" />
|
||
</el-form-item>
|
||
<el-form-item @click.prevent>
|
||
<template #label>
|
||
<div class="flex align-center">
|
||
<div class="mr-4">
|
||
<span>{{
|
||
$t('views.applicationWorkflow.nodes.aiChatNode.returnContent.label')
|
||
}}</span>
|
||
</div>
|
||
<el-tooltip effect="dark" placement="right" popper-class="max-w-200">
|
||
<template #content>
|
||
{{ $t('views.applicationWorkflow.nodes.aiChatNode.returnContent.tooltip') }}
|
||
</template>
|
||
<AppIcon iconName="app-warning" class="app-warning-icon"></AppIcon>
|
||
</el-tooltip>
|
||
</div>
|
||
</template>
|
||
<el-switch size="small" v-model="chat_data.is_result" />
|
||
</el-form-item>
|
||
</el-form>
|
||
</el-card>
|
||
|
||
<AIModeParamSettingDialog ref="AIModeParamSettingDialogRef" @refresh="refreshParam" />
|
||
<ReasoningParamSettingDialog
|
||
ref="ReasoningParamSettingDialogRef"
|
||
@refresh="submitReasoningDialog"
|
||
/>
|
||
<McpServersDialog ref="mcpServersDialogRef" @refresh="submitMcpServersDialog" />
|
||
</NodeContainer>
|
||
</template>
|
||
<script setup lang="ts">
|
||
import { cloneDeep, set, groupBy } from 'lodash'
|
||
import { app } from '@/main'
|
||
import NodeContainer from '@/workflow/common/NodeContainer.vue'
|
||
import type { FormInstance } from 'element-plus'
|
||
import { ref, computed, onMounted } from 'vue'
|
||
import applicationApi from '@/api/application/application'
|
||
import useStore from '@/stores'
|
||
import { isLastNode } from '@/workflow/common/data'
|
||
import AIModeParamSettingDialog from '@/views/application/component/AIModeParamSettingDialog.vue'
|
||
import { t } from '@/locales'
|
||
import ReasoningParamSettingDialog from '@/views/application/component/ReasoningParamSettingDialog.vue'
|
||
import McpServersDialog from '@/views/application/component/McpServersDialog.vue'
|
||
const { model } = useStore()
|
||
|
||
const wheel = (e: any) => {
|
||
if (e.ctrlKey === true) {
|
||
e.preventDefault()
|
||
return true
|
||
} else {
|
||
e.stopPropagation()
|
||
return true
|
||
}
|
||
}
|
||
|
||
function submitSystemDialog(val: string) {
|
||
set(props.nodeModel.properties.node_data, 'system', val)
|
||
}
|
||
|
||
function submitDialog(val: string) {
|
||
set(props.nodeModel.properties.node_data, 'prompt', val)
|
||
}
|
||
|
||
const model_change = (model_id?: string) => {
|
||
if (model_id) {
|
||
AIModeParamSettingDialogRef.value?.reset_default(model_id, id)
|
||
} else {
|
||
refreshParam({})
|
||
}
|
||
}
|
||
const {
|
||
params: { id }
|
||
} = app.config.globalProperties.$route as any
|
||
|
||
// @ts-ignore
|
||
const defaultPrompt = `${t('views.applicationWorkflow.nodes.aiChatNode.defaultPrompt')}：
|
||
{{${t('views.applicationWorkflow.nodes.searchDatasetNode.label')}.data}}
|
||
${t('views.problem.title')}：
|
||
{{${t('views.applicationWorkflow.nodes.startNode.label')}.question}}`
|
||
|
||
const form = {
|
||
model_id: '',
|
||
system: '',
|
||
prompt: defaultPrompt,
|
||
dialogue_number: 1,
|
||
is_result: false,
|
||
temperature: null,
|
||
max_tokens: null,
|
||
dialogue_type: 'WORKFLOW',
|
||
model_setting: {
|
||
reasoning_content_start: '<think>',
|
||
reasoning_content_end: '</think>',
|
||
reasoning_content_enable: false
|
||
}
|
||
}
|
||
|
||
const chat_data = computed({
|
||
get: () => {
|
||
if (props.nodeModel.properties.node_data) {
|
||
if (!props.nodeModel.properties.node_data.model_setting) {
|
||
set(props.nodeModel.properties.node_data, 'model_setting', {
|
||
reasoning_content_start: '<think>',
|
||
reasoning_content_end: '</think>',
|
||
reasoning_content_enable: false
|
||
})
|
||
}
|
||
return props.nodeModel.properties.node_data
|
||
} else {
|
||
set(props.nodeModel.properties, 'node_data', form)
|
||
}
|
||
|
||
return props.nodeModel.properties.node_data
|
||
},
|
||
set: (value) => {
|
||
set(props.nodeModel.properties, 'node_data', value)
|
||
}
|
||
})
|
||
const props = defineProps<{ nodeModel: any }>()
|
||
|
||
const aiChatNodeFormRef = ref<FormInstance>()
|
||
|
||
const modelOptions = ref<any>(null)
|
||
const AIModeParamSettingDialogRef = ref<InstanceType<typeof AIModeParamSettingDialog>>()
|
||
const ReasoningParamSettingDialogRef = ref<InstanceType<typeof ReasoningParamSettingDialog>>()
|
||
const validate = () => {
|
||
return aiChatNodeFormRef.value?.validate().catch((err) => {
|
||
return Promise.reject({ node: props.nodeModel, errMessage: err })
|
||
})
|
||
}
|
||
|
||
function getModel() {
|
||
if (id) {
|
||
applicationApi.getApplicationModel(id).then((res: any) => {
|
||
modelOptions.value = groupBy(res?.data, 'provider')
|
||
})
|
||
} else {
|
||
model.asyncGetModel().then((res: any) => {
|
||
modelOptions.value = groupBy(res?.data, 'provider')
|
||
})
|
||
}
|
||
}
|
||
|
||
const openAIParamSettingDialog = (modelId: string) => {
|
||
if (modelId) {
|
||
AIModeParamSettingDialogRef.value?.open(modelId, id, chat_data.value.model_params_setting)
|
||
}
|
||
}
|
||
|
||
const openReasoningParamSettingDialog = () => {
|
||
ReasoningParamSettingDialogRef.value?.open(chat_data.value.model_setting)
|
||
}
|
||
|
||
function refreshParam(data: any) {
|
||
set(props.nodeModel.properties.node_data, 'model_params_setting', data)
|
||
}
|
||
|
||
function submitReasoningDialog(val: any) {
|
||
let model_setting = cloneDeep(props.nodeModel.properties.node_data.model_setting)
|
||
model_setting = {
|
||
...model_setting,
|
||
...val
|
||
}
|
||
|
||
set(props.nodeModel.properties.node_data, 'model_setting', model_setting)
|
||
}
|
||
|
||
const mcpServersDialogRef = ref()
|
||
function openMcpServersDialog() {
|
||
const config = {
|
||
mcp_servers: chat_data.value.mcp_servers,
|
||
mcp_enable: chat_data.value.mcp_enable
|
||
}
|
||
mcpServersDialogRef.value.open(config)
|
||
}
|
||
|
||
function submitMcpServersDialog(config: any) {
|
||
set(props.nodeModel.properties.node_data, 'mcp_servers', config.mcp_servers)
|
||
set(props.nodeModel.properties.node_data, 'mcp_enable', config.mcp_enable)
|
||
}
|
||
|
||
onMounted(() => {
|
||
getModel()
|
||
if (typeof props.nodeModel.properties.node_data?.is_result === 'undefined') {
|
||
if (isLastNode(props.nodeModel)) {
|
||
set(props.nodeModel.properties.node_data, 'is_result', true)
|
||
}
|
||
}
|
||
set(props.nodeModel, 'validate', validate)
|
||
if (!chat_data.value.dialogue_type) {
|
||
chat_data.value.dialogue_type = 'WORKFLOW'
|
||
}
|
||
})
|
||
</script>
|
||
<style lang="scss" scoped></style>
|